[Sommaire] [Imprimer]

   1  <?php
   2  /***********************************************************************
   3  
   4    Copyright (C) 2002-2005  Rickard Andersson (rickard@punbb.org)
   5  
   6    This file is part of PunBB.
   7  
   8    PunBB is free software; you can redistribute it and/or modify it
   9    under the terms of the GNU General Public License as published
  10    by the Free Software Foundation; either version 2 of the License,
  11    or (at your option) any later version.
  12  
  13    PunBB is distributed in the hope that it will be useful, but
  14    WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17  
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 59 Temple Place, Suite 330, Boston,
  21    MA  02111-1307  USA
  22  
  23  ************************************************************************/
  24  
  25  
  26  // Make sure we have built in support for PostgreSQL
  27  if (!function_exists('pg_connect'))
  28      exit('This PHP environment doesn\'t have PostgreSQL support built in. PostgreSQL support is required if you want to use a PostgreSQL database to run this forum. Consult the PHP documentation for further assistance.');
  29  
  30  
  31  class DBLayer
  32  {
  33      var $prefix;
  34      var $link_id;
  35      var $query_result;
  36      var $last_query_text = array();
  37      var $in_transaction = 0;
  38  
  39      var $saved_queries = array();
  40      var $num_queries = 0;
  41  
  42      var $error_no = false;
  43      var $error_msg = 'Unknown';
  44  
  45  
  46  	function DBLayer($db_host, $db_username, $db_password, $db_name, $db_prefix, $p_connect)
  47      {
  48          $this->prefix = $db_prefix;
  49  
  50          if ($db_host != '')
  51          {
  52              if (strpos($db_host, ':') !== false)
  53              {
  54                  list($db_host, $dbport) = explode(':', $db_host);
  55                  $connect_str[] = 'host='.$db_host.' port='.$dbport;
  56              }
  57              else
  58              {
  59                  if ($db_host != 'localhost')
  60                      $connect_str[] = 'host='.$db_host;
  61              }
  62          }
  63  
  64          if ($db_name)
  65              $connect_str[] = 'dbname='.$db_name;
  66  
  67          if ($db_username != '')
  68              $connect_str[] = 'user='.$db_username;
  69  
  70          if ($db_password != '')
  71              $connect_str[] = 'password='.$db_password;
  72  
  73          if ($p_connect)
  74              $this->link_id = @pg_pconnect(implode(' ', $connect_str));
  75          else
  76              $this->link_id = @pg_connect(implode(' ', $connect_str));
  77  
  78          if (!$this->link_id)
  79              error('Unable to connect to PostgreSQL server', __FILE__, __LINE__);
  80          else
  81              return $this->link_id;
  82      }
  83  
  84  
  85  	function start_transaction()
  86      {
  87          ++$this->in_transaction;
  88  
  89          return (@pg_query($this->link_id, 'BEGIN')) ? true : false;
  90      }
  91  
  92  
  93  	function end_transaction()
  94      {
  95          --$this->in_transaction;
  96  
  97          if (@pg_query($this->link_id, 'COMMIT'))
  98              return true;
  99          else
 100          {
 101              @pg_query($this->link_id, 'ROLLBACK');
 102              return false;
 103          }
 104      }
 105  
 106  
 107  	function query($sql, $unbuffered = false)    // $unbuffered is ignored since there is no pgsql_unbuffered_query()
 108      {
 109          if (strrpos($sql, 'LIMIT') !== false)
 110              $sql = preg_replace('#LIMIT ([0-9]+),([ 0-9]+)#', 'LIMIT \\2 OFFSET \\1', $sql);
 111  
 112          if (defined('PUN_SHOW_QUERIES'))
 113              $q_start = get_microtime();
 114  
 115          @pg_send_query($this->link_id, $sql);
 116          $this->query_result = @pg_get_result($this->link_id);
 117  
 118          if (pg_result_status($this->query_result) != PGSQL_FATAL_ERROR)
 119          {
 120              if (defined('PUN_SHOW_QUERIES'))
 121                  $this->saved_queries[] = array($sql, sprintf('%.5f', get_microtime() - $q_start));
 122  
 123              ++$this->num_queries;
 124  
 125              $this->last_query_text[$this->query_result] = $sql;
 126  
 127              return $this->query_result;
 128          }
 129          else
 130          {
 131              if (defined('PUN_SHOW_QUERIES'))
 132                  $this->saved_queries[] = array($sql, 0);
 133  
 134              $this->error_msg = @pg_result_error($this->query_result);
 135  
 136              if ($this->in_transaction)
 137                  @pg_query($this->link_id, 'ROLLBACK');
 138  
 139              --$this->in_transaction;
 140  
 141              return false;
 142          }
 143      }
 144  
 145  
 146  	function result($query_id = 0, $row = 0)
 147      {
 148          return ($query_id) ? @pg_fetch_result($query_id, $row, 0) : false;
 149      }
 150  
 151  
 152  	function fetch_assoc($query_id = 0)
 153      {
 154          return ($query_id) ? @pg_fetch_assoc($query_id) : false;
 155      }
 156  
 157  
 158  	function fetch_row($query_id = 0)
 159      {
 160          return ($query_id) ? @pg_fetch_row($query_id) : false;
 161      }
 162  
 163  
 164  	function num_rows($query_id = 0)
 165      {
 166          return ($query_id) ? @pg_num_rows($query_id) : false;
 167      }
 168  
 169  
 170  	function affected_rows()
 171      {
 172          return ($this->query_result) ? @pg_affected_rows($this->query_result) : false;
 173      }
 174  
 175  
 176  	function insert_id()
 177      {
 178          $query_id = $this->query_result;
 179  
 180          if ($query_id && $this->last_query_text[$query_id] != '')
 181          {
 182              if (preg_match('/^INSERT INTO ([a-z0-9\_\-]+)/is', $this->last_query_text[$query_id], $table_name))
 183              {
 184                  // Hack (don't ask)
 185                  if (substr($table_name[1], -6) == 'groups')
 186                      $table_name[1] .= '_g';
 187  
 188                  $temp_q_id = @pg_query($this->link_id, 'SELECT currval(\''.$table_name[1].'_id_seq\')');
 189                  return ($temp_q_id) ? intval(@pg_fetch_result($temp_q_id, 0)) : false;
 190              }
 191          }
 192  
 193          return false;
 194      }
 195  
 196  
 197  	function get_num_queries()
 198      {
 199          return $this->num_queries;
 200      }
 201  
 202  
 203  	function get_saved_queries()
 204      {
 205          return $this->saved_queries;
 206      }
 207  
 208  
 209  	function free_result($query_id = false)
 210      {
 211          if (!$query_id)
 212              $query_id = $this->query_result;
 213  
 214          return ($query_id) ? @pg_free_result($query_id) : false;
 215      }
 216  
 217  
 218  	function escape($str)
 219      {
 220          return is_array($str) ? '' : pg_escape_string($str);
 221      }
 222  
 223  
 224  	function error()
 225      {
 226          $result['error_sql'] = @current(@end($this->saved_queries));
 227          $result['error_no'] = false;
 228  /*
 229          if (!empty($this->query_result))
 230          {
 231              $result['error_msg'] = trim(@pg_result_error($this->query_result));
 232              if ($result['error_msg'] != '')
 233                  return $result;
 234          }
 235  
 236          $result['error_msg'] = (!empty($this->link_id)) ? trim(@pg_last_error($this->link_id)) : trim(@pg_last_error());
 237  */
 238          $result['error_msg'] = $this->error_msg;
 239  
 240          return $result;
 241      }
 242  
 243  
 244  	function close()
 245      {
 246          if ($this->link_id)
 247          {
 248              if ($this->in_transaction)
 249              {
 250                  if (defined('PUN_SHOW_QUERIES'))
 251                      $this->saved_queries[] = array('COMMIT', 0);
 252  
 253                  @pg_query($this->link_id, 'COMMIT');
 254              }
 255  
 256              if ($this->query_result)
 257                  @pg_free_result($this->query_result);
 258  
 259              return @pg_close($this->link_id);
 260          }
 261          else
 262              return false;
 263      }
 264  }