include/poller.php

   1 <?php
   2 if (!file_exists("boot.php") AND (sizeof($_SERVER["argv"]) != 0)) {
   3         $directory = dirname($_SERVER["argv"][0]);
   4
   5         if (substr($directory, 0, 1) != "/")
   6                 $directory = $_SERVER["PWD"]."/".$directory;
   7
   8         $directory = realpath($directory."/..");
   9
  10         chdir($directory);
  11 }
  12
  13 require_once("boot.php");
  14
  15 function poller_run(&$argv, &$argc){
  16         global $a, $db;
  17
  18         if(is_null($a)) {
  19                 $a = new App;
  20         }
  21
  22         if(is_null($db)) {
  23                 @include(".htconfig.php");
  24                 require_once("include/dba.php");
  25                 $db = new dba($db_host, $db_user, $db_pass, $db_data);
  26                 unset($db_host, $db_user, $db_pass, $db_data);
  27         };
  28
  29         $load = current_load();
  30         if($load) {
  31                 $maxsysload = intval(get_config('system','maxloadavg'));
  32                 if($maxsysload < 1)
  33                         $maxsysload = 50;
  34
  35                 if(intval($load) > $maxsysload) {
  36                         logger('system: load ' . $load . ' too high. poller deferred to next scheduled run.');
  37                         return;
  38                 }
  39         }
  40
  41         // Checking the number of workers
  42         if (poller_too_much_workers(1))
  43                 return;
  44
  45         if(($argc <= 1) OR ($argv[1] != "no_cron")) {
  46                 // Run the cron job that calls all other jobs
  47                 proc_run("php","include/cron.php");
  48
  49                 // Run the cronhooks job separately from cron for being able to use a different timing
  50                 proc_run("php","include/cronhooks.php");
  51
  52                 // Cleaning dead processes
  53                 $r = q("SELECT DISTINCT(`pid`) FROM `workerqueue` WHERE `executed` != '0000-00-00 00:00:00'");
  54                 foreach($r AS $pid)
  55                         if (!posix_kill($pid["pid"], 0))
  56                                 q("UPDATE `workerqueue` SET `executed` = '0000-00-00 00:00:00', `pid` = 0 WHERE `pid` = %d",
  57                                         intval($pid["pid"]));
  58                         else {
  59                                 /// @TODO Kill long running processes
  60                                 /// But: Update processes (like the database update) mustn't be killed
  61                         }
  62
  63         } else
  64                 // Sleep four seconds before checking for running processes again to avoid having too many workers
  65                 sleep(4);
  66
  67         // Checking number of workers
  68         if (poller_too_much_workers(2))
  69                 return;
  70
  71         $starttime = time();
  72
  73         while ($r = q("SELECT * FROM `workerqueue` WHERE `executed` = '0000-00-00 00:00:00' ORDER BY `created` LIMIT 1")) {
  74
  75                 // Count active workers and compare them with a maximum value that depends on the load
  76                 if (poller_too_much_workers(3))
  77                         return;
  78
  79                 q("UPDATE `workerqueue` SET `executed` = '%s', `pid` = %d WHERE `id` = %d AND `executed` = '0000-00-00 00:00:00'",
  80                         dbesc(datetime_convert()),
  81                         intval(getmypid()),
  82                         intval($r[0]["id"]));
  83
  84                 // Assure that there are no tasks executed twice
  85                 $id = q("SELECT `id` FROM `workerqueue` WHERE `id` = %d AND `pid` = %d",
  86                         intval($r[0]["id"]),
  87                         intval(getmypid()));
  88                 if (!$id) {
  89                         logger("Queue item ".$r[0]["id"]." was executed multiple times - skip this execution", LOGGER_DEBUG);
  90                         continue;
  91                 }
  92
  93                 $argv = json_decode($r[0]["parameter"]);
  94
  95                 $argc = count($argv);
  96
  97                 // Check for existance and validity of the include file
  98                 $include = $argv[0];
  99
 100                 if (!validate_include($include)) {
 101                         logger("Include file ".$argv[0]." is not valid!");
 102                         q("DELETE FROM `workerqueue` WHERE `id` = %d", intval($r[0]["id"]));
 103                         continue;
 104                 }
 105
 106                 require_once($include);
 107
 108                 $funcname=str_replace(".php", "", basename($argv[0]))."_run";
 109
 110                 if (function_exists($funcname)) {
 111                         logger("Process ".getmypid()." - ID ".$r[0]["id"].": ".$funcname." ".$r[0]["parameter"]);
 112                         $funcname($argv, $argc);
 113
 114                         logger("Process ".getmypid()." - ID ".$r[0]["id"].": ".$funcname." - done");
 115
 116                         q("DELETE FROM `workerqueue` WHERE `id` = %d", intval($r[0]["id"]));
 117                 } else
 118                         logger("Function ".$funcname." does not exist");
 119
 120                 // Quit the poller once every hour
 121                 if (time() > ($starttime + 3600))
 122                         return;
 123         }
 124
 125 }
 126
 127 function poller_too_much_workers($stage) {
 128
 129         $queues = get_config("system", "worker_queues");
 130
 131         if ($queues == 0)
 132                 $queues = 4;
 133
 134         $active = poller_active_workers();
 135
 136         // Decrease the number of workers at higher load
 137         $load = current_load();
 138         if($load) {
 139                 $maxsysload = intval(get_config('system','maxloadavg'));
 140                 if($maxsysload < 1)
 141                         $maxsysload = 50;
 142
 143                 $maxworkers = $queues;
 144
 145                 // Some magical mathemathics to reduce the workers
 146                 $exponent = 3;
 147                 $slope = $maxworkers / pow($maxsysload, $exponent);
 148                 $queues = ceil($slope * pow(max(0, $maxsysload - $load), $exponent));
 149
 150                 logger("Current load stage ".$stage.": ".$load." - maximum: ".$maxsysload." - current queues: ".$active." - maximum: ".$queues, LOGGER_DEBUG);
 151
 152         }
 153
 154         return($active >= $queues);
 155 }
 156
 157 function poller_active_workers() {
 158         $workers = q("SELECT COUNT(*) AS `workers` FROM `workerqueue` WHERE `executed` != '0000-00-00 00:00:00'");
 159
 160         return($workers[0]["workers"]);
 161 }
 162
 163 if (array_search(__file__,get_included_files())===0){
 164   poller_run($_SERVER["argv"],$_SERVER["argc"]);
 165   killme();
 166 }
 167 ?>