lib-php/DB.php

   1 <?php
   2 /**
   3  * SPDX-License-Identifier: GPL-2.0-only
   4  *
   5  * This file is part of Nominatim. (https://nominatim.org)
   6  *
   7  * Copyright (C) 2022 by the Nominatim developer community.
   8  * For a full list of authors see the git log.
   9  */
  10
  11 namespace Nominatim;
  12
  13 require_once(CONST_LibDir.'/DatabaseError.php');
  14
  15 /**
  16  * Uses PDO to access the database specified in the CONST_Database_DSN
  17  * setting.
  18  */
  19 class DB
  20 {
  21     protected $connection;
  22
  23     public function __construct($sDSN = null)
  24     {
  25         $this->sDSN = $sDSN ?? getSetting('DATABASE_DSN');
  26     }
  27
  28     public function connect($bNew = false, $bPersistent = true)
  29     {
  30         if (isset($this->connection) && !$bNew) {
  31             return true;
  32         }
  33         $aConnOptions = array(
  34                          \PDO::ATTR_ERRMODE            => \PDO::ERRMODE_EXCEPTION,
  35                          \PDO::ATTR_DEFAULT_FETCH_MODE => \PDO::FETCH_ASSOC,
  36                          \PDO::ATTR_PERSISTENT         => $bPersistent
  37         );
  38
  39         // https://secure.php.net/manual/en/ref.pdo-pgsql.connection.php
  40         try {
  41             $this->connection = new \PDO($this->sDSN, null, null, $aConnOptions);
  42         } catch (\PDOException $e) {
  43             $sMsg = 'Failed to establish database connection:' . $e->getMessage();
  44             throw new \Nominatim\DatabaseError($sMsg, 500, null, $e->getMessage());
  45         }
  46
  47         $this->connection->exec("SET DateStyle TO 'sql,european'");
  48         $this->connection->exec("SET client_encoding TO 'utf-8'");
  49         // Disable JIT and parallel workers. They interfere badly with search SQL.
  50         $this->connection->exec("SET max_parallel_workers_per_gather TO 0");
  51         if ($this->getPostgresVersion() >= 11) {
  52             $this->connection->exec("SET jit_above_cost TO -1");
  53         }
  54
  55         $iMaxExecution = ini_get('max_execution_time');
  56         if ($iMaxExecution > 0) {
  57             $this->connection->setAttribute(\PDO::ATTR_TIMEOUT, $iMaxExecution); // seconds
  58         }
  59
  60         return true;
  61     }
  62
  63     // returns the number of rows that were modified or deleted by the SQL
  64     // statement. If no rows were affected returns 0.
  65     public function exec($sSQL, $aInputVars = null, $sErrMessage = 'Database query failed')
  66     {
  67         $val = null;
  68         try {
  69             if (isset($aInputVars)) {
  70                 $stmt = $this->connection->prepare($sSQL);
  71                 $stmt->execute($aInputVars);
  72             } else {
  73                 $val = $this->connection->exec($sSQL);
  74             }
  75         } catch (\PDOException $e) {
  76             throw new \Nominatim\DatabaseError($sErrMessage, 500, null, $e, $sSQL);
  77         }
  78         return $val;
  79     }
  80
  81     /**
  82      * Executes query. Returns first row as array.
  83      * Returns false if no result found.
  84      *
  85      * @param string  $sSQL
  86      *
  87      * @return array[]
  88      */
  89     public function getRow($sSQL, $aInputVars = null, $sErrMessage = 'Database query failed')
  90     {
  91         try {
  92             $stmt = $this->getQueryStatement($sSQL, $aInputVars, $sErrMessage);
  93             $row = $stmt->fetch();
  94         } catch (\PDOException $e) {
  95             throw new \Nominatim\DatabaseError($sErrMessage, 500, null, $e, $sSQL);
  96         }
  97         return $row;
  98     }
  99
 100     /**
 101      * Executes query. Returns first value of first result.
 102      * Returns false if no results found.
 103      *
 104      * @param string  $sSQL
 105      *
 106      * @return array[]
 107      */
 108     public function getOne($sSQL, $aInputVars = null, $sErrMessage = 'Database query failed')
 109     {
 110         try {
 111             $stmt = $this->getQueryStatement($sSQL, $aInputVars, $sErrMessage);
 112             $row = $stmt->fetch(\PDO::FETCH_NUM);
 113             if ($row === false) {
 114                 return false;
 115             }
 116         } catch (\PDOException $e) {
 117             throw new \Nominatim\DatabaseError($sErrMessage, 500, null, $e, $sSQL);
 118         }
 119         return $row[0];
 120     }
 121
 122     /**
 123      * Executes query. Returns array of results (arrays).
 124      * Returns empty array if no results found.
 125      *
 126      * @param string  $sSQL
 127      *
 128      * @return array[]
 129      */
 130     public function getAll($sSQL, $aInputVars = null, $sErrMessage = 'Database query failed')
 131     {
 132         try {
 133             $stmt = $this->getQueryStatement($sSQL, $aInputVars, $sErrMessage);
 134             $rows = $stmt->fetchAll();
 135         } catch (\PDOException $e) {
 136             throw new \Nominatim\DatabaseError($sErrMessage, 500, null, $e, $sSQL);
 137         }
 138         return $rows;
 139     }
 140
 141     /**
 142      * Executes query. Returns array of the first value of each result.
 143      * Returns empty array if no results found.
 144      *
 145      * @param string  $sSQL
 146      *
 147      * @return array[]
 148      */
 149     public function getCol($sSQL, $aInputVars = null, $sErrMessage = 'Database query failed')
 150     {
 151         $aVals = array();
 152         try {
 153             $stmt = $this->getQueryStatement($sSQL, $aInputVars, $sErrMessage);
 154
 155             while (($val = $stmt->fetchColumn(0)) !== false) { // returns first column or false
 156                 $aVals[] = $val;
 157             }
 158         } catch (\PDOException $e) {
 159             throw new \Nominatim\DatabaseError($sErrMessage, 500, null, $e, $sSQL);
 160         }
 161         return $aVals;
 162     }
 163
 164     /**
 165      * Executes query. Returns associate array mapping first value to second value of each result.
 166      * Returns empty array if no results found.
 167      *
 168      * @param string  $sSQL
 169      *
 170      * @return array[]
 171      */
 172     public function getAssoc($sSQL, $aInputVars = null, $sErrMessage = 'Database query failed')
 173     {
 174         try {
 175             $stmt = $this->getQueryStatement($sSQL, $aInputVars, $sErrMessage);
 176
 177             $aList = array();
 178             while ($aRow = $stmt->fetch(\PDO::FETCH_NUM)) {
 179                 $aList[$aRow[0]] = $aRow[1];
 180             }
 181         } catch (\PDOException $e) {
 182             throw new \Nominatim\DatabaseError($sErrMessage, 500, null, $e, $sSQL);
 183         }
 184         return $aList;
 185     }
 186
 187     /**
 188      * Executes query. Returns a PDO statement to iterate over.
 189      *
 190      * @param string  $sSQL
 191      *
 192      * @return PDOStatement
 193      */
 194     public function getQueryStatement($sSQL, $aInputVars = null, $sErrMessage = 'Database query failed')
 195     {
 196         try {
 197             if (isset($aInputVars)) {
 198                 $stmt = $this->connection->prepare($sSQL);
 199                 $stmt->execute($aInputVars);
 200             } else {
 201                 $stmt = $this->connection->query($sSQL);
 202             }
 203         } catch (\PDOException $e) {
 204             throw new \Nominatim\DatabaseError($sErrMessage, 500, null, $e, $sSQL);
 205         }
 206         return $stmt;
 207     }
 208
 209     /**
 210      * St. John's Way => 'St. John\'s Way'
 211      *
 212      * @param string  $sVal  Text to be quoted.
 213      *
 214      * @return string
 215      */
 216     public function getDBQuoted($sVal)
 217     {
 218         return $this->connection->quote($sVal);
 219     }
 220
 221     /**
 222      * Like getDBQuoted, but takes an array.
 223      *
 224      * @param array  $aVals  List of text to be quoted.
 225      *
 226      * @return array[]
 227      */
 228     public function getDBQuotedList($aVals)
 229     {
 230         return array_map(function ($sVal) {
 231             return $this->getDBQuoted($sVal);
 232         }, $aVals);
 233     }
 234
 235     /**
 236      * [1,2,'b'] => 'ARRAY[1,2,'b']''
 237      *
 238      * @param array  $aVals  List of text to be quoted.
 239      *
 240      * @return string
 241      */
 242     public function getArraySQL($a)
 243     {
 244         return 'ARRAY['.join(',', $a).']';
 245     }
 246
 247     /**
 248      * Check if a table exists in the database. Returns true if it does.
 249      *
 250      * @param string  $sTableName
 251      *
 252      * @return boolean
 253      */
 254     public function tableExists($sTableName)
 255     {
 256         $sSQL = 'SELECT count(*) FROM pg_tables WHERE tablename = :tablename';
 257         return ($this->getOne($sSQL, array(':tablename' => $sTableName)) == 1);
 258     }
 259
 260     /**
 261      * Deletes a table. Returns true if deleted or didn't exist.
 262      *
 263      * @param string  $sTableName
 264      *
 265      * @return boolean
 266      */
 267     public function deleteTable($sTableName)
 268     {
 269         return $this->exec('DROP TABLE IF EXISTS '.$sTableName.' CASCADE') == 0;
 270     }
 271
 272     /**
 273      * Tries to connect to the database but on failure doesn't throw an exception.
 274      *
 275      * @return boolean
 276      */
 277     public function checkConnection()
 278     {
 279         $bExists = true;
 280         try {
 281             $this->connect(true);
 282         } catch (\Nominatim\DatabaseError $e) {
 283             $bExists = false;
 284         }
 285         return $bExists;
 286     }
 287
 288     /**
 289      * e.g. 9.6, 10, 11.2
 290      *
 291      * @return float
 292      */
 293     public function getPostgresVersion()
 294     {
 295         $sVersionString = $this->getOne('SHOW server_version_num');
 296         preg_match('#([0-9]?[0-9])([0-9][0-9])[0-9][0-9]#', $sVersionString, $aMatches);
 297         return (float) ($aMatches[1].'.'.$aMatches[2]);
 298     }
 299
 300     /**
 301      * e.g. 2, 2.2
 302      *
 303      * @return float
 304      */
 305     public function getPostgisVersion()
 306     {
 307         $sVersionString = $this->getOne('select postgis_lib_version()');
 308         preg_match('#^([0-9]+)[.]([0-9]+)[.]#', $sVersionString, $aMatches);
 309         return (float) ($aMatches[1].'.'.$aMatches[2]);
 310     }
 311
 312     /**
 313      * Returns an associate array of postgresql database connection settings. Keys can
 314      * be 'database', 'hostspec', 'port', 'username', 'password'.
 315      * Returns empty array on failure, thus check if at least 'database' is set.
 316      *
 317      * @return array[]
 318      */
 319     public static function parseDSN($sDSN)
 320     {
 321         // https://secure.php.net/manual/en/ref.pdo-pgsql.connection.php
 322         $aInfo = array();
 323         if (preg_match('/^pgsql:(.+)$/', $sDSN, $aMatches)) {
 324             foreach (explode(';', $aMatches[1]) as $sKeyVal) {
 325                 list($sKey, $sVal) = explode('=', $sKeyVal, 2);
 326                 if ($sKey == 'host') {
 327                     $sKey = 'hostspec';
 328                 } elseif ($sKey == 'dbname') {
 329                     $sKey = 'database';
 330                 } elseif ($sKey == 'user') {
 331                     $sKey = 'username';
 332                 }
 333                 $aInfo[$sKey] = $sVal;
 334             }
 335         }
 336         return $aInfo;
 337     }
 338
 339     /**
 340      * Takes an array of settings and return the DNS string. Key names can be
 341      * 'database', 'hostspec', 'port', 'username', 'password' but aliases
 342      * 'dbname', 'host' and 'user' are also supported.
 343      *
 344      * @return string
 345      *
 346      */
 347     public static function generateDSN($aInfo)
 348     {
 349         $sDSN = sprintf(
 350             'pgsql:host=%s;port=%s;dbname=%s;user=%s;password=%s;',
 351             $aInfo['host'] ?? $aInfo['hostspec'] ?? '',
 352             $aInfo['port'] ?? '',
 353             $aInfo['dbname'] ?? $aInfo['database'] ?? '',
 354             $aInfo['user'] ?? '',
 355             $aInfo['password'] ?? ''
 356         );
 357         $sDSN = preg_replace('/\b\w+=;/', '', $sDSN);
 358         $sDSN = preg_replace('/;\Z/', '', $sDSN);
 359
 360         return $sDSN;
 361     }
 362 }