From 2332b2279d8a8599b4f041370315edc9544b1560 Mon Sep 17 00:00:00 2001
From: tbrehm <t.brehm@ispconfig.org>
Date: Fri, 02 Aug 2013 10:48:38 -0400
Subject: [PATCH] Added support for mongodb. Many thanks to MaddinXx for the patch. http://www.howtoforge.com/forums/showthread.php?t=62691

---
 server/mods-available/rescue_core_module.inc.php |  175 +++++++++++++++++++++++++++++++++++++++++-----------------
 1 files changed, 123 insertions(+), 52 deletions(-)

diff --git a/server/mods-available/rescue_core_module.inc.php b/server/mods-available/rescue_core_module.inc.php
index 49012d5..050d32f 100644
--- a/server/mods-available/rescue_core_module.inc.php
+++ b/server/mods-available/rescue_core_module.inc.php
@@ -35,7 +35,7 @@
 	/* No actions at this time. maybe later... */
 	var $actions_available = array();
 	/**
-	 * The monitoring-Data of this module. 
+	 * The monitoring-Data of this module.
 	 * [0] are the actual data, [1] are the data 1 minnute ago [2] are teh data 2 minuntes...
 	 */
 	private $_monitoringData = array();
@@ -73,7 +73,7 @@
 		 * do nothing, if the rescue-system is not enabled
 		 */
 		global $conf;
-		if ((!isset($conf['serverconfig']['rescue']['try_rescue'])) || 
+		if ((!isset($conf['serverconfig']['rescue']['try_rescue'])) ||
 				((isset($conf['serverconfig']['rescue']['try_rescue'])) && ($conf['serverconfig']['rescue']['try_rescue'] !='y'))){
 			return;
 		}
@@ -87,17 +87,22 @@
 		 * Next we get the rescue data needed for rescuing the system
 		 */
 		$this->_rescueData = $this->_getRescueData();
-		
+
+		/*
+		 * rescue MongoDB if needed
+		 */
+		$this->_rescueMongoDB();
+
 		/*
 		 * rescue mysql if needed (maybe httpd depends on mysql, so try this first!)
 		 */
 		$this->_rescueMySql();
-		
+
 		/*
 		 * rescue httpd if needed
 		 */
 		$this->_rescueHttpd();
-		
+
 		/*
 		 * The last step is to save the rescue-data
 		 */
@@ -112,12 +117,12 @@
 	 */
 	private function _getMonitoringData() {
 		global $app;
-		
+
 		$dataFilename = dirname(__FILE__) . "/../temp/rescue_module_monitoringdata.ser.txt";
-		
+
 		/*
 		 * If the file containing the data is too old (older than 5 minutes) it is better to
-		 * delete it, because it could be, that the server was down for some times and so the data 
+		 * delete it, because it could be, that the server was down for some times and so the data
 		 * are outdated
 		 */
 		if (file_exists($dataFilename) && (filemtime($dataFilename) < (time() - 5 * 60))) {
@@ -132,7 +137,7 @@
 		} else {
 			$data = array();
 		}
-		
+
 		/*
 		 * $temp[0] was the data of the last monitoring (means 1 minute ago), $temp[1] is the data
 		 * 2 minutes ago and so on. Now we have make place for the newest data...
@@ -144,32 +149,32 @@
 		for ($i = $max; $i > 0; $i--){
 			$data[$i] = $data[$i -1];
 		}
-		
+
 		/*
 		 * we need the monitoring tools
 		 */
 		$app->load('monitor_tools');
 		$tools = new monitor_tools();
-		
+
 		/*
 		 * Call the needed Monitoring-step and get the data
 		 */
 		$tmp[0] = $tools->monitorServices();
-		
+
 		/* Add the data at the FIRST position of the history */
 		$data[0] = $tmp;
-		
+
 		/*
 		 * We have the newest monitoring data. Save it!
 		 * (and protect it, because there may be sensible data in it)
 		 */
 		file_put_contents($dataFilename, serialize($data));
 		chmod($dataFilename, 0600);
-						
+
 		/* Thats it */
 		return $data;
 	}
-	
+
 	/**
 	 * This gets the rescue-Data, needed for rescuing the system.
 	 * Because we can not be 100% sure, that the mysql-DB is up and running, so we use the
@@ -181,7 +186,7 @@
 
 		/*
 		 * If the file containing the data is too old (older than 5 minutes) it is better to
-		 * delete it, because it could be, that the server was down for some times and so the data 
+		 * delete it, because it could be, that the server was down for some times and so the data
 		 * are outdated
 		 */
 		if (file_exists($dataFilename) && (filemtime($dataFilename) < (time() - 5 * 60))) {
@@ -196,7 +201,7 @@
 		} else {
 			$data = array();
 		}
-		
+
 		/* Thats it */
 		return $data;
 	}
