X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/1787892d32351988231a60b71900c6a623609c54..8f0c628310e9ad4160eb995f3ba53afa791a3ffc:/lib/cmd.php?ds=sidebyside diff --git a/lib/cmd.php b/lib/cmd.php index 1edf5dfd..44923618 100644 --- a/lib/cmd.php +++ b/lib/cmd.php @@ -134,7 +134,7 @@ function info($sMsg) echo date('Y-m-d H:i:s == ').$sMsg."\n"; } -$aWarnings = []; +$aWarnings = array(); function warn($sMsg) @@ -158,6 +158,16 @@ function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreError $aDSNInfo = DB::parseDSN(CONST_Database_DSN); if (!isset($aDSNInfo['port']) || !$aDSNInfo['port']) $aDSNInfo['port'] = 5432; $sCMD = 'psql -p '.$aDSNInfo['port'].' -d '.$aDSNInfo['database']; + if (isset($aDSNInfo['hostspec']) && $aDSNInfo['hostspec']) { + $sCMD .= ' -h ' . $aDSNInfo['hostspec']; + } + if (isset($aDSNInfo['username']) && $aDSNInfo['username']) { + $sCMD .= ' -U ' . $aDSNInfo['username']; + } + $aProcEnv = null; + if (isset($aDSNInfo['password']) && $aDSNInfo['password']) { + $aProcEnv = array_merge(array('PGPASSWORD' => $aDSNInfo['password']), $_ENV); + } if (!$bVerbose) { $sCMD .= ' -q'; } @@ -170,15 +180,19 @@ function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreError 2 => STDERR ); $ahPipes = null; - $hProcess = @proc_open($sCMD, $aDescriptors, $ahPipes); + $hProcess = @proc_open($sCMD, $aDescriptors, $ahPipes, null, $aProcEnv); if (!is_resource($hProcess)) { fail('unable to start pgsql'); } + if (!$bVerbose) { + fwrite($ahPipes[0], 'set client_min_messages to WARNING;'); + } + while (strlen($sScript)) { - $written = fwrite($ahPipes[0], $sScript); - if ($written <= 0) break; - $sScript = substr($sScript, $written); + $iWritten = fwrite($ahPipes[0], $sScript); + if ($iWritten <= 0) break; + $sScript = substr($sScript, $iWritten); } fclose($ahPipes[0]); $iReturn = proc_close($hProcess); @@ -186,3 +200,23 @@ function runSQLScript($sScript, $bfatal = true, $bVerbose = false, $bIgnoreError fail("pgsql returned with error code ($iReturn)"); } } + + +function runWithEnv($sCmd, $aEnv) +{ + $aFDs = array( + 0 => array('pipe', 'r'), + 1 => STDOUT, + 2 => STDERR + ); + $aPipes = null; + $hProc = @proc_open($sCmd, $aFDs, $aPipes, null, $aEnv); + if (!is_resource($hProc)) { + fail('unable to run command:' . $sCmd); + } + + fclose($aPipes[0]); // no stdin + + $iStat = proc_close($hProc); + return $iStat; +}