]> git.mjollnir.org Git - moodle.git/commitdiff
MDL-17020 dml: native pgsql driver - still buggy, works only partially
authorskodak <skodak>
Sun, 26 Oct 2008 22:36:13 +0000 (22:36 +0000)
committerskodak <skodak>
Sun, 26 Oct 2008 22:36:13 +0000 (22:36 +0000)
lib/dml/pgsql_native_moodle_database.php [new file with mode: 0644]
lib/dml/pgsql_native_moodle_recordset.php [new file with mode: 0644]

diff --git a/lib/dml/pgsql_native_moodle_database.php b/lib/dml/pgsql_native_moodle_database.php
new file mode 100644 (file)
index 0000000..6a3ee82
--- /dev/null
@@ -0,0 +1,965 @@
+<?php  //$Id$
+
+require_once($CFG->libdir.'/dml/moodle_database.php');
+require_once($CFG->libdir.'/dml/pgsql_native_moodle_recordset.php');
+
+/**
+ * Native pgsql class representing moodle database interface.
+ * @package dml
+ */
+class pgsql_native_moodle_database extends moodle_database {
+
+    protected $pgsql = null;
+    protected $debug  = false;
+
+    /**
+     * Detects if all needed PHP stuff installed.
+     * Note: can be used before connect()
+     * @return mixed true if ok, string if something
+     */
+    public function driver_installed() {
+        if (!extension_loaded('pgsql')) {
+            return get_string('pgsqlextensionisnotpresentinphp', 'install');
+        }
+        return true;
+    }
+
+    /**
+     * Returns database family type - describes SQL dialect
+     * Note: can be used before connect()
+     * @return string db family name (mysql, postgres, mssql, oracle, etc.)
+     */
+    public function get_dbfamily() {
+        return 'postgres';
+    }
+
+    /**
+     * Returns more specific database driver type
+     * Note: can be used before connect()
+     * @return string db type mysql, pgsql, postgres7
+     */
+    protected function get_dbtype() {
+        return 'pgsql';
+    }
+
+    /**
+     * Returns general database library name
+     * Note: can be used before connect()
+     * @return string db type adodb, pdo, native
+     */
+    protected function get_dblibrary() {
+        return 'native';
+    }
+
+    /**
+     * Returns localised database type name
+     * Note: can be used before connect()
+     * @return string
+     */
+    public function get_name() {
+        return get_string('nativepgsql', 'install'); // TODO: localise
+    }
+
+    /**
+     * Returns localised database description
+     * Note: can be used before connect()
+     * @return string
+     */
+    public function get_configuration_hints() {
+        return get_string('databasesettingssub_postgres7', 'install'); // TODO: improve
+    }
+
+    /**
+     * Connect to db
+     * Must be called before other methods.
+     * @param string $dbhost
+     * @param string $dbuser
+     * @param string $dbpass
+     * @param string $dbname
+     * @param bool $dbpersist
+     * @param mixed $prefix string means moodle db prefix, false used for external databases where prefix not used
+     * @param array $dboptions driver specific options
+     * @return bool success
+     */
+    public function connect($dbhost, $dbuser, $dbpass, $dbname, $dbpersist, $prefix, array $dboptions=null) {
+        global $CFG;
+
+        $this->store_settings($dbhost, $dbuser, $dbpass, $dbname, false, $prefix, $dboptions);
+
+        //TODO: handle both port and socket connection
+        $this->pgsql = pg_connect("host='{$this->dbhost}' user='{$this->dbuser}' password='{$this->dbpass}' dbname='{$this->dbname}'");
+        $status = pg_connection_status($this->pgsql);
+        if ($status === PGSQL_CONNECTION_BAD) {
+            $this->pgsql = null;
+            return false;
+        }
+        pg_set_client_encoding($this->pgsql, 'utf8');
+        return true;
+    }
+
+    /**
+     * Close database connection and release all resources
+     * and memory (especially circular memory references).
+     * Do NOT use connect() again, create a new instance if needed.
+     */
+    public function dispose() {
+        if ($this->pgsql) {
+            pg_close($this->pgsql);
+            $this->pgsql = null;
+        }
+        parent::dispose();
+    }
+
+    /**
+     * Returns database server info array
+     * @return array
+     */
+    public function get_server_info() {
+        static $info;
+        if (!$info) {
+            $info = pg_version($this->pgsql);
+        }
+        return array('description'=>$info['server'], 'version'=>$info['server']);
+    }
+
+    protected function is_min_version($version) {
+        $server = $this->get_server_info();
+        $server = $server['version'];
+        return version_compare($server, $version, '>=');
+    }
+
+    /**
+     * Returns supported query parameter types
+     * @return bitmask
+     */
+    protected function allowed_param_types() {
+        return SQL_PARAMS_DOLLAR;
+    }
+
+    /**
+     * Returns last error reported by database engine.
+     */
+    public function get_last_error() {
+        return pg_last_error($this->pgsql);
+    }
+
+    /**
+     * Return tables in database WITHOUT current prefix
+     * @return array of table names in lowercase and without prefix
+     */
+    public function get_tables() {
+        $this->reads++;
+        $tables = array();
+        $prefix = str_replace('_', '\\\\_', $this->prefix);
+        $sql = "SELECT tablename
+                  FROM pg_catalog.pg_tables
+                 WHERE tablename LIKE '$prefix%'";
+        if ($result = pg_query($this->pgsql, $sql)) {
+            while ($row = pg_fetch_row($result)) {
+                $tablename = reset($row);
+                if (strpos($tablename, $this->prefix) !== 0) {
+                    continue;
+                }
+                $tablename = substr($tablename, strlen($this->prefix));
+                $tables[$tablename] = $tablename;
+            }
+            pg_free_result($result);
+        }
+        return $tables;
+    }
+
+    /**
+     * Return table indexes - everything lowercased
+     * @return array of arrays
+     */
+    public function get_indexes($table) {
+        $indexes = array();
+        $tablename = $this->prefix.$table;
+
+        $sql = "SELECT *
+                  FROM pg_catalog.pg_indexes
+                 WHERE tablename = '$tablename'";
+        if ($result = pg_query($this->pgsql, $sql)) {
+            while ($row = pg_fetch_assoc($result)) {
+                if (!preg_match('/CREATE (|UNIQUE )INDEX ([^\s]+) ON '.$tablename.' USING ([^\s]+) \(([^\)]+)\)/i', $row['indexdef'], $matches)) {
+                    continue;
+                }
+                if ($matches[4] === 'id') {
+                    continue;
+                }
+                $indexes[$matches[2]] = array('unique'=>!empty($matches[1]),
+                                              'columns'=>explode(',', $matches[4]));
+            }
+            pg_free_result($result);
+        }
+        return $indexes;
+    }
+
+    /**
+     * Returns datailed information about columns in table. This information is cached internally.
+     * @param string $table name
+     * @param bool $usecache
+     * @return array array of database_column_info objects indexed with column names
+     */
+    public function get_columns($table, $usecache=true) {
+        if ($usecache and isset($this->columns[$table])) {
+            return $this->columns[$table];
+        }
+
+        $this->columns[$table] = array();
+
+        $tablename = $this->prefix.$table;
+
+        $sql = "SELECT a.attnum, a.attname AS field, t.typname AS type, a.attlen, a.atttypmod, a.attnotnull, a.atthasdef, d.adsrc
+                  FROM pg_catalog.pg_class c
+                  JOIN pg_catalog.pg_attribute a ON a.attrelid = c.oid 
+                  JOIN pg_catalog.pg_type t ON t.oid = a.atttypid
+             LEFT JOIN pg_catalog.pg_attrdef d ON (d.adrelid = c.oid AND d.adnum = a.attnum)
+                 WHERE relkind = 'r' AND c.relname = '$tablename' AND c.reltype > 0 AND a.attnum > 0
+              ORDER BY a.attnum";
+
+        if (!$result = pg_query($this->pgsql, $sql)) {
+            return array();
+        }
+        while ($rawcolumn = pg_fetch_object($result)) {
+
+            $info = new object();
+            $info->name = $rawcolumn->field;
+            $matches = null;
+
+            if ($rawcolumn->type === 'varchar') {
+                //TODO add some basic enum support here
+                $info->type          = 'varchar';
+                $info->meta_type     = 'C';
+                $info->max_length    = $rawcolumn->atttypmod - 4;
+                $info->scale         = null;
+                $info->not_null      = (bool)$rawcolumn->attnotnull;
+                $info->has_default   = (bool)$rawcolumn->atthasdef;
+                if ($info->has_default) {
+                    $parts = explode('::', $rawcolumn->adsrc);
+                    if (count($parts) > 1) {
+                        $info->default_value = reset($parts);
+                        $info->default_value = trim($info->default_value, "'");
+                    } else {
+                        $info->default_value = $rawcolumn->adsrc;
+                    }
+                } else {
+                    $info->default_value = null;
+                }
+                $info->primary_key   = false;
+                $info->binary        = false;
+                $info->unsigned      = null;
+                $info->auto_increment= false;
+                $info->unique        = null;
+
+            } else if (preg_match('/int(\d)/i', $rawcolumn->type, $matches)) {
+                $info->type = 'int';
+                if (strpos($rawcolumn->adsrc, 'nextval') === 0) {
+                    $info->primary_key   = true;
+                    $info->meta_type     = 'R';
+                    $info->unique        = true;
+                    $info->auto_increment= true;
+                    $info->has_default   = false;
+                } else {
+                    $info->primary_key   = false;
+                    $info->meta_type     = 'I';
+                    $info->unique        = null;
+                    $info->auto_increment= false;
+                    $info->has_default   = (bool)$rawcolumn->atthasdef;
+                }
+                $info->max_length    = $matches[1];
+                $info->scale         = null;
+                $info->not_null      = (bool)$rawcolumn->attnotnull;
+                if ($info->has_default) {
+                    $info->default_value = $rawcolumn->adsrc;
+                } else {
+                    $info->default_value = null;
+                }
+                $info->binary        = false;
+                $info->unsigned      = false;
+
+            } else if ($rawcolumn->type === 'numeric') {
+                $info->type = $rawcolumn->type;
+                $info->meta_type     = 'N';
+                $info->primary_key   = false;
+                $info->binary        = false;
+                $info->unsigned      = null;
+                $info->auto_increment= false;
+                $info->unique        = null;
+                $info->not_null      = (bool)$rawcolumn->attnotnull;
+                $info->has_default   = (bool)$rawcolumn->atthasdef;
+                if ($info->has_default) {
+                    $info->default_value = $rawcolumn->adsrc;
+                } else {
+                    $info->default_value = null;
+                }
+                $info->max_length    = $rawcolumn->atttypmod >> 16;
+                $info->scale         = ($rawcolumn->atttypmod & 0xFFFF) - 4;
+
+            } else if (preg_match('/float(\d)/i', $rawcolumn->type, $matches)) {
+                $info->type = 'float';
+                $info->meta_type     = 'N';
+                $info->primary_key   = false;
+                $info->binary        = false;
+                $info->unsigned      = null;
+                $info->auto_increment= false;
+                $info->unique        = null;
+                $info->not_null      = (bool)$rawcolumn->attnotnull;
+                $info->has_default   = (bool)$rawcolumn->atthasdef;
+                if ($info->has_default) {
+                    $info->default_value = $rawcolumn->adsrc;
+                } else {
+                    $info->default_value = null;
+                }
+                // just guess expected number of deciaml places :-(
+                if ($matches[1] == 8) {
+                    // total 15 digits
+                    $info->max_length = 8;
+                    $info->scale      = 7;
+                } else {
+                    // total 6 digits
+                    $info->max_length = 4;
+                    $info->scale      = 2;
+                }
+
+            } else if ($rawcolumn->type === 'text') {
+                $info->type          = $rawcolumn->type;
+                $info->meta_type     = 'X';
+                $info->max_length    = -1;
+                $info->scale         = null;
+                $info->not_null      = (bool)$rawcolumn->attnotnull;
+                $info->has_default   = (bool)$rawcolumn->atthasdef;
+                if ($info->has_default) {
+                    $parts = explode('::', $rawcolumn->adsrc);
+                    if (count($parts) > 1) {
+                        $info->default_value = reset($parts);
+                        $info->default_value = trim($info->default_value, "'");
+                    } else {
+                        $info->default_value = $rawcolumn->adsrc;
+                    }
+                } else {
+                    $info->default_value = null;
+                }
+                $info->primary_key   = false;
+                $info->binary        = false;
+                $info->unsigned      = null;
+                $info->auto_increment= false;
+                $info->unique        = null;
+
+            } else if ($rawcolumn->type === 'bytea') {
+                $info->type          = $rawcolumn->type;
+                $info->meta_type     = 'B';
+                $info->max_length    = -1;
+                $info->scale         = null;
+                $info->not_null      = (bool)$rawcolumn->attnotnull;
+                $info->has_default   = false;
+                $info->default_value = null;
+                $info->primary_key   = false;
+                $info->binary        = true;
+                $info->unsigned      = null;
+                $info->auto_increment= false;
+                $info->unique        = null;
+
+            }
+
+            $this->columns[$table][$info->name] = new database_column_info($info);
+        }
+
+        return $this->columns[$table];
+    }
+
+    /**
+     * Reset a sequence to the id field of a table.
+     * @param string $table name of table
+     * @return success
+     */
+    public function reset_sequence($table) {
+        if (!$this->get_manager()->table_exists($table)) {
+            return false;
+        }
+        $value = (int)$this->get_field_sql('SELECT MAX(id) FROM {'.$table.'}');
+        $value++;
+        return $this->change_database_structure("ALTER SEQUENCE $this->prefix{$table}_id_seq RESTART WITH $value");
+    }
+
+    /**
+     * Is db in unicode mode?
+     * @return bool
+     */
+    public function setup_is_unicodedb() {
+    /// Get PostgreSQL server_encoding value
+        $this->reads++;
+        if (!$result = pg_query($this->pgsql, "SHOW server_encoding")) {
+            return false;
+        }
+        $rawcolumn = pg_fetch_object($result);
+        $encoding = $rawcolumn->server_encoding;
+        pg_free_result($result);
+
+        return (strtoupper($encoding) == 'UNICODE' || strtoupper($encoding) == 'UTF8');
+    }
+
+    /**
+     * Enable/disable very detailed debugging
+     * @param bool $state
+     */
+    public function set_debug($state) {
+        $this->debug = $state;
+    }
+
+    /**
+     * Returns debug status
+     * @return bool $state
+     */
+    public function get_debug() {
+        return $this->debug;
+    }
+
+    /**
+     * Enable/disable detailed sql logging
+     * @param bool $state
+     */
+    public function set_logging($state) {
+        //TODO
+    }
+
+    /**
+     * Do NOT use in code, to be used by database_manager only!
+     * @param string $sql query
+     * @return bool success
+     */
+    public function change_database_structure($sql) {
+        $this->writes++;
+        $this->print_debug($sql);
+        $result = pg_query($this->pgsql, $sql);
+        $this->reset_columns();
+        if ($result === false) {
+            $this->report_error($sql);
+            return false;
+        }
+        // no need to free result, we do not expect any here
+        return true;
+    }
+
+    /**
+     * Execute general sql query. Should be used only when no other method suitable.
+     * Do NOT use this to make changes in db structure, use database_manager::execute_sql() instead!
+     * @param string $sql query
+     * @param array $params query parameters
+     * @return bool success
+     */
+    public function execute($sql, array $params=null) {
+        list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
+
+        if (strpos($sql, ';') !== false) {
+            debugging('Error: Multiple sql statements found or bound parameters not used properly in query!');
+            return false;
+        }
+
+        $this->writes++;
+        $this->print_debug($sql, $params);
+        $result = pg_query_params($this->pgsql, $sql, $params);
+
+        if ($result === false) {
+            $this->report_error($sql, $params);
+            return false;
+
+        }
+
+        return true;
+    }
+
+    /**
+     * Get a number of records as a moodle_recordset using a SQL statement.
+     *
+     * Since this method is a little less readable, use of it should be restricted to
+     * code where it's possible there might be large datasets being returned.  For known
+     * small datasets use get_records_sql - it leads to simpler code.
+     *
+     * The return type is as for @see function get_recordset.
+     *
+     * @param string $sql the SQL select query to execute.
+     * @param array $params array of sql parameters
+     * @param int $limitfrom return a subset of records, starting at this point (optional, required if $limitnum is set).
+     * @param int $limitnum return a subset comprising this many records (optional, required if $limitfrom is set).
+     * @return mixed an moodle_recorset object, or false if an error occured.
+     */
+    public function get_recordset_sql($sql, array $params=null, $limitfrom=0, $limitnum=0) {
+        if ($limitfrom or $limitnum) {
+            $limitfrom = (int)$limitfrom;
+            $limitnum  = (int)$limitnum;
+            if ($limitnum < 1) {
+                $limitnum = "18446744073709551615";
+            }
+            $sql .= " LIMIT $limitnum OFFSET $limitfrom";
+        }
+
+        list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
+
+        $this->reads++;
+        $this->print_debug($sql, $params);
+        $result = pg_query_params($this->pgsql, $sql, $params);
+
+        if ($result === false) {
+            $this->report_error($sql, $params);
+            return false;
+        }
+
+        return $this->create_recordset($result);
+    }
+
+    protected function create_recordset($result) {
+        return new pgsql_native_moodle_recordset($result);
+    }
+
+    /**
+     * Get a number of records as an array of objects using a SQL statement.
+     *
+     * Return value as for @see function get_records.
+     *
+     * @param string $sql the SQL select query to execute. The first column of this SELECT statement
+     *   must be a unique value (usually the 'id' field), as it will be used as the key of the
+     *   returned array.
+     * @param array $params array of sql parameters
+     * @param int $limitfrom return a subset of records, starting at this point (optional, required if $limitnum is set).
+     * @param int $limitnum return a subset comprising this many records (optional, required if $limitfrom is set).
+     * @return mixed an array of objects, or empty array if no records were found, or false if an error occured.
+     */
+    public function get_records_sql($sql, array $params=null, $limitfrom=0, $limitnum=0) {
+        if ($limitfrom or $limitnum) {
+            $limitfrom = (int)$limitfrom;
+            $limitnum  = (int)$limitnum;
+            if ($limitnum < 1) {
+                $limitnum = "18446744073709551615";
+            }
+            $sql .= " LIMIT $limitnum OFFSET $limitfrom";
+        }
+
+        list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
+        $this->reads++;
+        $this->print_debug($sql, $params);
+        $result = pg_query_params($this->pgsql, $sql, $params);
+
+        if ($result === false) {
+            $this->report_error($sql, $params);
+            return false;
+        }
+
+        $rows = pg_fetch_all($result);
+        pg_free_result($result);
+
+        $return = array();
+        if ($rows) {
+            foreach ($rows as $row) {
+                $id = reset($row);
+                $return[$id] = (object)$row;
+            }
+        }
+         
+        return $return;
+    }
+
+    /**
+     * Selects records and return values (first field) as an array using a SQL statement.
+     *
+     * @param string $sql The SQL query
+     * @param array $params array of sql parameters
+     * @return mixed array of values or false if an error occured
+     */
+    public function get_fieldset_sql($sql, array $params=null) {
+        list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
+
+        $this->reads++;
+        $this->print_debug($sql, $params);
+        $result = pg_query_params($this->pgsql, $sql, $params);
+
+        if ($result === false) {
+            $this->report_error($sql, $params);
+            return false;
+        }
+
+        $return = pg_fetch_all_columns($result, 0);
+        pg_free_result($result);
+         
+        return $return;
+    }
+
+    /**
+     * Insert new record into database, as fast as possible, no safety checks, lobs not supported.
+     * @param string $table name
+     * @param mixed $params data record as object or array
+     * @param bool $returnit return it of inserted record
+     * @param bool $bulk true means repeated inserts expected
+     * @param bool $customsequence true if 'id' included in $params, disables $returnid
+     * @return mixed success or new id
+     */
+    public function insert_record_raw($table, $params, $returnid=true, $bulk=false, $customsequence=false) {
+        if (!is_array($params)) {
+            $params = (array)$params;
+        }
+
+        $returning = "";
+
+        if ($customsequence) {
+            if (!isset($params['id'])) {
+                return false;
+            }
+            $returnid = false;
+        } else {
+            if ($returnid) {
+                if ($this->is_min_version('8.2.0')) {
+                    $returning = "RETURNING id";
+                    unset($params['id']);
+                } else {
+                    //ugly workaround for pg < 8.2
+                    $this->reads++;
+                    $seqsql = "SELECT NEXTVAL({$this->prefix}{$table}_id_seq) AS id";
+                    $result = pg_query($this->pgsql, $seqsql);
+                    if ($result === false) {
+                        throw new dml_exception('missingidsequence', "{$this->prefix}{$table}"); // TODO: add localised string
+                    }
+                    $row = pg_fetch_assoc($result);
+                    $params['id'] = reset($row);
+                    pg_free_result($result);
+                }
+            } else {
+                unset($params['id']);
+            }
+        }
+
+        if (empty($params)) {
+            return false;
+        }
+
+        $fields = implode(',', array_keys($params));
+        $values = array();
+        $count = count($params);
+        for ($i=1; $i<=$count; $i++) {
+            $values[] = "\$".$i;
+        } 
+        $values = implode(',', $values);
+
+        $sql = "INSERT INTO {$this->prefix}$table ($fields) VALUES($values) $returning";
+        $this->writes++;
+        $this->print_debug($sql, $params);
+        $result = pg_query_params($this->pgsql, $sql, $params);
+
+        if ($result === false) {
+            $this->report_error($sql, $params);
+            return false;
+        }
+
+        if ($returning !== "") {
+            $row = pg_fetch_assoc($result);
+            $params['id'] = reset($row);
+            pg_free_result($result);
+        }
+
+        if (!$returnid) {
+            return true;
+        }
+
+        return (int)$params['id'];
+    }
+
+    /**
+     * Insert a record into a table and return the "id" field if required.
+     *
+     * Some conversions and safety checks are carried out. Lobs are supported.
+     * If the return ID isn't required, then this just reports success as true/false.
+     * $data is an object containing needed data
+     * @param string $table The database table to be inserted into
+     * @param object $data A data object with values for one or more fields in the record
+     * @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.
+     * @return mixed success or new ID
+     */
+    public function insert_record($table, $dataobject, $returnid=true, $bulk=false) {
+        if (!is_object($dataobject)) {
+            $dataobject = (object)$dataobject;
+        }
+
+        $columns = $this->get_columns($table);
+
+        unset($dataobject->id);
+        $cleaned = array();
+        $blobs   = array();
+
+        foreach ($dataobject as $field=>$value) {
+            if (!isset($columns[$field])) {
+                continue;
+            }
+            $column = $columns[$field];
+            if ($column->meta_type == 'B') {
+                if (is_null($value)) {
+                    $cleaned[$field] = null;
+                } else {
+                    $blobs[$field] = $value;
+                    $cleaned[$field] = '@#BLOB#@';
+                }
+                continue;
+
+            } else if (is_bool($value)) {
+                $value = (int)$value; // prevent false '' problems
+
+            } else if ($value === '') {
+                if ($column->meta_type == 'I' or $column->meta_type == 'F' or $column->meta_type == 'N') {
+                    $value = 0; // prevent '' problems in numeric fields
+                }
+            }
+
+            $cleaned[$field] = $value;
+        }
+
+        if (empty($cleaned)) {
+            return false;
+        }
+
+        if (empty($blobs)) {
+            return $this->insert_record_raw($table, $cleaned, $returnid, $bulk);
+        }
+
+        if (!$id = $this->insert_record_raw($table, $cleaned, true, $bulk)) {
+            return false;
+        }
+
+        foreach ($blobs as $key=>$value) {
+            $this->writes++;
+            $value = pg_escape_bytea($this->pgsql, $value);
+            $sql = "UPDATE {$this->prefix}$table SET $key = '$value'::bytea WHERE id = $id";
+            pg_query($this->pgsql, $sql);
+        }
+
+        return ($returnid ? $id : true);
+
+    }
+
+    /**
+     * Import a record into a table, id field is required.
+     * Safety checks are NOT carried out. Lobs are supported.
+     *
+     * @param string $table name of database table to be inserted into
+     * @param object $dataobject A data object with values for one or more fields in the record
+     * @return bool success
+     */
+    public function import_record($table, $dataobject) {
+        $dataobject = (object)$dataobject;
+
+        if (empty($dataobject->id)) {
+            return false;
+        }
+
+        $columns = $this->get_columns($table);
+        $cleaned = array();
+
+        foreach ($dataobject as $field=>$value) {
+            if (!isset($columns[$field])) {
+                continue;
+            }
+            $cleaned[$field] = $value;
+        }
+
+        return $this->insert_record_raw($table, $cleaned, false, true, true);
+    }
+
+    /**
+     * Update record in database, as fast as possible, no safety checks, lobs not supported.
+     * @param string $table name
+     * @param mixed $params data record as object or array
+     * @param bool true means repeated updates expected
+     * @return bool success
+     */
+    public function update_record_raw($table, $params, $bulk=false) {
+        if (!is_array($params)) {
+            $params = (array)$params;
+        }
+        if (!isset($params['id'])) {
+            return false;
+        }
+        $id = $params['id'];
+        unset($params['id']);
+
+        if (empty($params)) {
+            return false;
+        }
+
+        $i = 1;
+
+        $sets = array();
+        foreach ($params as $field=>$value) {
+            $sets[] = "$field = \$".$i++;
+        }
+
+        $params[] = $id; // last ? in WHERE condition
+
+        $sets = implode(',', $sets);
+        $sql = "UPDATE {$this->prefix}$table SET $sets WHERE id=\$".$i;
+
+        $this->writes++;
+        $this->print_debug($sql, $params);
+        $result = pg_query_params($this->pgsql, $sql, $params);
+
+        if ($result === false) {
+            $this->report_error($sql, $params);
+            return false;
+        }
+
+        return true;
+    }
+
+    /**
+     * Update a record in a table
+     *
+     * $dataobject is an object containing needed data
+     * Relies on $dataobject having a variable "id" to
+     * specify the record to update
+     *
+     * @param string $table The database table to be checked against.
+     * @param object $dataobject An object with contents equal to fieldname=>fieldvalue. Must have an entry for 'id' to map to the table specified.
+     * @param bool true means repeated updates expected
+     * @return bool success
+     */
+    public function update_record($table, $dataobject, $bulk=false) {
+        if (!is_object($dataobject)) {
+            $dataobject = (object)$dataobject;
+        }
+
+        if (!isset($dataobject->id) ) {
+            return false;
+        }
+
+        $columns = $this->get_columns($table);
+        $cleaned = array();
+
+        foreach ($dataobject as $field=>$value) {
+            if (!isset($columns[$field])) {
+                continue;
+            }
+            if (is_bool($value)) {
+                $value = (int)$value; // prevent "false" problems
+            }
+            $cleaned[$field] = $value;
+        }
+
+        return $this->update_record_raw($table, $cleaned, $bulk);
+    }
+
+    /**
+     * Set a single field in every table record which match a particular WHERE clause.
+     *
+     * @param string $table The database table to be checked against.
+     * @param string $newfield the field to set.
+     * @param string $newvalue the value to set the field to.
+     * @param string $select A fragment of SQL to be used in a where clause in the SQL call.
+     * @param array $params array of sql parameters
+     * @return bool success
+     */
+    public function set_field_select($table, $newfield, $newvalue, $select, array $params=null) {
+        if ($select) {
+            $select = "WHERE $select";
+        }
+        if (is_null($params)) {
+            $params = array();
+        }
+        list($select, $params, $type) = $this->fix_sql_params($select, $params);
+        $i = count($params)+1;
+
+        if (is_bool($newvalue)) {
+            $newvalue = (int)$newvalue; // prevent "false" problems
+        }
+        if (is_null($newvalue)) {
+            $newfield = "$newfield = NULL";
+        } else {
+            $newfield = "$newfield = \$".$i;
+            $params[] = $newvalue;
+        }
+        $sql = "UPDATE {$this->prefix}$table SET $newfield $select";
+
+        $this->writes++;
+        $this->print_debug($sql, $params);
+        $result = pg_query_params($this->pgsql, $sql, $params);
+
+        if ($result === false) {
+            $this->report_error($sql, $params);
+            return false;
+        }
+
+        return true;
+    }
+
+    /**
+     * Delete one or more records from a table which match a particular WHERE clause.
+     *
+     * @param string $table The database table to be checked against.
+     * @param string $select A fragment of SQL to be used in a where clause in the SQL call (used to define the selection criteria).
+     * @param array $params array of sql parameters
+     * @return returns success.
+     */
+    public function delete_records_select($table, $select, array $params=null) {
+        if ($select) {
+            $select = "WHERE $select";
+        }
+        $sql = "DELETE FROM {$this->prefix}$table $select";
+
+        list($sql, $params, $type) = $this->fix_sql_params($sql, $params);
+
+        $this->writes++;
+        $this->print_debug($sql, $params);
+        $result = pg_query_params($this->pgsql, $sql, $params);
+
+        if ($result === false) {
+            $this->report_error($sql, $params);
+            return false;
+        }
+
+        return true;
+    }
+
+    public function sql_ilike() {
+        return 'ILIKE';
+    }
+
+    public function sql_bitxor($int1, $int2) {
+        return '(' . $this->sql_bitor($int1, $int2) . ' - ' . $this->sql_bitand($int1, $int2) . ')';
+    }
+
+    public function sql_cast_char2int($fieldname, $text=false) {
+        return ' CAST(' . $fieldname . ' AS INT) ';
+    }
+
+    public function sql_cast_char2real($fieldname, $text=false) {
+        return " $fieldname::real ";
+    }
+
+    public function sql_concat() {
+        $arr = func_get_args();
+        $s = implode(' || ', $arr);
+        if ($s === '') {
+            return " '' ";
+        }
+        return " $s ";
+    }
+
+    public function sql_concat_join($separator="' '", $elements=array()) {
+        for ($n=count($elements)-1; $n > 0 ; $n--) {
+            array_splice($elements, $n, 0, $separator);
+        }
+        $s = implode(' || ', $elements);
+        if ($s === '') {
+            return " '' ";
+        }
+        return " $s ";
+    }
+
+    public function sql_substr() {
+        return "SUBSTRING";
+    }
+
+    public function sql_regex_supported() {
+        return true;
+    }
+
+    public function sql_regex($positivematch=true) {
+        return $positivematch ? '~*' : '!~*';
+    }
+
+}
diff --git a/lib/dml/pgsql_native_moodle_recordset.php b/lib/dml/pgsql_native_moodle_recordset.php
new file mode 100644 (file)
index 0000000..30617e4
--- /dev/null
@@ -0,0 +1,56 @@
+<?php  //$Id$
+
+require_once($CFG->libdir.'/dml/moodle_recordset.php');
+
+class pgsql_native_moodle_recordset extends moodle_recordset {
+
+    protected $result;
+    protected $current; // current row as array
+
+    public function __construct($result) {
+        $this->result  = $result;
+        $this->current = $this->fetch_next();
+    }
+
+    public function __destruct() {
+        $this->close();
+    }
+
+    private function fetch_next() {
+        $row = pg_fetch_assoc($this->result);
+        return $row;
+    }
+
+    public function current() {
+        return (object)$this->current;
+    }
+
+    public function key() {
+    /// return first column value as key
+        if (!$this->current) {
+            return false;
+        }
+        $key = reset($this->current);
+        return $key;
+    }
+
+    public function next() {
+        $this->current = $this->fetch_next();
+    }
+
+    public function rewind() {
+        // we can not seek, sorry - let's ignore it ;-)
+    }
+
+    public function valid() {
+        return !empty($this->current);
+    }
+
+    public function close() {
+        if ($this->result) {
+            pg_free_result($this->result);
+            $this->result  = null;
+        }
+        $this->current = null;
+    }
+}