X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/cb40aa41d5b0efefdb2b9263e87ed016cce96277..4a321487f3d44241242a1729ab4d137df1f57bfb:/lib/db.php

diff --git a/lib/db.php b/lib/db.php
index c50b3c40..145c57d0 100644
--- a/lib/db.php
+++ b/lib/db.php
@@ -1,22 +1,38 @@
 <?php
-	require_once('DB.php');
 
-	function &getDB($bNew = false)
-	{
-		// Get the database object
-		$oDB =& DB::connect(CONST_Database_DSN.($bNew?'?new_link=true':''), false);
-		if (PEAR::IsError($oDB))
-		{
-			fail($oDB->getMessage(), 'Unable to connect to the database');
-		}
-		$oDB->setFetchMode(DB_FETCHMODE_ASSOC);
-		$oDB->query("SET DateStyle TO 'sql,european'");
-		$oDB->query("SET client_encoding TO 'utf-8'");
-		return $oDB;
-	}
+require_once('DB.php');
 
-	function getDBQuoted($s)
-	{
-		return "'".pg_escape_string($s)."'";
-	}
 
+function &getDB($bNew = false, $bPersistent = false)
+{
+    // Get the database object
+    $oDB = chksql(
+        DB::connect(CONST_Database_DSN.($bNew?'?new_link=true':''), $bPersistent),
+        "Failed to establish database connection"
+    );
+    $oDB->setFetchMode(DB_FETCHMODE_ASSOC);
+    $oDB->query("SET DateStyle TO 'sql,european'");
+    $oDB->query("SET client_encoding TO 'utf-8'");
+    $iMaxExecution = ini_get('max_execution_time') * 1000;
+    if ($iMaxExecution > 0) $oDB->query("SET statement_timeout TO $iMaxExecution");
+    return $oDB;
+}
+
+function getDBQuoted($s)
+{
+    return "'".pg_escape_string($s)."'";
+}
+
+function getPostgresVersion(&$oDB)
+{
+    $sVersionString = $oDB->getOne('select version()');
+    preg_match('#PostgreSQL ([0-9]+)[.]([0-9]+)[^0-9]#', $sVersionString, $aMatches);
+    return (float) ($aMatches[1].'.'.$aMatches[2]);
+}
+
+function getPostgisVersion(&$oDB)
+{
+    $sVersionString = $oDB->getOne('select postgis_full_version()');
+    preg_match('#POSTGIS="([0-9]+)[.]([0-9]+)[.]([0-9]+)( r([0-9]+))?"#', $sVersionString, $aMatches);
+    return (float) ($aMatches[1].'.'.$aMatches[2]);
+}