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         if(function_exists('sys_getloadavg')) {
  30                 $maxsysload = intval(get_config('system','maxloadavg'));
  31                 if($maxsysload < 1)
  32                         $maxsysload = 50;
  33
  34                 $load = sys_getloadavg();
  35                 if(intval($load[0]) > $maxsysload) {
  36                         logger('system: load ' . $load[0] . ' too high. poller deferred to next scheduled run.');
  37                         return;
  38                 }
  39         }
  40
  41         if(($argc <= 1) OR ($argv[1] != "no_cron")) {
  42                 // Run the cron job that calls all other jobs
  43                 proc_run("php","include/cron.php");
  44
  45                 // Run the cronhooks job separately from cron for being able to use a different timing
  46                 proc_run("php","include/cronhooks.php");
  47
  48                 // Cleaning dead processes
  49                 $r = q("SELECT DISTINCT(`pid`) FROM `workerqueue` WHERE `executed` != '0000-00-00 00:00:00'");
  50                 foreach($r AS $pid)
  51                         if (!posix_kill($pid["pid"], 0))
  52                                 q("UPDATE `workerqueue` SET `executed` = '0000-00-00 00:00:00', `pid` = 0 WHERE `pid` = %d",
  53                                         intval($pid["pid"]));
  54
  55         } else
  56                 // Sleep two seconds before checking for running processes to avoid having too many workers
  57                 sleep(4);
  58
  59         // Checking number of workers
  60         if (poller_too_much_workers())
  61                 return;
  62
  63         $starttime = time();
  64
  65         while ($r = q("SELECT * FROM `workerqueue` WHERE `executed` = '0000-00-00 00:00:00' ORDER BY `created` LIMIT 1")) {
  66
  67                 // Quit the poller once every hour
  68                 if (time() > ($starttime + 3600))
  69                         return;
  70
  71                 // Count active workers and compare them with a maximum value that depends on the load
  72                 if (poller_too_much_workers())
  73                         return;
  74
  75                 q("UPDATE `workerqueue` SET `executed` = '%s', `pid` = %d WHERE `id` = %d",
  76                         dbesc(datetime_convert()),
  77                         intval(getmypid()),
  78                         intval($r[0]["id"]));
  79
  80                 $argv = json_decode($r[0]["parameter"]);
  81
  82                 $argc = count($argv);
  83
  84                 // Check for existance and validity of the include file
  85                 $include = $argv[0];
  86
  87                 if (!validate_include($include)) {
  88                         logger("Include file ".$argv[0]." is not valid!");
  89                         q("DELETE FROM `workerqueue` WHERE `id` = %d", intval($r[0]["id"]));
  90                         continue;
  91                 }
  92
  93                 require_once($include);
  94
  95                 $funcname=str_replace(".php", "", basename($argv[0]))."_run";
  96
  97                 if (function_exists($funcname)) {
  98                         logger("Process ".getmypid().": ".$funcname." ".$r[0]["parameter"]);
  99                         $funcname($argv, $argc);
 100
 101                         logger("Process ".getmypid().": ".$funcname." - done");
 102
 103                         q("DELETE FROM `workerqueue` WHERE `id` = %d", intval($r[0]["id"]));
 104                 } else
 105                         logger("Function ".$funcname." does not exist");
 106         }
 107
 108 }
 109
 110 function poller_too_much_workers() {
 111
 112         $queues = get_config("system", "worker_queues");
 113
 114         if ($queues == 0)
 115                 $queues = 4;
 116
 117         $active = poller_active_workers();
 118
 119         // Decrease the number of workers at higher load
 120         if(function_exists('sys_getloadavg')) {
 121                 $load = max(sys_getloadavg());
 122
 123                 $maxsysload = intval(get_config('system','maxloadavg'));
 124                 if($maxsysload < 1)
 125                         $maxsysload = 50;
 126
 127                 $queues = max(0, ceil($queues * (($maxsysload - $load) / $maxsysload)));
 128
 129                 logger("Current load: ".$load." - maximum: ".$maxsysload." - current queues: ".$active." - maximum: ".$queues, LOGGER_DEBUG);
 130
 131         }
 132
 133         return($active >= $queues);
 134 }
 135
 136 function poller_active_workers() {
 137         $workers = q("SELECT COUNT(*) AS `workers` FROM `workerqueue` WHERE `executed` != '0000-00-00 00:00:00'");
 138
 139         return($workers[0]["workers"]);
 140 }
 141
 142 if (array_search(__file__,get_included_files())===0){
 143   poller_run($_SERVER["argv"],$_SERVER["argc"]);
 144   killme();
 145 }
 146 ?>