Refactor munin command to use Mojo::Pg

This commit is contained in:
Daniel Friesel 2019-04-17 13:36:01 -04:00
parent 8ad05720a2
commit 065c844fde

View file

@ -8,37 +8,40 @@ has description => 'Generate statistics for munin-node';
has usage => sub { shift->extract_usage }; has usage => sub { shift->extract_usage };
sub query_to_munin { sub query_to_munin {
my ( $label, $query, @args ) = @_; my ( $label, $value ) = @_;
$query->execute(@args); if ( defined $value ) {
my $rows = $query->fetchall_arrayref; printf( "%s.value %d\n", $label, $value );
if ( @{$rows} ) {
printf( "%s.value %d\n", $label, $rows->[0][0] );
} }
} }
sub run { sub run {
my ($self, $filename) = @_; my ( $self, $filename ) = @_;
my $dbh = $self->app->dbh; my $db = $self->app->pg->db;
my $now = DateTime->now( time_zone => 'Europe/Berlin' ); my $now = DateTime->now( time_zone => 'Europe/Berlin' );
my $checkin_window_query my $checkin_window_query
= $dbh->prepare( = qq{select count(*) as count from user_actions where action_id = 1 and action_time > to_timestamp(?);};
qq{select count(*) from user_actions where action_id = 1 and action_time > to_timestamp(?);}
);
query_to_munin( 'reg_user_count', query_to_munin( 'reg_user_count',
$dbh->prepare(qq{select count(*) from users where status = 1;}) ); $db->select( 'users', 'count(*) as count', { status => 1 } )
query_to_munin( 'checkins_24h', $checkin_window_query, ->hash->{count} );
$now->subtract( hours => 24 )->epoch ); query_to_munin(
query_to_munin( 'checkins_7d', $checkin_window_query, 'checkins_24h',
$now->subtract( days => 7 )->epoch ); $db->query( $checkin_window_query,
query_to_munin( 'checkins_30d', $checkin_window_query, $now->subtract( hours => 24 )->epoch )->hash->{count}
$now->subtract( days => 30 )->epoch ); );
query_to_munin( 'checkins_7d',
$dbh->disconnect; $db->query( $checkin_window_query, $now->subtract( days => 7 )->epoch )
->hash->{count} );
query_to_munin(
'checkins_30d',
$db->query(
$checkin_window_query, $now->subtract( days => 30 )->epoch
)->hash->{count}
);
} }
1; 1;