updated

Author: gbxyz

bin/stats.pl

@@ -1,23 +1,29 @@
 #!/usr/bin/env perl
-use Data::Printer;
 use DBI;
+use Fcntl qw(:flock);
 use HTTP::Request::Common qw(GET DELETE);
 use JSON::XS;
 use LWP::UserAgent;
 use constant URL => q{https://rdap.org/stats};
 use vars qw(@SERIES $DB);
 use common::sense;
 
+if (!flock(DATA, LOCK_EX | LOCK_NB)) {
+    say STDERR q{waiting for existing process to finish...};
+    flock(DATA, LOCK_EX);
+}
+
 @SERIES = qw(status type user_agent network);
 
+say STDERR q{opening database connection...};
 my $DB = DBI->connect(sprintf(
     q{dbi:SQLite:dbname=%s},
     $ARGV[0] || q{./stats.db}
 ));
 
 $DB->do(q{CREATE TABLE IF NOT EXISTS total_queries (
     id          INTEGER PRIMARY KEY,
-    timestamp   INTEGER,
+    timestamp   INTEGER UNIQUE,
     count       INTEGER
 )});
 
@@ -28,6 +34,13 @@
         `%s`        TEXT,
         `count`     INTEGER
     )}, $column, $column));
+
+    $DB->do(sprintf(
+        q{CREATE UNIQUE INDEX IF NOT EXISTS `queries_by_%s_index` ON `queries_by_%s`(`timestamp`, `%s`)},
+        $column,
+        $column,
+        $column,
+    ));
 }
 
 my $ua = LWP::UserAgent->new;
@@ -38,6 +51,9 @@
 
 my $req1 = GET(URL);
 $req1->header(authorization => sprintf(q{Bearer %s}, $ENV{STATS_TOKEN}));
+
+say STDERR q{getting stats...};
+
 my $res1 = $ua->request($req1);
 
 die($res1->status_line) unless ($res1->is_success);
@@ -48,6 +64,7 @@
 my $req2 = DELETE(URL);
 $req2->header(authorization => sprintf(q{Bearer %s}, $ENV{STATS_TOKEN}));
 
+say STDERR q{purging stats...};
 my $res2 = $ua->request($req2);
 
 die($res2->status_line) unless ($res2->is_success);
@@ -60,6 +77,8 @@
 
 my $timestamp = delete($stats->{timestamp});
 
+say STDERR q{updating database...};
+
 $DB->prepare(q{
     INSERT INTO `total_queries`
     (`timestamp`, `count`)
@@ -90,3 +109,7 @@
         );
     }
 }
+
+say STDERR q{done};
+
+__DATA__