// Improve DbQuery
This commit is contained in:
rMalie
@@ -0,0 +1,186 @@
|
||||
<?php
|
||||
/*
|
||||
* 2007-2011 PrestaShop
|
||||
*
|
||||
* NOTICE OF LICENSE
|
||||
*
|
||||
* This source file is subject to the Open Software License (OSL 3.0)
|
||||
* that is bundled with this package in the file LICENSE.txt.
|
||||
* It is also available through the world-wide-web at this URL:
|
||||
* http://opensource.org/licenses/osl-3.0.php
|
||||
* If you did not receive a copy of the license and are unable to
|
||||
* obtain it through the world-wide-web, please send an email
|
||||
* to license@prestashop.com so we can send you a copy immediately.
|
||||
*
|
||||
* DISCLAIMER
|
||||
*
|
||||
* Do not edit or add to this file if you wish to upgrade PrestaShop to newer
|
||||
* versions in the future. If you wish to customize PrestaShop for your
|
||||
* needs please refer to http://www.prestashop.com for more information.
|
||||
*
|
||||
* @author PrestaShop SA <contact@prestashop.com>
|
||||
* @copyright 2007-2011 PrestaShop SA
|
||||
* @version Release: $Revision$
|
||||
* @license http://opensource.org/licenses/osl-3.0.php Open Software License (OSL 3.0)
|
||||
* International Registered Trademark & Property of PrestaShop SA
|
||||
*/
|
||||
|
||||
/**
|
||||
* This class is currently only here for tests
|
||||
*
|
||||
* @since 1.5.0
|
||||
*/
|
||||
class DbPDOCore extends Db
|
||||
{
|
||||
/**
|
||||
* @see DbCore::connect()
|
||||
*/
|
||||
public function connect()
|
||||
{
|
||||
try
|
||||
{
|
||||
$this->_link = new PDO('mysql:dbname='.$this->_database.';host='.$this->_server, $this->_user, $this->_password);
|
||||
}
|
||||
catch (PDOException $e)
|
||||
{
|
||||
die(Tools::displayError('Link to database cannot be established. ('.$e->getMessage().')'));
|
||||
}
|
||||
|
||||
// UTF-8 support
|
||||
if ($this->_link->exec('SET NAMES \'utf8\'') === false)
|
||||
die(Tools::displayError('PrestaShop Fatal error: no utf-8 support. Please check your server configuration.'));
|
||||
|
||||
return $this->_link;
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::disconnect()
|
||||
*/
|
||||
public function disconnect()
|
||||
{
|
||||
unset($this->_link);
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::_query()
|
||||
*/
|
||||
protected function _query($sql)
|
||||
{
|
||||
return $this->_link->query($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::nextRow()
|
||||
*/
|
||||
public function nextRow($result = false)
|
||||
{
|
||||
if (!$result)
|
||||
$result = $this->_result;
|
||||
return $result->fetch();
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::_numRows()
|
||||
*/
|
||||
protected function _numRows($result)
|
||||
{
|
||||
return $result->rowCount();;
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::Insert_ID()
|
||||
*/
|
||||
public function Insert_ID()
|
||||
{
|
||||
return $this->_link->lastInsertId();
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::Affected_Rows()
|
||||
*/
|
||||
public function Affected_Rows()
|
||||
{
|
||||
return $this->_result->rowCount();
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::getMsgError()
|
||||
*/
|
||||
public function getMsgError($query = false)
|
||||
{
|
||||
$error = $this->_link->errorInfo();
|
||||
return $error[2];
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::getNumberError()
|
||||
*/
|
||||
public function getNumberError()
|
||||
{
|
||||
$error = $this->_link->errorInfo();
|
||||
return $error[1];
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::getVersion()
|
||||
*/
|
||||
public function getVersion()
|
||||
{
|
||||
return $this->getValue('SELECT VERSION()');
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::_escape()
|
||||
*/
|
||||
public function _escape($str)
|
||||
{
|
||||
$search = array("\\", "\0", "\n", "\r", "\x1a", "'", '"');
|
||||
$replace = array("\\\\", "\\0", "\\n", "\\r", "\Z", "\'", '\"');
|
||||
|
||||
return str_replace($search, $replace, $str);
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::set_db()
|
||||
*/
|
||||
public function set_db($db_name)
|
||||
{
|
||||
return $this->_link->exec('USE '.pSQL($db_name));
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::tryToConnect()
|
||||
*/
|
||||
static public function tryToConnect($server, $user, $pwd, $db)
|
||||
{
|
||||
try
|
||||
{
|
||||
$test = new PDO('mysql:dbname='.$db.';host='.$server, $user, $pwd);
|
||||
}
|
||||
catch (PDOException $e)
|
||||
{
|
||||
return 1;
|
||||
}
|
||||
unset($test);
|
||||
return 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* @see DbCore::tryUTF8()
|
||||
*/
|
||||
static public function tryUTF8($server, $user, $pwd)
|
||||
{
|
||||
try
|
||||
{
|
||||
$test = new PDO('mysql:dbname='.$db.';host='.$server, $user, $pwd);
|
||||
}
|
||||
catch (PDOException $e)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
$result = $test->exec('SET NAMES \'utf8\'');
|
||||
unset($test);
|
||||
|
||||
return ($result === false) ? false : true;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,216 @@
|
||||
<?php
|
||||
/*
|
||||
* 2007-2011 PrestaShop
|
||||
*
|
||||
* NOTICE OF LICENSE
|
||||
*
|
||||
* This source file is subject to the Open Software License (OSL 3.0)
|
||||
* that is bundled with this package in the file LICENSE.txt.
|
||||
* It is also available through the world-wide-web at this URL:
|
||||
* http://opensource.org/licenses/osl-3.0.php
|
||||
* If you did not receive a copy of the license and are unable to
|
||||
* obtain it through the world-wide-web, please send an email
|
||||
* to license@prestashop.com so we can send you a copy immediately.
|
||||
*
|
||||
* DISCLAIMER
|
||||
*
|
||||
* Do not edit or add to this file if you wish to upgrade PrestaShop to newer
|
||||
* versions in the future. If you wish to customize PrestaShop for your
|
||||
* needs please refer to http://www.prestashop.com for more information.
|
||||
*
|
||||
* @author PrestaShop SA <contact@prestashop.com>
|
||||
* @copyright 2007-2011 PrestaShop SA
|
||||
* @version Release: $Revision$
|
||||
* @license http://opensource.org/licenses/osl-3.0.php Open Software License (OSL 3.0)
|
||||
* International Registered Trademark & Property of PrestaShop SA
|
||||
*/
|
||||
|
||||
/**
|
||||
* SQL query builder
|
||||
*
|
||||
* @since 1.5.0
|
||||
*/
|
||||
class DbQueryCore
|
||||
{
|
||||
/**
|
||||
* @var array list of data to build the query
|
||||
*/
|
||||
protected $query = array(
|
||||
'select' => array(),
|
||||
'from' => '',
|
||||
'join' => array(),
|
||||
'where' => array(),
|
||||
'group' => array(),
|
||||
'having' => array(),
|
||||
'order' => array(),
|
||||
'limit' => array('offset' => 0, 'limit' => 0),
|
||||
);
|
||||
|
||||
/**
|
||||
* Add fields in query selection
|
||||
*
|
||||
* @param string $fields List of fields to concat to other fields
|
||||
* @return DbQuery
|
||||
*/
|
||||
public function select($fields)
|
||||
{
|
||||
$this->query['select'][] = $fields;
|
||||
return $this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Set table for FROM clause
|
||||
*
|
||||
* @param string $table Table name
|
||||
* @return DbQuery
|
||||
*/
|
||||
public function from($table)
|
||||
{
|
||||
$this->query['from'] = _DB_PREFIX_.$table;
|
||||
return $this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Add JOIN clause
|
||||
*
|
||||
* @param string $type Join type : left|right|inner|cross|union|natural
|
||||
* @return DbQuery
|
||||
*/
|
||||
public function join($type, $join)
|
||||
{
|
||||
$type = strtolower($type);
|
||||
$types = array(
|
||||
'left' => 'LEFT JOIN',
|
||||
'right' => 'RIGHT JOIN',
|
||||
'inner' => 'INNER JOIN',
|
||||
'cross' => 'CROSS JOIN',
|
||||
'union' => 'UNION JOIN',
|
||||
'natural' => 'NATURAL JOIN',
|
||||
);
|
||||
|
||||
if (!isset($types[$type]))
|
||||
die('Bad type in DbQuery->join()');
|
||||
|
||||
$this->query['join'][] = $types[$type].' '._DB_PREFIX_.$join;
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function leftJoin($join)
|
||||
{
|
||||
return $this->join('left', $join);
|
||||
}
|
||||
|
||||
public function innerJoin($table)
|
||||
{
|
||||
return $this->join('inner', $join);
|
||||
}
|
||||
|
||||
/**
|
||||
* Add a restriction in WHERE clause (each restriction will be separated by AND statement)
|
||||
*
|
||||
* @param string $restriction
|
||||
* @return DbQuery
|
||||
*/
|
||||
public function where($restriction)
|
||||
{
|
||||
$this->query['where'][] = $restriction;
|
||||
return $this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Add a restriction in HAVING clause (each restriction will be separated by AND statement)
|
||||
*
|
||||
* @param string $restriction
|
||||
* @return DbQuery
|
||||
*/
|
||||
public function having($restriction)
|
||||
{
|
||||
$this->query['having'][] = $restriction;
|
||||
return $this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Add an ORDER B restriction
|
||||
*
|
||||
* @param string $fields List of fields to sort. E.g. $this->order('myField, b.mySecondField DESC')
|
||||
* @return DbQuery
|
||||
*/
|
||||
public function order($fields)
|
||||
{
|
||||
$this->query['order'][] = $fields;
|
||||
return $this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Add a GROUP BY restriction
|
||||
*
|
||||
* @param string $fields List of fields to sort. E.g. $this->group('myField, b.mySecondField DESC')
|
||||
* @return DbQuery
|
||||
*/
|
||||
public function group($fields)
|
||||
{
|
||||
$this->query['group'][] = $fields;
|
||||
return $this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Limit results in query
|
||||
*
|
||||
* @param string $fields List of fields to sort. E.g. $this->order('myField, b.mySecondField DESC')
|
||||
* @return DbQuery
|
||||
*/
|
||||
public function limit($limit, $offset = 0)
|
||||
{
|
||||
$offset = (int)$offset;
|
||||
if ($offset < 0)
|
||||
$offset = 0;
|
||||
|
||||
$this->query['limit'] = array(
|
||||
'offset' => $offset,
|
||||
'limit' => (int)$limit,
|
||||
);
|
||||
return $this;
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate and get the query
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
public function build()
|
||||
{
|
||||
$sql = 'SELECT '.((($this->query['select'])) ? implode(",\n", $this->query['select']) : '*')."\n";
|
||||
|
||||
if (!$this->query['from'])
|
||||
die('DbQuery->build() missing from clause');
|
||||
$sql .= 'FROM '.$this->query['from']."\n";
|
||||
|
||||
if ($this->query['join'])
|
||||
$sql .= implode("\n", $this->query['join'])."\n";
|
||||
|
||||
if ($this->query['where'])
|
||||
$sql .= 'WHERE '.implode(' AND ', $this->query['where'])."\n";
|
||||
|
||||
if ($this->query['group'])
|
||||
$sql .= 'GROUP BY '.implode(', ', $this->query['group'])."\n";
|
||||
|
||||
if ($this->query['having'])
|
||||
$sql .= 'HAVING '.implode(' AND ', $this->query['having'])."\n";
|
||||
|
||||
if ($this->query['order'])
|
||||
$sql .= 'ORDER BY '.implode(', ', $this->query['order'])."\n";
|
||||
|
||||
if ($this->query['limit']['limit'])
|
||||
{
|
||||
$limit = $this->query['limit'];
|
||||
$sql .= 'LIMIT '.(($limit['offset']) ? $limit['offset'].', '.$limit['limit'] : $limit['limit']);
|
||||
}
|
||||
|
||||
return $sql;
|
||||
}
|
||||
|
||||
public function __toString()
|
||||
{
|
||||
return $this->build();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user