Page MenuHomePhorge
Files F3310394

rcube_mdb2.inc
No OneTemporary
Actions

Size
8 KB
Referenced Files
None
Subscribers
None

rcube_mdb2.inc
View Options

<?php
/*
+-----------------------------------------------------------------------+
| program/include/rcube_db.inc |
| |
| This file is part of the RoundCube Webmail client |
| Copyright (C) 2005, RoundCube Dev. - Switzerland |
| Licensed under the GNU GPL |
| |
| PURPOSE: |
| PEAR:DB wrapper class that implements PEAR DB functions |
| See http://pear.php.net/package/DB |
| |
+-----------------------------------------------------------------------+
| Author: David Saez Padros <david@ols.es> |
+-----------------------------------------------------------------------+
$Id$
*/
require_once('MDB2.php');
class rcube_db
{
var $db_dsnw; // DSN for write operations
var $db_dsnr; // DSN for read operations
var $db_connected=false; // Already connected ?
var $db_mode=''; // Connection mode
var $db_handle=0; // Connection handle
var $a_query_results = array('dummy');
var $last_res_id = 0;
// PHP 5 constructor
function __construct($db_dsnw,$db_dsnr='')
{
if ($db_dsnr=='') $db_dsnr=$db_dsnw;
$this->db_dsnw = $db_dsnw;
$this->db_dsnr = $db_dsnr;
$dsn_array = MDB2::parseDSN($db_dsnw);
$this->db_provider = $dsn_array['phptype'];
}
// PHP 4 compatibility
function rcube_db($db_dsnw,$db_dsnr='')
{
$this->__construct($db_dsnw,$db_dsnr);
}
// Connect to specific database
function dsn_connect($dsn)
{
// Use persistent connections if available
$dbh = MDB2::factory($dsn, array('persistent' => $true));
if (PEAR::isError($dbh))
raise_error(array('code' => 500,
'type' => 'db',
'line' => __LINE__,
'file' => __FILE__,
'message' => $dbh->getMessage()), TRUE, FALSE);
else if ($this->db_provider=='sqlite')
{
$dsn_array = MDB2::parseDSN($dsn);
if (!filesize($dsn_array['database']) && !empty($this->sqlite_initials))
$this->_sqlite_create_database($dbh, $this->sqlite_initials);
}
return $dbh;
}
// Connect to appropiate databse
function db_connect ($mode)
{
$this->db_mode = $mode;
// Already connected
if ($this->db_connected)
{
// no replication, current connection is ok
if ($this->db_dsnw==$this->db_dsnr) return;
// connected to master, current connection is ok
if ($this->db_mode=='w') return;
// Same mode, current connection is ok
if ($this->db_mode==$mode) return;
}
if ($mode=='r')
$dsn=$this->db_dsnr;
else
$dsn=$this->db_dsnw;
$this->db_handle = $this->dsn_connect($dsn);
$this->db_connected = true;
}
// Query database
function query()
{
$params = func_get_args();
$query = array_shift($params);
return $this->_query($query, 0, 0, $params);
}
function limitquery()
{
$params = func_get_args();
$query = array_shift($params);
$offset = array_shift($params);
$numrows = array_shift($params);
return $this->_query($query, $offset, $numrows, $params);
}
function _query($query, $offset, $numrows, $params)
{
// Read or write ?
if (strtolower(trim(substr($query,0,6)))=='select')
$mode='r';
else
$mode='w';
$this->db_connect($mode);
if ($this->db_provider == 'sqlite')
$query = $this->_sqlite_prepare_query($query);
$this->db_handle->row_offset = $offset;
$this->db_handle->row_limit = $numrows;
//$result = $this->db_handle->query($query,$params);
$q = $this->db_handle->prepare($query);
$q->bindParamArray($params);
$result = $q->execute();
if (PEAR::isError($result))
raise_error(array('code' => 500,
'type' => 'db',
'line' => __LINE__,
'file' => __FILE__,
'message' => $result->getMessage()), TRUE, FALSE);
return $this->_add_result($result, $query);
}
function num_rows($res_id=NULL)
{
if (!$this->db_handle)
return FALSE;
$result = $this->_get_result($res_id);
if ($result)
return $result->numRows();
else
return FALSE;
}
function affected_rows($res_id=NULL)
{
if (!$this->db_handle)
return FALSE;
return $this->db_handle->affectedRows();
}
function insert_id($sequence = '')
{
if (!$this->db_handle || $this->db_mode=='r')
return FALSE;
return $this->db_handle->lastInsertID();
}
function fetch_assoc($res_id=NULL)
{
$result = $this->_get_result($res_id);
if (PEAR::isError($result))
{
raise_error( array('code' => 500, 'type' => 'db', 'line' => __LINE__, 'file' => __FILE__,
'message' => $this->db_link->getMessage()), TRUE, FALSE);
return FALSE;
}
return $result->fetchRow(MDB2_FETCHMODE_ASSOC);
}
function quote($input, $type=null)
{
if (!$this->db_handle)
$this->db_connect('r');
return $this->db_handle->quote($input, $type);
}
function quoteIdentifier($str)
{
if (!$this->db_handle)
$this->db_connect('r');
return $this->db_handle->quoteIdentifier($str);
}
function quote_identifier($str)
{
return $this->quoteIdentifier($str);
}
function unixtimestamp($field)
{
switch($this->db_provider)
{
case 'pgsql':
return "EXTRACT (EPOCH FROM $field)";
break;
default:
return "UNIX_TIMESTAMP($field)";
}
}
function format_date($timestamp)
{
switch($this->db_provider)
{
case 'mysqli':
case 'mysql':
return "FROM_UNIXTIME($timestamp)";
break;
case 'sqlite':
return "datetime('$timestamp')";
break;
default:
return date("Y-m-d H:i:s", $timestamp);
}
}
function _add_result($res, $query)
{
// sql error occured
if (PEAR::isError($res))
{
raise_error(array('code' => 500, 'type' => 'db', 'line' => __LINE__, 'file' => __FILE__, 'message' => $res->getMessage() . " Query: " . substr(preg_replace('/[\r\n]+\s*/', ' ', $query), 0, 1024)), TRUE, FALSE);
return FALSE;
}
else
{
$res_id = sizeof($this->a_query_results);
$this->a_query_results[$res_id] = $res;
$this->last_res_id = $res_id;
return $res_id;
}
}
function _get_result($res_id)
{
if ($res_id==NULL)
$res_id = $this->last_res_id;
if ($res_id && isset($this->a_query_results[$res_id]))
return $this->a_query_results[$res_id];
else
return FALSE;
}
// create a sqlite database from a file
function _sqlite_create_database($dbh, $fileName)
{
if (empty($fileName) || !is_string($fileName))
return ;
$data = '';
if ($fd = fopen($fileName, 'r'))
{
$data = fread($fd, filesize($fileName));
fclose($fd);
}
if (strlen($data))
sqlite_exec($dbh->connection, $data);
}
// transform a query so that it is sqlite2 compliant
function _sqlite_prepare_query($query)
{
if (!is_string($query))
return ($query);
$search = array('/NOW\(\)/i', '/`/');
$replace = array("datetime('now')", '"');
$query = preg_replace($search, $replace, $query);
return ($query);
}
}
?>

File Metadata

Mime Type
text/x-php
Expires
Thu, Apr 9, 12:26 PM (14 h, 9 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
540396
Default Alt Text
rcube_mdb2.inc (8 KB)

Event Timeline