From: Sarah Hoffmann Date: Tue, 3 Oct 2017 08:16:17 +0000 (+0200) Subject: move psqlRunScript implementation into cmd lib X-Git-Tag: v3.1.0~60^2~1 X-Git-Url: https://git.openstreetmap.org/nominatim.git/commitdiff_plain/8e2ef2842ef6d854e87a27ff57043971765fdd8a move psqlRunScript implementation into cmd lib Function needed for update.php as well. --- diff --git a/lib/cmd.php b/lib/cmd.php index d7ba285d..898fbe95 100644 --- a/lib/cmd.php +++ b/lib/cmd.php @@ -136,15 +136,53 @@ function info($sMsg) $aWarnings = []; + function warn($sMsg) { $GLOBALS['aWarnings'][] = $sMsg; echo date('Y-m-d H:i:s == ').'WARNING: '.$sMsg."\n"; } + function repeatWarnings() { foreach ($GLOBALS['aWarnings'] as $sMsg) { echo ' * ',$sMsg."\n"; } } + + +function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreErrors = false) +{ + // Convert database DSN to psql parameters + $aDSNInfo = DB::parseDSN(CONST_Database_DSN); + if (!isset($aDSNInfo['port']) || !$aDSNInfo['port']) $aDSNInfo['port'] = 5432; + $sCMD = 'psql -p '.$aDSNInfo['port'].' -d '.$aDSNInfo['database']; + if (!$bVerbose) { + $sCMD .= ' -q'; + } + if ($bfatal && !$bIgnoreErrors) { + $sCMD .= ' -v ON_ERROR_STOP=1'; + } + $aDescriptors = array( + 0 => array('pipe', 'r'), + 1 => STDOUT, + 2 => STDERR + ); + $ahPipes = null; + $hProcess = @proc_open($sCMD, $aDescriptors, $ahPipes); + if (!is_resource($hProcess)) { + fail('unable to start pgsql'); + } + + while (strlen($sScript)) { + $written = fwrite($ahPipes[0], $sScript); + if ($written <= 0) break; + $sScript = substr($sScript, $written); + } + fclose($ahPipes[0]); + $iReturn = proc_close($hProcess); + if ($bfatal && $iReturn > 0) { + fail("pgsql returned with error code ($iReturn)"); + } +} diff --git a/utils/setup.php b/utils/setup.php index 2369b183..624c49a3 100755 --- a/utils/setup.php +++ b/utils/setup.php @@ -765,34 +765,12 @@ function pgsqlRunScriptFile($sFilename) function pgsqlRunScript($sScript, $bfatal = true) { global $aCMDResult; - // Convert database DSN to psql parameters - $aDSNInfo = DB::parseDSN(CONST_Database_DSN); - if (!isset($aDSNInfo['port']) || !$aDSNInfo['port']) $aDSNInfo['port'] = 5432; - $sCMD = 'psql -p '.$aDSNInfo['port'].' -d '.$aDSNInfo['database']; - if (!$aCMDResult['verbose']) { - $sCMD .= ' -q'; - } - if ($bfatal && !$aCMDResult['ignore-errors']) - $sCMD .= ' -v ON_ERROR_STOP=1'; - $aDescriptors = array( - 0 => array('pipe', 'r'), - 1 => STDOUT, - 2 => STDERR - ); - $ahPipes = null; - $hProcess = @proc_open($sCMD, $aDescriptors, $ahPipes); - if (!is_resource($hProcess)) fail('unable to start pgsql'); - - while (strlen($sScript)) { - $written = fwrite($ahPipes[0], $sScript); - if ($written <= 0) break; - $sScript = substr($sScript, $written); - } - fclose($ahPipes[0]); - $iReturn = proc_close($hProcess); - if ($bfatal && $iReturn > 0) { - fail("pgsql returned with error code ($iReturn)"); - } + runSQLScript( + $sScript, + $bfatal, + $aCMDResult['verbose'], + $aCMDResult['ignore-errors'] + ); } function pgsqlRunPartitionScript($sTemplate)