@@ -222,14 +227,14 @@
 	 */
 	private function _rescueHttpd(){
 		global $app, $conf;
-		
+
 		/*
 		 * do nothing, if it is not allowed to rescue httpd
 		 */
 		if ((isset($conf['serverconfig']['rescue']['do_not_try_rescue_httpd']) && ($conf['serverconfig']['rescue']['do_not_try_rescue_httpd']) == 'y')){
 			return;
 		}
-		
+
 		/*
 		 * if the service is up and running, or the service is not installed there is nothing to do...
 		 */
@@ -238,42 +243,42 @@
 			$this->_rescueData['webserver']['try_counter'] = 0;
 			return;
 		}
-		
+
 		/*
 		 * OK, the service is installed and down.
 		 * Maybe this is because of a restart of the service by the admin.
 		 * This means, we check the data 1 minute ago
 		 */
-		if ((!isset($this->_monitoringData[1][0]['data']['webserver'])) || 
+		if ((!isset($this->_monitoringData[1][0]['data']['webserver'])) ||
 				((isset($this->_monitoringData[1][0]['data']['webserver'])) && ($this->_monitoringData[1][0]['data']['webserver'] != 0))){
-			/* 
-			 * We do NOT have this data or we have this data, but the webserver was not down 1 minute ago. 
-			 * This means, it could be, that the admin is restarting the server. 
+			/*
+			 * We do NOT have this data or we have this data, but the webserver was not down 1 minute ago.
+			 * This means, it could be, that the admin is restarting the server.
 			 * We wait one more minute...
 			 */
 			return;
 		}
-		
+
 		/*#####
 		 * The service is down and it was down 1 minute ago.
 		 * We try to rescue it
 		 *#####*/
-		
+
 		/* Get the try counter */
 		$tryCount = (!isset($this->_rescueData['webserver']['try_counter']))? 1 : $this->_rescueData['webserver']['try_counter'] + 1;
-		
+
 		/* Set the new try counter */
 		$this->_rescueData['webserver']['try_counter'] = $tryCount;
-		
+
 		/* if 5 times will not work, we have to give up... */
 		if ($tryCount > 5){
 			$app->log('httpd is down! Rescue will not help!', LOGLEVEL_ERROR);
 			return;
 		}
-		
-		
+
+
 		$app->log('httpd is down! Try rescue httpd (try:' . $tryCount . ')...', LOGLEVEL_WARN);
-		
+
 		if($conf['serverconfig']['web']['server_type'] == 'nginx'){
 			$daemon = 'nginx';
 		} else {
@@ -285,23 +290,89 @@
 				$daemon = 'apache2';
 			}
 		}
-		
+
 		$this->_rescueDaemon($daemon);
 	}
-	
+
+	/**
+	 * restarts MongoDB, if needed
+	 */
+	private function _rescueMongoDB(){
+		global $app, $conf;
+
+		/*
+		 * do nothing, if it is not allowed to rescue mysql
+		 */
+		if ((isset($conf['serverconfig']['rescue']['do_not_try_rescue_mongodb']) && ($conf['serverconfig']['rescue']['do_not_try_rescue_mongodb']) == 'y')){
+			return;
+		}
+
+		/*
+		 * if the service is up and running, or the service is not installed there is nothing to do...
+		 */
+		if ($this->_monitoringData[0][0]['data']['mongodbserver'] != 0){
+			/* Clear the try counter, because we do not have to try to rescue the service */
+			$this->_rescueData['mongodbserver']['try_counter'] = 0;
+			return;
+		}
+
+		/*
+		 * OK, the service is installed and down.
+		 * Maybe this is because of a restart of the service by the admin.
+		 * This means, we check the data 1 minute ago
+		 */
+		if ((!isset($this->_monitoringData[1][0]['data']['mongodbserver'])) ||
+				((isset($this->_monitoringData[1][0]['data']['mongodbserver'])) && ($this->_monitoringData[1][0]['data']['mongodbserver'] != 0))){
+			/*
+			 * We do NOT have this data or we have this data, but the webserver was not down 1 minute ago.
+			 * This means, it could be, that the admin is restarting the server.
+			 * We wait one more minute...
+			 */
+			return;
+		}
+
+		/*#####
+		 * The service is down and it was down 1 minute ago.
+		 * We try to rescue it
+		 *#####*/
+
+		/* Get the try counter */
+		$tryCount = (!isset($this->_rescueData['mongodbserver']['try_counter']))? 1 : $this->_rescueData['mongodbserver']['try_counter'] + 1;
+
+		/* Set the new try counter */
+		$this->_rescueData['mongodbserver']['try_counter'] = $tryCount;
+
+		/* if 5 times will not work, we have to give up... */
+		if ($tryCount > 5){
+			$app->log('MongoDB is down! Rescue will not help!', LOGLEVEL_ERROR);
+			return;
+		}
+
+
+		$app->log('MongoDB is down! Try rescue MongoDB (try:' . $tryCount . ')...', LOGLEVEL_WARN);
+
+		if(is_file($conf['init_scripts'] . '/' . 'mongodb')) {
+			$daemon = 'mongodb';
+		} else {
+			$daemon = 'mongodb';
+		}
+
+		$this->_rescueDaemon($daemon);
+	}
+
 	/**
 	 * restarts mysql, if needed
 	 */
 	private function _rescueMySql(){
 		global $app, $conf;
-		
+
 		/*
 		 * do nothing, if it is not allowed to rescue mysql
 		 */
 		if ((isset($conf['serverconfig']['rescue']['do_not_try_rescue_mysql']) && ($conf['serverconfig']['rescue']['do_not_try_rescue_mysql']) == 'y')){
 			return;
 		}
-		
+
 		/*
 		 * if the service is up and running, or the service is not installed there is nothing to do...
 		 */
@@ -310,40 +381,40 @@
 			$this->_rescueData['mysqlserver']['try_counter'] = 0;
 			return;
 		}
