Merge pull request #41 from statusengine/ITC-3481

ITC-3481 CustomalertModule: Statusengine, don't truncate whole *stat…

Author: nook24

etc/config.yml.example

@@ -363,3 +363,9 @@ age_perfdata: 90
 # Enable (1) this option to clear proxy environment variables (For Statusengine only)
 # Disable (0) and Statusengine will use the proxy out of your environment
 disable_http_proxy: 1
+
+# This option is only for users of openITCOCKPIT (https://openitcockpit.io)
+# It enables specific performance optimizations, that can only be used with openITCOCKPIT.
+# Enable (1) If Statusengine is running with openITCOCKPIT
+# Disable (0) you are a happy user of standalone Statusengine :)
+is_openitcockpit: 0

src/Backends/MySQL/MySQL.php

@@ -231,15 +231,15 @@ public function deleteNodeByName($nodeName) {
     }
 
     public function dispatch() {
-            /*$bt = debug_backtrace(DEBUG_BACKTRACE_IGNORE_ARGS);
+        /*$bt = debug_backtrace(DEBUG_BACKTRACE_IGNORE_ARGS);
 
-            if($bt[3]['function'] === 'forkServicestatusChild'){
-              // echo "============ START ========== \n";
-              // foreach($bt as $line){
-              //     printf("File: %s Line: %s function: %s\n", $line['file'], $line['line'], $line['function']);
-              // }
-              // echo "============ END ========== \n\n";
-            }*/
+        if($bt[3]['function'] === 'forkServicestatusChild'){
+          // echo "============ START ========== \n";
+          // foreach($bt as $line){
+          //     printf("File: %s Line: %s function: %s\n", $line['file'], $line['line'], $line['function']);
+          // }
+          // echo "============ END ========== \n\n";
+        }*/
 
 
         if ($this->BulkInsertObjectStore->hasRaisedTimeout()) {
@@ -672,10 +672,18 @@ public function getServiceScheduleddowntimeBackend() {
     public function monitoringengineWasRestarted() {
         $this->connect();
         $Hoststatus = new MysqlHoststatus($this, $this->BulkInsertObjectStore, $this->nodeName);
-        $Hoststatus->truncate();
+        if ($this->Config->isOpenITCOCKPIT() === true) {
+            $Hoststatus->truncateForOpenITCOCKPIT();
+        } else {
+            $Hoststatus->truncate();
+        }
 
         $Servicestatus = new MysqlServicestatus($this, $this->BulkInsertObjectStore, $this->nodeName);
-        $Servicestatus->truncate();
+        if ($this->Config->isOpenITCOCKPIT() === true) {
+            $Servicestatus->truncateForOpenITCOCKPIT();
+        } else {
+            $Servicestatus->truncate();
+        }
         $this->disconnect();
     }
 

src/Backends/MySQL/SqlObjects/MysqlHoststatus.php

@@ -152,4 +152,20 @@ public function truncate($isRecursion = false){
         }
     }
 
+    /**
+     * @param bool $isRecursion
+     * @return bool
+     */
+    public function truncateForOpenITCOCKPIT($isRecursion = false){
+        $query = $this->MySQL->prepare('DELETE FROM statusengine_hoststatus WHERE NOT EXISTS (SELECT hosts.uuid FROM hosts WHERE statusengine_hoststatus.hostname = hosts.uuid AND hosts.disabled=0)');
+        try {
+            return $this->MySQL->executeQuery($query, 'MysqlHoststatus');
+        } catch (StorageBackendUnavailableExceptions $Exceptions) {
+            //Retry
+            if ($isRecursion === false) {
+                $this->insert(true);
+            }
+        }
+    }
+
 }

src/Backends/MySQL/SqlObjects/MysqlServicestatus.php

@@ -61,7 +61,7 @@ class MysqlServicestatus extends MysqlModel {
      * @param BulkInsertObjectStore $BulkInsertObjectStore
      * @param string $nodeName
      */
-    public function __construct(MySQL $MySQL, BulkInsertObjectStore $BulkInsertObjectStore, $nodeName){
+    public function __construct(MySQL $MySQL, BulkInsertObjectStore $BulkInsertObjectStore, $nodeName) {
         $this->MySQL = $MySQL;
         $this->BulkInsertObjectStore = $BulkInsertObjectStore;
         $this->nodeName = $nodeName;
@@ -71,7 +71,7 @@ public function __construct(MySQL $MySQL, BulkInsertObjectStore $BulkInsertObjec
      * @param bool $isRecursion
      * @return bool
      */
-    public function insert($isRecursion = false){
+    public function insert($isRecursion = false) {
         /**
          * @var Servicestatus $Servicestatus
          */
@@ -142,7 +142,7 @@ public function insert($isRecursion = false){
      * @param bool $isRecursion
      * @return bool
      */
-    public function truncate($isRecursion = false){
+    public function truncate($isRecursion = false) {
         $query = $this->MySQL->prepare('DELETE FROM statusengine_servicestatus WHERE node_name=?');
         $query->bindValue(1, $this->nodeName);
         try {
@@ -155,4 +155,17 @@ public function truncate($isRecursion = false){
         }
     }
 
+    public function truncateForOpenITCOCKPIT($isRecursion = false) {
+        $query = $this->MySQL->prepare('DELETE FROM statusengine_servicestatus WHERE NOT EXISTS (SELECT services.uuid FROM services WHERE statusengine_servicestatus.service_description = services.uuid AND services.disabled=0)');
+        try {
+            return $this->MySQL->executeQuery($query, 'MysqlServicestatus');
+        } catch (StorageBackendUnavailableExceptions $Exceptions) {
+            //Retry
+            if ($isRecursion === false) {
+                $this->insert(true);
+            }
+        }
+    }
+
+
 }

src/Config.php

@@ -959,4 +959,12 @@ public function isDumpOfMysqlQueryParametersEnabled() {
         return $default;
     }
 
+    public function isOpenITCOCKPIT() {
+        $default = false;
+        $default = Env::get('SE_IS_OPENITCOCKPIT', $default, Env::VALUE_BOOL);
+        if (isset($this->config['is_openitcockpit'])) {
+            return (bool)$this->config['is_openitcockpit'];
+        }
+        return $default;
+    }
 }