408 lines
15 KiB
PHP
Executable file
408 lines
15 KiB
PHP
Executable file
#!/usr/bin/php
|
||
<?php
|
||
|
||
/*
|
||
This program is free software: you can redistribute it and/or modify
|
||
it under the terms of the GNU General Public License as published by
|
||
the Free Software Foundation, either version 3 of the License, or
|
||
(at your option) any later version.
|
||
|
||
This program is distributed in the hope that it will be useful,
|
||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
GNU General Public License for more details.
|
||
|
||
You should have received a copy of the GNU General Public License
|
||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
*/
|
||
|
||
define('N',"\n");
|
||
define('SNAME',basename(__FILE__));
|
||
define('FNAME',preg_replace('/\.[^.]*$/','',SNAME));
|
||
define('CHILD','getinstinfo.php');
|
||
define('LIBDP','/../site/mustard/include');
|
||
|
||
require(__DIR__.LIBDP.'/ght.php');
|
||
|
||
use function mysqli_real_escape_string as myesc;
|
||
|
||
declare(ticks=1);
|
||
if (function_exists('pcntl_signal')) {
|
||
function signalHandler($signal) {
|
||
echo(N);
|
||
mexit('received signal «'.$signal.'», shutting down.'.N,0,true);
|
||
}
|
||
pcntl_signal(SIGTERM,'signalHandler');// Termination ('kill' was called)
|
||
pcntl_signal(SIGHUP,'signalHandler');// Terminal log-out
|
||
pcntl_signal(SIGINT,'signalHandler');// Interrupted (Ctrl-C is pressed)
|
||
}
|
||
|
||
$msglevs=['debug', 'info', 'warning', 'error', 'none'];
|
||
|
||
$opts=[
|
||
'poolsize'=>20,
|
||
'moreclauses'=>'',
|
||
'peersfp'=>null,
|
||
'dontrestore'=>false,
|
||
'ignorelock'=>false,
|
||
'logminmsglev'=>1,
|
||
'tuiminmsglev'=>1
|
||
];
|
||
|
||
$ghtsa=[[' day',' days'],[' hour',' hours'],[' minute',' minutes'],[' second',' seconds']];
|
||
|
||
$help='SYNOPSIS
|
||
|
||
'.SNAME.' [options]
|
||
|
||
DESCRIPTION
|
||
|
||
This script coordinates the parallel execution of a definable number of
|
||
'.CHILD.' processes “against” all the alive instances which are already
|
||
present in mastostart’s database, plus optionally those listed in a
|
||
specifiable file (typically the output file from a peerscrawl.php run).
|
||
|
||
OPTIONS
|
||
|
||
-
|
||
Everything after a single dash will be passed to '.CHILD.' processes as is.
|
||
-p, --peersfp <file>
|
||
Defines the path to a file containing a list of instances to consider in
|
||
addition to those which are already present in the database. Note that this
|
||
option is ignored if the script will restore a previous unfinished session.
|
||
-P, --poolsize <number>
|
||
The number of slots in the processes pool, that is the number of '.CHILD.'
|
||
processes the script will run in parallel. Note that this option is ignored
|
||
if the script will restore a previous unfinished session.
|
||
DEFAULT: '.$opts['poolsize'].'
|
||
-I, --ignorelock
|
||
Normally, if its lockfile exists, the script will exit with an error.
|
||
If this option is set, instead, the lockfile existence will be ignored.
|
||
Please check that the script is actually not running before using it.
|
||
-R, --dontrestore
|
||
If this option is set and «instances.job» and «status.job» files from
|
||
a previous unfinished session are present in the «run» subdirectory inside
|
||
the directory where the script resides, the script will ignore them and
|
||
start a new session; otherwise the script will restore the previous,
|
||
unfinished session.
|
||
-m, --moreclauses <more SQL clauses>
|
||
If this option is set, whatever one writes as argument to the option will
|
||
be added to the main query for instances’ records, which is «SELECT URI FROM
|
||
Instances WHERE Dead=0».
|
||
-L, --logminmsglev <«debug»|«info»|«warning»|«error»|«none»>
|
||
Defines the minimum “importance level” of messages to be written into the
|
||
log file «run/[instance hostname].log». There are 4 “importance levels”, in
|
||
this order of importance: «debug», «info», «warning», «error».
|
||
Setting this option to any of these values will write into the logfile all
|
||
the messages with the specified or a greater level; setting it to the
|
||
special value «none» will completely disable logging to file.
|
||
DEFAULT: '.$msglevs[$opts['logminmsglev']].'
|
||
-T, --tuiminmsglev <«debug»|«info»|«warning»|«error»|«none»>
|
||
Defines the minimum “importance level” of messages to be written to the
|
||
terminal. See the option above to understand how this works.
|
||
DEFAULT: '.$msglevs[$opts['tuiminmsglev']].'
|
||
-h, --help
|
||
When this option is specified, the script will show this help text and exit.
|
||
|
||
LICENSE
|
||
|
||
This program comes with ABSOLUTELY NO WARRANTY; for details see the source.
|
||
This is free software, and you are welcome to redistribute it under certain
|
||
conditions; see <http://www.gnu.org/licenses/> for details.'.N;
|
||
|
||
$childopts='';
|
||
|
||
for ($i=1; $i<$argc; $i++) {
|
||
if ($argv[$i]=='-') {
|
||
if ($i<$argc-1) {
|
||
$i++;
|
||
while ($i<$argc) {
|
||
$childopts.=' '.$argv[$i];
|
||
$i++;
|
||
}
|
||
} else {
|
||
eecho(2,'you have specified «-» as last argument...'.N);
|
||
}
|
||
} elseif ($argv[$i]=='-p' || $argv[$i]=='--peersfp') {
|
||
if ($i+1>=$argc || !file_exists($argv[$i+1]) || !is_file($argv[$i+1]) || !is_readable($argv[$i+1]))
|
||
mexit('option «'.$argv[$i].'» requires an existing and readable file as an argument (use «-h» to read help).'.N,1,false);
|
||
$i++;
|
||
$opts['peersfp']=$argv[$i];
|
||
} elseif ($argv[$i]=='-P' || $argv[$i]=='--poolsize') {
|
||
if ($i+1>=$argc || preg_match('/\d+/',$argv[$i+1])!==1 || $argv[$i+1]+0<1)
|
||
mexit('option «'.$argv[$i].'» requires an integer number greater than 0 as an argument (use «-h» to read help).'.N,1,false);
|
||
$i++;
|
||
$opts['poolsize']=$argv[$i]+0;
|
||
} elseif ($argv[$i]=='-R' || $argv[$i]=='--dontrestore') {
|
||
$opts['dontrestore']=true;
|
||
} elseif ($argv[$i]=='-I' || $argv[$i]=='--ignorelock') {
|
||
$opts['ignorelock']=true;
|
||
} elseif ($argv[$i]=='-m' || $argv[$i]=='--moreclauses') {
|
||
if ($i+1>=$argc)
|
||
mexit('option «'.$argv[$i].'» requires some SQL clause as argument (use «-h» to read help).'.N,1,false);
|
||
$i++;
|
||
$opts['moreclauses']=$argv[$i];
|
||
} elseif ($argv[$i]=='-L' || $argv[$i]=='--logminmsglev') {
|
||
if ($i+1>=$argc || !in_array(strtolower($argv[$i+1]),$msglevs))
|
||
mexit('option «'.$argv[$i].'» requires a “log level” value as an argument (use «-h» to read help).'.N,1);
|
||
$i++;
|
||
$opts['logminmsglev']=array_search(strtolower($argv[$i]),$msglevs);
|
||
} elseif ($argv[$i]=='-T' || $argv[$i]=='--tuiminmsglev') {
|
||
if ($i+1>=$argc || !in_array(strtolower($argv[$i+1]),$msglevs))
|
||
mexit('option «'.$argv[$i].'» requires a “log level” value as an argument (use «-h» to read help).'.N,1);
|
||
$i++;
|
||
$opts['tuiminmsglev']=array_search(strtolower($argv[$i]),$msglevs);
|
||
} elseif ($argv[$i]=='-h' || $argv[$i]=='--help') {
|
||
echo($help);
|
||
exit(0);
|
||
} else {
|
||
mexit('don’t know how to interpret «'.$argv[$i].'» (you can read the help text using «-h» or «--help»).'.N,1,false);
|
||
}
|
||
}
|
||
|
||
foreach ($msglevs as $key=>$val) $msglevs[$key]=ucfirst($val);
|
||
|
||
$rundirpath=__DIR__.'/run';
|
||
|
||
$lockfp=$rundirpath.'/'.FNAME.'.lock';
|
||
if (file_exists($lockfp) && !$opts['ignorelock']) {
|
||
eecho(3,'lock file «'.$lockfp.'» exists (if you are sure '.SNAME.' is not already running you can use option «-I» to force execution).'.N);
|
||
exit(1);
|
||
}
|
||
if (@touch($lockfp)===false) {
|
||
eecho(3,'could not touch file «'.$lockfp.'».'.N);
|
||
exit(1);
|
||
}
|
||
|
||
if (file_exists($rundirpath) && !is_dir($rundirpath))
|
||
mexit('«'.$rundirpath.'» is not a directory.'.N,1,false);
|
||
elseif (file_exists($rundirpath) && (!is_readable($rundirpath) || !is_writeable($rundirpath)))
|
||
mexit('«'.$rundirpath.'» is not readable and writeable.'.N,1,false);
|
||
elseif (!file_exists($rundirpath))
|
||
if (@mkdir($rundirpath)===false)
|
||
mexit('could not create directory «'.$rundirpath.'».'.N,1,false);
|
||
|
||
$instsjfp=$rundirpath.'/'.FNAME.'_instances.job';
|
||
$statusjfp=$rundirpath.'/'.FNAME.'_status.job';
|
||
|
||
(!$opts['dontrestore'] && file_exists($instsjfp) && file_exists($statusjfp)) ? $restore=true : $restore=false;
|
||
|
||
if ($opts['logminmsglev']<4) {
|
||
$logfp=$rundirpath.'/'.FNAME.'.log';
|
||
($restore) ? $mode='a' : $mode='w';
|
||
$logf=fopen($logfp,$mode);
|
||
if ($logf===false) mexit('could not open log file «'.$logfp.'» for writing.'.N,1,true);
|
||
}
|
||
|
||
($restore) ? eecho(1,'--- restarting ---'.N) : eecho(1,'--- starting ---'.N);
|
||
|
||
if ($restore) {
|
||
eecho(0,'looks like previous session was interrupted, trying to restore it...'.N);
|
||
$insts=@file($instsjfp,FILE_IGNORE_NEW_LINES|FILE_SKIP_EMPTY_LINES);
|
||
if ($insts===false) mexit('could not open file «'.$instsjfp.'» for reading.'.N,1,true);
|
||
$cinsts=count($insts);
|
||
eecho(1,'loaded '.$cinsts.' hostnames from previous session file.'.N);
|
||
$buf=@file($statusjfp,FILE_IGNORE_NEW_LINES|FILE_SKIP_EMPTY_LINES);
|
||
if ($buf===false) mexit('could not open file «'.$statusjfp.'» for reading.'.N,1,true);
|
||
if (count($buf)<2) mexit('file «'.$statusjfp.'»: wrong format (1).'.N,1,true);
|
||
$buf[0]=explode("\t",$buf[0]);
|
||
if (count($buf[0])!=4 ||
|
||
preg_match('/^\d+$/',$buf[0][0])!==1 ||
|
||
preg_match('/^\d+$/',$buf[0][1])!==1 ||
|
||
preg_match('/^\d+(\.\d+)?$/',$buf[0][2])!==1 ||
|
||
preg_match('/^\d+$/',$buf[0][3])!==1)
|
||
mexit('file «'.$statusjfp.'»: wrong format (2).'.N,1,true);
|
||
$opts['poolsize']=$buf[0][0]+0;
|
||
$instk=$buf[0][1]+0;
|
||
$toff=$buf[0][2]+0;
|
||
$done=$buf[0][3]+0;
|
||
//eecho(0,'poolsize: '.$opts['poolsize'].'; instk: '.$instk.'; eta: '.$tet.'; done: '.$done.'.'.N);
|
||
for ($i=1; $i<count($buf); $i++) {
|
||
if (preg_match('/^\d+$/',$buf[$i])!==1) mexit('file «'.$statusjfp.'»: wrong format (3).'.N,1,true);
|
||
//eecho(0,$i.': '.$buf[$i].'.'.N);
|
||
$host=$insts[$buf[$i]+0];
|
||
eecho(1,'bootstrapping processes pool, adding host «'.$host.'».'.N);
|
||
$descspecs=[ 0=>['pipe','r'], 1=>['pipe','w'], 2=>['file',$rundirpath.'/'.$host.'.stderr.log','w'] ];
|
||
$procs[]=['proc'=>proc_open(cmd($childopts,$host),$descspecs,$pipes), 'instk'=>$buf[$i]+0, 'host'=>$host, 'begts'=>microtime(true)];
|
||
}
|
||
eecho(1,'restored previous session.'.N);
|
||
} else {
|
||
|
||
$inifp=__DIR__.'/../conf/mustard.ini';
|
||
$iniarr=@parse_ini_file($inifp);
|
||
if ($iniarr===false) mexit('could not open config file «'.$inifp.'»'.N,1,true);
|
||
|
||
try { $link=@mysqli_connect($iniarr['db_host'],$iniarr['db_admin_name'],$iniarr['db_admin_password'],$iniarr['db_name'],$iniarr['db_port'],$iniarr['db_socket']); }
|
||
catch (Exception $error) { mexit('could not connect to MySQL server: '.mysqli_connect_error().'.'.N,1,true); }
|
||
// for php versions < 8
|
||
if ($link===false) mexit('could not connect to MySQL server: '.mysqli_connect_error().'.'.N,1,true);
|
||
try { $res=mysqli_set_charset($link,'utf8mb4'); }
|
||
catch (Exception $error) { mexit('could not set «utf8mb4» charset for MySQL: '.mysqli_error($link).'.'.N,1,true); }
|
||
// for php versions < 8
|
||
if ($res===false) mexit('could not set MySQL charset: '.mysqli_errno($link).': '.mysqli_error($link).'.'.N,1,true);
|
||
|
||
$insts=[];
|
||
eecho(0,'loading known, alive instances from the database...'.N);
|
||
$res=myq($link,'SELECT URI FROM Instances WHERE Dead=0'.$opts['moreclauses'],__LINE__);
|
||
while($row=mysqli_fetch_assoc($res))
|
||
if (!in_array($row['URI'],$insts))
|
||
$insts[]=$row['URI'];
|
||
eecho(1,'loaded '.count($insts).' known, alive instances from the database.'.N);
|
||
|
||
if (!is_null($opts['peersfp'])) {
|
||
eecho(0,'loading dead instances from the database...'.N);
|
||
$res=myq($link,'SELECT URI FROM Instances WHERE Dead=1',__LINE__);
|
||
$deadinsts=[];
|
||
while($row=mysqli_fetch_assoc($res))
|
||
if (!in_array($row['URI'],$deadinsts))
|
||
$deadinsts[]=$row['URI'];
|
||
eecho(1,'loaded '.count($deadinsts).' dead instances from the database.'.N);
|
||
eecho(0,'loading instances from «'.$opts['peersfp'].'»...'.N);
|
||
$peers=@file($opts['peersfp'],FILE_IGNORE_NEW_LINES|FILE_SKIP_EMPTY_LINES);
|
||
if ($peers===false) mexit('could not open «'.$opts['peersfp'].'» for reading.'.N,1,true);
|
||
$i=0;
|
||
foreach ($peers as $pdom) {
|
||
if (!in_array($pdom,$insts)) {
|
||
if (!in_array($pdom,$deadinsts)) {
|
||
$i++;
|
||
$insts[]=$pdom;
|
||
} else {
|
||
eecho(1,'ignoring instance «'.$pdom.'» from peers file because it’s dead.'.N);
|
||
}
|
||
}
|
||
}
|
||
eecho(1,'loaded '.$i.' more instances from «'.$opts['peersfp'].'».'.N);
|
||
}
|
||
|
||
mysqli_close($link);
|
||
unset($link);
|
||
|
||
unset($deadinsts);
|
||
shuffle($insts);
|
||
$cinsts=count($insts);
|
||
eecho(1,$cinsts.' instances to be checked.'.N);
|
||
|
||
$instsf=@fopen($instsjfp,'w');
|
||
if ($instsf===false) mexit('could not open «'.$instsjfp.'» for writing.'.N,1,true);
|
||
foreach ($insts as $host) fwrite($instsf,$host.N);
|
||
fclose($instsf);
|
||
|
||
$toff=0;
|
||
$done=0;
|
||
$procs=[];
|
||
|
||
for ($instk=0; $instk<$opts['poolsize'] && $instk<$cinsts; $instk++) {
|
||
$host=$insts[$instk];
|
||
eecho(1,'bootstrapping processes pool, adding host «'.$host.'».'.N);
|
||
$descspecs=[ 0=>['pipe','r'], 1=>['pipe','w'], 2=>['file',$rundirpath.'/'.$host.'.stderr.log','w'] ];
|
||
$procs[]=['proc'=>proc_open(cmd($childopts,$host),$descspecs,$pipes), 'instk'=>$instk, 'host'=>$host, 'begts'=>microtime(true)];
|
||
}
|
||
$instk--;
|
||
|
||
}
|
||
|
||
$tini=microtime(true);
|
||
$rundone=false;
|
||
do {
|
||
$now=microtime(true);
|
||
$tet=$now-$tini+$toff;
|
||
eecho(0,'[[[ CHECKING PROCESSES POOL ]]]'.N);
|
||
$somerun=false;
|
||
foreach ($procs as $key=>$proc) {
|
||
if (!is_null($proc)) {
|
||
$pstat=proc_get_status($proc['proc']);
|
||
if (!$pstat['running']) {
|
||
$done++;
|
||
$out='proc slot '.$key.': finished running on «'.$proc['host'].'» (exit code: '.$pstat['exitcode'].')';
|
||
if ($instk<$cinsts-1) {
|
||
$instk++;
|
||
$host=$insts[$instk];
|
||
$descspecs=[ 0=>['pipe','r'], 1=>['pipe','w'], 2=>['file',$rundirpath.'/'.$host.'.stderr.log','w'] ];
|
||
$procs[$key]=['proc'=>proc_open(cmd($childopts,$host),$descspecs,$pipes), 'instk'=>$instk, 'host'=>$host, 'begts'=>$now];
|
||
$out.='; started a new process on «'.$host.'».'.N;
|
||
} else {
|
||
$out.='; no more hosts to check.'.N;
|
||
$procs[$key]=null;
|
||
}
|
||
eecho(1,$out);
|
||
} else {
|
||
eecho(0,'proc slot '.$key.': been running on «'.$proc['host'].'» for '.ght($now-$proc['begts']).'.'.N);
|
||
$somerun=true;
|
||
}
|
||
}
|
||
}
|
||
$out=$done.'/'.$cinsts.' ('.round(100/$cinsts*$done).'%); elapsed time: '.ght($tet);
|
||
if ($done>0) $out.='; estimated time remaining: '.ght($cinsts*$tet/$done-$tet);
|
||
eecho(1,$out.'.'.N);
|
||
if ($somerun) {
|
||
writestatus($statusjfp,$opts,$instk,$tet,$done,$procs);
|
||
sleep(1);
|
||
} else {
|
||
$rundone=true;
|
||
}
|
||
} while (!$rundone);
|
||
|
||
unlink($instsjfp);
|
||
unlink($statusjfp);
|
||
unlink($lockfp);
|
||
eecho(1,'done :-)'.N);
|
||
if (isset($logf)) fclose($logf);
|
||
exit(0);
|
||
|
||
|
||
// functions
|
||
|
||
function writestatus(&$statusjfp,&$opts,&$instk,&$tet,&$done,&$procs) {
|
||
$f=@fopen($statusjfp,'w');
|
||
if ($f===false) mexit('could not open «'.$statusjfp.'» for writing.'.N,2,true);
|
||
fwrite($f,$opts['poolsize']."\t".$instk."\t".$tet."\t".$done.N);
|
||
foreach ($procs as $proc)
|
||
if (!is_null($proc))
|
||
fwrite($f,$proc['instk'].N);
|
||
fclose($f);
|
||
}
|
||
|
||
function cmd(&$childopts, &$host) {
|
||
return(__DIR__.'/'.CHILD.$childopts.' '.escapeshellarg($host));
|
||
}
|
||
|
||
function eecho($lev,$msg) {
|
||
global $logf, $opts, $msglevs;
|
||
$time=microtime(false);
|
||
$time=explode(' ',$time);
|
||
$time=date('Y-m-d H:i:s',$time[1]).'.'.substr($time[0],2);
|
||
$msg=$time.' '.$msglevs[$lev].': '.$msg;
|
||
if ($lev>=$opts['tuiminmsglev']) {
|
||
if ($lev<2)
|
||
echo($msg);
|
||
else
|
||
fwrite(STDERR,$msg);
|
||
}
|
||
if ($lev>=$opts['logminmsglev'] && isset($logf) && $logf!==false) fwrite($logf,$msg);
|
||
}
|
||
|
||
function myq(&$link,$query,$line) {
|
||
try {
|
||
$res=mysqli_query($link,$query);
|
||
}
|
||
catch (Exception $error) {
|
||
mexit('query «'.$query.'» (line '.$line.') failed: '.$error->getMessage().N,3,true);
|
||
}
|
||
// for older php versions < 8, which seem to not catch mysql exceptions
|
||
if ($res===false) mexit('query «'.$query.'» (line '.$line.') failed: '.mysqli_errno($link).': '.mysqli_error($link).'.'.N,3,true);
|
||
return($res);
|
||
}
|
||
|
||
function mexit($msg,$code,$remlock) {
|
||
global $link, $logf, $lockfp;
|
||
if (isset($link) && $link!==false) mysqli_close($link);
|
||
if ($remlock && isset($lockfp) && is_file($lockfp)) unlink($lockfp);
|
||
if ($code!=0)
|
||
eecho(3,$msg);
|
||
else
|
||
eecho(1,$msg);
|
||
if (isset($logf) && $logf!==false) fclose($logf);
|
||
exit($code);
|
||
}
|
||
|
||
?>
|