#336 new "stats" command for netdisco-do

lib/App/Netdisco.pm

@@ -222,6 +222,7 @@ run for each installation:
  
  # apply database schema updates, update MIBs and Vendor MACs
  ~/bin/netdisco-deploy
+ ~/bin/netdisco-do stats
  
  # restart web service (if you run it)
  ~/bin/netdisco-web restart

lib/App/Netdisco/Backend/Worker/Poller/Expiry.pm

@@ -3,8 +3,8 @@ package App::Netdisco::Backend::Worker::Poller::Expiry;
 use Dancer qw/:moose :syntax :script/;
 use Dancer::Plugin::DBIC 'schema';
 
-use Time::Piece;
 use App::Netdisco::Backend::Util ':all';
+use App::Netdisco::Util::Statistics 'update_stats';
 
 use Role::Tiny;
 use namespace::clean;
@@ -51,66 +51,12 @@ sub expire {
       });
   }
 
-  # now update stats
-  my $schema = schema('netdisco');
-  eval { require SNMP::Info };
-  my $snmpinfo_ver = ($@ ? 'n/a' : $SNMP::Info::VERSION);
-
-  # TODO: (when we have the capabilities table?)
-  #  $stats{waps} = sql_scalar('device',['COUNT(*)'], {"model"=>"AIR%"});
-
-  $schema->txn_do(sub {
-    $schema->resultset('Statistics')->update_or_create({
-      day => localtime->ymd,
-
-      device_count =>
-        $schema->resultset('Device')->count_rs->as_query,
-      device_ip_count =>
-        $schema->resultset('DeviceIp')->count_rs->as_query,
-      device_link_count =>
-        $schema->resultset('Virtual::DeviceLinks')
-          ->count_rs({'me.left_ip' => {'>', \'me.right_ip'}})->as_query,
-      device_port_count =>
-        $schema->resultset('DevicePort')->count_rs->as_query,
-      device_port_up_count =>
-        $schema->resultset('DevicePort')->count_rs({up => 'up'})->as_query,
-      ip_table_count =>
-        $schema->resultset('NodeIp')->count_rs->as_query,
-      ip_active_count =>
-        $schema->resultset('NodeIp')->search({-bool => 'active'},
-          {columns => 'ip', distinct => 1})->count_rs->as_query,
-      node_table_count =>
-        $schema->resultset('Node')->count_rs->as_query,
-      node_active_count =>
-        $schema->resultset('Node')->search({-bool => 'active'},
-          {columns => 'mac', distinct => 1})->count_rs->as_query,
-
-      netdisco_ver => pretty_version($App::Netdisco::VERSION, 3),
-      snmpinfo_ver => $snmpinfo_ver,
-      schema_ver   => $schema->schema_version,
-      perl_ver     => pretty_version($], 3),
-      pg_ver       =>
-        pretty_version($schema->storage->dbh->{pg_server_version}, 2),
-
-    }, { key => 'primary' });
-  });
+  # now update stats
+  update_stats();
 
   return job_done("Checked expiry and updated stats");
 }
 
-# take perl or pg versions and make pretty
-sub pretty_version {
-  my ($version, $seglen) = @_;
-  return unless $version and $seglen;
-  return $version if $version !~ m/^[0-9.]+$/;
-  $version =~ s/\.//g;
-  $version = (join '.', reverse map {scalar reverse}
-    unpack("(A${seglen})*", reverse $version));
-  $version =~ s/\.000/.0/g;
-  $version =~ s/\.0+([1-9]+)/.$1/g;
-  return $version;
-}
-
 # expire nodes for a specific device
 sub expirenodes {
   my ($self, $job) = @_;

lib/App/Netdisco/Util/Statistics.pm

@@ -0,0 +1,91 @@
+package App::Netdisco::Util::Statistics;
+
+use Dancer qw/:syntax :script/;
+use Dancer::Plugin::DBIC 'schema';
+
+use Time::Piece; # for OO localtime
+
+use base 'Exporter';
+our @EXPORT = ();
+our @EXPORT_OK = qw/update_stats/;
+our %EXPORT_TAGS = (all => \@EXPORT_OK);
+
+=head1 NAME
+
+App::Netdisco::Util::Statistics
+
+=head1 DESCRIPTION
+
+Update the Netdisco statistics.
+
+There are no default exports, however the C<:all> tag will export all
+subroutines.
+
+=head1 EXPORT_OK
+
+=head2 update_stats()
+
+Update the Netdisco statistics, either new for today or updating today's
+figures.
+
+=cut
+
+sub update_stats {
+  my $schema = schema('netdisco');
+  eval { require SNMP::Info };
+  my $snmpinfo_ver = ($@ ? 'n/a' : $SNMP::Info::VERSION);
+
+  # TODO: (when we have the capabilities table?)
+  #  $stats{waps} = sql_scalar('device',['COUNT(*)'], {"model"=>"AIR%"});
+
+  $schema->txn_do(sub {
+    $schema->resultset('Statistics')->update_or_create({
+      day => localtime->ymd,
+
+      device_count =>
+        $schema->resultset('Device')->count_rs->as_query,
+      device_ip_count =>
+        $schema->resultset('DeviceIp')->count_rs->as_query,
+      device_link_count =>
+        $schema->resultset('Virtual::DeviceLinks')
+          ->count_rs({'me.left_ip' => {'>', \'me.right_ip'}})->as_query,
+      device_port_count =>
+        $schema->resultset('DevicePort')->count_rs->as_query,
+      device_port_up_count =>
+        $schema->resultset('DevicePort')->count_rs({up => 'up'})->as_query,
+      ip_table_count =>
+        $schema->resultset('NodeIp')->count_rs->as_query,
+      ip_active_count =>
+        $schema->resultset('NodeIp')->search({-bool => 'active'},
+          {columns => 'ip', distinct => 1})->count_rs->as_query,
+      node_table_count =>
+        $schema->resultset('Node')->count_rs->as_query,
+      node_active_count =>
+        $schema->resultset('Node')->search({-bool => 'active'},
+          {columns => 'mac', distinct => 1})->count_rs->as_query,
+
+      netdisco_ver => pretty_version($App::Netdisco::VERSION, 3),
+      snmpinfo_ver => $snmpinfo_ver,
+      schema_ver   => $schema->schema_version,
+      perl_ver     => pretty_version($], 3),
+      pg_ver       =>
+        pretty_version($schema->storage->dbh->{pg_server_version}, 2),
+
+    }, { key => 'primary' });
+  });
+}
+
+# take perl or pg versions and make pretty
+sub pretty_version {
+  my ($version, $seglen) = @_;
+  return unless $version and $seglen;
+  return $version if $version !~ m/^[0-9.]+$/;
+  $version =~ s/\.//g;
+  $version = (join '.', reverse map {scalar reverse}
+    unpack("(A${seglen})*", reverse $version));
+  $version =~ s/\.000/.0/g;
+  $version =~ s/\.0+([1-9]+)/.$1/g;
+  return $version;
+}
+
+true;