Add healthcheck to status page

Author: carlbennett

controllers/Status.php

@@ -2,15 +2,19 @@
 
 namespace BNETDocs\Controllers;
 
+use \BNETDocs\Libraries\Cache;
 use \BNETDocs\Libraries\Common;
 use \BNETDocs\Libraries\Controller;
+use \BNETDocs\Libraries\Database;
+use \BNETDocs\Libraries\DatabaseDriver;
 use \BNETDocs\Libraries\Exceptions\UnspecifiedViewException;
 use \BNETDocs\Libraries\Router;
 use \BNETDocs\Models\Status as StatusModel;
 use \BNETDocs\Views\StatusJSON as StatusJSONView;
 use \BNETDocs\Views\StatusPlain as StatusPlainView;
 use \DateTime;
 use \DateTimeZone;
+use \StdClass;
 
 class Status extends Controller {
 
@@ -26,21 +30,34 @@ public function run(Router &$router) {
         throw new UnspecifiedViewException();
     }
     $model = new StatusModel();
-    $this->getStatus($model);
+    $code  = (!$this->getStatus($model) ? 500 : 200);
     ob_start();
     $view->render($model);
-    $router->setResponseCode(200);
+    $router->setResponseCode($code);
     $router->setResponseTTL(300);
     $router->setResponseHeader("Content-Type", $view->getMimeType());
     $router->setResponseContent(ob_get_contents());
     ob_end_clean();
   }
 
   protected function getStatus(StatusModel &$model) {
+    if (!isset(Common::$database)) {
+      Common::$database = DatabaseDriver::getDatabaseObject();
+    }
+
+    $healthcheck           = new StdClass();
+    $healthcheck->database = (Common::$database instanceof Database);
+    $healthcheck->memcache = (Common::$cache    instanceof Cache   );
+
+    $model->healthcheck    = $healthcheck;
     $model->remote_address = getenv("REMOTE_ADDR");
     $model->remote_geoinfo = geoip_record_by_name($model->remote_address);
     $model->timestamp      = new DateTime("now", new DateTimeZone("UTC"));
     $model->version_info   = Common::$version;
+
+    foreach ($healthcheck as $key => $val) {
+      if (is_bool($val) && !$val) return false;
+    }
     return true;
   }
 

views/StatusJSON.php

@@ -20,6 +20,7 @@ public function render(Model &$model) {
       throw new IncorrectModelException();
     }
     echo json_encode([
+      "healthcheck"    => $model->healthcheck,
       "remote_address" => $model->remote_address,
       "remote_geoinfo" => $model->remote_geoinfo,
       "timestamp"      => $model->timestamp->format("r"),

views/StatusPlain.php

@@ -19,6 +19,16 @@ public function render(Model &$model) {
     if (!$model instanceof StatusModel) {
       throw new IncorrectModelException();
     }
+    foreach ($model->healthcheck as $key => $val) {
+      if (is_bool($val))
+        echo "healthcheck_" . $key . " " . ($val ? "true" : "false") . "\n";
+      else if (is_null($val))
+        echo "healthcheck_" . $key . " null\n";
+      else if (is_scalar($val))
+        echo "healthcheck_" . $key . " " . $val . "\n";
+      else
+        echo "healthcheck_" . $key . " " . gettype($val) . "\n";
+    }
     echo "remote_address " . $model->remote_address . "\n";
     if ($model->remote_geoinfo) {
       foreach ($model->remote_geoinfo as $key => $val) {