-		
+
 		/*
 		 * OK, the service is installed and down.
 		 * Maybe this is because of a restart of the service by the admin.
 		 * This means, we check the data 1 minute ago
 		 */
-		if ((!isset($this->_monitoringData[1][0]['data']['mysqlserver'])) || 
+		if ((!isset($this->_monitoringData[1][0]['data']['mysqlserver'])) ||
 				((isset($this->_monitoringData[1][0]['data']['mysqlserver'])) && ($this->_monitoringData[1][0]['data']['mysqlserver'] != 0))){
-			/* 
-			 * We do NOT have this data or we have this data, but the webserver was not down 1 minute ago. 
-			 * This means, it could be, that the admin is restarting the server. 
+			/*
+			 * We do NOT have this data or we have this data, but the webserver was not down 1 minute ago.
+			 * This means, it could be, that the admin is restarting the server.
 			 * We wait one more minute...
 			 */
 			return;
 		}
-		
+
 		/*#####
 		 * The service is down and it was down 1 minute ago.
 		 * We try to rescue it
 		 *#####*/
-		
+
 		/* Get the try counter */
 		$tryCount = (!isset($this->_rescueData['mysqlserver']['try_counter']))? 1 : $this->_rescueData['mysqlserver']['try_counter'] + 1;
-		
+
 		/* Set the new try counter */
 		$this->_rescueData['mysqlserver']['try_counter'] = $tryCount;
-		
+
 		/* if 5 times will not work, we have to give up... */
 		if ($tryCount > 5){
 			$app->log('MySQL is down! Rescue will not help!', LOGLEVEL_ERROR);
 			return;
 		}
-		
-		
+
+
 		$app->log('MySQL is down! Try rescue mysql (try:' . $tryCount . ')...', LOGLEVEL_WARN);
 
 		if(is_file($conf['init_scripts'] . '/' . 'mysqld')) {
@@ -351,23 +422,23 @@
 		} else {
 			$daemon = 'mysql';
 		}
-		
+
 		$this->_rescueDaemon($daemon);
 	}
 
 	/**
 	 * Tries to stop and then restart the daemon
-	 * 
+	 *
 	 * @param type $daemon the name of the daemon
 	 */
 	private function _rescueDaemon($daemon){
 		global $conf;
-		
+
 		// if you need to find all restarts search for "['init_scripts']"
 		/*
 		 * First we stop the running service "normally"
 		 */
-		
+
 		/*
 		 * ATTENTION!
 		 * The service hangs. this means it could be, that "stop" will hang also.
@@ -375,13 +446,13 @@
 		 * of the service
 		 */
 		exec($conf['init_scripts'] . '/' . $daemon . ' stop && (sleep 3; kill $!; sleep 2; kill -9 $!) &> /dev/null');
-		
+
 		/*
 		 * OK, we tryed to stop it normally, maybe this worked maybe not. So we have to look
 		 * if the service is already running or not. If so, we have to kill them hard
 		 */
 		exec("kill -9 `ps -A | grep " . $daemon . "| grep -v grep | awk '{print $1}'` &> /dev/null");
-		
+
 		/*
 		 * There are no more zombies left. Lets start the service..
 		 */

--
Gitblit v1.9.1