[Summary view] [Print] [Text view]

   1  <?php
   2  // This file is part of Moodle - http://moodle.org/
   3  //
   4  // Moodle is free software: you can redistribute it and/or modify
   5  // it under the terms of the GNU General Public License as published by
   6  // the Free Software Foundation, either version 3 of the License, or
   7  // (at your option) any later version.
   8  //
   9  // Moodle is distributed in the hope that it will be useful,
  10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  // GNU General Public License for more details.
  13  //
  14  // You should have received a copy of the GNU General Public License
  15  // along with Moodle.  If not, see <http://www.gnu.org/licenses/>.
  16  
  17  /**
  18   * Native postgresql recordset.
  19   *
  20   * @package    core_dml
  21   * @copyright  2008 Petr Skoda (http://skodak.org)
  22   * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
  23   */
  24  
  25  defined('MOODLE_INTERNAL') || die();
  26  
  27  require_once (__DIR__.'/moodle_recordset.php');
  28  
  29  /**
  30   * pgsql specific moodle recordset class
  31   *
  32   * @package    core_dml
  33   * @copyright  2008 Petr Skoda (http://skodak.org)
  34   * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
  35   */
  36  class pgsql_native_moodle_recordset extends moodle_recordset {
  37  
  38      protected $result;
  39      /** @var current row as array.*/
  40      protected $current;
  41      protected $blobs = array();
  42  
  43      /**
  44       * Build a new recordset to iterate over.
  45       *
  46       * @param resource $result A pg_query() result object to create a recordset from.
  47       */
  48      public function __construct($result) {
  49          $this->result = $result;
  50  
  51          // Find out if there are any blobs.
  52          $numfields = pg_num_fields($result);
  53          for ($i = 0; $i < $numfields; $i++) {
  54              $type = pg_field_type($result, $i);
  55              if ($type == 'bytea') {
  56                  $this->blobs[] = pg_field_name($result, $i);
  57              }
  58          }
  59  
  60          $this->current = $this->fetch_next();
  61      }
  62  
  63      public function __destruct() {
  64          $this->close();
  65      }
  66  
  67      private function fetch_next() {
  68          if (!$this->result) {
  69              return false;
  70          }
  71          if (!$row = pg_fetch_assoc($this->result)) {
  72              pg_free_result($this->result);
  73              $this->result = null;
  74              return false;
  75          }
  76  
  77          if ($this->blobs) {
  78              foreach ($this->blobs as $blob) {
  79                  $row[$blob] = $row[$blob] !== null ? pg_unescape_bytea($row[$blob]) : null;
  80              }
  81          }
  82  
  83          return $row;
  84      }
  85  
  86      public function current() {
  87          return (object)$this->current;
  88      }
  89  
  90      public function key() {
  91          // return first column value as key
  92          if (!$this->current) {
  93              return false;
  94          }
  95          $key = reset($this->current);
  96          return $key;
  97      }
  98  
  99      public function next() {
 100          $this->current = $this->fetch_next();
 101      }
 102  
 103      public function valid() {
 104          return !empty($this->current);
 105      }
 106  
 107      public function close() {
 108          if ($this->result) {
 109              pg_free_result($this->result);
 110              $this->result  = null;
 111          }
 112          $this->current = null;
 113          $this->blobs   = null;
 114      }
 115  }