refactor: convert heartbeat event handlers to console commands

JarvusInnovations · Jul 21, 2020 · 2832e23 · 2832e23
1 parent 563f338
commit 2832e23
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 15 deletions.
diff --git a/console-commands/gk/flush-metrics.php b/console-commands/gk/flush-metrics.php
@@ -0,0 +1,9 @@
+<?php
+
+$logger = $_COMMAND['LOGGER'];
+
+$logger->info('Flushing metrics');
+
+$flushed = Gatekeeper\Metrics\Metrics::flushMetricSamples();
+
+$logger->info("{$flushed} metrics flushed");
diff --git a/console-commands/gk/ping-endpoints.php b/console-commands/gk/ping-endpoints.php
@@ -0,0 +1,9 @@
+<?php
+
+$logger = $_COMMAND['LOGGER'];
+
+$logger->info('Pinging endpoints...');
+
+$pinged = Gatekeeper\Endpoints\Pinger::pingOverdueEndpoints($logger);
+
+$logger->info("{$pinged} endpoints pinged");
diff --git a/event-handlers/Gatekeeper/heartbeat/40_flush-metrics.php b/event-handlers/Gatekeeper/heartbeat/40_flush-metrics.php
diff --git a/event-handlers/Gatekeeper/heartbeat/50_ping-endpoints.php b/event-handlers/Gatekeeper/heartbeat/50_ping-endpoints.php
diff --git a/php-classes/Gatekeeper/Endpoints/Pinger.php b/php-classes/Gatekeeper/Endpoints/Pinger.php
@@ -6,14 +6,16 @@
 use Cache;
 use HttpProxy;
 
+use Psr\Log\LoggerInterface;
+
 use Gatekeeper\ApiRequestHandler;
 use Gatekeeper\Alerts\TestFailed;
 use Gatekeeper\Transactions\PingTransaction;
 
 
 class Pinger
 {
-    public static function pingOverdueEndpoints($verbose = false)
+    public static function pingOverdueEndpoints(LoggerInterface $logger = null)
     {
         $endpoints = Endpoint::getAllByWhere('PingFrequency IS NOT NULL');
 
@@ -24,14 +26,16 @@ public static function pingOverdueEndpoints($verbose = false)
                 $lastPing === false
                 || $Endpoint->PingFrequency*60 < time()-$lastPing['time']
             ) {
-                static::pingEndpoint($Endpoint, $verbose);
+                static::pingEndpoint($Endpoint, $logger);
             }
         }
+
+        return count($endpoints);
     }
 
-    public static function pingEndpoint(Endpoint $Endpoint, $verbose = false)
+    public static function pingEndpoint(Endpoint $Endpoint, LoggerInterface $logger = null)
     {
-        $verbose && printf('Testing endpoint: /%s...', $Endpoint->Path) && flush();
+        $logger && $logger->info("/{$Endpoint->Path} is being pinged");
 
 
         // execute and capture request
@@ -108,7 +112,7 @@ public static function pingEndpoint(Endpoint $Endpoint, $verbose = false)
         }
 
 
-        $verbose && printf("%s\n", $testPassed ? 'passed' : 'failed') && flush();
+        $logger && $logger->info("/{$Endpoint->Path} ". ($testPassed ? 'passed' : 'failed'));
 
 
         // cache result and timestamp