9e7df66858ce56c7c601db174d6acd7557c80e8b
[moodle.git] / lib / dml / pgsql_native_moodle_database.php
1 <?php
3 // This file is part of Moodle - http://moodle.org/
4 //
5 // Moodle is free software: you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation, either version 3 of the License, or
8 // (at your option) any later version.
9 //
10 // Moodle is distributed in the hope that it will be useful,
11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 // GNU General Public License for more details.
14 //
15 // You should have received a copy of the GNU General Public License
16 // along with Moodle.  If not, see <http://www.gnu.org/licenses/>.
19 /**
20  * Native pgsql class representing moodle database interface.
21  *
22  * @package    core
23  * @subpackage dml_driver
24  * @copyright  2008 Petr Skoda (http://skodak.org)
25  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
26  */
28 defined('MOODLE_INTERNAL') || die();
30 require_once($CFG->libdir.'/dml/moodle_database.php');
31 require_once($CFG->libdir.'/dml/pgsql_native_moodle_recordset.php');
32 require_once($CFG->libdir.'/dml/pgsql_native_moodle_temptables.php');
34 /**
35  * Native pgsql class representing moodle database interface.
36  *
37  * @package    core
38  * @subpackage dml_driver
39  * @copyright  2008 Petr Skoda (http://skodak.org)
40  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
41  */
42 class pgsql_native_moodle_database extends moodle_database {
44     protected $pgsql     = null;
45     protected $bytea_oid = null;
47     protected $last_error_reporting; // To handle pgsql driver default verbosity
49     /**
50      * Detects if all needed PHP stuff installed.
51      * Note: can be used before connect()
52      * @return mixed true if ok, string if something
53      */
54     public function driver_installed() {
55         if (!extension_loaded('pgsql')) {
56             return get_string('pgsqlextensionisnotpresentinphp', 'install');
57         }
58         return true;
59     }
61     /**
62      * Returns database family type - describes SQL dialect
63      * Note: can be used before connect()
64      * @return string db family name (mysql, postgres, mssql, oracle, etc.)
65      */
66     public function get_dbfamily() {
67         return 'postgres';
68     }
70     /**
71      * Returns more specific database driver type
72      * Note: can be used before connect()
73      * @return string db type mysqli, pgsql, oci, mssql, sqlsrv
74      */
75     protected function get_dbtype() {
76         return 'pgsql';
77     }
79     /**
80      * Returns general database library name
81      * Note: can be used before connect()
82      * @return string db type pdo, native
83      */
84     protected function get_dblibrary() {
85         return 'native';
86     }
88     /**
89      * Returns localised database type name
90      * Note: can be used before connect()
91      * @return string
92      */
93     public function get_name() {
94         return get_string('nativepgsql', 'install');
95     }
97     /**
98      * Returns localised database configuration help.
99      * Note: can be used before connect()
100      * @return string
101      */
102     public function get_configuration_help() {
103         return get_string('nativepgsqlhelp', 'install');
104     }
106     /**
107      * Returns localised database description
108      * Note: can be used before connect()
109      * @return string
110      */
111     public function get_configuration_hints() {
112         return get_string('databasesettingssub_postgres7', 'install');
113     }
115     /**
116      * Connect to db
117      * Must be called before other methods.
118      * @param string $dbhost The database host.
119      * @param string $dbuser The database username.
120      * @param string $dbpass The database username's password.
121      * @param string $dbname The name of the database being connected to.
122      * @param mixed $prefix string means moodle db prefix, false used for external databases where prefix not used
123      * @param array $dboptions driver specific options
124      * @return bool true
125      * @throws dml_connection_exception if error
126      */
127     public function connect($dbhost, $dbuser, $dbpass, $dbname, $prefix, array $dboptions=null) {
128         if ($prefix == '' and !$this->external) {
129             //Enforce prefixes for everybody but mysql
130             throw new dml_exception('prefixcannotbeempty', $this->get_dbfamily());
131         }
133         $driverstatus = $this->driver_installed();
135         if ($driverstatus !== true) {
136             throw new dml_exception('dbdriverproblem', $driverstatus);
137         }
139         $this->store_settings($dbhost, $dbuser, $dbpass, $dbname, $prefix, $dboptions);
141         $pass = addcslashes($this->dbpass, "'\\");
143         // Unix socket connections should have lower overhead
144         if (!empty($this->dboptions['dbsocket']) and ($this->dbhost === 'localhost' or $this->dbhost === '127.0.0.1')) {
145             $connection = "user='$this->dbuser' password='$pass' dbname='$this->dbname'";
146             if (strpos($this->dboptions['dbsocket'], '/') !== false) {
147                 $connection = $connection." host='".$this->dboptions['dbsocket']."'";
148             }
149         } else {
150             $this->dboptions['dbsocket'] = '';
151             if (empty($this->dbname)) {
152                 // probably old style socket connection - do not add port
153                 $port = "";
154             } else if (empty($this->dboptions['dbport'])) {
155                 $port = "port ='5432'";
156             } else {
157                 $port = "port ='".$this->dboptions['dbport']."'";
158             }
159             $connection = "host='$this->dbhost' $port user='$this->dbuser' password='$pass' dbname='$this->dbname'";
160         }
162         ob_start();
163         if (empty($this->dboptions['dbpersist'])) {
164             $this->pgsql = pg_connect($connection, PGSQL_CONNECT_FORCE_NEW);
165         } else {
166             $this->pgsql = pg_pconnect($connection, PGSQL_CONNECT_FORCE_NEW);
167         }
168         $dberr = ob_get_contents();
169         ob_end_clean();
171         $status = pg_connection_status($this->pgsql);
173         if ($status === false or $status === PGSQL_CONNECTION_BAD) {
174             $this->pgsql = null;
175             throw new dml_connection_exception($dberr);
176         }
178         $this->query_start("--pg_set_client_encoding()", null, SQL_QUERY_AUX);
179         pg_set_client_encoding($this->pgsql, 'utf8');
180         $this->query_end(true);
182         // find out the bytea oid
183         $sql = "SELECT oid FROM pg_type WHERE typname = 'bytea'";
184         $this->query_start($sql, null, SQL_QUERY_AUX);
185         $result = pg_query($this->pgsql, $sql);
186         $this->query_end($result);
188         $this->bytea_oid = pg_fetch_result($result, 0, 0);
189         pg_free_result($result);
190         if ($this->bytea_oid === false) {
191             $this->pgsql = null;
192             throw new dml_connection_exception('Can not read bytea type.');
193         }
195         // Connection stabilised and configured, going to instantiate the temptables controller
196         $this->temptables = new pgsql_native_moodle_temptables($this);
198         return true;
199     }
201     /**
202      * Close database connection and release all resources
203      * and memory (especially circular memory references).
204      * Do NOT use connect() again, create a new instance if needed.
205      */
206     public function dispose() {
207         parent::dispose(); // Call parent dispose to write/close session and other common stuff before closing connection
208         if ($this->pgsql) {
209             pg_close($this->pgsql);
210             $this->pgsql = null;
211         }
212     }
215     /**
216      * Called before each db query.
217      * @param string $sql
218      * @param array array of parameters
219      * @param int $type type of query
220      * @param mixed $extrainfo driver specific extra information
221      * @return void
222      */
223     protected function query_start($sql, array $params=null, $type, $extrainfo=null) {
224         parent::query_start($sql, $params, $type, $extrainfo);
225         // pgsql driver tents to send debug to output, we do not need that ;-)
226         $this->last_error_reporting = error_reporting(0);
227     }
229     /**
230      * Called immediately after each db query.
231      * @param mixed db specific result
232      * @return void
233      */
234     protected function query_end($result) {
235         // reset original debug level
236         error_reporting($this->last_error_reporting);
237         parent::query_end($result);
238     }
240     /**
241      * Returns database server info array
242      * @return array Array containing 'description' and 'version' info
243      */
244     public function get_server_info() {
245         static $info;
246         if (!$info) {
247             $this->query_start("--pg_version()", null, SQL_QUERY_AUX);
248             $info = pg_version($this->pgsql);
249             $this->query_end(true);
250         }
251         return array('description'=>$info['server'], 'version'=>$info['server']);
252     }
254     protected function is_min_version($version) {
255         $server = $this->get_server_info();
256         $server = $server['version'];
257         return version_compare($server, $version, '>=');
258     }
260     /**
261      * Returns supported query parameter types
262      * @return int bitmask of accepted SQL_PARAMS_*
263      */
264     protected function allowed_param_types() {
265         return SQL_PARAMS_DOLLAR;
266     }
268     /**
269      * Returns last error reported by database engine.
270      * @return string error message
271      */
272     public function get_last_error() {
273         return pg_last_error($this->pgsql);
274     }
276     /**
277      * Return tables in database WITHOUT current prefix.
278      * @param bool $usecache if true, returns list of cached tables.
279      * @return array of table names in lowercase and without prefix
280      */
281     public function get_tables($usecache=true) {
282         if ($usecache and $this->tables !== null) {
283             return $this->tables;
284         }
285         $this->tables = array();
286         $prefix = str_replace('_', '|_', $this->prefix);
287         // Get them from information_schema instead of catalog as far as
288         // we want to get only own session temp objects (catalog returns all)
289         $sql = "SELECT table_name
290                   FROM information_schema.tables
291                  WHERE table_name LIKE '$prefix%' ESCAPE '|'
292                    AND table_type IN ('BASE TABLE', 'LOCAL TEMPORARY')";
293         $this->query_start($sql, null, SQL_QUERY_AUX);
294         $result = pg_query($this->pgsql, $sql);
295         $this->query_end($result);
297         if ($result) {
298             while ($row = pg_fetch_row($result)) {
299                 $tablename = reset($row);
300                 if (strpos($tablename, $this->prefix) !== 0) {
301                     continue;
302                 }
303                 $tablename = substr($tablename, strlen($this->prefix));
304                 $this->tables[$tablename] = $tablename;
305             }
306             pg_free_result($result);
307         }
308         return $this->tables;
309     }
311     /**
312      * Return table indexes - everything lowercased.
313      * @param string $table The table we want to get indexes from.
314      * @return array of arrays
315      */
316     public function get_indexes($table) {
317         $indexes = array();
318         $tablename = $this->prefix.$table;
320         $sql = "SELECT *
321                   FROM pg_catalog.pg_indexes
322                  WHERE tablename = '$tablename'";
324         $this->query_start($sql, null, SQL_QUERY_AUX);
325         $result = pg_query($this->pgsql, $sql);
326         $this->query_end($result);
328         if ($result) {
329             while ($row = pg_fetch_assoc($result)) {
330                 if (!preg_match('/CREATE (|UNIQUE )INDEX ([^\s]+) ON '.$tablename.' USING ([^\s]+) \(([^\)]+)\)/i', $row['indexdef'], $matches)) {
331                     continue;
332                 }
333                 if ($matches[4] === 'id') {
334                     continue;
335                 }
336                 $columns = explode(',', $matches[4]);
337                 $columns = array_map(array($this, 'trim_quotes'), $columns);
338                 $indexes[$row['indexname']] = array('unique'=>!empty($matches[1]),
339                                               'columns'=>$columns);
340             }
341             pg_free_result($result);
342         }
343         return $indexes;
344     }
346     /**
347      * Returns detailed information about columns in table. This information is cached internally.
348      * @param string $table name
349      * @param bool $usecache
350      * @return array array of database_column_info objects indexed with column names
351      */
352     public function get_columns($table, $usecache=true) {
353         if ($usecache and isset($this->columns[$table])) {
354             return $this->columns[$table];
355         }
357         $this->columns[$table] = array();
359         $tablename = $this->prefix.$table;
361         $sql = "SELECT a.attnum, a.attname AS field, t.typname AS type, a.attlen, a.atttypmod, a.attnotnull, a.atthasdef, d.adsrc
362                   FROM pg_catalog.pg_class c
363                   JOIN pg_catalog.pg_attribute a ON a.attrelid = c.oid
364                   JOIN pg_catalog.pg_type t ON t.oid = a.atttypid
365              LEFT JOIN pg_catalog.pg_attrdef d ON (d.adrelid = c.oid AND d.adnum = a.attnum)
366                  WHERE relkind = 'r' AND c.relname = '$tablename' AND c.reltype > 0 AND a.attnum > 0
367               ORDER BY a.attnum";
369         $this->query_start($sql, null, SQL_QUERY_AUX);
370         $result = pg_query($this->pgsql, $sql);
371         $this->query_end($result);
373         if (!$result) {
374             return array();
375         }
376         while ($rawcolumn = pg_fetch_object($result)) {
378             $info = new stdClass();
379             $info->name = $rawcolumn->field;
380             $matches = null;
382             if ($rawcolumn->type === 'varchar') {
383                 $info->type          = 'varchar';
384                 $info->meta_type     = 'C';
385                 $info->max_length    = $rawcolumn->atttypmod - 4;
386                 $info->scale         = null;
387                 $info->not_null      = ($rawcolumn->attnotnull === 't');
388                 $info->has_default   = ($rawcolumn->atthasdef === 't');
389                 if ($info->has_default) {
390                     $parts = explode('::', $rawcolumn->adsrc);
391                     if (count($parts) > 1) {
392                         $info->default_value = reset($parts);
393                         $info->default_value = trim($info->default_value, "'");
394                     } else {
395                         $info->default_value = $rawcolumn->adsrc;
396                     }
397                 } else {
398                     $info->default_value = null;
399                 }
400                 $info->primary_key   = false;
401                 $info->binary        = false;
402                 $info->unsigned      = null;
403                 $info->auto_increment= false;
404                 $info->unique        = null;
406             } else if (preg_match('/int(\d)/i', $rawcolumn->type, $matches)) {
407                 $info->type = 'int';
408                 if (strpos($rawcolumn->adsrc, 'nextval') === 0) {
409                     $info->primary_key   = true;
410                     $info->meta_type     = 'R';
411                     $info->unique        = true;
412                     $info->auto_increment= true;
413                     $info->has_default   = false;
414                 } else {
415                     $info->primary_key   = false;
416                     $info->meta_type     = 'I';
417                     $info->unique        = null;
418                     $info->auto_increment= false;
419                     $info->has_default   = ($rawcolumn->atthasdef === 't');
420                 }
421                 $info->max_length    = $matches[1];
422                 $info->scale         = null;
423                 $info->not_null      = ($rawcolumn->attnotnull === 't');
424                 if ($info->has_default) {
425                     $info->default_value = trim($rawcolumn->adsrc, '()');
426                 } else {
427                     $info->default_value = null;
428                 }
429                 $info->binary        = false;
430                 $info->unsigned      = false;
432             } else if ($rawcolumn->type === 'numeric') {
433                 $info->type = $rawcolumn->type;
434                 $info->meta_type     = 'N';
435                 $info->primary_key   = false;
436                 $info->binary        = false;
437                 $info->unsigned      = null;
438                 $info->auto_increment= false;
439                 $info->unique        = null;
440                 $info->not_null      = ($rawcolumn->attnotnull === 't');
441                 $info->has_default   = ($rawcolumn->atthasdef === 't');
442                 if ($info->has_default) {
443                     $info->default_value = trim($rawcolumn->adsrc, '()');
444                 } else {
445                     $info->default_value = null;
446                 }
447                 $info->max_length    = $rawcolumn->atttypmod >> 16;
448                 $info->scale         = ($rawcolumn->atttypmod & 0xFFFF) - 4;
450             } else if (preg_match('/float(\d)/i', $rawcolumn->type, $matches)) {
451                 $info->type = 'float';
452                 $info->meta_type     = 'N';
453                 $info->primary_key   = false;
454                 $info->binary        = false;
455                 $info->unsigned      = null;
456                 $info->auto_increment= false;
457                 $info->unique        = null;
458                 $info->not_null      = ($rawcolumn->attnotnull === 't');
459                 $info->has_default   = ($rawcolumn->atthasdef === 't');
460                 if ($info->has_default) {
461                     $info->default_value = trim($rawcolumn->adsrc, '()');
462                 } else {
463                     $info->default_value = null;
464                 }
465                 // just guess expected number of deciaml places :-(
466                 if ($matches[1] == 8) {
467                     // total 15 digits
468                     $info->max_length = 8;
469                     $info->scale      = 7;
470                 } else {
471                     // total 6 digits
472                     $info->max_length = 4;
473                     $info->scale      = 2;
474                 }
476             } else if ($rawcolumn->type === 'text') {
477                 $info->type          = $rawcolumn->type;
478                 $info->meta_type     = 'X';
479                 $info->max_length    = -1;
480                 $info->scale         = null;
481                 $info->not_null      = ($rawcolumn->attnotnull === 't');
482                 $info->has_default   = ($rawcolumn->atthasdef === 't');
483                 if ($info->has_default) {
484                     $parts = explode('::', $rawcolumn->adsrc);
485                     if (count($parts) > 1) {
486                         $info->default_value = reset($parts);
487                         $info->default_value = trim($info->default_value, "'");
488                     } else {
489                         $info->default_value = $rawcolumn->adsrc;
490                     }
491                 } else {
492                     $info->default_value = null;
493                 }
494                 $info->primary_key   = false;
495                 $info->binary        = false;
496                 $info->unsigned      = null;
497                 $info->auto_increment= false;
498                 $info->unique        = null;
500             } else if ($rawcolumn->type === 'bytea') {
501                 $info->type          = $rawcolumn->type;
502                 $info->meta_type     = 'B';
503                 $info->max_length    = -1;
504                 $info->scale         = null;
505                 $info->not_null      = ($rawcolumn->attnotnull === 't');
506                 $info->has_default   = false;
507                 $info->default_value = null;
508                 $info->primary_key   = false;
509                 $info->binary        = true;
510                 $info->unsigned      = null;
511                 $info->auto_increment= false;
512                 $info->unique        = null;
514             }
516             $this->columns[$table][$info->name] = new database_column_info($info);
517         }
519         pg_free_result($result);
521         return $this->columns[$table];
522     }
524     /**
525      * Normalise values based in RDBMS dependencies (booleans, LOBs...)
526      *
527      * @param database_column_info $column column metadata corresponding with the value we are going to normalise
528      * @param mixed $value value we are going to normalise
529      * @return mixed the normalised value
530      */
531     protected function normalise_value($column, $value) {
532         if (is_bool($value)) { // Always, convert boolean to int
533             $value = (int)$value;
535         } else if ($column->meta_type === 'B') { // BLOB detected, we return 'blob' array instead of raw value to allow
536             if (!is_null($value)) {             // binding/executing code later to know about its nature
537                 $value = array('blob' => $value);
538             }
540         } else if ($value === '') {
541             if ($column->meta_type === 'I' or $column->meta_type === 'F' or $column->meta_type === 'N') {
542                 $value = 0; // prevent '' problems in numeric fields
543             }
544         }
545         return $value;
546     }
548     /**
549      * Is db in unicode mode?
550      * @return bool
551      */
552     public function setup_is_unicodedb() {
553     /// Get PostgreSQL server_encoding value
554         $sql = "SHOW server_encoding";
555         $this->query_start($sql, null, SQL_QUERY_AUX);
556         $result = pg_query($this->pgsql, $sql);
557         $this->query_end($result);
559         if (!$result) {
560             return false;
561         }
562         $rawcolumn = pg_fetch_object($result);
563         $encoding = $rawcolumn->server_encoding;
564         pg_free_result($result);
566         return (strtoupper($encoding) == 'UNICODE' || strtoupper($encoding) == 'UTF8');
567     }
569     /**
570      * Do NOT use in code, to be used by database_manager only!
571      * @param string $sql query
572      * @return bool true
573      * @throws dml_exception A DML specific exception is thrown for any errors.
574      */
575     public function change_database_structure($sql) {
576         $this->reset_caches();
578         $this->query_start($sql, null, SQL_QUERY_STRUCTURE);
579         $result = pg_query($this->pgsql, $sql);
580         $this->query_end($result);
582         pg_free_result($result);
583         return true;
584     }
586     /**
587      * Execute general sql query. Should be used only when no other method suitable.
588      * Do NOT use this to make changes in db structure, use database_manager::execute_sql() instead!
589      * @param string $sql query
590      * @param array $params query parameters
591      * @return bool true
592      * @throws dml_exception A DML specific exception is thrown for any errors.
593      */
594     public function execute($sql, array $params=null) {
595         list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
597         if (strpos($sql, ';') !== false) {
598             throw new coding_exception('moodle_database::execute() Multiple sql statements found or bound parameters not used properly in query!');
599         }
601         $this->query_start($sql, $params, SQL_QUERY_UPDATE);
602         $result = pg_query_params($this->pgsql, $sql, $params);
603         $this->query_end($result);
605         pg_free_result($result);
606         return true;
607     }
609     /**
610      * Get a number of records as a moodle_recordset using a SQL statement.
611      *
612      * Since this method is a little less readable, use of it should be restricted to
613      * code where it's possible there might be large datasets being returned.  For known
614      * small datasets use get_records_sql - it leads to simpler code.
615      *
616      * The return type is as for @see function get_recordset.
617      *
618      * @param string $sql the SQL select query to execute.
619      * @param array $params array of sql parameters
620      * @param int $limitfrom return a subset of records, starting at this point (optional, required if $limitnum is set).
621      * @param int $limitnum return a subset comprising this many records (optional, required if $limitfrom is set).
622      * @return moodle_recordset instance
623      * @throws dml_exception A DML specific exception is thrown for any errors.
624      */
625     public function get_recordset_sql($sql, array $params=null, $limitfrom=0, $limitnum=0) {
626         $limitfrom = (int)$limitfrom;
627         $limitnum  = (int)$limitnum;
628         $limitfrom = ($limitfrom < 0) ? 0 : $limitfrom;
629         $limitnum  = ($limitnum < 0)  ? 0 : $limitnum;
630         if ($limitfrom or $limitnum) {
631             if ($limitnum < 1) {
632                 $limitnum = "ALL";
633             } else if (PHP_INT_MAX - $limitnum < $limitfrom) {
634                 // this is a workaround for weird max int problem
635                 $limitnum = "ALL";
636             }
637             $sql .= " LIMIT $limitnum OFFSET $limitfrom";
638         }
640         list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
642         $this->query_start($sql, $params, SQL_QUERY_SELECT);
643         $result = pg_query_params($this->pgsql, $sql, $params);
644         $this->query_end($result);
646         return $this->create_recordset($result);
647     }
649     protected function create_recordset($result) {
650         return new pgsql_native_moodle_recordset($result, $this->bytea_oid);
651     }
653     /**
654      * Get a number of records as an array of objects using a SQL statement.
655      *
656      * Return value as for @see function get_records.
657      *
658      * @param string $sql the SQL select query to execute. The first column of this SELECT statement
659      *   must be a unique value (usually the 'id' field), as it will be used as the key of the
660      *   returned array.
661      * @param array $params array of sql parameters
662      * @param int $limitfrom return a subset of records, starting at this point (optional, required if $limitnum is set).
663      * @param int $limitnum return a subset comprising this many records (optional, required if $limitfrom is set).
664      * @return array of objects, or empty array if no records were found
665      * @throws dml_exception A DML specific exception is thrown for any errors.
666      */
667     public function get_records_sql($sql, array $params=null, $limitfrom=0, $limitnum=0) {
668         $limitfrom = (int)$limitfrom;
669         $limitnum  = (int)$limitnum;
670         $limitfrom = ($limitfrom < 0) ? 0 : $limitfrom;
671         $limitnum  = ($limitnum < 0)  ? 0 : $limitnum;
672         if ($limitfrom or $limitnum) {
673             if ($limitnum < 1) {
674                 $limitnum = "ALL";
675             } else if (PHP_INT_MAX - $limitnum < $limitfrom) {
676                 // this is a workaround for weird max int problem
677                 $limitnum = "ALL";
678             }
679             $sql .= " LIMIT $limitnum OFFSET $limitfrom";
680         }
682         list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
683         $this->query_start($sql, $params, SQL_QUERY_SELECT);
684         $result = pg_query_params($this->pgsql, $sql, $params);
685         $this->query_end($result);
687         // find out if there are any blobs
688         $numrows = pg_num_fields($result);
689         $blobs = array();
690         for($i=0; $i<$numrows; $i++) {
691             $type_oid = pg_field_type_oid($result, $i);
692             if ($type_oid == $this->bytea_oid) {
693                 $blobs[] = pg_field_name($result, $i);
694             }
695         }
697         $rows = pg_fetch_all($result);
698         pg_free_result($result);
700         $return = array();
701         if ($rows) {
702             foreach ($rows as $row) {
703                 $id = reset($row);
704                 if ($blobs) {
705                     foreach ($blobs as $blob) {
706                         // note: in PostgreSQL 9.0 the returned blobs are hexencoded by default - see http://www.postgresql.org/docs/9.0/static/runtime-config-client.html#GUC-BYTEA-OUTPUT
707                         $row[$blob] = $row[$blob] !== null ? pg_unescape_bytea($row[$blob]) : null;
708                     }
709                 }
710                 if (isset($return[$id])) {
711                     $colname = key($row);
712                     debugging("Did you remember to make the first column something unique in your call to get_records? Duplicate value '$id' found in column '$colname'.", DEBUG_DEVELOPER);
713                 }
714                 $return[$id] = (object)$row;
715             }
716         }
718         return $return;
719     }
721     /**
722      * Selects records and return values (first field) as an array using a SQL statement.
723      *
724      * @param string $sql The SQL query
725      * @param array $params array of sql parameters
726      * @return array of values
727      * @throws dml_exception A DML specific exception is thrown for any errors.
728      */
729     public function get_fieldset_sql($sql, array $params=null) {
730         list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
732         $this->query_start($sql, $params, SQL_QUERY_SELECT);
733         $result = pg_query_params($this->pgsql, $sql, $params);
734         $this->query_end($result);
736         $return = pg_fetch_all_columns($result, 0);
737         pg_free_result($result);
739         return $return;
740     }
742     /**
743      * Insert new record into database, as fast as possible, no safety checks, lobs not supported.
744      * @param string $table name
745      * @param mixed $params data record as object or array
746      * @param bool $returnit return it of inserted record
747      * @param bool $bulk true means repeated inserts expected
748      * @param bool $customsequence true if 'id' included in $params, disables $returnid
749      * @return bool|int true or new id
750      * @throws dml_exception A DML specific exception is thrown for any errors.
751      */
752     public function insert_record_raw($table, $params, $returnid=true, $bulk=false, $customsequence=false) {
753         if (!is_array($params)) {
754             $params = (array)$params;
755         }
757         $returning = "";
759         if ($customsequence) {
760             if (!isset($params['id'])) {
761                 throw new coding_exception('moodle_database::insert_record_raw() id field must be specified if custom sequences used.');
762             }
763             $returnid = false;
764         } else {
765             if ($returnid) {
766                 $returning = "RETURNING id";
767                 unset($params['id']);
768             } else {
769                 unset($params['id']);
770             }
771         }
773         if (empty($params)) {
774             throw new coding_exception('moodle_database::insert_record_raw() no fields found.');
775         }
777         $fields = implode(',', array_keys($params));
778         $values = array();
779         $count = count($params);
780         for ($i=1; $i<=$count; $i++) {
781             $values[] = "\$".$i;
782         }
783         $values = implode(',', $values);
785         $sql = "INSERT INTO {$this->prefix}$table ($fields) VALUES($values) $returning";
786         $this->query_start($sql, $params, SQL_QUERY_INSERT);
787         $result = pg_query_params($this->pgsql, $sql, $params);
788         $this->query_end($result);
790         if ($returning !== "") {
791             $row = pg_fetch_assoc($result);
792             $params['id'] = reset($row);
793         }
794         pg_free_result($result);
796         if (!$returnid) {
797             return true;
798         }
800         return (int)$params['id'];
801     }
803     /**
804      * Insert a record into a table and return the "id" field if required.
805      *
806      * Some conversions and safety checks are carried out. Lobs are supported.
807      * If the return ID isn't required, then this just reports success as true/false.
808      * $data is an object containing needed data
809      * @param string $table The database table to be inserted into
810      * @param object $data A data object with values for one or more fields in the record
811      * @param bool $returnid Should the id of the newly created record entry be returned? If this option is not requested then true/false is returned.
812      * @return bool|int true or new id
813      * @throws dml_exception A DML specific exception is thrown for any errors.
814      */
815     public function insert_record($table, $dataobject, $returnid=true, $bulk=false) {
816         $dataobject = (array)$dataobject;
818         $columns = $this->get_columns($table);
819         $cleaned = array();
820         $blobs   = array();
822         foreach ($dataobject as $field=>$value) {
823             if ($field === 'id') {
824                 continue;
825             }
826             if (!isset($columns[$field])) {
827                 continue;
828             }
829             $column = $columns[$field];
830             $normalised_value = $this->normalise_value($column, $value);
831             if (is_array($normalised_value) && array_key_exists('blob', $normalised_value)) {
832                 $cleaned[$field] = '@#BLOB#@';
833                 $blobs[$field] = $normalised_value['blob'];
834             } else {
835                 $cleaned[$field] = $normalised_value;
836             }
837         }
839         if (empty($blobs)) {
840             return $this->insert_record_raw($table, $cleaned, $returnid, $bulk);
841         }
843         $id = $this->insert_record_raw($table, $cleaned, true, $bulk);
845         foreach ($blobs as $key=>$value) {
846             $value = pg_escape_bytea($this->pgsql, $value);
847             $sql = "UPDATE {$this->prefix}$table SET $key = '$value'::bytea WHERE id = $id";
848             $this->query_start($sql, NULL, SQL_QUERY_UPDATE);
849             $result = pg_query($this->pgsql, $sql);
850             $this->query_end($result);
851             if ($result !== false) {
852                 pg_free_result($result);
853             }
854         }
856         return ($returnid ? $id : true);
858     }
860     /**
861      * Import a record into a table, id field is required.
862      * Safety checks are NOT carried out. Lobs are supported.
863      *
864      * @param string $table name of database table to be inserted into
865      * @param object $dataobject A data object with values for one or more fields in the record
866      * @return bool true
867      * @throws dml_exception A DML specific exception is thrown for any errors.
868      */
869     public function import_record($table, $dataobject) {
870         $dataobject = (array)$dataobject;
872         $columns = $this->get_columns($table);
873         $cleaned = array();
874         $blobs   = array();
876         foreach ($dataobject as $field=>$value) {
877             if (!isset($columns[$field])) {
878                 continue;
879             }
880             if ($columns[$field]->meta_type === 'B') {
881                 if (!is_null($value)) {
882                     $cleaned[$field] = '@#BLOB#@';
883                     $blobs[$field] = $value;
884                     continue;
885                 }
886             }
888             $cleaned[$field] = $value;
889         }
891         $this->insert_record_raw($table, $cleaned, false, true, true);
892         $id = $dataobject['id'];
894         foreach ($blobs as $key=>$value) {
895             $value = pg_escape_bytea($this->pgsql, $value);
896             $sql = "UPDATE {$this->prefix}$table SET $key = '$value'::bytea WHERE id = $id";
897             $this->query_start($sql, NULL, SQL_QUERY_UPDATE);
898             $result = pg_query($this->pgsql, $sql);
899             $this->query_end($result);
900             if ($result !== false) {
901                 pg_free_result($result);
902             }
903         }
905         return true;
906     }
908     /**
909      * Update record in database, as fast as possible, no safety checks, lobs not supported.
910      * @param string $table name
911      * @param mixed $params data record as object or array
912      * @param bool true means repeated updates expected
913      * @return bool true
914      * @throws dml_exception A DML specific exception is thrown for any errors.
915      */
916     public function update_record_raw($table, $params, $bulk=false) {
917         $params = (array)$params;
919         if (!isset($params['id'])) {
920             throw new coding_exception('moodle_database::update_record_raw() id field must be specified.');
921         }
922         $id = $params['id'];
923         unset($params['id']);
925         if (empty($params)) {
926             throw new coding_exception('moodle_database::update_record_raw() no fields found.');
927         }
929         $i = 1;
931         $sets = array();
932         foreach ($params as $field=>$value) {
933             $sets[] = "$field = \$".$i++;
934         }
936         $params[] = $id; // last ? in WHERE condition
938         $sets = implode(',', $sets);
939         $sql = "UPDATE {$this->prefix}$table SET $sets WHERE id=\$".$i;
941         $this->query_start($sql, $params, SQL_QUERY_UPDATE);
942         $result = pg_query_params($this->pgsql, $sql, $params);
943         $this->query_end($result);
945         pg_free_result($result);
946         return true;
947     }
949     /**
950      * Update a record in a table
951      *
952      * $dataobject is an object containing needed data
953      * Relies on $dataobject having a variable "id" to
954      * specify the record to update
955      *
956      * @param string $table The database table to be checked against.
957      * @param object $dataobject An object with contents equal to fieldname=>fieldvalue. Must have an entry for 'id' to map to the table specified.
958      * @param bool true means repeated updates expected
959      * @return bool true
960      * @throws dml_exception A DML specific exception is thrown for any errors.
961      */
962     public function update_record($table, $dataobject, $bulk=false) {
963         $dataobject = (array)$dataobject;
965         $columns = $this->get_columns($table);
966         $cleaned = array();
967         $blobs   = array();
969         foreach ($dataobject as $field=>$value) {
970             if (!isset($columns[$field])) {
971                 continue;
972             }
973             $column = $columns[$field];
974             $normalised_value = $this->normalise_value($column, $value);
975             if (is_array($normalised_value) && array_key_exists('blob', $normalised_value)) {
976                 $cleaned[$field] = '@#BLOB#@';
977                 $blobs[$field] = $normalised_value['blob'];
978             } else {
979                 $cleaned[$field] = $normalised_value;
980             }
981         }
983         $this->update_record_raw($table, $cleaned, $bulk);
985         if (empty($blobs)) {
986             return true;
987         }
989         $id = (int)$dataobject['id'];
991         foreach ($blobs as $key=>$value) {
992             $value = pg_escape_bytea($this->pgsql, $value);
993             $sql = "UPDATE {$this->prefix}$table SET $key = '$value'::bytea WHERE id = $id";
994             $this->query_start($sql, NULL, SQL_QUERY_UPDATE);
995             $result = pg_query($this->pgsql, $sql);
996             $this->query_end($result);
998             pg_free_result($result);
999         }
1001         return true;
1002     }
1004     /**
1005      * Set a single field in every table record which match a particular WHERE clause.
1006      *
1007      * @param string $table The database table to be checked against.
1008      * @param string $newfield the field to set.
1009      * @param string $newvalue the value to set the field to.
1010      * @param string $select A fragment of SQL to be used in a where clause in the SQL call.
1011      * @param array $params array of sql parameters
1012      * @return bool true
1013      * @throws dml_exception A DML specific exception is thrown for any errors.
1014      */
1015     public function set_field_select($table, $newfield, $newvalue, $select, array $params=null) {
1017         if ($select) {
1018             $select = "WHERE $select";
1019         }
1020         if (is_null($params)) {
1021             $params = array();
1022         }
1023         list($select, $params, $type) = $this->fix_sql_params($select, $params);
1024         $i = count($params)+1;
1026     /// Get column metadata
1027         $columns = $this->get_columns($table);
1028         $column = $columns[$newfield];
1030         $normalised_value = $this->normalise_value($column, $newvalue);
1031         if (is_array($normalised_value) && array_key_exists('blob', $normalised_value)) {
1032         /// Update BYTEA and return
1033             $normalised_value = pg_escape_bytea($this->pgsql, $normalised_value['blob']);
1034             $sql = "UPDATE {$this->prefix}$table SET $newfield = '$normalised_value'::bytea $select";
1035             $this->query_start($sql, NULL, SQL_QUERY_UPDATE);
1036             $result = pg_query_params($this->pgsql, $sql, $params);
1037             $this->query_end($result);
1038             pg_free_result($result);
1039             return true;
1040         }
1042         if (is_null($normalised_value)) {
1043             $newfield = "$newfield = NULL";
1044         } else {
1045             $newfield = "$newfield = \$".$i;
1046             $params[] = $normalised_value;
1047         }
1048         $sql = "UPDATE {$this->prefix}$table SET $newfield $select";
1050         $this->query_start($sql, $params, SQL_QUERY_UPDATE);
1051         $result = pg_query_params($this->pgsql, $sql, $params);
1052         $this->query_end($result);
1054         pg_free_result($result);
1056         return true;
1057     }
1059     /**
1060      * Delete one or more records from a table which match a particular WHERE clause.
1061      *
1062      * @param string $table The database table to be checked against.
1063      * @param string $select A fragment of SQL to be used in a where clause in the SQL call (used to define the selection criteria).
1064      * @param array $params array of sql parameters
1065      * @return bool true
1066      * @throws dml_exception A DML specific exception is thrown for any errors.
1067      */
1068     public function delete_records_select($table, $select, array $params=null) {
1069         if ($select) {
1070             $select = "WHERE $select";
1071         }
1072         $sql = "DELETE FROM {$this->prefix}$table $select";
1074         list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
1076         $this->query_start($sql, $params, SQL_QUERY_UPDATE);
1077         $result = pg_query_params($this->pgsql, $sql, $params);
1078         $this->query_end($result);
1080         pg_free_result($result);
1082         return true;
1083     }
1085     /**
1086      * Returns 'LIKE' part of a query.
1087      *
1088      * @param string $fieldname usually name of the table column
1089      * @param string $param usually bound query parameter (?, :named)
1090      * @param bool $casesensitive use case sensitive search
1091      * @param bool $accensensitive use accent sensitive search (not all databases support accent insensitive)
1092      * @param bool $notlike true means "NOT LIKE"
1093      * @param string $escapechar escape char for '%' and '_'
1094      * @return string SQL code fragment
1095      */
1096     public function sql_like($fieldname, $param, $casesensitive = true, $accentsensitive = true, $notlike = false, $escapechar = '\\') {
1097         if (strpos($param, '%') !== false) {
1098             debugging('Potential SQL injection detected, sql_ilike() expects bound parameters (? or :named)');
1099         }
1100         $escapechar = pg_escape_string($this->pgsql, $escapechar); // prevents problems with C-style escapes of enclosing '\'
1102         // postgresql does not support accent insensitive text comparisons, sorry
1103         if ($casesensitive) {
1104             $LIKE = $notlike ? 'NOT LIKE' : 'LIKE';
1105         } else {
1106             $LIKE = $notlike ? 'NOT ILIKE' : 'ILIKE';
1107         }
1108         return "$fieldname $LIKE $param ESCAPE '$escapechar'";
1109     }
1111     public function sql_ilike() {
1112         debugging('sql_ilike() is deprecated, please use sql_like() instead');
1113         return 'ILIKE';
1114     }
1116     public function sql_bitxor($int1, $int2) {
1117         return '((' . $int1 . ') # (' . $int2 . '))';
1118     }
1120     public function sql_cast_char2int($fieldname, $text=false) {
1121         return ' CAST(' . $fieldname . ' AS INT) ';
1122     }
1124     public function sql_cast_char2real($fieldname, $text=false) {
1125         return " $fieldname::real ";
1126     }
1128     public function sql_concat() {
1129         $arr = func_get_args();
1130         $s = implode(' || ', $arr);
1131         if ($s === '') {
1132             return " '' ";
1133         }
1134         // Add always empty string element so integer-exclusive concats
1135         // will work without needing to cast each element explicity
1136         return " '' || $s ";
1137     }
1139     public function sql_concat_join($separator="' '", $elements=array()) {
1140         for ($n=count($elements)-1; $n > 0 ; $n--) {
1141             array_splice($elements, $n, 0, $separator);
1142         }
1143         $s = implode(' || ', $elements);
1144         if ($s === '') {
1145             return " '' ";
1146         }
1147         return " $s ";
1148     }
1150     public function sql_regex_supported() {
1151         return true;
1152     }
1154     public function sql_regex($positivematch=true) {
1155         return $positivematch ? '~*' : '!~*';
1156     }
1158 /// session locking
1159     public function session_lock_supported() {
1160         return true;
1161     }
1163     /**
1164      * Obtain session lock
1165      * @param int $rowid id of the row with session record
1166      * @param int $timeout max allowed time to wait for the lock in seconds
1167      * @return bool success
1168      */
1169     public function get_session_lock($rowid, $timeout) {
1170         // NOTE: there is a potential locking problem for database running
1171         //       multiple instances of moodle, we could try to use pg_advisory_lock(int, int),
1172         //       luckily there is not a big chance that they would collide
1173         if (!$this->session_lock_supported()) {
1174             return;
1175         }
1177         parent::get_session_lock($rowid, $timeout);
1179         $timeoutmilli = $timeout * 1000;
1181         $sql = "SET statement_timeout TO $timeoutmilli";
1182         $this->query_start($sql, null, SQL_QUERY_AUX);
1183         $result = pg_query($this->pgsql, $sql);
1184         $this->query_end($result);
1186         if ($result) {
1187             pg_free_result($result);
1188         }
1190         $sql = "SELECT pg_advisory_lock($rowid)";
1191         $this->query_start($sql, null, SQL_QUERY_AUX);
1192         $start = time();
1193         $result = pg_query($this->pgsql, $sql);
1194         $end = time();
1195         try {
1196             $this->query_end($result);
1197         } catch (dml_exception $ex) {
1198             if ($end - $start >= $timeout) {
1199                 throw new dml_sessionwait_exception();
1200             } else {
1201                 throw $ex;
1202             }
1203         }
1205         if ($result) {
1206             pg_free_result($result);
1207         }
1209         $sql = "SET statement_timeout TO DEFAULT";
1210         $this->query_start($sql, null, SQL_QUERY_AUX);
1211         $result = pg_query($this->pgsql, $sql);
1212         $this->query_end($result);
1214         if ($result) {
1215             pg_free_result($result);
1216         }
1217     }
1219     public function release_session_lock($rowid) {
1220         if (!$this->session_lock_supported()) {
1221             return;
1222         }
1223         parent::release_session_lock($rowid);
1225         $sql = "SELECT pg_advisory_unlock($rowid)";
1226         $this->query_start($sql, null, SQL_QUERY_AUX);
1227         $result = pg_query($this->pgsql, $sql);
1228         $this->query_end($result);
1230         if ($result) {
1231             pg_free_result($result);
1232         }
1233     }
1235 /// transactions
1236     /**
1237      * Driver specific start of real database transaction,
1238      * this can not be used directly in code.
1239      * @return void
1240      */
1241     protected function begin_transaction() {
1242         $sql = "BEGIN ISOLATION LEVEL READ COMMITTED";
1243         $this->query_start($sql, NULL, SQL_QUERY_AUX);
1244         $result = pg_query($this->pgsql, $sql);
1245         $this->query_end($result);
1247         pg_free_result($result);
1248     }
1250     /**
1251      * Driver specific commit of real database transaction,
1252      * this can not be used directly in code.
1253      * @return void
1254      */
1255     protected function commit_transaction() {
1256         $sql = "COMMIT";
1257         $this->query_start($sql, NULL, SQL_QUERY_AUX);
1258         $result = pg_query($this->pgsql, $sql);
1259         $this->query_end($result);
1261         pg_free_result($result);
1262     }
1264     /**
1265      * Driver specific abort of real database transaction,
1266      * this can not be used directly in code.
1267      * @return void
1268      */
1269     protected function rollback_transaction() {
1270         $sql = "ROLLBACK";
1271         $this->query_start($sql, NULL, SQL_QUERY_AUX);
1272         $result = pg_query($this->pgsql, $sql);
1273         $this->query_end($result);
1275         pg_free_result($result);
1276     }
1278     /**
1279      * Helper function trimming (whitespace + quotes) any string
1280      * needed because PG uses to enclose with double quotes some
1281      * fields in indexes definition and others
1282      *
1283      * @param string $str string to apply whitespace + quotes trim
1284      * @return string trimmed string
1285      */
1286     private function trim_quotes($str) {
1287         return trim(trim($str), "'\"");
1288     }