First commit

This commit is contained in:
pezcurrel 2023-12-27 00:46:32 +01:00
parent c799f73f05
commit 7840524d0f

View file

@ -0,0 +1,132 @@
#!/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('CONFIGFP',__DIR__.'/../../conf/mustard.ini');
$help='SYNOPSYS
'.SNAME.' [options]
DESCRIPTION
This is a script to set the «Threads» column of «Instances» table according
to them suspending or limiting threads, or not doing it, or the case its not
known.
OPTIONS
-h, --help
Show this help text and exit.'.N;
for ($i=1; $i<$argc; $i++) {
if ($argv[$i]=='-h' || $argv[$i]=='--help') {
mexit($help,0);
} else {
mexit('Dont know how to interpret «'.$argv[$i].'», please read the help text using «-h» or «--help» option.'.N,1);
}
}
use function mysqli_real_escape_string as myesc;
$iniarr=@parse_ini_file(CONFIGFP)
or mexit('Could not open config file «'.CONFIGFP.'».'.N,1);
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=[];
$res=myq($link,'SELECT * FROM Instances');
while ($row=mysqli_fetch_assoc($res)) $insts[]=$row;
$cinsts=count($insts);
$i=0;
$now=time();
foreach ($insts as $inst) {
$i++;
echo('Working on instance '.$i.'/'.$cinsts.' ('.round(100/$cinsts*$i,2).'%) with ID = '.$inst['ID'].' and URI = «'.$inst['URI'].'».'.N);
$res=myq($link,'SELECT * FROM InstBlocks WHERE InstID='.$inst['ID']);
if (mysqli_num_rows($res)>0) {
$threads='"accessible"';
while ($row=mysqli_fetch_assoc($res)) {
if (preg_match('#^(threads.net|.*\.threads.net)$#i',$row['Domain'])===1) {
if ($row['Severity']=='silence')
$threads='"limited"';
elseif ($row['Severity']=='suspend')
$threads='"suspended"';
else
$threads='"'.$row['Severity'].'"';
break;
}
}
} else {
$threads='NULL';
}
$que='UPDATE Instances SET Threads='.$threads.' WHERE ID='.$inst['ID'];
echo 'Executing query «'.$que.'» ...'.N;
myq($link,$que);
}
mysqli_close($link);
echo('Done updating '.$cinsts.' «Instances» records :-).'.N);
exit(0);
// functions
function myq(&$l,$q) {
try {
$res=mysqli_query($l,$q);
}
catch (Exception $e) {
echo('query «'.$q.'» failed: '.$e->getMessage().' (error code: '.$e->getCode().').'.N);
exit(3);
}
if ($res===false) {
echo('query «'.$q.'» failed: '.mysqli_errno($l).': '.mysqli_error($l).'.'.N);
exit(3);
}
return($res);
}
function mexit($msg,$code) {
global $link;
if (isset($link) && $link!==false) mysqli_close($link);
if ($code>0)
fwrite(STDERR,$msg);
else
echo($msg);
exit($code);
}
function utstd($val) {
if (is_null($val)) return(null);
$val=round($val);
return(date('Y-m-d H:i:s',$val));
}
function pr($val) {
if (is_null($val)) return('NULL');
return($val);
}
function eecho($msg) {
echo($msg);
}
?>