moodle/lib/dml/pgsql_native_moodle_recordset.php

103 lines
2.7 KiB
PHP

<?php
// This file is part of Moodle - http://moodle.org/
//
// Moodle is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Moodle is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Moodle. If not, see <http://www.gnu.org/licenses/>.
/**
* Native postgresql recordset.
*
* @package core
* @subpackage dml
* @copyright 2008 Petr Skoda (http://skodak.org)
* @license http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
*/
defined('MOODLE_INTERNAL') || die();
require_once($CFG->libdir.'/dml/moodle_recordset.php');
/**
* pgsql specific moodle recordset class
*/
class pgsql_native_moodle_recordset extends moodle_recordset {
protected $result;
protected $current; // current row as array
protected $bytea_oid;
protected $blobs = array();
public function __construct($result, $bytea_oid) {
$this->result = $result;
$this->bytea_oid = $bytea_oid;
// find out if there are any blobs
$numrows = pg_num_fields($result);
for($i=0; $i<$numrows; $i++) {
$type_oid = pg_field_type_oid($result, $i);
if ($type_oid == $this->bytea_oid) {
$this->blobs[] = pg_field_name($result, $i);
}
}
$this->current = $this->fetch_next();
}
public function __destruct() {
$this->close();
}
private function fetch_next() {
$row = pg_fetch_assoc($this->result);
if ($this->blobs) {
foreach ($this->blobs as $blob) {
$row[$blob] = $row[$blob] !== null ? pg_unescape_bytea($row[$blob]) : null;
}
}
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 valid() {
return !empty($this->current);
}
public function close() {
if ($this->result) {
pg_free_result($this->result);
$this->result = null;
}
$this->current = null;
$this->blobs = null;
}
}