mirror of
https://gitee.com/fastadminnet/framework.git
synced 2026-07-01 20:52:48 +08:00
数据库中间层改为PDO
增加Think\Db\Lite 用于操作原生SQL需求
This commit is contained in:
thinkphp
879
Think/Db.php
879
Think/Db.php
@@ -19,861 +19,94 @@ namespace Think;
|
||||
* @author liu21st <liu21st@gmail.com>
|
||||
*/
|
||||
class Db {
|
||||
// 数据库类型
|
||||
protected $dbType = null;
|
||||
// 是否自动释放查询结果
|
||||
protected $autoFree = false;
|
||||
// 当前操作所属的模型名
|
||||
protected $model = '_think_';
|
||||
// 是否使用永久连接
|
||||
protected $pconnect = false;
|
||||
// 当前SQL指令
|
||||
protected $queryStr = '';
|
||||
protected $modelSql = [];
|
||||
// 最后插入ID
|
||||
protected $lastInsID = null;
|
||||
// 返回或者影响记录数
|
||||
protected $numRows = 0;
|
||||
// 返回字段数
|
||||
protected $numCols = 0;
|
||||
// 事务指令数
|
||||
protected $transTimes = 0;
|
||||
// 错误信息
|
||||
protected $error = '';
|
||||
// 数据库连接ID 支持多个连接
|
||||
protected $linkID = [];
|
||||
// 当前连接ID
|
||||
protected $_linkID = null;
|
||||
// 当前查询ID
|
||||
protected $queryID = null;
|
||||
// 是否已经连接数据库
|
||||
protected $connected = false;
|
||||
// 数据库连接参数配置
|
||||
protected $config = '';
|
||||
// 数据库表达式
|
||||
protected $comparison = ['eq'=>'=','neq'=>'<>','gt'=>'>','egt'=>'>=','lt'=>'<','elt'=>'<=','notlike'=>'NOT LIKE','like'=>'LIKE','in'=>'IN','notin'=>'NOT IN'];
|
||||
// 查询表达式
|
||||
protected $selectSql = 'SELECT%DISTINCT% %FIELD% FROM %TABLE%%JOIN%%WHERE%%GROUP%%HAVING%%ORDER%%LIMIT% %UNION%%COMMENT%';
|
||||
protected $conf = [];
|
||||
|
||||
static private $instance = [];
|
||||
/**
|
||||
* 取得数据库类实例
|
||||
* @static
|
||||
* @access public
|
||||
* @return mixed 返回数据库驱动类
|
||||
*/
|
||||
public static function getInstance($config) {
|
||||
static $_instance = [];
|
||||
public static function getInstance($config=[]) {
|
||||
$md5 = md5(serialize($config));
|
||||
if(!isset(self::$instance[$md5])) {
|
||||
$options = self::parseConfig($config);
|
||||
$class = 'Think\\Db\\Driver\\'.ucwords($options['dbms']);
|
||||
if(class_exists($class)) {
|
||||
self::$instance[$md5] = new $class($options);
|
||||
}else{
|
||||
Error::halt('_DB_TYPE_INVALID_:'.$options['dbms']);
|
||||
}
|
||||
}
|
||||
return self::$instance[$md5];
|
||||
}
|
||||
|
||||
/**
|
||||
* Lite版本数据库引擎 仅支持原生SQL 包括query和execute方法
|
||||
* @static
|
||||
* @access public
|
||||
* @return mixed 返回数据库驱动类
|
||||
*/
|
||||
public static function lite($config=[]) {
|
||||
static $_instance = [];
|
||||
$md5 = md5(serialize($config));
|
||||
if(!isset($_instance[$md5])) {
|
||||
$Db = new Db();
|
||||
$_instance[$md5] = $Db->factory($config);
|
||||
$_instance[$md5] = new Think\Db\Lite(self::parseConfig($config));
|
||||
}
|
||||
return $_instance[$md5];
|
||||
}
|
||||
|
||||
/**
|
||||
* 加载数据库 支持配置文件或者 DSN
|
||||
* @access public
|
||||
* @param mixed $db_config 数据库配置信息
|
||||
* @return string
|
||||
*/
|
||||
public function factory($db_config='') {
|
||||
$this->conf = Config::get();
|
||||
// 读取数据库配置
|
||||
$db_config = $this->parseConfig($db_config);
|
||||
if(empty($db_config['dbms']))
|
||||
throw_exception(Lang::get('_NO_DB_CONFIG_'));
|
||||
// 数据库类型
|
||||
$db_type = strtolower($db_config['dbms']);
|
||||
$class = '\Think\Db\Driver\\'. ucwords($db_type);
|
||||
// 检查驱动类
|
||||
if(class_exists($class)) {
|
||||
$db = new $class($db_config);
|
||||
// 获取当前的数据库类型
|
||||
if( 'pdo' != $db_type )
|
||||
$db->dbType = strtoupper($db_type);
|
||||
else
|
||||
$db->dbType = $this->_getDsnType($db_config['dsn']);
|
||||
}else {
|
||||
// 类没有定义
|
||||
throw_exception(Lang::get('_NO_DB_DRIVER_').': ' . $class);
|
||||
static public function parseConfig($config=[]){
|
||||
if(empty($config)) {
|
||||
$config = Config::get();
|
||||
}
|
||||
return $db;
|
||||
}
|
||||
|
||||
/**
|
||||
* 根据DSN获取数据库类型 返回大写
|
||||
* @access protected
|
||||
* @param string $dsn dsn字符串
|
||||
* @return string
|
||||
*/
|
||||
protected function _getDsnType($dsn) {
|
||||
$match = explode(':',$dsn);
|
||||
$dbType = strtoupper(trim($match[0]));
|
||||
return $dbType;
|
||||
}
|
||||
|
||||
/**
|
||||
* 分析数据库配置信息，支持数组和DSN
|
||||
* @access private
|
||||
* @param mixed $db_config 数据库配置信息
|
||||
* @return string
|
||||
*/
|
||||
private function parseConfig($db_config='') {
|
||||
if ( !empty($db_config) && is_string($db_config)) {
|
||||
// 如果DSN字符串则进行解析
|
||||
$db_config = $this->parseDSN($db_config);
|
||||
}elseif(is_array($db_config)) { // 数组配置
|
||||
$db_config = array_change_key_case($db_config);
|
||||
$db_config = [
|
||||
'dbms' => $db_config['db_type'],
|
||||
'username' => $db_config['db_user'],
|
||||
'password' => $db_config['db_pwd'],
|
||||
'hostname' => $db_config['db_host'],
|
||||
'hostport' => $db_config['db_port'],
|
||||
'database' => $db_config['db_name'],
|
||||
'dsn' => $db_config['db_dsn'],
|
||||
'params' => $db_config['db_params'],
|
||||
];
|
||||
}elseif(empty($db_config)) {
|
||||
// 如果配置为空，读取配置文件设置
|
||||
if( isset($this->conf['db_dsn']) && $this->conf['db_dsn'] && 'pdo' != strtolower($this->conf['db_type']) ) { // 如果设置了DB_DSN 则优先
|
||||
$db_config = $this->parseDSN($this->conf['db_dsn']);
|
||||
}else{
|
||||
$db_config = array (
|
||||
'dbms' => $this->conf['db_type'],
|
||||
'username' => $this->conf['db_user'],
|
||||
'password' => $this->conf['db_pwd'],
|
||||
'hostname' => $this->conf['db_host'],
|
||||
'hostport' => $this->conf['db_port'],
|
||||
'database' => $this->conf['db_name'],
|
||||
'dsn' => isset($this->conf['db_dsn'])?$this->conf['db_dsn']:'',
|
||||
'params' => isset($this->conf['db_params'])?$this->conf['db_params']:'',
|
||||
);
|
||||
}
|
||||
if(is_string($config)) {
|
||||
return self::parseDsn($config);
|
||||
}
|
||||
return $db_config;
|
||||
}
|
||||
|
||||
/**
|
||||
* 初始化数据库连接
|
||||
* @access protected
|
||||
* @param boolean $master 主服务器
|
||||
* @return void
|
||||
*/
|
||||
protected function initConnect($master=true) {
|
||||
if(empty($this->conf)) {
|
||||
$this->conf = Config::get();
|
||||
}
|
||||
|
||||
if(1 == $this->conf['db_deploy_type'])
|
||||
// 采用分布式数据库
|
||||
$this->_linkID = $this->multiConnect($master);
|
||||
else
|
||||
// 默认单数据库
|
||||
if ( !$this->connected ) $this->_linkID = $this->connect();
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接分布式服务器
|
||||
* @access protected
|
||||
* @param boolean $master 主服务器
|
||||
* @return void
|
||||
*/
|
||||
protected function multiConnect($master=false) {
|
||||
static $_config = [];
|
||||
if(empty($_config)) {
|
||||
// 缓存分布式数据库配置解析
|
||||
foreach ($this->config as $key=>$val){
|
||||
$_config[$key] = explode(',',$val);
|
||||
}
|
||||
}
|
||||
// 数据库读写是否分离
|
||||
if($this->conf['db_rw_separate']){
|
||||
// 主从式采用读写分离
|
||||
if($master)
|
||||
// 主服务器写入
|
||||
$r = floor(mt_rand(0,$this->conf['db_master_num']-1));
|
||||
else{
|
||||
if(is_numeric($this->conf['db_slave_no'])) {// 指定服务器读
|
||||
$r = $this->conf['db_slave_no'];
|
||||
}else{
|
||||
// 读操作连接从服务器
|
||||
$r = floor(mt_rand($this->conf['db_master_num'],count($_config['hostname'])-1)); // 每次随机连接的数据库
|
||||
}
|
||||
}
|
||||
}else{
|
||||
// 读写操作不区分服务器
|
||||
$r = floor(mt_rand(0,count($_config['hostname'])-1)); // 每次随机连接的数据库
|
||||
}
|
||||
$db_config = [
|
||||
'username' => isset($_config['username'][$r])?$_config['username'][$r]:$_config['username'][0],
|
||||
'password' => isset($_config['password'][$r])?$_config['password'][$r]:$_config['password'][0],
|
||||
'hostname' => isset($_config['hostname'][$r])?$_config['hostname'][$r]:$_config['hostname'][0],
|
||||
'hostport' => isset($_config['hostport'][$r])?$_config['hostport'][$r]:$_config['hostport'][0],
|
||||
'database' => isset($_config['database'][$r])?$_config['database'][$r]:$_config['database'][0],
|
||||
'dsn' => isset($_config['dsn'][$r])?$_config['dsn'][$r]:$_config['dsn'][0],
|
||||
'params' => isset($_config['params'][$r])?$_config['params'][$r]:$_config['params'][0],
|
||||
];
|
||||
return $this->connect($db_config,$r);
|
||||
return [
|
||||
'dbms' => $config['db_type'],
|
||||
'dsn' => $config['db_dsn'],
|
||||
'username' => $config['db_user'],
|
||||
'password' => $config['db_pwd'],
|
||||
'hostname' => $config['db_host'],
|
||||
'hostport' => $config['db_port'],
|
||||
'database' => $config['db_name'],
|
||||
'params' => $config['db_params'],
|
||||
'charset' => $config['db_charset'],
|
||||
'deploy' => $config['db_deploy'],
|
||||
'socket' => $config['db_unix_socket'],
|
||||
];
|
||||
}
|
||||
|
||||
/**
|
||||
* DSN解析
|
||||
* 格式： mysql://username:passwd@localhost:3306/DbName
|
||||
* 格式： mysql://username:passwd@localhost:3306/DbName?param1=val1¶m2=val2#utf8
|
||||
* @static
|
||||
* @access public
|
||||
* @param string $dsnStr
|
||||
* @return array
|
||||
*/
|
||||
public function parseDSN($dsnStr) {
|
||||
static public function parseDsn($dsnStr) {
|
||||
if( empty($dsnStr) ){return false;}
|
||||
$info = parse_url($dsnStr);
|
||||
if($info['scheme']){
|
||||
$dsn = [
|
||||
if(!$info) {
|
||||
return false;
|
||||
}
|
||||
$dsn = [
|
||||
'dbms' => $info['scheme'],
|
||||
'username' => isset($info['user']) ? $info['user'] : '',
|
||||
'password' => isset($info['pass']) ? $info['pass'] : '',
|
||||
'hostname' => isset($info['host']) ? $info['host'] : '',
|
||||
'hostport' => isset($info['port']) ? $info['port'] : '',
|
||||
'database' => isset($info['path']) ? substr($info['path'],1) : ''
|
||||
];
|
||||
}else {
|
||||
preg_match('/^(.*?)\:\/\/(.*?)\:(.*?)\@(.*?)\:([0-9]{1, 6})\/(.*?)$/',trim($dsnStr),$matches);
|
||||
$dsn = [
|
||||
'dbms' => $matches[1],
|
||||
'username' => $matches[2],
|
||||
'password' => $matches[3],
|
||||
'hostname' => $matches[4],
|
||||
'hostport' => $matches[5],
|
||||
'database' => $matches[6]
|
||||
];
|
||||
}
|
||||
'database' => isset($info['path']) ? substr($info['path'],1) : '',
|
||||
'charset' => isset($info['fragment'])?$info['fragment']:'',
|
||||
];
|
||||
$dsn['dsn'] = ''; // 兼容配置信息数组
|
||||
if(isset($info['query'])) {
|
||||
parse_str($info['query'],$dsn['params']);
|
||||
}else{
|
||||
$dsn['params'] = [];
|
||||
}
|
||||
return $dsn;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库调试 记录当前SQL
|
||||
* @access protected
|
||||
*/
|
||||
protected function debug() {
|
||||
$this->modelSql[$this->model] = $this->queryStr;
|
||||
$this->model = '_think_';
|
||||
// 记录操作结束时间
|
||||
Debug::remark('queryEndTime','time');
|
||||
Log::record($this->queryStr.' [ RunTime:'.Debug::getUseTime('queryStartTime','queryEndTime').'s ]','SQL');
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置锁机制
|
||||
* @access protected
|
||||
* @return string
|
||||
*/
|
||||
protected function parseLock($lock=false) {
|
||||
if(!$lock) return '';
|
||||
if('ORACLE' == $this->dbType) {
|
||||
return ' FOR UPDATE NOWAIT ';
|
||||
}
|
||||
return ' FOR UPDATE ';
|
||||
}
|
||||
|
||||
/**
|
||||
* set分析
|
||||
* @access protected
|
||||
* @param array $data
|
||||
* @return string
|
||||
*/
|
||||
protected function parseSet($data) {
|
||||
foreach ($data as $key=>$val){
|
||||
$value = $this->parseValue($val);
|
||||
if(is_scalar($value)) // 过滤非标量数据
|
||||
$set[] = $this->parseKey($key).'='.$value;
|
||||
}
|
||||
return ' SET '.implode(',',$set);
|
||||
}
|
||||
|
||||
/**
|
||||
* 字段名分析
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @return string
|
||||
*/
|
||||
protected function parseKey(&$key) {
|
||||
return $key;
|
||||
}
|
||||
|
||||
/**
|
||||
* value分析
|
||||
* @access protected
|
||||
* @param mixed $value
|
||||
* @return string
|
||||
*/
|
||||
protected function parseValue($value) {
|
||||
if(is_string($value)) {
|
||||
$value = '\''.$this->escapeString($value).'\'';
|
||||
}elseif(isset($value[0]) && is_string($value[0]) && strtolower($value[0]) == 'exp'){
|
||||
$value = $this->escapeString($value[1]);
|
||||
}elseif(is_array($value)) {
|
||||
$value = array_map(array($this, 'parseValue'),$value);
|
||||
}elseif(is_bool($value)){
|
||||
$value = $value ? '1' : '0';
|
||||
}elseif(is_null($value)){
|
||||
$value = 'null';
|
||||
}
|
||||
return $value;
|
||||
}
|
||||
|
||||
/**
|
||||
* field分析
|
||||
* @access protected
|
||||
* @param mixed $fields
|
||||
* @return string
|
||||
*/
|
||||
protected function parseField($fields) {
|
||||
if(is_string($fields) && strpos($fields,',')) {
|
||||
$fields = explode(',',$fields);
|
||||
}
|
||||
if(is_array($fields)) {
|
||||
// 完善数组方式传字段名的支持
|
||||
// 支持 'field1'=>'field2' 这样的字段别名定义
|
||||
$array = [];
|
||||
foreach ($fields as $key=>$field){
|
||||
if(!is_numeric($key))
|
||||
$array[] = $this->parseKey($key).' AS '.$this->parseKey($field);
|
||||
else
|
||||
$array[] = $this->parseKey($field);
|
||||
}
|
||||
$fieldsStr = implode(',', $array);
|
||||
}elseif(is_string($fields) && !empty($fields)) {
|
||||
$fieldsStr = $this->parseKey($fields);
|
||||
}else{
|
||||
$fieldsStr = '*';
|
||||
}
|
||||
//TODO 如果是查询全部字段，并且是join的方式，那么就把要查的表加个别名，以免字段被覆盖
|
||||
return $fieldsStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* table分析
|
||||
* @access protected
|
||||
* @param mixed $table
|
||||
* @return string
|
||||
*/
|
||||
protected function parseTable($tables) {
|
||||
if(is_array($tables)) {// 支持别名定义
|
||||
$array = [];
|
||||
foreach ($tables as $table=>$alias){
|
||||
if(!is_numeric($table))
|
||||
$array[] = $this->parseKey($table).' '.$this->parseKey($alias);
|
||||
else
|
||||
$array[] = $this->parseKey($table);
|
||||
}
|
||||
$tables = $array;
|
||||
}elseif(is_string($tables)){
|
||||
$tables = explode(',',$tables);
|
||||
array_walk($tables, array(&$this, 'parseKey'));
|
||||
}
|
||||
return implode(',',$tables);
|
||||
}
|
||||
|
||||
/**
|
||||
* where分析
|
||||
* @access protected
|
||||
* @param mixed $where
|
||||
* @return string
|
||||
*/
|
||||
protected function parseWhere($where) {
|
||||
$whereStr = '';
|
||||
if(is_string($where)) {
|
||||
// 直接使用字符串条件
|
||||
$whereStr = $where;
|
||||
}else{ // 使用数组表达式
|
||||
$operate = isset($where['_logic'])?strtoupper($where['_logic']):'';
|
||||
if(in_array($operate,['AND','OR','XOR'])){
|
||||
// 定义逻辑运算规则 例如 OR XOR AND NOT
|
||||
$operate = ' '.$operate.' ';
|
||||
unset($where['_logic']);
|
||||
}else{
|
||||
// 默认进行 AND 运算
|
||||
$operate = ' AND ';
|
||||
}
|
||||
foreach ($where as $key=>$val){
|
||||
$whereStr .= '( ';
|
||||
if(0===strpos($key,'_')) {
|
||||
// 解析特殊条件表达式
|
||||
$whereStr .= $this->parseThinkWhere($key,$val);
|
||||
}else{
|
||||
// 查询字段的安全过滤
|
||||
if(!preg_match('/^[A-Z_\|\&\-.a-z0-9\(\)\,]+$/',trim($key))){
|
||||
throw_exception(L('_EXPRESS_ERROR_').':'.$key);
|
||||
}
|
||||
// 多条件支持
|
||||
$multi = is_array($val) && isset($val['_multi']);
|
||||
$key = trim($key);
|
||||
if(strpos($key,'|')) { // 支持 name|title|nickname 方式定义查询字段
|
||||
$array = explode('|',$key);
|
||||
$str = [];
|
||||
foreach ($array as $m=>$k){
|
||||
$v = $multi?$val[$m]:$val;
|
||||
$str[] = '('.$this->parseWhereItem($this->parseKey($k),$v).')';
|
||||
}
|
||||
$whereStr .= implode(' OR ',$str);
|
||||
}elseif(strpos($key,'&')){
|
||||
$array = explode('&',$key);
|
||||
$str = [];
|
||||
foreach ($array as $m=>$k){
|
||||
$v = $multi?$val[$m]:$val;
|
||||
$str[] = '('.$this->parseWhereItem($this->parseKey($k),$v).')';
|
||||
}
|
||||
$whereStr .= implode(' AND ',$str);
|
||||
}else{
|
||||
$whereStr .= $this->parseWhereItem($this->parseKey($key),$val);
|
||||
}
|
||||
}
|
||||
$whereStr .= ' )'.$operate;
|
||||
}
|
||||
$whereStr = substr($whereStr,0,-strlen($operate));
|
||||
}
|
||||
return empty($whereStr)?'':' WHERE '.$whereStr;
|
||||
}
|
||||
|
||||
// where子单元分析
|
||||
protected function parseWhereItem($key,$val) {
|
||||
$whereStr = '';
|
||||
if(is_array($val)) {
|
||||
if(is_string($val[0])) {
|
||||
if(preg_match('/^(EQ|NEQ|GT|EGT|LT|ELT)$/i',$val[0])) { // 比较运算
|
||||
$whereStr .= $key.' '.$this->comparison[strtolower($val[0])].' '.$this->parseValue($val[1]);
|
||||
}elseif(preg_match('/^(NOTLIKE|LIKE)$/i',$val[0])){// 模糊查找
|
||||
if(is_array($val[1])) {
|
||||
$likeLogic = isset($val[2])?strtoupper($val[2]):'OR';
|
||||
if(in_array($likeLogic,['AND','OR','XOR'])){
|
||||
$likeStr = $this->comparison[strtolower($val[0])];
|
||||
$like = [];
|
||||
foreach ($val[1] as $item){
|
||||
$like[] = $key.' '.$likeStr.' '.$this->parseValue($item);
|
||||
}
|
||||
$whereStr .= '('.implode(' '.$likeLogic.' ',$like).')';
|
||||
}
|
||||
}else{
|
||||
$whereStr .= $key.' '.$this->comparison[strtolower($val[0])].' '.$this->parseValue($val[1]);
|
||||
}
|
||||
}elseif('exp'==strtolower($val[0])){ // 使用表达式
|
||||
$whereStr .= ' ('.$key.' '.$val[1].') ';
|
||||
}elseif(preg_match('/IN/i',$val[0])){ // IN 运算
|
||||
if(isset($val[2]) && 'exp'==$val[2]) {
|
||||
$whereStr .= $key.' '.strtoupper($val[0]).' '.$val[1];
|
||||
}else{
|
||||
if(is_string($val[1])) {
|
||||
$val[1] = explode(',',$val[1]);
|
||||
}
|
||||
$zone = implode(',',$this->parseValue($val[1]));
|
||||
$whereStr .= $key.' '.strtoupper($val[0]).' ('.$zone.')';
|
||||
}
|
||||
}elseif(preg_match('/BETWEEN/i',$val[0])){ // BETWEEN运算
|
||||
$data = is_string($val[1])? explode(',',$val[1]):$val[1];
|
||||
$whereStr .= ' ('.$key.' '.strtoupper($val[0]).' '.$this->parseValue($data[0]).' AND '.$this->parseValue($data[1]).' )';
|
||||
}else{
|
||||
throw_exception(L('_EXPRESS_ERROR_').':'.$val[0]);
|
||||
}
|
||||
}else {
|
||||
$count = count($val);
|
||||
$rule = isset($val[$count-1])?strtoupper($val[$count-1]):'';
|
||||
if(in_array($rule,['AND','OR','XOR'])) {
|
||||
$count = $count -1;
|
||||
}else{
|
||||
$rule = 'AND';
|
||||
}
|
||||
for($i=0;$i<$count;$i++) {
|
||||
$data = is_array($val[$i])?$val[$i][1]:$val[$i];
|
||||
if('exp'==strtolower($val[$i][0])) {
|
||||
$whereStr .= '('.$key.' '.$data.') '.$rule.' ';
|
||||
}else{
|
||||
$op = is_array($val[$i])?$this->comparison[strtolower($val[$i][0])]:'=';
|
||||
$whereStr .= '('.$key.' '.$op.' '.$this->parseValue($data).') '.$rule.' ';
|
||||
}
|
||||
}
|
||||
$whereStr = substr($whereStr,0,-4);
|
||||
}
|
||||
}else {
|
||||
//对字符串类型字段采用模糊匹配
|
||||
if($this->conf['db_like_fields'] && preg_match('/('.$this->conf['db_like_fields'].')/i',$key)) {
|
||||
$val = '%'.$val.'%';
|
||||
$whereStr .= $key.' LIKE '.$this->parseValue($val);
|
||||
}else {
|
||||
$whereStr .= $key.' = '.$this->parseValue($val);
|
||||
}
|
||||
}
|
||||
return $whereStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* 特殊条件分析
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @param mixed $val
|
||||
* @return string
|
||||
*/
|
||||
protected function parseThinkWhere($key,$val) {
|
||||
$whereStr = '';
|
||||
switch($key) {
|
||||
case '_string':
|
||||
// 字符串模式查询条件
|
||||
$whereStr = $val;
|
||||
break;
|
||||
case '_complex':
|
||||
// 复合查询条件
|
||||
$whereStr = substr($this->parseWhere($val),6);
|
||||
break;
|
||||
case '_query':
|
||||
// 字符串模式查询条件
|
||||
parse_str($val,$where);
|
||||
if(isset($where['_logic'])) {
|
||||
$op = ' '.strtoupper($where['_logic']).' ';
|
||||
unset($where['_logic']);
|
||||
}else{
|
||||
$op = ' AND ';
|
||||
}
|
||||
$array = [];
|
||||
foreach ($where as $field=>$data)
|
||||
$array[] = $this->parseKey($field).' = '.$this->parseValue($data);
|
||||
$whereStr = implode($op,$array);
|
||||
break;
|
||||
}
|
||||
return $whereStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* limit分析
|
||||
* @access protected
|
||||
* @param mixed $lmit
|
||||
* @return string
|
||||
*/
|
||||
protected function parseLimit($limit) {
|
||||
return !empty($limit)? ' LIMIT '.$limit.' ':'';
|
||||
}
|
||||
|
||||
/**
|
||||
* join分析
|
||||
* @access protected
|
||||
* @param mixed $join
|
||||
* @return string
|
||||
*/
|
||||
protected function parseJoin($join) {
|
||||
$joinStr = '';
|
||||
if(!empty($join)) {
|
||||
if(is_array($join)) {
|
||||
foreach ($join as $key=>$_join){
|
||||
if(false !== stripos($_join,'JOIN'))
|
||||
$joinStr .= ' '.$_join;
|
||||
else
|
||||
$joinStr .= ' LEFT JOIN ' .$_join;
|
||||
}
|
||||
}else{
|
||||
$joinStr .= ' LEFT JOIN ' .$join;
|
||||
}
|
||||
}
|
||||
//将__TABLE_NAME__这样的字符串替换成正规的表名,并且带上前缀和后缀
|
||||
$joinStr = preg_replace("/__([A-Z_-]+)__/esU",Config::get('db_prefix').".strtolower('$1')",$joinStr);
|
||||
return $joinStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* order分析
|
||||
* @access protected
|
||||
* @param mixed $order
|
||||
* @return string
|
||||
*/
|
||||
protected function parseOrder($order) {
|
||||
if(is_array($order)) {
|
||||
$array = [];
|
||||
foreach ($order as $key=>$val){
|
||||
if(is_numeric($key)) {
|
||||
$array[] = $this->parseKey($val);
|
||||
}else{
|
||||
$array[] = $this->parseKey($key).' '.$val;
|
||||
}
|
||||
}
|
||||
$order = implode(',',$array);
|
||||
}
|
||||
return !empty($order)? ' ORDER BY '.$order:'';
|
||||
}
|
||||
|
||||
/**
|
||||
* group分析
|
||||
* @access protected
|
||||
* @param mixed $group
|
||||
* @return string
|
||||
*/
|
||||
protected function parseGroup($group) {
|
||||
return !empty($group)? ' GROUP BY '.$group:'';
|
||||
}
|
||||
|
||||
/**
|
||||
* having分析
|
||||
* @access protected
|
||||
* @param string $having
|
||||
* @return string
|
||||
*/
|
||||
protected function parseHaving($having) {
|
||||
return !empty($having)? ' HAVING '.$having:'';
|
||||
}
|
||||
|
||||
/**
|
||||
* comment分析
|
||||
* @access protected
|
||||
* @param string $comment
|
||||
* @return string
|
||||
*/
|
||||
protected function parseComment($comment) {
|
||||
return !empty($comment)? ' /* '.$comment.' */':'';
|
||||
}
|
||||
|
||||
/**
|
||||
* distinct分析
|
||||
* @access protected
|
||||
* @param mixed $distinct
|
||||
* @return string
|
||||
*/
|
||||
protected function parseDistinct($distinct) {
|
||||
return !empty($distinct)? ' DISTINCT ' :'';
|
||||
}
|
||||
|
||||
/**
|
||||
* union分析
|
||||
* @access protected
|
||||
* @param mixed $union
|
||||
* @return string
|
||||
*/
|
||||
protected function parseUnion($union) {
|
||||
if(empty($union)) return '';
|
||||
if(isset($union['_all'])) {
|
||||
$str = 'UNION ALL ';
|
||||
unset($union['_all']);
|
||||
}else{
|
||||
$str = 'UNION ';
|
||||
}
|
||||
foreach ($union as $u){
|
||||
$sql[] = $str.(is_array($u)?$this->buildSelectSql($u):$u);
|
||||
}
|
||||
return implode(' ',$sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 插入记录
|
||||
* @access public
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 参数表达式
|
||||
* @param boolean $replace 是否replace
|
||||
* @return false | integer
|
||||
*/
|
||||
public function insert($data,$options=[],$replace=false) {
|
||||
$values = $fields = [];
|
||||
$this->model = $options['model'];
|
||||
foreach ($data as $key=>$val){
|
||||
$value = $this->parseValue($val);
|
||||
if(is_scalar($value)) { // 过滤非标量数据
|
||||
$values[] = $value;
|
||||
$fields[] = $this->parseKey($key);
|
||||
}
|
||||
}
|
||||
$sql = ($replace?'REPLACE':'INSERT').' INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES ('.implode(',', $values).')';
|
||||
$sql .= $this->parseLock(isset($options['lock'])?$options['lock']:false);
|
||||
$sql .= $this->parseComment(!empty($options['comment'])?$options['comment']:'');
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 通过Select方式插入记录
|
||||
* @access public
|
||||
* @param string $fields 要插入的数据表字段名
|
||||
* @param string $table 要插入的数据表名
|
||||
* @param array $option 查询数据参数
|
||||
* @return false | integer
|
||||
*/
|
||||
public function selectInsert($fields,$table,$options=[]) {
|
||||
$this->model = $options['model'];
|
||||
if(is_string($fields)) $fields = explode(',',$fields);
|
||||
array_walk($fields, [$this, 'parseKey']);
|
||||
$sql = 'INSERT INTO '.$this->parseTable($table).' ('.implode(',', $fields).') ';
|
||||
$sql .= $this->buildSelectSql($options);
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新记录
|
||||
* @access public
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function update($data,$options) {
|
||||
$this->model = $options['model'];
|
||||
$sql = 'UPDATE '
|
||||
.$this->parseTable($options['table'])
|
||||
.$this->parseSet($data)
|
||||
.$this->parseWhere(!empty($options['where'])?$options['where']:'')
|
||||
.$this->parseOrder(!empty($options['order'])?$options['order']:'')
|
||||
.$this->parseLimit(!empty($options['limit'])?$options['limit']:'')
|
||||
.$this->parseLock(isset($options['lock'])?$options['lock']:false)
|
||||
.$this->parseComment(!empty($options['comment'])?$options['comment']:'');
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function delete($options=[]) {
|
||||
$this->model = $options['model'];
|
||||
$sql = 'DELETE FROM '
|
||||
.$this->parseTable($options['table'])
|
||||
.$this->parseWhere(!empty($options['where'])?$options['where']:'')
|
||||
.$this->parseOrder(!empty($options['order'])?$options['order']:'')
|
||||
.$this->parseLimit(!empty($options['limit'])?$options['limit']:'')
|
||||
.$this->parseLock(isset($options['lock'])?$options['lock']:false)
|
||||
.$this->parseComment(!empty($options['comment'])?$options['comment']:'');
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return mixed
|
||||
*/
|
||||
public function select($options=[]) {
|
||||
$this->model = $options['model'];
|
||||
$sql = $this->buildSelectSql($options);
|
||||
$cache = isset($options['cache'])?$options['cache']:false;
|
||||
if($cache) { // 查询缓存检测
|
||||
$key = is_string($cache['key'])?$cache['key']:md5($sql);
|
||||
$value = S($key,'',$cache);
|
||||
if(false !== $value) {
|
||||
return $value;
|
||||
}
|
||||
}
|
||||
$result = $this->query($sql);
|
||||
if($cache && false !== $result ) { // 查询缓存写入
|
||||
S($key,$result,$cache);
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成查询SQL
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return string
|
||||
*/
|
||||
public function buildSelectSql($options=[]) {
|
||||
if(isset($options['page'])) {
|
||||
// 根据页数计算limit
|
||||
if(strpos($options['page'],',')) {
|
||||
list($page,$listRows) = explode(',',$options['page']);
|
||||
}else{
|
||||
$page = $options['page'];
|
||||
}
|
||||
$page = $page?$page:1;
|
||||
$listRows= isset($listRows)?$listRows:(is_numeric($options['limit'])?$options['limit']:20);
|
||||
$offset = $listRows*((int)$page-1);
|
||||
$options['limit'] = $offset.','.$listRows;
|
||||
}
|
||||
$sql = $this->parseSql($this->selectSql,$options);
|
||||
$sql .= $this->parseLock(isset($options['lock'])?$options['lock']:false);
|
||||
return $sql;
|
||||
}
|
||||
|
||||
/**
|
||||
* 替换SQL语句中表达式
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return string
|
||||
*/
|
||||
public function parseSql($sql,$options=[]){
|
||||
$sql = str_replace(
|
||||
['%TABLE%','%DISTINCT%','%FIELD%','%JOIN%','%WHERE%','%GROUP%','%HAVING%','%ORDER%','%LIMIT%','%UNION%','%COMMENT%'],
|
||||
[
|
||||
$this->parseTable($options['table']),
|
||||
$this->parseDistinct(isset($options['distinct'])?$options['distinct']:false),
|
||||
$this->parseField(!empty($options['field'])?$options['field']:'*'),
|
||||
$this->parseJoin(!empty($options['join'])?$options['join']:''),
|
||||
$this->parseWhere(!empty($options['where'])?$options['where']:''),
|
||||
$this->parseGroup(!empty($options['group'])?$options['group']:''),
|
||||
$this->parseHaving(!empty($options['having'])?$options['having']:''),
|
||||
$this->parseOrder(!empty($options['order'])?$options['order']:''),
|
||||
$this->parseLimit(!empty($options['limit'])?$options['limit']:''),
|
||||
$this->parseUnion(!empty($options['union'])?$options['union']:''),
|
||||
$this->parseComment(!empty($options['comment'])?$options['comment']:'')
|
||||
],$sql);
|
||||
return $sql;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最近一次查询的sql语句
|
||||
* @param string $model 模型名
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getLastSql($model='') {
|
||||
return $model?$this->modelSql[$model]:$this->queryStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最近插入的ID
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getLastInsID() {
|
||||
return $this->lastInsID;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最近的错误信息
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getError() {
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* SQL指令安全过滤
|
||||
* @access public
|
||||
* @param string $str SQL字符串
|
||||
* @return string
|
||||
*/
|
||||
public function escapeString($str) {
|
||||
return addslashes($str);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置当前操作模型
|
||||
* @access public
|
||||
* @param string $model 模型名
|
||||
* @return void
|
||||
*/
|
||||
public function setModel($model){
|
||||
$this->model = $model;
|
||||
}
|
||||
|
||||
/**
|
||||
* 析构方法
|
||||
* @access public
|
||||
*/
|
||||
public function __destruct() {
|
||||
// 释放查询
|
||||
if ($this->queryID){
|
||||
$this->free();
|
||||
}
|
||||
// 关闭连接
|
||||
$this->close();
|
||||
}
|
||||
|
||||
// 关闭数据库 由驱动类定义
|
||||
public function close(){}
|
||||
}
|
||||
943
Think/Db/Driver.php
Normal file
943
Think/Db/Driver.php
Normal file
@@ -0,0 +1,943 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | TOPThink [ WE CAN DO IT JUST THINK ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2011 http://topthink.com All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
// $Id$
|
||||
namespace Think\Db;
|
||||
use Think\Config;
|
||||
use Think\Debug;
|
||||
use Think\Log;
|
||||
use PDO;
|
||||
class Driver {
|
||||
// PDO操作实例
|
||||
protected $PDOStatement = null;
|
||||
// 当前操作所属的模型名
|
||||
protected $model = '_think_';
|
||||
// 当前SQL指令
|
||||
protected $queryStr = '';
|
||||
protected $modelSql = [];
|
||||
// 最后插入ID
|
||||
protected $lastInsID = null;
|
||||
// 返回或者影响记录数
|
||||
protected $numRows = 0;
|
||||
// 返回字段数
|
||||
protected $numCols = 0;
|
||||
// 事务指令数
|
||||
protected $transTimes = 0;
|
||||
// 错误信息
|
||||
protected $error = '';
|
||||
// 数据库连接ID 支持多个连接
|
||||
protected $linkID = [];
|
||||
// 当前连接ID
|
||||
protected $_linkID = null;
|
||||
// 当前查询ID
|
||||
protected $queryID = null;
|
||||
// 数据库连接参数配置
|
||||
protected $config = [];
|
||||
// 数据库表达式
|
||||
protected $comparison = ['eq'=>'=','neq'=>'<>','gt'=>'>','egt'=>'>=','lt'=>'<','elt'=>'<=','notlike'=>'NOT LIKE','like'=>'LIKE','in'=>'IN','notin'=>'NOT IN'];
|
||||
// 查询表达式
|
||||
protected $selectSql = 'SELECT%DISTINCT% %FIELD% FROM %TABLE%%JOIN%%WHERE%%GROUP%%HAVING%%ORDER%%LIMIT% %UNION%%COMMENT%';
|
||||
protected $queryTimes = 0;
|
||||
protected $executeTimes = 0;
|
||||
// PDO连接参数
|
||||
protected $options = [
|
||||
PDO::ATTR_CASE => PDO::CASE_LOWER,
|
||||
PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
|
||||
PDO::ATTR_ORACLE_NULLS => PDO::NULL_NATURAL,
|
||||
PDO::ATTR_STRINGIFY_FETCHES => false,
|
||||
];
|
||||
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config=''){
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = [];
|
||||
}
|
||||
$this->config['params'] = $this->options+$this->config['params'];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
try{
|
||||
if(empty($config['dsn'])) {
|
||||
$config['dsn'] = $config['dbms'].':dbname='.$config['database'].';host='.$config['hostname'];
|
||||
if(!empty($config['hostport'])) {
|
||||
$config['dsn'] .= ';port='.$config['hostport'];
|
||||
}elseif(!empty($config['socket'])){
|
||||
$config['dsn'] .= ';unix_socket='.$config['socket'];
|
||||
}
|
||||
}
|
||||
$this->linkID[$linkNum] = new PDO( $config['dsn'], $config['username'], $config['password'],$config['params']);
|
||||
}catch (\PDOException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
if(!empty($config['charset'])) {
|
||||
$this->linkID[$linkNum]->exec('SET NAMES '.$config['charset']);
|
||||
}
|
||||
// 注销数据库连接配置信息
|
||||
if(1 != $config['deploy']) $this->config = [];
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
$this->PDOStatement = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str,$bind=[]) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( !empty($this->PDOStatement) ) $this->free();
|
||||
$this->queryTimes++;
|
||||
// 记录开始执行时间
|
||||
Debug::remark('queryStartTime','time');
|
||||
$this->PDOStatement = $this->_linkID->prepare($str);
|
||||
if(false === $this->PDOStatement)
|
||||
throw_exception($this->error());
|
||||
$result = $this->PDOStatement->execute($bind);
|
||||
$this->debug();
|
||||
if ( false === $result ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
return $this->getResult();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str,$bind=[]) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( !empty($this->PDOStatement) ) $this->free();
|
||||
$this->executeTimes++;
|
||||
// 记录开始执行时间
|
||||
Debug::remark('queryStartTime','time');
|
||||
$this->PDOStatement = $this->_linkID->prepare($str);
|
||||
if(false === $this->PDOStatement) {
|
||||
throw_exception($this->error());
|
||||
}
|
||||
$result = $this->PDOStatement->execute($bind);
|
||||
$this->debug();
|
||||
if ( false === $result) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = $this->PDOStatement->rowCount();
|
||||
if(preg_match("/^\s*(INSERT\s+INTO|REPLACE\s+INTO)\s+/i", $str)) {
|
||||
$this->lastInsID = $this->getLastInsertId();
|
||||
}
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最后插入id
|
||||
* @access public
|
||||
* @return integer
|
||||
*/
|
||||
public function getLastInsertId() {
|
||||
return $this->_linkID->lastInsertId();
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
$this->_linkID->beginTransaction();
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = $this->_linkID->commit();
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = $this->_linkID->rollback();
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getResult() {
|
||||
//返回数据集
|
||||
$result = $this->PDOStatement->fetchAll(PDO::FETCH_ASSOC);
|
||||
$this->numRows = count( $result );
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error() {
|
||||
if($this->PDOStatement) {
|
||||
$error = $this->PDOStatement->errorInfo();
|
||||
$this->error = $error[2];
|
||||
}else{
|
||||
$this->error = '';
|
||||
}
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
Log::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置锁机制
|
||||
* @access protected
|
||||
* @return string
|
||||
*/
|
||||
protected function parseLock($lock=false) {
|
||||
return $lock? ' FOR UPDATE ' : '';
|
||||
}
|
||||
|
||||
/**
|
||||
* set分析
|
||||
* @access protected
|
||||
* @param array $data
|
||||
* @return string
|
||||
*/
|
||||
protected function parseSet($data) {
|
||||
foreach ($data as $key=>$val){
|
||||
$value = $this->parseValue($val);
|
||||
if(is_scalar($value)) // 过滤非标量数据
|
||||
$set[] = $this->parseKey($key).'='.$value;
|
||||
}
|
||||
return ' SET '.implode(',',$set);
|
||||
}
|
||||
|
||||
/**
|
||||
* 字段名分析
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @return string
|
||||
*/
|
||||
protected function parseKey(&$key) {
|
||||
return $key;
|
||||
}
|
||||
|
||||
/**
|
||||
* value分析
|
||||
* @access protected
|
||||
* @param mixed $value
|
||||
* @return string
|
||||
*/
|
||||
protected function parseValue($value) {
|
||||
if(is_string($value)) {
|
||||
$value = '\''.$this->escapeString($value).'\'';
|
||||
}elseif(isset($value[0]) && is_string($value[0]) && strtolower($value[0]) == 'exp'){
|
||||
$value = $this->escapeString($value[1]);
|
||||
}elseif(is_array($value)) {
|
||||
$value = array_map([$this, 'parseValue'],$value);
|
||||
}elseif(is_bool($value)){
|
||||
$value = $value ? '1' : '0';
|
||||
}elseif(is_null($value)){
|
||||
$value = 'null';
|
||||
}
|
||||
return $value;
|
||||
}
|
||||
|
||||
/**
|
||||
* field分析
|
||||
* @access protected
|
||||
* @param mixed $fields
|
||||
* @return string
|
||||
*/
|
||||
protected function parseField($fields) {
|
||||
if(is_string($fields) && strpos($fields,',')) {
|
||||
$fields = explode(',',$fields);
|
||||
}
|
||||
if(is_array($fields)) {
|
||||
// 完善数组方式传字段名的支持
|
||||
// 支持 'field1'=>'field2' 这样的字段别名定义
|
||||
$array = [];
|
||||
foreach ($fields as $key=>$field){
|
||||
if(!is_numeric($key))
|
||||
$array[] = $this->parseKey($key).' AS '.$this->parseKey($field);
|
||||
else
|
||||
$array[] = $this->parseKey($field);
|
||||
}
|
||||
$fieldsStr = implode(',', $array);
|
||||
}elseif(is_string($fields) && !empty($fields)) {
|
||||
$fieldsStr = $this->parseKey($fields);
|
||||
}else{
|
||||
$fieldsStr = '*';
|
||||
}
|
||||
//TODO 如果是查询全部字段，并且是join的方式，那么就把要查的表加个别名，以免字段被覆盖
|
||||
return $fieldsStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* table分析
|
||||
* @access protected
|
||||
* @param mixed $table
|
||||
* @return string
|
||||
*/
|
||||
protected function parseTable($tables) {
|
||||
if(is_array($tables)) {// 支持别名定义
|
||||
$array = [];
|
||||
foreach ($tables as $table=>$alias){
|
||||
if(!is_numeric($table))
|
||||
$array[] = $this->parseKey($table).' '.$this->parseKey($alias);
|
||||
else
|
||||
$array[] = $this->parseKey($table);
|
||||
}
|
||||
$tables = $array;
|
||||
}elseif(is_string($tables)){
|
||||
$tables = explode(',',$tables);
|
||||
array_walk($tables, [&$this, 'parseKey']);
|
||||
}
|
||||
return implode(',',$tables);
|
||||
}
|
||||
|
||||
/**
|
||||
* where分析
|
||||
* @access protected
|
||||
* @param mixed $where
|
||||
* @return string
|
||||
*/
|
||||
protected function parseWhere($where) {
|
||||
$whereStr = '';
|
||||
if(is_string($where)) {
|
||||
// 直接使用字符串条件
|
||||
$whereStr = $where;
|
||||
}else{ // 使用数组表达式
|
||||
$operate = isset($where['_logic'])?strtoupper($where['_logic']):'';
|
||||
if(in_array($operate,['AND','OR','XOR'])){
|
||||
// 定义逻辑运算规则 例如 OR XOR AND NOT
|
||||
$operate = ' '.$operate.' ';
|
||||
unset($where['_logic']);
|
||||
}else{
|
||||
// 默认进行 AND 运算
|
||||
$operate = ' AND ';
|
||||
}
|
||||
foreach ($where as $key=>$val){
|
||||
$whereStr .= '( ';
|
||||
if(0===strpos($key,'_')) {
|
||||
// 解析特殊条件表达式
|
||||
$whereStr .= $this->parseThinkWhere($key,$val);
|
||||
}else{
|
||||
// 查询字段的安全过滤
|
||||
if(!preg_match('/^[A-Z_\|\&\-.a-z0-9\(\)\,]+$/',trim($key))){
|
||||
throw_exception(L('_EXPRESS_ERROR_').':'.$key);
|
||||
}
|
||||
// 多条件支持
|
||||
$multi = is_array($val) && isset($val['_multi']);
|
||||
$key = trim($key);
|
||||
if(strpos($key,'|')) { // 支持 name|title|nickname 方式定义查询字段
|
||||
$array = explode('|',$key);
|
||||
$str = [];
|
||||
foreach ($array as $m=>$k){
|
||||
$v = $multi?$val[$m]:$val;
|
||||
$str[] = '('.$this->parseWhereItem($this->parseKey($k),$v).')';
|
||||
}
|
||||
$whereStr .= implode(' OR ',$str);
|
||||
}elseif(strpos($key,'&')){
|
||||
$array = explode('&',$key);
|
||||
$str = [];
|
||||
foreach ($array as $m=>$k){
|
||||
$v = $multi?$val[$m]:$val;
|
||||
$str[] = '('.$this->parseWhereItem($this->parseKey($k),$v).')';
|
||||
}
|
||||
$whereStr .= implode(' AND ',$str);
|
||||
}else{
|
||||
$whereStr .= $this->parseWhereItem($this->parseKey($key),$val);
|
||||
}
|
||||
}
|
||||
$whereStr .= ' )'.$operate;
|
||||
}
|
||||
$whereStr = substr($whereStr,0,-strlen($operate));
|
||||
}
|
||||
return empty($whereStr)?'':' WHERE '.$whereStr;
|
||||
}
|
||||
|
||||
// where子单元分析
|
||||
protected function parseWhereItem($key,$val) {
|
||||
$whereStr = '';
|
||||
if(is_array($val)) {
|
||||
if(is_string($val[0])) {
|
||||
if(preg_match('/^(EQ|NEQ|GT|EGT|LT|ELT)$/i',$val[0])) { // 比较运算
|
||||
$whereStr .= $key.' '.$this->comparison[strtolower($val[0])].' '.$this->parseValue($val[1]);
|
||||
}elseif(preg_match('/^(NOTLIKE|LIKE)$/i',$val[0])){// 模糊查找
|
||||
if(is_array($val[1])) {
|
||||
$likeLogic = isset($val[2])?strtoupper($val[2]):'OR';
|
||||
if(in_array($likeLogic,['AND','OR','XOR'])){
|
||||
$likeStr = $this->comparison[strtolower($val[0])];
|
||||
$like = [];
|
||||
foreach ($val[1] as $item){
|
||||
$like[] = $key.' '.$likeStr.' '.$this->parseValue($item);
|
||||
}
|
||||
$whereStr .= '('.implode(' '.$likeLogic.' ',$like).')';
|
||||
}
|
||||
}else{
|
||||
$whereStr .= $key.' '.$this->comparison[strtolower($val[0])].' '.$this->parseValue($val[1]);
|
||||
}
|
||||
}elseif('exp'==strtolower($val[0])){ // 使用表达式
|
||||
$whereStr .= ' ('.$key.' '.$val[1].') ';
|
||||
}elseif(preg_match('/IN/i',$val[0])){ // IN 运算
|
||||
if(isset($val[2]) && 'exp'==$val[2]) {
|
||||
$whereStr .= $key.' '.strtoupper($val[0]).' '.$val[1];
|
||||
}else{
|
||||
if(is_string($val[1])) {
|
||||
$val[1] = explode(',',$val[1]);
|
||||
}
|
||||
$zone = implode(',',$this->parseValue($val[1]));
|
||||
$whereStr .= $key.' '.strtoupper($val[0]).' ('.$zone.')';
|
||||
}
|
||||
}elseif(preg_match('/BETWEEN/i',$val[0])){ // BETWEEN运算
|
||||
$data = is_string($val[1])? explode(',',$val[1]):$val[1];
|
||||
$whereStr .= ' ('.$key.' '.strtoupper($val[0]).' '.$this->parseValue($data[0]).' AND '.$this->parseValue($data[1]).' )';
|
||||
}else{
|
||||
throw_exception(L('_EXPRESS_ERROR_').':'.$val[0]);
|
||||
}
|
||||
}else {
|
||||
$count = count($val);
|
||||
$rule = isset($val[$count-1])?strtoupper($val[$count-1]):'';
|
||||
if(in_array($rule,['AND','OR','XOR'])) {
|
||||
$count = $count -1;
|
||||
}else{
|
||||
$rule = 'AND';
|
||||
}
|
||||
for($i=0;$i<$count;$i++) {
|
||||
$data = is_array($val[$i])?$val[$i][1]:$val[$i];
|
||||
if('exp'==strtolower($val[$i][0])) {
|
||||
$whereStr .= '('.$key.' '.$data.') '.$rule.' ';
|
||||
}else{
|
||||
$op = is_array($val[$i])?$this->comparison[strtolower($val[$i][0])]:'=';
|
||||
$whereStr .= '('.$key.' '.$op.' '.$this->parseValue($data).') '.$rule.' ';
|
||||
}
|
||||
}
|
||||
$whereStr = substr($whereStr,0,-4);
|
||||
}
|
||||
}else {
|
||||
//对字符串类型字段采用模糊匹配
|
||||
if($this->conf['db_like_fields'] && preg_match('/('.$this->conf['db_like_fields'].')/i',$key)) {
|
||||
$val = '%'.$val.'%';
|
||||
$whereStr .= $key.' LIKE '.$this->parseValue($val);
|
||||
}else {
|
||||
$whereStr .= $key.' = '.$this->parseValue($val);
|
||||
}
|
||||
}
|
||||
return $whereStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* 特殊条件分析
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @param mixed $val
|
||||
* @return string
|
||||
*/
|
||||
protected function parseThinkWhere($key,$val) {
|
||||
$whereStr = '';
|
||||
switch($key) {
|
||||
case '_string':
|
||||
// 字符串模式查询条件
|
||||
$whereStr = $val;
|
||||
break;
|
||||
case '_complex':
|
||||
// 复合查询条件
|
||||
$whereStr = substr($this->parseWhere($val),6);
|
||||
break;
|
||||
case '_query':
|
||||
// 字符串模式查询条件
|
||||
parse_str($val,$where);
|
||||
if(isset($where['_logic'])) {
|
||||
$op = ' '.strtoupper($where['_logic']).' ';
|
||||
unset($where['_logic']);
|
||||
}else{
|
||||
$op = ' AND ';
|
||||
}
|
||||
$array = [];
|
||||
foreach ($where as $field=>$data)
|
||||
$array[] = $this->parseKey($field).' = '.$this->parseValue($data);
|
||||
$whereStr = implode($op,$array);
|
||||
break;
|
||||
}
|
||||
return $whereStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* limit分析
|
||||
* @access protected
|
||||
* @param mixed $lmit
|
||||
* @return string
|
||||
*/
|
||||
protected function parseLimit($limit) {
|
||||
return !empty($limit)? ' LIMIT '.$limit.' ':'';
|
||||
}
|
||||
|
||||
/**
|
||||
* join分析
|
||||
* @access protected
|
||||
* @param mixed $join
|
||||
* @return string
|
||||
*/
|
||||
protected function parseJoin($join) {
|
||||
$joinStr = '';
|
||||
if(!empty($join)) {
|
||||
if(is_array($join)) {
|
||||
foreach ($join as $key=>$_join){
|
||||
if(false !== stripos($_join,'JOIN'))
|
||||
$joinStr .= ' '.$_join;
|
||||
else
|
||||
$joinStr .= ' LEFT JOIN ' .$_join;
|
||||
}
|
||||
}else{
|
||||
$joinStr .= ' LEFT JOIN ' .$join;
|
||||
}
|
||||
}
|
||||
//将__TABLE_NAME__这样的字符串替换成正规的表名,并且带上前缀和后缀
|
||||
$joinStr = preg_replace("/__([A-Z_-]+)__/esU",Config::get('db_prefix').".strtolower('$1')",$joinStr);
|
||||
return $joinStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* order分析
|
||||
* @access protected
|
||||
* @param mixed $order
|
||||
* @return string
|
||||
*/
|
||||
protected function parseOrder($order) {
|
||||
if(is_array($order)) {
|
||||
$array = [];
|
||||
foreach ($order as $key=>$val){
|
||||
if(is_numeric($key)) {
|
||||
$array[] = $this->parseKey($val);
|
||||
}else{
|
||||
$array[] = $this->parseKey($key).' '.$val;
|
||||
}
|
||||
}
|
||||
$order = implode(',',$array);
|
||||
}
|
||||
return !empty($order)? ' ORDER BY '.$order:'';
|
||||
}
|
||||
|
||||
/**
|
||||
* group分析
|
||||
* @access protected
|
||||
* @param mixed $group
|
||||
* @return string
|
||||
*/
|
||||
protected function parseGroup($group) {
|
||||
return !empty($group)? ' GROUP BY '.$group:'';
|
||||
}
|
||||
|
||||
/**
|
||||
* having分析
|
||||
* @access protected
|
||||
* @param string $having
|
||||
* @return string
|
||||
*/
|
||||
protected function parseHaving($having) {
|
||||
return !empty($having)? ' HAVING '.$having:'';
|
||||
}
|
||||
|
||||
/**
|
||||
* comment分析
|
||||
* @access protected
|
||||
* @param string $comment
|
||||
* @return string
|
||||
*/
|
||||
protected function parseComment($comment) {
|
||||
return !empty($comment)? ' /* '.$comment.' */':'';
|
||||
}
|
||||
|
||||
/**
|
||||
* distinct分析
|
||||
* @access protected
|
||||
* @param mixed $distinct
|
||||
* @return string
|
||||
*/
|
||||
protected function parseDistinct($distinct) {
|
||||
return !empty($distinct)? ' DISTINCT ' :'';
|
||||
}
|
||||
|
||||
/**
|
||||
* union分析
|
||||
* @access protected
|
||||
* @param mixed $union
|
||||
* @return string
|
||||
*/
|
||||
protected function parseUnion($union) {
|
||||
if(empty($union)) return '';
|
||||
if(isset($union['_all'])) {
|
||||
$str = 'UNION ALL ';
|
||||
unset($union['_all']);
|
||||
}else{
|
||||
$str = 'UNION ';
|
||||
}
|
||||
foreach ($union as $u){
|
||||
$sql[] = $str.(is_array($u)?$this->buildSelectSql($u):$u);
|
||||
}
|
||||
return implode(' ',$sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 插入记录
|
||||
* @access public
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 参数表达式
|
||||
* @param boolean $replace 是否replace
|
||||
* @return false | integer
|
||||
*/
|
||||
public function insert($data,$options=[],$replace=false) {
|
||||
$values = $fields = [];
|
||||
$this->model = $options['model'];
|
||||
foreach ($data as $key=>$val){
|
||||
$value = $this->parseValue($val);
|
||||
if(is_scalar($value)) { // 过滤非标量数据
|
||||
$values[] = $value;
|
||||
$fields[] = $this->parseKey($key);
|
||||
}
|
||||
}
|
||||
$sql = ($replace?'REPLACE':'INSERT').' INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES ('.implode(',', $values).')';
|
||||
$sql .= $this->parseLock(isset($options['lock'])?$options['lock']:false);
|
||||
$sql .= $this->parseComment(!empty($options['comment'])?$options['comment']:'');
|
||||
return $this->execute($sql,!empty($options['bind'])?$options['bind']:[]);
|
||||
}
|
||||
|
||||
/**
|
||||
* 通过Select方式插入记录
|
||||
* @access public
|
||||
* @param string $fields 要插入的数据表字段名
|
||||
* @param string $table 要插入的数据表名
|
||||
* @param array $option 查询数据参数
|
||||
* @return false | integer
|
||||
*/
|
||||
public function selectInsert($fields,$table,$options=[]) {
|
||||
$this->model = $options['model'];
|
||||
if(is_string($fields)) $fields = explode(',',$fields);
|
||||
array_walk($fields, [$this, 'parseKey']);
|
||||
$sql = 'INSERT INTO '.$this->parseTable($table).' ('.implode(',', $fields).') ';
|
||||
$sql .= $this->buildSelectSql($options);
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新记录
|
||||
* @access public
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function update($data,$options) {
|
||||
$this->model = $options['model'];
|
||||
$sql = 'UPDATE '
|
||||
.$this->parseTable($options['table'])
|
||||
.$this->parseSet($data)
|
||||
.$this->parseWhere(!empty($options['where'])?$options['where']:'')
|
||||
.$this->parseOrder(!empty($options['order'])?$options['order']:'')
|
||||
.$this->parseLimit(!empty($options['limit'])?$options['limit']:'')
|
||||
.$this->parseLock(isset($options['lock'])?$options['lock']:false)
|
||||
.$this->parseComment(!empty($options['comment'])?$options['comment']:'');
|
||||
return $this->execute($sql,!empty($options['bind'])?$options['bind']:[]);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function delete($options=[]) {
|
||||
$this->model = $options['model'];
|
||||
$sql = 'DELETE FROM '
|
||||
.$this->parseTable($options['table'])
|
||||
.$this->parseWhere(!empty($options['where'])?$options['where']:'')
|
||||
.$this->parseOrder(!empty($options['order'])?$options['order']:'')
|
||||
.$this->parseLimit(!empty($options['limit'])?$options['limit']:'')
|
||||
.$this->parseLock(isset($options['lock'])?$options['lock']:false)
|
||||
.$this->parseComment(!empty($options['comment'])?$options['comment']:'');
|
||||
return $this->execute($sql,!empty($options['bind'])?$options['bind']:[]);
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return mixed
|
||||
*/
|
||||
public function select($options=[]) {
|
||||
$this->model = $options['model'];
|
||||
$sql = $this->buildSelectSql($options);
|
||||
$cache = isset($options['cache'])?$options['cache']:false;
|
||||
if($cache) { // 查询缓存检测
|
||||
$key = is_string($cache['key'])?$cache['key']:md5($sql);
|
||||
$value = S($key,'',$cache);
|
||||
if(false !== $value) {
|
||||
return $value;
|
||||
}
|
||||
}
|
||||
$result = $this->query($sql,!empty($options['bind'])?$options['bind']:[]);
|
||||
if($cache && false !== $result ) { // 查询缓存写入
|
||||
S($key,$result,$cache);
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成查询SQL
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return string
|
||||
*/
|
||||
public function buildSelectSql($options=[]) {
|
||||
if(isset($options['page'])) {
|
||||
// 根据页数计算limit
|
||||
if(strpos($options['page'],',')) {
|
||||
list($page,$listRows) = explode(',',$options['page']);
|
||||
}else{
|
||||
$page = $options['page'];
|
||||
}
|
||||
$page = $page?$page:1;
|
||||
$listRows= isset($listRows)?$listRows:(is_numeric($options['limit'])?$options['limit']:20);
|
||||
$offset = $listRows*((int)$page-1);
|
||||
$options['limit'] = $offset.','.$listRows;
|
||||
}
|
||||
$sql = $this->parseSql($this->selectSql,$options);
|
||||
$sql .= $this->parseLock(isset($options['lock'])?$options['lock']:false);
|
||||
return $sql;
|
||||
}
|
||||
|
||||
/**
|
||||
* 替换SQL语句中表达式
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return string
|
||||
*/
|
||||
public function parseSql($sql,$options=[]){
|
||||
$sql = str_replace(
|
||||
['%TABLE%','%DISTINCT%','%FIELD%','%JOIN%','%WHERE%','%GROUP%','%HAVING%','%ORDER%','%LIMIT%','%UNION%','%COMMENT%'],
|
||||
[
|
||||
$this->parseTable($options['table']),
|
||||
$this->parseDistinct(isset($options['distinct'])?$options['distinct']:false),
|
||||
$this->parseField(!empty($options['field'])?$options['field']:'*'),
|
||||
$this->parseJoin(!empty($options['join'])?$options['join']:''),
|
||||
$this->parseWhere(!empty($options['where'])?$options['where']:''),
|
||||
$this->parseGroup(!empty($options['group'])?$options['group']:''),
|
||||
$this->parseHaving(!empty($options['having'])?$options['having']:''),
|
||||
$this->parseOrder(!empty($options['order'])?$options['order']:''),
|
||||
$this->parseLimit(!empty($options['limit'])?$options['limit']:''),
|
||||
$this->parseUnion(!empty($options['union'])?$options['union']:''),
|
||||
$this->parseComment(!empty($options['comment'])?$options['comment']:'')
|
||||
],$sql);
|
||||
return $sql;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最近一次查询的sql语句
|
||||
* @param string $model 模型名
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getLastSql($model='') {
|
||||
return $model?$this->modelSql[$model]:$this->queryStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最近插入的ID
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getLastInsID() {
|
||||
return $this->lastInsID;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最近的错误信息
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getError() {
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* SQL指令安全过滤
|
||||
* @access public
|
||||
* @param string $str SQL字符串
|
||||
* @return string
|
||||
*/
|
||||
public function escapeString($str) {
|
||||
return addslashes($str);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置当前操作模型
|
||||
* @access public
|
||||
* @param string $model 模型名
|
||||
* @return void
|
||||
*/
|
||||
public function setModel($model){
|
||||
$this->model = $model;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库调试 记录当前SQL
|
||||
* @access protected
|
||||
*/
|
||||
protected function debug() {
|
||||
$this->modelSql[$this->model] = $this->queryStr;
|
||||
$this->model = '_think_';
|
||||
// 记录操作结束时间
|
||||
Debug::remark('queryEndTime','time');
|
||||
Log::record($this->queryStr.' [ RunTime:'.Debug::getUseTime('queryStartTime','queryEndTime').'s ]','SQL');
|
||||
}
|
||||
|
||||
/**
|
||||
* 初始化数据库连接
|
||||
* @access protected
|
||||
* @param boolean $master 主服务器
|
||||
* @return void
|
||||
*/
|
||||
protected function initConnect($master=true) {
|
||||
if(1 == $this->config['deploy'])
|
||||
// 采用分布式数据库
|
||||
$this->_linkID = $this->multiConnect($master);
|
||||
else
|
||||
// 默认单数据库
|
||||
if ( !$this->_linkID ) $this->_linkID = $this->connect();
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接分布式服务器
|
||||
* @access protected
|
||||
* @param boolean $master 主服务器
|
||||
* @return void
|
||||
*/
|
||||
protected function multiConnect($master=false) {
|
||||
static $_config = [];
|
||||
if(empty($_config)) {
|
||||
// 缓存分布式数据库配置解析
|
||||
foreach ($this->config as $key=>$val){
|
||||
$_config[$key] = explode(',',$val);
|
||||
}
|
||||
}
|
||||
// 数据库读写是否分离
|
||||
if(Config::get('db_rw_separate')){
|
||||
// 主从式采用读写分离
|
||||
if($master)
|
||||
// 主服务器写入
|
||||
$r = floor(mt_rand(0,Config::get('db_master_num')-1));
|
||||
else{
|
||||
if(is_numeric(Config::get('db_slave_no'))) {// 指定服务器读
|
||||
$r = Config::get('db_slave_no');
|
||||
}else{
|
||||
// 读操作连接从服务器
|
||||
$r = floor(mt_rand(Config::get('db_master_num'),count($_config['hostname'])-1)); // 每次随机连接的数据库
|
||||
}
|
||||
}
|
||||
}else{
|
||||
// 读写操作不区分服务器
|
||||
$r = floor(mt_rand(0,count($_config['hostname'])-1)); // 每次随机连接的数据库
|
||||
}
|
||||
$db_config = [
|
||||
'username' => isset($_config['username'][$r])?$_config['username'][$r]:$_config['username'][0],
|
||||
'password' => isset($_config['password'][$r])?$_config['password'][$r]:$_config['password'][0],
|
||||
'hostname' => isset($_config['hostname'][$r])?$_config['hostname'][$r]:$_config['hostname'][0],
|
||||
'hostport' => isset($_config['hostport'][$r])?$_config['hostport'][$r]:$_config['hostport'][0],
|
||||
'database' => isset($_config['database'][$r])?$_config['database'][$r]:$_config['database'][0],
|
||||
'dsn' => isset($_config['dsn'][$r])?$_config['dsn'][$r]:$_config['dsn'][0],
|
||||
'params' => isset($_config['params'][$r])?$_config['params'][$r]:$_config['params'][0],
|
||||
];
|
||||
return $this->connect($db_config,$r);
|
||||
}
|
||||
|
||||
/**
|
||||
* 析构方法
|
||||
* @access public
|
||||
*/
|
||||
public function __destruct() {
|
||||
// 释放查询
|
||||
if ($this->queryID){
|
||||
$this->free();
|
||||
}
|
||||
// 关闭连接
|
||||
$this->close();
|
||||
}
|
||||
}
|
||||
@@ -1,339 +0,0 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006-2012 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
/**
|
||||
* Firebird数据库驱动
|
||||
* @category Extend
|
||||
* @package Extend
|
||||
* @subpackage Driver.Db
|
||||
* @author 剑雷
|
||||
*/
|
||||
class Ibase extends Db{
|
||||
|
||||
protected $selectSql = 'SELECT %LIMIT% %DISTINCT% %FIELD% FROM %TABLE%%JOIN%%WHERE%%GROUP%%HAVING%%ORDER%';
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config='') {
|
||||
if ( !extension_loaded('interbase') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').':Interbase or Firebird');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = array();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
* @throws ThinkExecption
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
$pconnect = !empty($config['params']['persist'])? $config['params']['persist']:$this->pconnect;
|
||||
$conn = $pconnect ? 'ibase_pconnect':'ibase_connect';
|
||||
// 处理不带端口号的socket连接情况
|
||||
$host = $config['hostname'].($config['hostport']?"/{$config['hostport']}":'');
|
||||
$this->linkID[$linkNum] = $conn($host.':'.$config['database'], $config['username'], $config['password'],C('DB_CHARSET'),0,3);
|
||||
if ( !$this->linkID[$linkNum]) {
|
||||
throw_exception(ibase_errmsg());
|
||||
}
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
// 注销数据库连接配置信息
|
||||
if(1 != Config::get('DB_DEPLOY_TYPE')) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
ibase_free_result($this->queryID);
|
||||
$this->queryID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->queryID = ibase_query($this->_linkID, $str);
|
||||
$this->debug();
|
||||
if ( false === $this->queryID ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
return $this->getAll();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_write',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = ibase_query($this->_linkID, $str) ;
|
||||
$this->debug();
|
||||
if ( false === $result) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = ibase_affected_rows($this->_linkID);
|
||||
$this->lastInsID =0;
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
ibase_trans( IBASE_DEFAULT, $this->_linkID);
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = ibase_commit($this->_linkID);
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result =ibase_rollback($this->_linkID);
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* BLOB字段解密函数 Firebird特有
|
||||
* @access public
|
||||
* @param $blob 待解密的BLOB
|
||||
* @return 二进制数据
|
||||
*/
|
||||
public function BlobDecode($blob) {
|
||||
$maxblobsize = 262144;
|
||||
$blob_data = ibase_blob_info($this->_linkID, $blob );
|
||||
$blobid = ibase_blob_open($this->_linkID, $blob );
|
||||
if( $blob_data[0] > $maxblobsize ) {
|
||||
$realblob = ibase_blob_get($blobid, $maxblobsize);
|
||||
while($string = ibase_blob_get($blobid, 8192)){
|
||||
$realblob .= $string;
|
||||
}
|
||||
} else {
|
||||
$realblob = ibase_blob_get($blobid, $blob_data[0]);
|
||||
}
|
||||
ibase_blob_close( $blobid );
|
||||
return( $realblob );
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getAll() {
|
||||
//返回数据集
|
||||
$result = array();
|
||||
while ( $row = ibase_fetch_assoc($this->queryID)) {
|
||||
$result[] = $row;
|
||||
}
|
||||
//剑雷 2007.12.30 自动解密BLOB字段
|
||||
//取BLOB字段清单
|
||||
$bloblist = array();
|
||||
$fieldCount = ibase_num_fields($this->queryID);
|
||||
for ($i = 0; $i < $fieldCount; $i++) {
|
||||
$col_info = ibase_field_info($this->queryID, $i);
|
||||
if ($col_info['type']=='BLOB') {
|
||||
$bloblist[]=trim($col_info['name']);
|
||||
}
|
||||
}
|
||||
//如果有BLOB字段,就进行解密处理
|
||||
if (!empty($bloblist)) {
|
||||
$i=0;
|
||||
foreach ($result as $row) {
|
||||
foreach($bloblist as $field) {
|
||||
if (!empty($row[$field])) $result[$i][$field]=$this->BlobDecode($row[$field]);
|
||||
}
|
||||
$i++;
|
||||
}
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
*/
|
||||
public function getFields($tableName) {
|
||||
$result = $this->query('SELECT RDB$FIELD_NAME AS FIELD, RDB$DEFAULT_VALUE AS DEFAULT1, RDB$NULL_FLAG AS NULL1 FROM RDB$RELATION_FIELDS WHERE RDB$RELATION_NAME=UPPER(\''.$tableName.'\') ORDER By RDB$FIELD_POSITION');
|
||||
$info = array();
|
||||
if($result) {
|
||||
foreach ($result as $key => $val) {
|
||||
$info[trim($val['FIELD'])] = array(
|
||||
'name' => trim($val['FIELD']),
|
||||
'type' => '',
|
||||
'notnull' => (bool) ($val['NULL1'] ==1), // 1表示不为Null
|
||||
'default' => $val['DEFAULT1'],
|
||||
'primary' => false,
|
||||
'autoinc' => false,
|
||||
);
|
||||
}
|
||||
}
|
||||
//剑雷 取表字段类型
|
||||
$sql='select first 1 * from '. $tableName;
|
||||
$rs_temp = ibase_query ($this->_linkID, $sql);
|
||||
$fieldCount = ibase_num_fields($rs_temp);
|
||||
|
||||
for ($i = 0; $i < $fieldCount; $i++)
|
||||
{
|
||||
$col_info = ibase_field_info($rs_temp, $i);
|
||||
$info[trim($col_info['name'])]['type']=$col_info['type'];
|
||||
}
|
||||
ibase_free_result ($rs_temp);
|
||||
|
||||
//剑雷 取表的主键
|
||||
$sql='select b.rdb$field_name as FIELD_NAME from rdb$relation_constraints a join rdb$index_segments b
|
||||
on a.rdb$index_name=b.rdb$index_name
|
||||
where a.rdb$constraint_type=\'PRIMARY KEY\' and a.rdb$relation_name=UPPER(\''.$tableName.'\')';
|
||||
$rs_temp = ibase_query ($this->_linkID, $sql);
|
||||
while ($row=ibase_fetch_object($rs_temp)) {
|
||||
$info[trim($row->FIELD_NAME)]['primary']=True;
|
||||
}
|
||||
ibase_free_result ($rs_temp);
|
||||
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据库的表信息
|
||||
* @access public
|
||||
*/
|
||||
public function getTables($dbName='') {
|
||||
$sql='SELECT DISTINCT RDB$RELATION_NAME FROM RDB$RELATION_FIELDS WHERE RDB$SYSTEM_FLAG=0';
|
||||
$result = $this->query($sql);
|
||||
$info = array();
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$key] = trim(current($val));
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
if ($this->_linkID){
|
||||
ibase_close($this->_linkID);
|
||||
}
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error() {
|
||||
$this->error = ibase_errmsg();
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
ThinkLog::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* SQL指令安全过滤
|
||||
* @access public
|
||||
* @param string $str SQL指令
|
||||
* @return string
|
||||
*/
|
||||
public function escapeString($str) {
|
||||
return str_replace("'", "''", $str);
|
||||
}
|
||||
|
||||
/**
|
||||
* limit
|
||||
* @access public
|
||||
* @param $limit limit表达式
|
||||
* @return string
|
||||
*/
|
||||
public function parseLimit($limit) {
|
||||
$limitStr = '';
|
||||
if(!empty($limit)) {
|
||||
$limit = explode(',',$limit);
|
||||
if(count($limit)>1) {
|
||||
$limitStr = ' FIRST '.($limit[1]-$limit[0]).' SKIP '.$limit[0].' ';
|
||||
}else{
|
||||
$limitStr = ' FIRST '.$limit[0].' ';
|
||||
}
|
||||
}
|
||||
return $limitStr;
|
||||
}
|
||||
}
|
||||
@@ -1,752 +0,0 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | TOPThink [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006-2012 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
/**
|
||||
* Mongo数据库驱动 必须配合MongoModel使用
|
||||
* @category Extend
|
||||
* @package Extend
|
||||
* @subpackage Driver.Db
|
||||
* @author liu21st <liu21st@gmail.com>
|
||||
*/
|
||||
class Mongo extends Db{
|
||||
|
||||
protected $_mongo = null; // MongoDb Object
|
||||
protected $_collection = null; // MongoCollection Object
|
||||
protected $_dbName = ''; // dbName
|
||||
protected $_collectionName = ''; // collectionName
|
||||
protected $_cursor = null; // MongoCursor Object
|
||||
protected $comparison = array('neq'=>'ne','ne'=>'ne','gt'=>'gt','egt'=>'gte','gte'=>'gte','lt'=>'lt','elt'=>'lte','lte'=>'lte','in'=>'in','not in'=>'nin','nin'=>'nin');
|
||||
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config=''){
|
||||
if ( !class_exists('mongo') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').':mongo');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = array();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
$host = 'mongodb://'.($config['username']?"{$config['username']}":'').($config['password']?":{$config['password']}@":'').$config['hostname'].($config['hostport']?":{$config['hostport']}":'').'/'.($config['database']?"{$config['database']}":'');
|
||||
try{
|
||||
$this->linkID[$linkNum] = new mongo( $host,$config['params']);
|
||||
}catch (MongoConnectionException $e){
|
||||
throw_exception($e->getmessage());
|
||||
}
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
// 注销数据库连接配置信息
|
||||
if(1 != C('DB_DEPLOY_TYPE')) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 切换当前操作的Db和Collection
|
||||
* @access public
|
||||
* @param string $collection collection
|
||||
* @param string $db db
|
||||
* @param boolean $master 是否主服务器
|
||||
* @return void
|
||||
*/
|
||||
public function switchCollection($collection,$db='',$master=true){
|
||||
// 当前没有连接 则首先进行数据库连接
|
||||
if ( !$this->_linkID ) $this->initConnect($master);
|
||||
try{
|
||||
if(!empty($db)) { // 传人Db则切换数据库
|
||||
// 当前MongoDb对象
|
||||
$this->_dbName = $db;
|
||||
$this->_mongo = $this->_linkID->selectDb($db);
|
||||
}
|
||||
// 当前MongoCollection对象
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.getCollection('.$collection.')';
|
||||
}
|
||||
if($this->_collectionName != $collection) {
|
||||
N('db_read',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->_collection = $this->_mongo->selectCollection($collection);
|
||||
$this->debug();
|
||||
$this->_collectionName = $collection; // 记录当前Collection名称
|
||||
}
|
||||
}catch (MongoException $e){
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
$this->_cursor = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行命令
|
||||
* @access public
|
||||
* @param array $command 指令
|
||||
* @return array
|
||||
*/
|
||||
public function command($command=array()) {
|
||||
N('db_write',1);
|
||||
$this->queryStr = 'command:'.json_encode($command);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_mongo->command($command);
|
||||
$this->debug();
|
||||
if(!$result['ok']) {
|
||||
throw_exception($result['errmsg']);
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $code sql指令
|
||||
* @param array $args 参数
|
||||
* @return mixed
|
||||
*/
|
||||
public function execute($code,$args=array()) {
|
||||
N('db_write',1);
|
||||
$this->queryStr = 'execute:'.$code;
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_mongo->execute($code,$args);
|
||||
$this->debug();
|
||||
if($result['ok']) {
|
||||
return $result['retval'];
|
||||
}else{
|
||||
throw_exception($result['errmsg']);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
if($this->_linkID) {
|
||||
$this->_linkID->close();
|
||||
$this->_linkID = null;
|
||||
$this->_mongo = null;
|
||||
$this->_collection = null;
|
||||
$this->_cursor = null;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error() {
|
||||
$this->error = $this->_mongo->lastError();
|
||||
ThinkLog::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* 插入记录
|
||||
* @access public
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 参数表达式
|
||||
* @param boolean $replace 是否replace
|
||||
* @return false | integer
|
||||
*/
|
||||
public function insert($data,$options=array(),$replace=false) {
|
||||
if(isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
N('db_write',1);
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.'.$this->_collectionName.'.insert(';
|
||||
$this->queryStr .= $data?json_encode($data):'{}';
|
||||
$this->queryStr .= ')';
|
||||
}
|
||||
try{
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $replace? $this->_collection->save($data,true): $this->_collection->insert($data,true);
|
||||
$this->debug();
|
||||
if($result) {
|
||||
$_id = $data['_id'];
|
||||
if(is_object($_id)) {
|
||||
$_id = $_id->__toString();
|
||||
}
|
||||
$this->lastInsID = $_id;
|
||||
}
|
||||
return $result;
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 插入多条记录
|
||||
* @access public
|
||||
* @param array $dataList 数据
|
||||
* @param array $options 参数表达式
|
||||
* @return bool
|
||||
*/
|
||||
public function insertAll($dataList,$options=array()) {
|
||||
if(isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
N('db_write',1);
|
||||
try{
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_collection->batchInsert($dataList);
|
||||
$this->debug();
|
||||
return $result;
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 生成下一条记录ID 用于自增非MongoId主键
|
||||
* @access public
|
||||
* @param string $pk 主键名
|
||||
* @return integer
|
||||
*/
|
||||
public function mongo_next_id($pk) {
|
||||
N('db_read',1);
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.'.$this->_collectionName.'.find({},{'.$pk.':1}).sort({'.$pk.':-1}).limit(1)';
|
||||
}
|
||||
try{
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_collection->find(array(),array($pk=>1))->sort(array($pk=>-1))->limit(1);
|
||||
$this->debug();
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
$data = $result->getNext();
|
||||
return isset($data[$pk])?$data[$pk]+1:1;
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新记录
|
||||
* @access public
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 表达式
|
||||
* @return bool
|
||||
*/
|
||||
public function update($data,$options) {
|
||||
if(isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
N('db_write',1);
|
||||
$query = $this->parseWhere($options['where']);
|
||||
$set = $this->parseSet($data);
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.'.$this->_collectionName.'.update(';
|
||||
$this->queryStr .= $query?json_encode($query):'{}';
|
||||
$this->queryStr .= ','.json_encode($set).')';
|
||||
}
|
||||
try{
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_collection->update($query,$set,array("multiple" => true));
|
||||
$this->debug();
|
||||
return $result;
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function delete($options=array()) {
|
||||
if(isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$query = $this->parseWhere($options['where']);
|
||||
$this->model = $options['model'];
|
||||
N('db_write',1);
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.'.$this->_collectionName.'.remove('.json_encode($query).')';
|
||||
}
|
||||
try{
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_collection->remove($query);
|
||||
$this->debug();
|
||||
return $result;
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 清空记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function clear($options=array()){
|
||||
if(isset($options['table'])) {
|
||||
$this->switchCollection($options['table']);
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
N('db_write',1);
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.'.$this->_collectionName.'.remove({})';
|
||||
}
|
||||
try{
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_collection->drop();
|
||||
$this->debug();
|
||||
return $result;
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return iterator
|
||||
*/
|
||||
public function select($options=array()) {
|
||||
if(isset($options['table'])) {
|
||||
$this->switchCollection($options['table'],'',false);
|
||||
}
|
||||
$cache = isset($options['cache'])?$options['cache']:false;
|
||||
if($cache) { // 查询缓存检测
|
||||
$key = is_string($cache['key'])?$cache['key']:md5(serialize($options));
|
||||
$value = S($key,'','',$cache['type']);
|
||||
if(false !== $value) {
|
||||
return $value;
|
||||
}
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
N('db_query',1);
|
||||
$query = $this->parseWhere($options['where']);
|
||||
$field = $this->parseField($options['field']);
|
||||
try{
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.'.$this->_collectionName.'.find(';
|
||||
$this->queryStr .= $query? json_encode($query):'{}';
|
||||
$this->queryStr .= $field? ','.json_encode($field):'';
|
||||
$this->queryStr .= ')';
|
||||
}
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$_cursor = $this->_collection->find($query,$field);
|
||||
if($options['order']) {
|
||||
$order = $this->parseOrder($options['order']);
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr .= '.sort('.json_encode($order).')';
|
||||
}
|
||||
$_cursor = $_cursor->sort($order);
|
||||
}
|
||||
if(isset($options['page'])) { // 根据页数计算limit
|
||||
if(strpos($options['page'],',')) {
|
||||
list($page,$length) = explode(',',$options['page']);
|
||||
}else{
|
||||
$page = $options['page'];
|
||||
}
|
||||
$page = $page?$page:1;
|
||||
$length = isset($length)?$length:(is_numeric($options['limit'])?$options['limit']:20);
|
||||
$offset = $length*((int)$page-1);
|
||||
$options['limit'] = $offset.','.$length;
|
||||
}
|
||||
if(isset($options['limit'])) {
|
||||
list($offset,$length) = $this->parseLimit($options['limit']);
|
||||
if(!empty($offset)) {
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr .= '.skip('.intval($offset).')';
|
||||
}
|
||||
$_cursor = $_cursor->skip(intval($offset));
|
||||
}
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr .= '.limit('.intval($length).')';
|
||||
}
|
||||
$_cursor = $_cursor->limit(intval($length));
|
||||
}
|
||||
$this->debug();
|
||||
$this->_cursor = $_cursor;
|
||||
$resultSet = iterator_to_array($_cursor);
|
||||
if($cache && $resultSet ) { // 查询缓存写入
|
||||
S($key,$resultSet,$cache['expire'],$cache['type']);
|
||||
}
|
||||
return $resultSet;
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 查找某个记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return array
|
||||
*/
|
||||
public function find($options=array()){
|
||||
if(isset($options['table'])) {
|
||||
$this->switchCollection($options['table'],'',false);
|
||||
}
|
||||
$cache = isset($options['cache'])?$options['cache']:false;
|
||||
if($cache) { // 查询缓存检测
|
||||
$key = is_string($cache['key'])?$cache['key']:md5(serialize($options));
|
||||
$value = S($key,'','',$cache['type']);
|
||||
if(false !== $value) {
|
||||
return $value;
|
||||
}
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
N('db_query',1);
|
||||
$query = $this->parseWhere($options['where']);
|
||||
$fields = $this->parseField($options['field']);
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.'.$this->_collectionName.'.findOne(';
|
||||
$this->queryStr .= $query?json_encode($query):'{}';
|
||||
$this->queryStr .= $fields?','.json_encode($fields):'';
|
||||
$this->queryStr .= ')';
|
||||
}
|
||||
try{
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_collection->findOne($query,$fields);
|
||||
$this->debug();
|
||||
if($cache && $result ) { // 查询缓存写入
|
||||
S($key,$result,$cache['expire'],$cache['type']);
|
||||
}
|
||||
return $result;
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 统计记录数
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return iterator
|
||||
*/
|
||||
public function count($options=array()){
|
||||
if(isset($options['table'])) {
|
||||
$this->switchCollection($options['table'],'',false);
|
||||
}
|
||||
$this->model = $options['model'];
|
||||
N('db_query',1);
|
||||
$query = $this->parseWhere($options['where']);
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.'.$this->_collectionName;
|
||||
$this->queryStr .= $query?'.find('.json_encode($query).')':'';
|
||||
$this->queryStr .= '.count()';
|
||||
}
|
||||
try{
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$count = $this->_collection->count($query);
|
||||
$this->debug();
|
||||
return $count;
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
public function group($keys,$initial,$reduce,$options=array()){
|
||||
$this->_collection->group($keys,$initial,$reduce,$options);
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getFields($collection=''){
|
||||
if(!empty($collection) && $collection != $this->_collectionName) {
|
||||
$this->switchCollection($collection,'',false);
|
||||
}
|
||||
N('db_query',1);
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.'.$this->_collectionName.'.findOne()';
|
||||
}
|
||||
try{
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_collection->findOne();
|
||||
$this->debug();
|
||||
} catch (MongoCursorException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
if($result) { // 存在数据则分析字段
|
||||
$info = array();
|
||||
foreach ($result as $key=>$val){
|
||||
$info[$key] = array(
|
||||
'name'=>$key,
|
||||
'type'=>getType($val),
|
||||
);
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
// 暂时没有数据 返回false
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得当前数据库的collection信息
|
||||
* @access public
|
||||
*/
|
||||
public function getTables(){
|
||||
if(C('DB_SQL_LOG')) {
|
||||
$this->queryStr = $this->_dbName.'.getCollenctionNames()';
|
||||
}
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$list = $this->_mongo->listCollections();
|
||||
$this->debug();
|
||||
$info = array();
|
||||
foreach ($list as $collection){
|
||||
$info[] = $collection->getName();
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* set分析
|
||||
* @access protected
|
||||
* @param array $data
|
||||
* @return string
|
||||
*/
|
||||
protected function parseSet($data) {
|
||||
$result = array();
|
||||
foreach ($data as $key=>$val){
|
||||
if(is_array($val)) {
|
||||
switch($val[0]) {
|
||||
case 'inc':
|
||||
$result['$inc'][$key] = (int)$val[1];
|
||||
break;
|
||||
case 'set':
|
||||
case 'unset':
|
||||
case 'push':
|
||||
case 'pushall':
|
||||
case 'addtoset':
|
||||
case 'pop':
|
||||
case 'pull':
|
||||
case 'pullall':
|
||||
$result['$'.$val[0]][$key] = $val[1];
|
||||
break;
|
||||
default:
|
||||
$result['$set'][$key] = $val;
|
||||
}
|
||||
}else{
|
||||
$result['$set'][$key] = $val;
|
||||
}
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* order分析
|
||||
* @access protected
|
||||
* @param mixed $order
|
||||
* @return array
|
||||
*/
|
||||
protected function parseOrder($order) {
|
||||
if(is_string($order)) {
|
||||
$array = explode(',',$order);
|
||||
$order = array();
|
||||
foreach ($array as $key=>$val){
|
||||
$arr = explode(' ',trim($val));
|
||||
if(isset($arr[1])) {
|
||||
$arr[1] = $arr[1]=='asc'?1:-1;
|
||||
}else{
|
||||
$arr[1] = 1;
|
||||
}
|
||||
$order[$arr[0]] = $arr[1];
|
||||
}
|
||||
}
|
||||
return $order;
|
||||
}
|
||||
|
||||
/**
|
||||
* limit分析
|
||||
* @access protected
|
||||
* @param mixed $limit
|
||||
* @return array
|
||||
*/
|
||||
protected function parseLimit($limit) {
|
||||
if(strpos($limit,',')) {
|
||||
$array = explode(',',$limit);
|
||||
}else{
|
||||
$array = array(0,$limit);
|
||||
}
|
||||
return $array;
|
||||
}
|
||||
|
||||
/**
|
||||
* field分析
|
||||
* @access protected
|
||||
* @param mixed $fields
|
||||
* @return array
|
||||
*/
|
||||
public function parseField($fields){
|
||||
if(empty($fields)) {
|
||||
$fields = array();
|
||||
}
|
||||
if(is_string($fields)) {
|
||||
$fields = explode(',',$fields);
|
||||
}
|
||||
return $fields;
|
||||
}
|
||||
|
||||
/**
|
||||
* where分析
|
||||
* @access protected
|
||||
* @param mixed $where
|
||||
* @return array
|
||||
*/
|
||||
public function parseWhere($where){
|
||||
$query = array();
|
||||
foreach ($where as $key=>$val){
|
||||
if('_id' != $key && 0===strpos($key,'_')) {
|
||||
// 解析特殊条件表达式
|
||||
$query = $this->parseThinkWhere($key,$val);
|
||||
}else{
|
||||
// 查询字段的安全过滤
|
||||
if(!preg_match('/^[A-Z_\|\&\-.a-z0-9]+$/',trim($key))){
|
||||
throw_exception(L('_ERROR_QUERY_').':'.$key);
|
||||
}
|
||||
$key = trim($key);
|
||||
if(strpos($key,'|')) {
|
||||
$array = explode('|',$key);
|
||||
$str = array();
|
||||
foreach ($array as $k){
|
||||
$str[] = $this->parseWhereItem($k,$val);
|
||||
}
|
||||
$query['$or'] = $str;
|
||||
}elseif(strpos($key,'&')){
|
||||
$array = explode('&',$key);
|
||||
$str = array();
|
||||
foreach ($array as $k){
|
||||
$str[] = $this->parseWhereItem($k,$val);
|
||||
}
|
||||
$query = array_merge($query,$str);
|
||||
}else{
|
||||
$str = $this->parseWhereItem($key,$val);
|
||||
$query = array_merge($query,$str);
|
||||
}
|
||||
}
|
||||
}
|
||||
return $query;
|
||||
}
|
||||
|
||||
/**
|
||||
* 特殊条件分析
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @param mixed $val
|
||||
* @return string
|
||||
*/
|
||||
protected function parseThinkWhere($key,$val) {
|
||||
$query = array();
|
||||
switch($key) {
|
||||
case '_query': // 字符串模式查询条件
|
||||
parse_str($val,$query);
|
||||
if(isset($query['_logic']) && strtolower($query['_logic']) == 'or' ) {
|
||||
unset($query['_logic']);
|
||||
$query['$or'] = $query;
|
||||
}
|
||||
break;
|
||||
case '_string':// MongoCode查询
|
||||
$query['$where'] = new MongoCode($val);
|
||||
break;
|
||||
}
|
||||
return $query;
|
||||
}
|
||||
|
||||
/**
|
||||
* where子单元分析
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @param mixed $val
|
||||
* @return array
|
||||
*/
|
||||
protected function parseWhereItem($key,$val) {
|
||||
$query = array();
|
||||
if(is_array($val)) {
|
||||
if(is_string($val[0])) {
|
||||
$con = strtolower($val[0]);
|
||||
if(in_array($con,array('neq','ne','gt','egt','gte','lt','lte','elt'))) { // 比较运算
|
||||
$k = '$'.$this->comparison[$con];
|
||||
$query[$key] = array($k=>$val[1]);
|
||||
}elseif('like'== $con){ // 模糊查询 采用正则方式
|
||||
$query[$key] = new MongoRegex("/".$val[1]."/");
|
||||
}elseif('mod'==$con){ // mod 查询
|
||||
$query[$key] = array('$mod'=>$val[1]);
|
||||
}elseif('regex'==$con){ // 正则查询
|
||||
$query[$key] = new MongoRegex($val[1]);
|
||||
}elseif(in_array($con,array('in','nin','not in'))){ // IN NIN 运算
|
||||
$data = is_string($val[1])? explode(',',$val[1]):$val[1];
|
||||
$k = '$'.$this->comparison[$con];
|
||||
$query[$key] = array($k=>$data);
|
||||
}elseif('all'==$con){ // 满足所有指定条件
|
||||
$data = is_string($val[1])? explode(',',$val[1]):$val[1];
|
||||
$query[$key] = array('$all'=>$data);
|
||||
}elseif('between'==$con){ // BETWEEN运算
|
||||
$data = is_string($val[1])? explode(',',$val[1]):$val[1];
|
||||
$query[$key] = array('$gte'=>$data[0],'$lte'=>$data[1]);
|
||||
}elseif('not between'==$con){
|
||||
$data = is_string($val[1])? explode(',',$val[1]):$val[1];
|
||||
$query[$key] = array('$lt'=>$data[0],'$gt'=>$data[1]);
|
||||
}elseif('exp'==$con){ // 表达式查询
|
||||
$query['$where'] = new MongoCode($val[1]);
|
||||
}elseif('exists'==$con){ // 字段是否存在
|
||||
$query[$key] =array('$exists'=>(bool)$val[1]);
|
||||
}elseif('size'==$con){ // 限制属性大小
|
||||
$query[$key] =array('$size'=>intval($val[1]));
|
||||
}elseif('type'==$con){ // 限制字段类型 1 浮点型 2 字符型 3 对象或者MongoDBRef 5 MongoBinData 7 MongoId 8 布尔型 9 MongoDate 10 NULL 15 MongoCode 16 32位整型 17 MongoTimestamp 18 MongoInt64 如果是数组的话判断元素的类型
|
||||
$query[$key] =array('$type'=>intval($val[1]));
|
||||
}else{
|
||||
$query[$key] = $val;
|
||||
}
|
||||
return $query;
|
||||
}
|
||||
}
|
||||
$query[$key] = $val;
|
||||
return $query;
|
||||
}
|
||||
}
|
||||
@@ -1,334 +0,0 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006-2012 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
/**
|
||||
* MSsql数据库驱动 要求sqlserver2005
|
||||
* @category Extend
|
||||
* @package Extend
|
||||
* @subpackage Driver.Db
|
||||
* @author liu21st <liu21st@gmail.com>
|
||||
*/
|
||||
class Mssql extends Db{
|
||||
protected $selectSql = 'SELECT T1.* FROM (SELECT thinkphp.*, ROW_NUMBER() OVER (%ORDER%) AS ROW_NUMBER FROM (SELECT %DISTINCT% %FIELD% FROM %TABLE%%JOIN%%WHERE%%GROUP%%HAVING%) AS thinkphp) AS T1 %LIMIT%%COMMENT%';
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config=''){
|
||||
if ( !function_exists('mssql_connect') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').':mssql');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = array();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
$pconnect = !empty($config['params']['persist'])? $config['params']['persist']:$this->pconnect;
|
||||
$conn = $pconnect ? 'mssql_pconnect':'mssql_connect';
|
||||
// 处理不带端口号的socket连接情况
|
||||
$sepr = IS_WIN ? ',' : ':';
|
||||
$host = $config['hostname'].($config['hostport']?$sepr."{$config['hostport']}":'');
|
||||
$this->linkID[$linkNum] = $conn( $host, $config['username'], $config['password']);
|
||||
if ( !$this->linkID[$linkNum] ) throw_exception("Couldn't connect to SQL Server on $host");
|
||||
if ( !empty($config['database']) && !mssql_select_db($config['database'], $this->linkID[$linkNum]) ) {
|
||||
throw_exception("Couldn't open database '".$config['database']);
|
||||
}
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
//注销数据库安全信息
|
||||
if(1 != C('DB_DEPLOY_TYPE')) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
mssql_free_result($this->queryID);
|
||||
$this->queryID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->queryID = mssql_query($str, $this->_linkID);
|
||||
$this->debug();
|
||||
if ( false === $this->queryID ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = mssql_num_rows($this->queryID);
|
||||
return $this->getAll();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_write',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = mssql_query($str, $this->_linkID);
|
||||
$this->debug();
|
||||
if ( false === $result ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = mssql_rows_affected($this->_linkID);
|
||||
$this->lastInsID = $this->mssql_insert_id();
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于获取最后插入的ID
|
||||
* @access public
|
||||
* @return integer
|
||||
*/
|
||||
public function mssql_insert_id() {
|
||||
$query = "SELECT @@IDENTITY as last_insert_id";
|
||||
$result = mssql_query($query, $this->_linkID);
|
||||
list($last_insert_id) = mssql_fetch_row($result);
|
||||
mssql_free_result($result);
|
||||
return $last_insert_id;
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
mssql_query('BEGIN TRAN', $this->_linkID);
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = mssql_query('COMMIT TRAN', $this->_linkID);
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = mssql_query('ROLLBACK TRAN', $this->_linkID);
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getAll() {
|
||||
//返回数据集
|
||||
$result = array();
|
||||
if($this->numRows >0) {
|
||||
while($row = mssql_fetch_assoc($this->queryID))
|
||||
$result[] = $row;
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getFields($tableName) {
|
||||
$result = $this->query("SELECT column_name, data_type, column_default, is_nullable
|
||||
FROM information_schema.tables AS t
|
||||
JOIN information_schema.columns AS c
|
||||
ON t.table_catalog = c.table_catalog
|
||||
AND t.table_schema = c.table_schema
|
||||
AND t.table_name = c.table_name
|
||||
WHERE t.table_name = '$tableName'");
|
||||
$info = array();
|
||||
if($result) {
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$val['column_name']] = array(
|
||||
'name' => $val['column_name'],
|
||||
'type' => $val['data_type'],
|
||||
'notnull' => (bool) ($val['is_nullable'] === ''), // not null is empty, null is yes
|
||||
'default' => $val['column_default'],
|
||||
'primary' => false,
|
||||
'autoinc' => false,
|
||||
);
|
||||
}
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getTables($dbName='') {
|
||||
$result = $this->query("SELECT TABLE_NAME
|
||||
FROM INFORMATION_SCHEMA.TABLES
|
||||
WHERE TABLE_TYPE = 'BASE TABLE'
|
||||
");
|
||||
$info = array();
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$key] = current($val);
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* order分析
|
||||
* @access protected
|
||||
* @param mixed $order
|
||||
* @return string
|
||||
*/
|
||||
protected function parseOrder($order) {
|
||||
return !empty($order)? ' ORDER BY '.$order:' ORDER BY rand()';
|
||||
}
|
||||
|
||||
/**
|
||||
* limit
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function parseLimit($limit) {
|
||||
if(empty($limit)) return '';
|
||||
$limit = explode(',',$limit);
|
||||
if(count($limit)>1)
|
||||
$limitStr = '(T1.ROW_NUMBER BETWEEN '.$limit[0].' + 1 AND '.$limit[0].' + '.$limit[1].')';
|
||||
else
|
||||
$limitStr = '(T1.ROW_NUMBER BETWEEN 1 AND '.$limit[0].")";
|
||||
return 'WHERE '.$limitStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新记录
|
||||
* @access public
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function update($data,$options) {
|
||||
$this->model = $options['model'];
|
||||
$sql = 'UPDATE '
|
||||
.$this->parseTable($options['table'])
|
||||
.$this->parseSet($data)
|
||||
.$this->parseWhere(!empty($options['where'])?$options['where']:'')
|
||||
.$this->parseLock(isset($options['lock'])?$options['lock']:false)
|
||||
.$this->parseComment(!empty($options['comment'])?$options['comment']:'');
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除记录
|
||||
* @access public
|
||||
* @param array $options 表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function delete($options=array()) {
|
||||
$this->model = $options['model'];
|
||||
$sql = 'DELETE FROM '
|
||||
.$this->parseTable($options['table'])
|
||||
.$this->parseWhere(!empty($options['where'])?$options['where']:'')
|
||||
.$this->parseLock(isset($options['lock'])?$options['lock']:false)
|
||||
.$this->parseComment(!empty($options['comment'])?$options['comment']:'');
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
if ($this->_linkID){
|
||||
mssql_close($this->_linkID);
|
||||
}
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error() {
|
||||
$this->error = mssql_get_last_message();
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
ThinkLog::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
}
|
||||
@@ -8,210 +8,24 @@
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
|
||||
use Think\Db\Driver;
|
||||
/**
|
||||
* Mysql数据库驱动类
|
||||
* @category Think
|
||||
* @package Think
|
||||
* PDO数据库驱动
|
||||
* @category Extend
|
||||
* @package Extend
|
||||
* @subpackage Driver.Db
|
||||
* @author liu21st <liu21st@gmail.com>
|
||||
*/
|
||||
class Mysql extends Db{
|
||||
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config=''){
|
||||
if ( !extension_loaded('mysql') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').':mysql');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = '';
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
* @throws ThinkExecption
|
||||
*/
|
||||
public function connect($config='',$linkNum=0,$force=false) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
// 处理不带端口号的socket连接情况
|
||||
$host = $config['hostname'].($config['hostport']?":{$config['hostport']}":'');
|
||||
// 是否长连接
|
||||
$pconnect = !empty($config['params']['persist'])? $config['params']['persist']:$this->pconnect;
|
||||
if($pconnect) {
|
||||
$this->linkID[$linkNum] = mysql_pconnect( $host, $config['username'], $config['password'],131072);
|
||||
}else{
|
||||
$this->linkID[$linkNum] = mysql_connect( $host, $config['username'], $config['password'],true,131072);
|
||||
}
|
||||
if ( !$this->linkID[$linkNum] || (!empty($config['database']) && !mysql_select_db($config['database'], $this->linkID[$linkNum])) ) {
|
||||
throw_exception(mysql_error());
|
||||
}
|
||||
$dbVersion = mysql_get_server_info($this->linkID[$linkNum]);
|
||||
//使用UTF8存取数据库
|
||||
mysql_query("SET NAMES '".$this->conf['db_charset']."'", $this->linkID[$linkNum]);
|
||||
//设置 sql_model
|
||||
if($dbVersion >'5.0.1'){
|
||||
mysql_query("SET sql_mode=''",$this->linkID[$linkNum]);
|
||||
}
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
// 注销数据库连接配置信息
|
||||
if(1 != $this->conf['db_deploy_type']) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
mysql_free_result($this->queryID);
|
||||
$this->queryID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str) {
|
||||
if(0===stripos($str, 'call')){ // 存储过程查询支持
|
||||
$this->close();
|
||||
}
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) { $this->free(); }
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
\Think\Debug::remark('queryStartTime','time');
|
||||
$this->queryID = mysql_query($str, $this->_linkID);
|
||||
$this->debug();
|
||||
if ( false === $this->queryID ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = mysql_num_rows($this->queryID);
|
||||
return $this->getAll();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer|false
|
||||
*/
|
||||
public function execute($str) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) { $this->free(); }
|
||||
N('db_write',1);
|
||||
// 记录开始执行时间
|
||||
\Think\Debug::remark('queryStartTime','time');
|
||||
$result = mysql_query($str, $this->_linkID) ;
|
||||
$this->debug();
|
||||
if ( false === $result) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = mysql_affected_rows($this->_linkID);
|
||||
$this->lastInsID = mysql_insert_id($this->_linkID);
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
mysql_query('START TRANSACTION', $this->_linkID);
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = mysql_query('COMMIT', $this->_linkID);
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = mysql_query('ROLLBACK', $this->_linkID);
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getAll() {
|
||||
//返回数据集
|
||||
$result = array();
|
||||
if($this->numRows >0) {
|
||||
while($row = mysql_fetch_assoc($this->queryID)){
|
||||
$result[] = $row;
|
||||
}
|
||||
mysql_data_seek($this->queryID,0);
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
class Mysql extends Driver{
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getFields($tableName) {
|
||||
$result = $this->query('SHOW COLUMNS FROM '.$this->parseKey($tableName));
|
||||
$this->initConnect(true);
|
||||
$sql = 'SHOW COLUMNS FROM `'.$tableName.'`';
|
||||
$result = $this->query($sql);
|
||||
$info = array();
|
||||
if($result) {
|
||||
foreach ($result as $key => $val) {
|
||||
@@ -231,15 +45,10 @@ class Mysql extends Db{
|
||||
/**
|
||||
* 取得数据库的表信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getTables($dbName='') {
|
||||
if(!empty($dbName)) {
|
||||
$sql = 'SHOW TABLES FROM '.$dbName;
|
||||
}else{
|
||||
$sql = 'SHOW TABLES ';
|
||||
}
|
||||
$result = $this->query($sql);
|
||||
$sql = !empty($dbName)?'SHOW TABLES FROM '.$dbName:'SHOW TABLES ';
|
||||
$result = $this->query($sql);
|
||||
$info = array();
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$key] = current($val);
|
||||
@@ -248,94 +57,7 @@ class Mysql extends Db{
|
||||
}
|
||||
|
||||
/**
|
||||
* 替换记录
|
||||
* @access public
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 参数表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function replace($data,$options=array()) {
|
||||
foreach ($data as $key=>$val){
|
||||
$value = $this->parseValue($val);
|
||||
if(is_scalar($value)) { // 过滤非标量数据
|
||||
$values[] = $value;
|
||||
$fields[] = $this->parseKey($key);
|
||||
}
|
||||
}
|
||||
$sql = 'REPLACE INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES ('.implode(',', $values).')';
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 插入记录
|
||||
* @access public
|
||||
* @param mixed $datas 数据
|
||||
* @param array $options 参数表达式
|
||||
* @param boolean $replace 是否replace
|
||||
* @return false | integer
|
||||
*/
|
||||
public function insertAll($datas,$options=array(),$replace=false) {
|
||||
if(!is_array($datas[0])) return false;
|
||||
$fields = array_keys($datas[0]);
|
||||
array_walk($fields, array($this, 'parseKey'));
|
||||
$values = array();
|
||||
foreach ($datas as $data){
|
||||
$value = array();
|
||||
foreach ($data as $key=>$val){
|
||||
$val = $this->parseValue($val);
|
||||
if(is_scalar($val)) { // 过滤非标量数据
|
||||
$value[] = $val;
|
||||
}
|
||||
}
|
||||
$values[] = '('.implode(',', $value).')';
|
||||
}
|
||||
$sql = ($replace?'REPLACE':'INSERT').' INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES '.implode(',',$values);
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function close() {
|
||||
if ($this->_linkID){
|
||||
mysql_close($this->_linkID);
|
||||
}
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error() {
|
||||
$this->error = mysql_error($this->_linkID);
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
\Think\Log::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* SQL指令安全过滤
|
||||
* @access public
|
||||
* @param string $str SQL字符串
|
||||
* @return string
|
||||
*/
|
||||
public function escapeString($str) {
|
||||
if($this->_linkID) {
|
||||
return mysql_real_escape_string($str,$this->_linkID);
|
||||
}else{
|
||||
return mysql_escape_string($str);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 字段和表名处理添加`
|
||||
* 字段和表名处理
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @return string
|
||||
@@ -345,6 +67,7 @@ class Mysql extends Db{
|
||||
if(!preg_match('/[,\'\"\*\(\)`.\s]/',$key)) {
|
||||
$key = '`'.$key.'`';
|
||||
}
|
||||
return $key;
|
||||
return $key;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -1,345 +0,0 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2007 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
/**
|
||||
* Mysqli数据库驱动类
|
||||
* @category Think
|
||||
* @package Think
|
||||
* @subpackage Driver.Db
|
||||
* @author liu21st <liu21st@gmail.com>
|
||||
*/
|
||||
class Mysqli extends Db{
|
||||
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config=''){
|
||||
if ( !extension_loaded('mysqli') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').':mysqli');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = '';
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
* @throws ThinkExecption
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
$this->linkID[$linkNum] = new mysqli($config['hostname'],$config['username'],$config['password'],$config['database'],$config['hostport']?intval($config['hostport']):3306);
|
||||
if (mysqli_connect_errno()) throw_exception(mysqli_connect_error());
|
||||
$dbVersion = $this->linkID[$linkNum]->server_version;
|
||||
|
||||
// 设置数据库编码
|
||||
$this->linkID[$linkNum]->query("SET NAMES '".C('DB_CHARSET')."'");
|
||||
//设置 sql_model
|
||||
if($dbVersion >'5.0.1'){
|
||||
$this->linkID[$linkNum]->query("SET sql_mode=''");
|
||||
}
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
//注销数据库安全信息
|
||||
if(1 != C('DB_DEPLOY_TYPE')) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
$this->queryID->free_result();
|
||||
$this->queryID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->queryID = $this->_linkID->query($str);
|
||||
// 对存储过程改进
|
||||
if( $this->_linkID->more_results() ){
|
||||
while (($res = $this->_linkID->next_result()) != NULL) {
|
||||
$res->free_result();
|
||||
}
|
||||
}
|
||||
$this->debug();
|
||||
if ( false === $this->queryID ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = $this->queryID->num_rows;
|
||||
$this->numCols = $this->queryID->field_count;
|
||||
return $this->getAll();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_write',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = $this->_linkID->query($str);
|
||||
$this->debug();
|
||||
if ( false === $result ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = $this->_linkID->affected_rows;
|
||||
$this->lastInsID = $this->_linkID->insert_id;
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
$this->_linkID->autocommit(false);
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = $this->_linkID->commit();
|
||||
$this->_linkID->autocommit( true);
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = $this->_linkID->rollback();
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @param string $sql sql语句
|
||||
* @return array
|
||||
*/
|
||||
private function getAll() {
|
||||
//返回数据集
|
||||
$result = array();
|
||||
if($this->numRows>0) {
|
||||
//返回数据集
|
||||
for($i=0;$i<$this->numRows ;$i++ ){
|
||||
$result[$i] = $this->queryID->fetch_assoc();
|
||||
}
|
||||
$this->queryID->data_seek(0);
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getFields($tableName) {
|
||||
$result = $this->query('SHOW COLUMNS FROM '.$this->parseKey($tableName));
|
||||
$info = array();
|
||||
if($result) {
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$val['Field']] = array(
|
||||
'name' => $val['Field'],
|
||||
'type' => $val['Type'],
|
||||
'notnull' => (bool) ($val['Null'] === ''), // not null is empty, null is yes
|
||||
'default' => $val['Default'],
|
||||
'primary' => (strtolower($val['Key']) == 'pri'),
|
||||
'autoinc' => (strtolower($val['Extra']) == 'auto_increment'),
|
||||
);
|
||||
}
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getTables($dbName='') {
|
||||
$sql = !empty($dbName)?'SHOW TABLES FROM '.$dbName:'SHOW TABLES ';
|
||||
$result = $this->query($sql);
|
||||
$info = array();
|
||||
if($result) {
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$key] = current($val);
|
||||
}
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* 替换记录
|
||||
* @access public
|
||||
* @param mixed $data 数据
|
||||
* @param array $options 参数表达式
|
||||
* @return false | integer
|
||||
*/
|
||||
public function replace($data,$options=array()) {
|
||||
foreach ($data as $key=>$val){
|
||||
$value = $this->parseValue($val);
|
||||
if(is_scalar($value)) { // 过滤非标量数据
|
||||
$values[] = $value;
|
||||
$fields[] = $this->parseKey($key);
|
||||
}
|
||||
}
|
||||
$sql = 'REPLACE INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES ('.implode(',', $values).')';
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 插入记录
|
||||
* @access public
|
||||
* @param mixed $datas 数据
|
||||
* @param array $options 参数表达式
|
||||
* @param boolean $replace 是否replace
|
||||
* @return false | integer
|
||||
*/
|
||||
public function insertAll($datas,$options=array(),$replace=false) {
|
||||
if(!is_array($datas[0])) return false;
|
||||
$fields = array_keys($datas[0]);
|
||||
array_walk($fields, array($this, 'parseKey'));
|
||||
$values = array();
|
||||
foreach ($datas as $data){
|
||||
$value = array();
|
||||
foreach ($data as $key=>$val){
|
||||
$val = $this->parseValue($val);
|
||||
if(is_scalar($val)) { // 过滤非标量数据
|
||||
$value[] = $val;
|
||||
}
|
||||
}
|
||||
$values[] = '('.implode(',', $value).')';
|
||||
}
|
||||
$sql = ($replace?'REPLACE':'INSERT').' INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES '.implode(',',$values);
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
* @return volid
|
||||
*/
|
||||
public function close() {
|
||||
if ($this->_linkID){
|
||||
$this->_linkID->close();
|
||||
}
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @static
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error() {
|
||||
$this->error = $this->_linkID->error;
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
ThinkLog::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* SQL指令安全过滤
|
||||
* @static
|
||||
* @access public
|
||||
* @param string $str SQL指令
|
||||
* @return string
|
||||
*/
|
||||
public function escapeString($str) {
|
||||
if($this->_linkID) {
|
||||
return $this->_linkID->real_escape_string($str);
|
||||
}else{
|
||||
return addslashes($str);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 字段和表名处理添加`
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @return string
|
||||
*/
|
||||
protected function parseKey(&$key) {
|
||||
$key = trim($key);
|
||||
if(!preg_match('/[,\'\"\*\(\)`.\s]/',$key)) {
|
||||
$key = '`'.$key.'`';
|
||||
}
|
||||
return $key;
|
||||
}
|
||||
}
|
||||
@@ -9,7 +9,7 @@
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
use Think\Db\Driver;
|
||||
/**
|
||||
* Oracle数据库驱动
|
||||
* @category Extend
|
||||
@@ -17,193 +17,49 @@ use Think\Db;
|
||||
* @subpackage Driver.Db
|
||||
* @author ZhangXuehun <zhangxuehun@sohu.com>
|
||||
*/
|
||||
class Oracle extends Db{
|
||||
class Oracle extends Driver{
|
||||
|
||||
private $mode = OCI_COMMIT_ON_SUCCESS;
|
||||
private $table = '';
|
||||
protected $selectSql = 'SELECT * FROM (SELECT thinkphp.*, rownum AS numrow FROM (SELECT %DISTINCT% %FIELD% FROM %TABLE%%JOIN%%WHERE%%GROUP%%HAVING%%ORDER%) thinkphp ) %LIMIT%%COMMENT%';
|
||||
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config=''){
|
||||
putenv("NLS_LANG=AMERICAN_AMERICA.UTF8");
|
||||
if ( !extension_loaded('oci8') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').'oracle');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = array();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
$pconnect = !empty($config['params']['persist'])? $config['params']['persist']:$this->pconnect;
|
||||
$conn = $pconnect ? 'oci_pconnect':'oci_new_connect';
|
||||
$this->linkID[$linkNum] = $conn($config['username'], $config['password'],$config['database']);//modify by wyfeng at 2008.12.19
|
||||
|
||||
if (!$this->linkID[$linkNum]){
|
||||
$this->error(false);
|
||||
}
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
//注销数据库安全信息
|
||||
if(1 != C('DB_DEPLOY_TYPE')) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
oci_free_statement($this->queryID);
|
||||
$this->queryID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//更改事务模式
|
||||
$this->mode = OCI_COMMIT_ON_SUCCESS;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->queryID = oci_parse($this->_linkID,$str);
|
||||
$this->debug();
|
||||
if (false === oci_execute($this->queryID, $this->mode)) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
return $this->getAll();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str) {
|
||||
public function execute($str,$bind=[]) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
// 判断新增操作
|
||||
$flag = false;
|
||||
if(preg_match("/^\s*(INSERT\s+INTO)\s+(\w+)\s+/i", $this->queryStr, $match)) {
|
||||
$this->table = C("DB_SEQUENCE_PREFIX") .str_ireplace(C("DB_PREFIX"), "", $match[2]);
|
||||
$this->table = C("DB_SEQUENCE_PREFIX").str_ireplace(C("DB_PREFIX"), "", $match[2]);
|
||||
$flag = (boolean)$this->query("SELECT * FROM user_sequences WHERE sequence_name='" . strtoupper($this->table) . "'");
|
||||
}//modify by wyfeng at 2009.08.28
|
||||
|
||||
//更改事务模式
|
||||
$this->mode = OCI_COMMIT_ON_SUCCESS;
|
||||
}
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_write',1);
|
||||
if ( !empty($this->PDOStatement) ) $this->free();
|
||||
$this->executeTimes++;
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$stmt = oci_parse($this->_linkID,$str);
|
||||
Debug::remark('queryStartTime','time');
|
||||
$this->PDOStatement = $this->_linkID->prepare($str);
|
||||
if(false === $this->PDOStatement) {
|
||||
throw_exception($this->error());
|
||||
}
|
||||
$result = $this->PDOStatement->execute($bind);
|
||||
$this->debug();
|
||||
if (false === oci_execute($stmt)) {
|
||||
if ( false === $result) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = oci_num_rows($stmt);
|
||||
$this->lastInsID = $flag?$this->insertLastId():0;//modify by wyfeng at 2009.08.28
|
||||
$this->numRows = $this->PDOStatement->rowCount();
|
||||
if($flag || preg_match("/^\s*(INSERT\s+INTO|REPLACE\s+INTO)\s+/i", $str)) {
|
||||
$this->lastInsID = $this->getLastInsertId();
|
||||
}
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
$this->mode = OCI_DEFAULT;
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit(){
|
||||
if ($this->transTimes > 0) {
|
||||
$result = oci_commit($this->_linkID);
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
$this->transTimes = 0;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback(){
|
||||
if ($this->transTimes > 0) {
|
||||
$result = oci_rollback($this->_linkID);
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
$this->transTimes = 0;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getAll() {
|
||||
//返回数据集
|
||||
$result = array();
|
||||
$this->numRows = oci_fetch_all($this->queryID, $result, 0, -1, OCI_FETCHSTATEMENT_BY_ROW);
|
||||
//add by wyfeng at 2008-12-23 强制将字段名转换为小写，以配合Model类函数如count等
|
||||
if(C("DB_CASE_LOWER")) {
|
||||
foreach($result as $k=>$v) {
|
||||
$result[$k] = array_change_key_case($result[$k], CASE_LOWER);
|
||||
}
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
@@ -242,39 +98,6 @@ class Oracle extends Db{
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
if($this->_linkID){
|
||||
oci_close($this->_linkID);
|
||||
}
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error($result = true) {
|
||||
if($result){
|
||||
$error = oci_error($this->queryID);
|
||||
}elseif(!$this->_linkID){
|
||||
$error = oci_error();
|
||||
}else{
|
||||
$error = oci_error($this->_linkID);
|
||||
}
|
||||
if('' != $this->queryStr){
|
||||
$error['message'] .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
$result? ThinkLog::record($error['message'],'ERR'):throw_exception($error['message'],'',$error['code']);
|
||||
$this->error = $error['message'];
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* SQL指令安全过滤
|
||||
* @access public
|
||||
@@ -309,7 +132,7 @@ class Oracle extends Db{
|
||||
* @access public
|
||||
* @return integer
|
||||
*/
|
||||
public function insertLastId() {
|
||||
public function getLastInsertId() {
|
||||
if(empty($this->table)) {
|
||||
return 0;
|
||||
}
|
||||
|
||||
@@ -1,447 +0,0 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2006-2012 http://thinkphp.cn All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
/**
|
||||
* PDO数据库驱动
|
||||
* @category Extend
|
||||
* @package Extend
|
||||
* @subpackage Driver.Db
|
||||
* @author liu21st <liu21st@gmail.com>
|
||||
*/
|
||||
class Pdo extends Db{
|
||||
|
||||
protected $PDOStatement = null;
|
||||
private $table = '';
|
||||
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config=''){
|
||||
if ( !class_exists('PDO') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').':PDO');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = array();
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
if($this->pconnect) {
|
||||
$config['params'][PDO::ATTR_PERSISTENT] = true;
|
||||
}
|
||||
//$config['params'][PDO::ATTR_CASE] = C("DB_CASE_LOWER")?PDO::CASE_LOWER:PDO::CASE_UPPER;
|
||||
try{
|
||||
$this->linkID[$linkNum] = new PDO( $config['dsn'], $config['username'], $config['password'],$config['params']);
|
||||
}catch (PDOException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
// 因为PDO的连接切换可能导致数据库类型不同，因此重新获取下当前的数据库类型
|
||||
$this->dbType = $this->_getDsnType($config['dsn']);
|
||||
if(in_array($this->dbType,array('MSSQL','ORACLE','IBASE','OCI'))) {
|
||||
// 由于PDO对于以上的数据库支持不够完美，所以屏蔽了 如果仍然希望使用PDO 可以注释下面一行代码
|
||||
throw_exception('由于目前PDO暂时不能完美支持'.$this->dbType.' 请使用官方的'.$this->dbType.'驱动');
|
||||
}
|
||||
$this->linkID[$linkNum]->exec('SET NAMES '.C('DB_CHARSET'));
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
// 注销数据库连接配置信息
|
||||
if(1 != C('DB_DEPLOY_TYPE')) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
$this->PDOStatement = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( !empty($this->PDOStatement) ) $this->free();
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->PDOStatement = $this->_linkID->prepare($str);
|
||||
if(false === $this->PDOStatement)
|
||||
throw_exception($this->error());
|
||||
$result = $this->PDOStatement->execute();
|
||||
$this->debug();
|
||||
if ( false === $result ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
return $this->getAll();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
$flag = false;
|
||||
if($this->dbType == 'OCI')
|
||||
{
|
||||
if(preg_match("/^\s*(INSERT\s+INTO)\s+(\w+)\s+/i", $this->queryStr, $match)) {
|
||||
$this->table = C("DB_SEQUENCE_PREFIX").str_ireplace(C("DB_PREFIX"), "", $match[2]);
|
||||
$flag = (boolean)$this->query("SELECT * FROM user_sequences WHERE sequence_name='" . strtoupper($this->table) . "'");
|
||||
}
|
||||
}//modify by wyfeng at 2009.08.28
|
||||
//释放前次的查询结果
|
||||
if ( !empty($this->PDOStatement) ) $this->free();
|
||||
N('db_write',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->PDOStatement = $this->_linkID->prepare($str);
|
||||
if(false === $this->PDOStatement) {
|
||||
throw_exception($this->error());
|
||||
}
|
||||
$result = $this->PDOStatement->execute();
|
||||
$this->debug();
|
||||
if ( false === $result) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = $this->PDOStatement->rowCount();
|
||||
if($flag || preg_match("/^\s*(INSERT\s+INTO|REPLACE\s+INTO)\s+/i", $str)) {
|
||||
$this->lastInsID = $this->getLastInsertId();
|
||||
}
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
$this->_linkID->beginTransaction();
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = $this->_linkID->commit();
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = $this->_linkID->rollback();
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getAll() {
|
||||
//返回数据集
|
||||
$result = $this->PDOStatement->fetchAll(PDO::FETCH_ASSOC);
|
||||
$this->numRows = count( $result );
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
*/
|
||||
public function getFields($tableName) {
|
||||
$this->initConnect(true);
|
||||
if(C('DB_DESCRIBE_TABLE_SQL')) {
|
||||
// 定义特殊的字段查询SQL
|
||||
$sql = str_replace('%table%',$tableName,C('DB_DESCRIBE_TABLE_SQL'));
|
||||
}else{
|
||||
switch($this->dbType) {
|
||||
case 'MSSQL':
|
||||
case 'SQLSRV':
|
||||
$sql = "SELECT column_name as 'Name', data_type as 'Type', column_default as 'Default', is_nullable as 'Null'
|
||||
FROM information_schema.tables AS t
|
||||
JOIN information_schema.columns AS c
|
||||
ON t.table_catalog = c.table_catalog
|
||||
AND t.table_schema = c.table_schema
|
||||
AND t.table_name = c.table_name
|
||||
WHERE t.table_name = '$tableName'";
|
||||
break;
|
||||
case 'SQLITE':
|
||||
$sql = 'PRAGMA table_info ('.$tableName.') ';
|
||||
break;
|
||||
case 'ORACLE':
|
||||
case 'OCI':
|
||||
$sql = "SELECT a.column_name \"Name\",data_type \"Type\",decode(nullable,'Y',0,1) notnull,data_default \"Default\",decode(a.column_name,b.column_name,1,0) \"pk\" "
|
||||
."FROM user_tab_columns a,(SELECT column_name FROM user_constraints c,user_cons_columns col "
|
||||
."WHERE c.constraint_name=col.constraint_name AND c.constraint_type='P' and c.table_name='".strtoupper($tableName)
|
||||
."') b where table_name='".strtoupper($tableName)."' and a.column_name=b.column_name(+)";
|
||||
break;
|
||||
case 'PGSQL':
|
||||
$sql = 'select fields_name as "Name",fields_type as "Type",fields_not_null as "Null",fields_key_name as "Key",fields_default as "Default",fields_default as "Extra" from table_msg('.$tableName.');';
|
||||
break;
|
||||
case 'IBASE':
|
||||
break;
|
||||
case 'MYSQL':
|
||||
default:
|
||||
$sql = 'DESCRIBE '.$tableName;//备注: 驱动类不只针对mysql，不能加``
|
||||
}
|
||||
}
|
||||
$result = $this->query($sql);
|
||||
$info = array();
|
||||
if($result) {
|
||||
foreach ($result as $key => $val) {
|
||||
$val = array_change_key_case($val);
|
||||
$val['name'] = isset($val['name'])?$val['name']:"";
|
||||
$val['type'] = isset($val['type'])?$val['type']:"";
|
||||
$name = isset($val['field'])?$val['field']:$val['name'];
|
||||
$info[$name] = array(
|
||||
'name' => $name ,
|
||||
'type' => $val['type'],
|
||||
'notnull' => (bool)(((isset($val['null'])) && ($val['null'] === '')) || ((isset($val['notnull'])) && ($val['notnull'] === ''))), // not null is empty, null is yes
|
||||
'default' => isset($val['default'])? $val['default'] :(isset($val['dflt_value'])?$val['dflt_value']:""),
|
||||
'primary' => isset($val['dey'])?strtolower($val['dey']) == 'pri':(isset($val['pk'])?$val['pk']:false),
|
||||
'autoinc' => isset($val['extra'])?strtolower($val['extra']) == 'auto_increment':(isset($val['key'])?$val['key']:false),
|
||||
);
|
||||
}
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据库的表信息
|
||||
* @access public
|
||||
*/
|
||||
public function getTables($dbName='') {
|
||||
if(C('DB_FETCH_TABLES_SQL')) {
|
||||
// 定义特殊的表查询SQL
|
||||
$sql = str_replace('%db%',$dnName,C('DB_FETCH_TABLES_SQL'));
|
||||
}else{
|
||||
switch($this->dbType) {
|
||||
case 'ORACLE':
|
||||
case 'OCI':
|
||||
$sql = 'SELECT table_name FROM user_tables';
|
||||
break;
|
||||
case 'MSSQL':
|
||||
case 'SQLSRV':
|
||||
$sql = "SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_TYPE = 'BASE TABLE'";
|
||||
break;
|
||||
case 'PGSQL':
|
||||
$sql = "select tablename as Tables_in_test from pg_tables where schemaname ='public'";
|
||||
break;
|
||||
case 'IBASE':
|
||||
// 暂时不支持
|
||||
throw_exception(L('_NOT_SUPPORT_DB_').':IBASE');
|
||||
break;
|
||||
case 'SQLITE':
|
||||
$sql = "SELECT name FROM sqlite_master WHERE type='table' "
|
||||
. "UNION ALL SELECT name FROM sqlite_temp_master "
|
||||
. "WHERE type='table' ORDER BY name";
|
||||
break;
|
||||
case 'MYSQL':
|
||||
default:
|
||||
if(!empty($dbName)) {
|
||||
$sql = 'SHOW TABLES FROM '.$dbName;
|
||||
}else{
|
||||
$sql = 'SHOW TABLES ';
|
||||
}
|
||||
}
|
||||
}
|
||||
$result = $this->query($sql);
|
||||
$info = array();
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$key] = current($val);
|
||||
}
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* limit分析
|
||||
* @access protected
|
||||
* @param mixed $lmit
|
||||
* @return string
|
||||
*/
|
||||
protected function parseLimit($limit) {
|
||||
$limitStr = '';
|
||||
if(!empty($limit)) {
|
||||
switch($this->dbType){
|
||||
case 'PGSQL':
|
||||
case 'SQLITE':
|
||||
$limit = explode(',',$limit);
|
||||
if(count($limit)>1) {
|
||||
$limitStr .= ' LIMIT '.$limit[1].' OFFSET '.$limit[0].' ';
|
||||
}else{
|
||||
$limitStr .= ' LIMIT '.$limit[0].' ';
|
||||
}
|
||||
break;
|
||||
case 'MSSQL':
|
||||
case 'SQLSRV':
|
||||
break;
|
||||
case 'IBASE':
|
||||
// 暂时不支持
|
||||
break;
|
||||
case 'ORACLE':
|
||||
case 'OCI':
|
||||
break;
|
||||
case 'MYSQL':
|
||||
default:
|
||||
$limitStr .= ' LIMIT '.$limit.' ';
|
||||
}
|
||||
}
|
||||
return $limitStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* 字段和表名处理
|
||||
* @access protected
|
||||
* @param string $key
|
||||
* @return string
|
||||
*/
|
||||
protected function parseKey(&$key) {
|
||||
if($this->dbType=='MYSQL'){
|
||||
$key = trim($key);
|
||||
if(!preg_match('/[,\'\"\*\(\)`.\s]/',$key)) {
|
||||
$key = '`'.$key.'`';
|
||||
}
|
||||
return $key;
|
||||
}else{
|
||||
return parent::parseKey($key);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error() {
|
||||
if($this->PDOStatement) {
|
||||
$error = $this->PDOStatement->errorInfo();
|
||||
$this->error = $error[2];
|
||||
}else{
|
||||
$this->error = '';
|
||||
}
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
ThinkLog::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* SQL指令安全过滤
|
||||
* @access public
|
||||
* @param string $str SQL指令
|
||||
* @return string
|
||||
*/
|
||||
public function escapeString($str) {
|
||||
switch($this->dbType) {
|
||||
case 'PGSQL':
|
||||
case 'MSSQL':
|
||||
case 'SQLSRV':
|
||||
case 'MYSQL':
|
||||
return addslashes($str);
|
||||
case 'IBASE':
|
||||
case 'SQLITE':
|
||||
case 'ORACLE':
|
||||
case 'OCI':
|
||||
return str_ireplace("'", "''", $str);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最后插入id
|
||||
* @access public
|
||||
* @return integer
|
||||
*/
|
||||
public function getLastInsertId() {
|
||||
switch($this->dbType) {
|
||||
case 'PGSQL':
|
||||
case 'SQLITE':
|
||||
case 'MSSQL':
|
||||
case 'SQLSRV':
|
||||
case 'IBASE':
|
||||
case 'MYSQL':
|
||||
return $this->_linkID->lastInsertId();
|
||||
case 'ORACLE':
|
||||
case 'OCI':
|
||||
$sequenceName = $this->table;
|
||||
$vo = $this->query("SELECT {$sequenceName}.currval currval FROM dual");
|
||||
return $vo?$vo[0]["currval"]:0;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -9,7 +9,7 @@
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
use Think\Db\Driver;
|
||||
/**
|
||||
* Pgsql数据库驱动
|
||||
* @category Extend
|
||||
@@ -17,214 +17,25 @@ use Think\Db;
|
||||
* @subpackage Driver.Db
|
||||
* @author liu21st <liu21st@gmail.com>
|
||||
*/
|
||||
class Pgsql extends Db{
|
||||
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config='') {
|
||||
if ( !extension_loaded('pgsql') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').':pgsql');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = array();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
$pconnect = !empty($config['params']['persist'])? $config['params']['persist']:$this->pconnect;
|
||||
$conn = $pconnect ? 'pg_pconnect':'pg_connect';
|
||||
$this->linkID[$linkNum] = $conn('host='.$config['hostname'].' port='.$config['hostport'].' dbname='.$config['database'].' user='.$config['username'].' password='.$config['password']);
|
||||
if (0 !== pg_connection_status($this->linkID[$linkNum])){
|
||||
throw_exception($this->error(false));
|
||||
}
|
||||
//设置编码
|
||||
pg_set_client_encoding($this->linkID[$linkNum], C('DB_CHARSET'));
|
||||
//$pgInfo = pg_version($this->linkID[$linkNum]);
|
||||
//$dbVersion = $pgInfo['server'];
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
//注销数据库安全信息
|
||||
if(1 != C('DB_DEPLOY_TYPE')) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
pg_free_result($this->queryID);
|
||||
$this->queryID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->queryID = pg_query($this->_linkID,$str);
|
||||
$this->debug();
|
||||
if ( false === $this->queryID ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = pg_num_rows($this->queryID);
|
||||
return $this->getAll();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_write',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = pg_query($this->_linkID,$str);
|
||||
$this->debug();
|
||||
if ( false === $result ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = pg_affected_rows($result);
|
||||
$this->lastInsID = $this->last_insert_id();
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于获取最后插入的ID
|
||||
* @access public
|
||||
* @return integer
|
||||
*/
|
||||
public function last_insert_id() {
|
||||
$query = "SELECT LASTVAL() AS insert_id";
|
||||
$result = pg_query($this->_linkID,$query);
|
||||
list($last_insert_id) = pg_fetch_array($result,null,PGSQL_ASSOC);
|
||||
pg_free_result($result);
|
||||
return $last_insert_id;
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
pg_exec($this->_linkID,'begin;');
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = pg_exec($this->_linkID,'end;');
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
$this->transTimes = 0;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = pg_exec($this->_linkID,'abort;');
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
$this->transTimes = 0;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getAll() {
|
||||
//返回数据集
|
||||
$result = pg_fetch_all($this->queryID);
|
||||
pg_result_seek($this->queryID,0);
|
||||
return $result;
|
||||
}
|
||||
class Pgsql extends Driver{
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getFields($tableName) {
|
||||
$result = $this->query("select a.attname as \"Field\",
|
||||
t.typname as \"Type\",
|
||||
a.attnotnull as \"Null\",
|
||||
i.indisprimary as \"Key\",
|
||||
d.adsrc as \"Default\"
|
||||
from pg_class c
|
||||
inner join pg_attribute a on a.attrelid = c.oid
|
||||
inner join pg_type t on a.atttypid = t.oid
|
||||
left join pg_attrdef d on a.attrelid=d.adrelid and d.adnum=a.attnum
|
||||
left join pg_index i on a.attnum=ANY(i.indkey) and c.oid = i.indrelid
|
||||
where (c.relname='{$tableName}' or c.relname = lower('{$tableName}')) AND a.attnum > 0
|
||||
order by a.attnum asc;");
|
||||
$result = $this->query('select fields_name as "Field",fields_type as "Type",fields_not_null as "Null",fields_key_name as "Key",fields_default as "Default",fields_default as "Extra" from table_msg('.$tableName.');');
|
||||
$info = array();
|
||||
if($result) {
|
||||
if($result){
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$val['Field']] = array(
|
||||
'name' => $val['Field'],
|
||||
'type' => $val['Type'],
|
||||
'notnull' => (bool) ($val['Null'] == 't'?1:0), // 't' is 'not null'
|
||||
'default' => $val['Default'],
|
||||
'primary' => (strtolower($val['Key']) == 't'),
|
||||
'autoinc' => (strtolower($val['Default']) == "nextval('{$tableName}_id_seq'::regclass)"),
|
||||
'name' => $val['Field'],
|
||||
'type' => $val['Type'],
|
||||
'notnull' => (bool) ($val['Null'] === ''), // not null is empty, null is yes
|
||||
'default' => $val['Default'],
|
||||
'primary' => (strtolower($val['Key']) == 'pri'),
|
||||
'autoinc' => (strtolower($val['Extra']) == 'auto_increment'),
|
||||
);
|
||||
}
|
||||
}
|
||||
@@ -234,9 +45,10 @@ class Pgsql extends Db{
|
||||
/**
|
||||
* 取得数据库的表信息
|
||||
* @access public
|
||||
* @return array
|
||||
*/
|
||||
public function getTables($dbName='') {
|
||||
$result = $this->query("select tablename as Tables_in_test from pg_tables where schemaname ='public'");
|
||||
$result = $this->query("select tablename as Tables_in_test from pg_tables where schemaname ='public'");
|
||||
$info = array();
|
||||
foreach ($result as $key => $val) {
|
||||
$info[$key] = current($val);
|
||||
@@ -245,44 +57,9 @@ class Pgsql extends Db{
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
if($this->_linkID){
|
||||
pg_close($this->_linkID);
|
||||
}
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error($result = true) {
|
||||
$this->error = $result?pg_result_error($this->queryID): pg_last_error($this->_linkID);
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
ThinkLog::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* SQL指令安全过滤
|
||||
* @access public
|
||||
* @param string $str SQL指令
|
||||
* @return string
|
||||
*/
|
||||
public function escapeString($str) {
|
||||
return pg_escape_string($str);
|
||||
}
|
||||
|
||||
/**
|
||||
* limit
|
||||
* @access public
|
||||
* limit分析
|
||||
* @access protected
|
||||
* @param mixed $lmit
|
||||
* @return string
|
||||
*/
|
||||
public function parseLimit($limit) {
|
||||
@@ -297,4 +74,5 @@ class Pgsql extends Db{
|
||||
}
|
||||
return $limitStr;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -9,7 +9,7 @@
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
use Think\Db\Driver;
|
||||
/**
|
||||
* Sqlite数据库驱动
|
||||
* @category Extend
|
||||
@@ -17,177 +17,7 @@ use Think\Db;
|
||||
* @subpackage Driver.Db
|
||||
* @author liu21st <liu21st@gmail.com>
|
||||
*/
|
||||
class Sqlite extends Db {
|
||||
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config='') {
|
||||
if ( !extension_loaded('sqlite') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').':sqlite');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
if(!isset($config['mode'])) {
|
||||
$config['mode'] = 0666;
|
||||
}
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = array();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
$pconnect = !empty($config['params']['persist'])? $config['params']['persist']:$this->pconnect;
|
||||
$conn = $pconnect ? 'sqlite_popen':'sqlite_open';
|
||||
$this->linkID[$linkNum] = $conn($config['database'],$config['mode']);
|
||||
if ( !$this->linkID[$linkNum]) {
|
||||
throw_exception(sqlite_error_string());
|
||||
}
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
//注销数据库安全信息
|
||||
if(1 != C('DB_DEPLOY_TYPE')) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
$this->queryID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->queryID = sqlite_query($this->_linkID,$str);
|
||||
$this->debug();
|
||||
if ( false === $this->queryID ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = sqlite_num_rows($this->queryID);
|
||||
return $this->getAll();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_write',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$result = sqlite_exec($this->_linkID,$str);
|
||||
$this->debug();
|
||||
if ( false === $result ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = sqlite_changes($this->_linkID);
|
||||
$this->lastInsID = sqlite_last_insert_rowid($this->_linkID);
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
sqlite_query($this->_linkID,'BEGIN TRANSACTION');
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = sqlite_query($this->_linkID,'COMMIT TRANSACTION');
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
$this->transTimes = 0;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = sqlite_query($this->_linkID,'ROLLBACK TRANSACTION');
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
$this->transTimes = 0;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getAll() {
|
||||
//返回数据集
|
||||
$result = array();
|
||||
if($this->numRows >0) {
|
||||
for($i=0;$i<$this->numRows ;$i++ ){
|
||||
// 返回数组集
|
||||
$result[$i] = sqlite_fetch_array($this->queryID,SQLITE_ASSOC);
|
||||
}
|
||||
sqlite_seek($this->queryID,0);
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
class Sqlite extends Driver {
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
@@ -228,32 +58,6 @@ class Sqlite extends Db {
|
||||
return $info;
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
if ($this->_linkID){
|
||||
sqlite_close($this->_linkID);
|
||||
}
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error() {
|
||||
$this->error = sqlite_error_string(sqlite_last_error($this->_linkID));
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
ThinkLog::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* SQL指令安全过滤
|
||||
* @access public
|
||||
@@ -261,7 +65,7 @@ class Sqlite extends Db {
|
||||
* @return string
|
||||
*/
|
||||
public function escapeString($str) {
|
||||
return sqlite_escape_string($str);
|
||||
return str_ireplace("'", "''", $str);
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -10,7 +10,7 @@
|
||||
// +----------------------------------------------------------------------
|
||||
|
||||
namespace Think\Db\Driver;
|
||||
use Think\Db;
|
||||
use Think\Db\Driver;
|
||||
/**
|
||||
* Sqlsrv数据库驱动
|
||||
* @category Extend
|
||||
@@ -18,180 +18,8 @@ use Think\Db;
|
||||
* @subpackage Driver.Db
|
||||
* @author liu21st <liu21st@gmail.com>
|
||||
*/
|
||||
class Sqlsrv extends Db{
|
||||
class Sqlsrv extends Driver{
|
||||
protected $selectSql = 'SELECT T1.* FROM (SELECT thinkphp.*, ROW_NUMBER() OVER (%ORDER%) AS ROW_NUMBER FROM (SELECT %DISTINCT% %FIELD% FROM %TABLE%%JOIN%%WHERE%%GROUP%%HAVING%) AS thinkphp) AS T1 %LIMIT%%COMMENT%';
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config='') {
|
||||
if ( !function_exists('sqlsrv_connect') ) {
|
||||
throw_exception(L('_NOT_SUPPERT_').':sqlsrv');
|
||||
}
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
$host = $config['hostname'].($config['hostport']?",{$config['hostport']}":'');
|
||||
$connectInfo = array('Database'=>$config['database'],'UID'=>$config['username'],'PWD'=>$config['password'],'CharacterSet' => C('DEFAULT_CHARSET'));
|
||||
$this->linkID[$linkNum] = sqlsrv_connect( $host, $connectInfo);
|
||||
if ( !$this->linkID[$linkNum] ) $this->error(false);
|
||||
// 标记连接成功
|
||||
$this->connected = true;
|
||||
//注销数据库安全信息
|
||||
if(1 != C('DB_DEPLOY_TYPE')) unset($this->config);
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
sqlsrv_free_stmt($this->queryID);
|
||||
$this->queryID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_query',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->queryID = sqlsrv_query($this->_linkID,$str,array(), array( "Scrollable" => SQLSRV_CURSOR_KEYSET));
|
||||
$this->debug();
|
||||
if ( false === $this->queryID ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = sqlsrv_num_rows($this->queryID);
|
||||
return $this->getAll();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( $this->queryID ) $this->free();
|
||||
N('db_write',1);
|
||||
// 记录开始执行时间
|
||||
G('queryStartTime');
|
||||
$this->queryID= sqlsrv_query($this->_linkID,$str);
|
||||
$this->debug();
|
||||
if ( false === $this->queryID ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = sqlsrv_rows_affected($this->queryID);
|
||||
$this->lastInsID = $this->mssql_insert_id();
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于获取最后插入的ID
|
||||
* @access public
|
||||
* @return integer
|
||||
*/
|
||||
public function mssql_insert_id() {
|
||||
$query = "SELECT @@IDENTITY as last_insert_id";
|
||||
$result = sqlsrv_query($this->_linkID,$query);
|
||||
list($last_insert_id) = sqlsrv_fetch_array($result);
|
||||
sqlsrv_free_stmt($result);
|
||||
return $last_insert_id;
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
sqlsrv_begin_transaction($this->_linkID);
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = sqlsrv_commit($this->_linkID);
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = sqlsrv_rollback($this->_linkID);
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getAll() {
|
||||
//返回数据集
|
||||
$result = array();
|
||||
if($this->numRows >0) {
|
||||
while($row = sqlsrv_fetch_array($this->queryID,SQLSRV_FETCH_ASSOC))
|
||||
$result[] = $row;
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 取得数据表的字段信息
|
||||
@@ -299,33 +127,4 @@ class Sqlsrv extends Db{
|
||||
return $this->execute($sql);
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
if ($this->_linkID){
|
||||
sqlsrv_close($this->_linkID);
|
||||
}
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error($result = true) {
|
||||
$errors = sqlsrv_errors();
|
||||
$this->error = '';
|
||||
foreach( $errors as $error ) {
|
||||
$this->error .= $error['message'];
|
||||
}
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
$result? ThinkLog::record($error['message'],'ERR'):throw_exception($this->error);
|
||||
return $this->error;
|
||||
}
|
||||
}
|
||||
387
Think/Db/Lite.php
Normal file
387
Think/Db/Lite.php
Normal file
@@ -0,0 +1,387 @@
|
||||
<?php
|
||||
// +----------------------------------------------------------------------
|
||||
// | TOPThink [ WE CAN DO IT JUST THINK ]
|
||||
// +----------------------------------------------------------------------
|
||||
// | Copyright (c) 2011 http://topthink.com All rights reserved.
|
||||
// +----------------------------------------------------------------------
|
||||
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
|
||||
// +----------------------------------------------------------------------
|
||||
// | Author: liu21st <liu21st@gmail.com>
|
||||
// +----------------------------------------------------------------------
|
||||
// $Id$
|
||||
namespace Think\Db;
|
||||
use Think\Config;
|
||||
use Think\Debug;
|
||||
use Think\Log;
|
||||
use PDO;
|
||||
class Lite {
|
||||
// PDO操作实例
|
||||
protected $PDOStatement = null;
|
||||
// 当前操作所属的模型名
|
||||
protected $model = '_think_';
|
||||
// 当前SQL指令
|
||||
protected $queryStr = '';
|
||||
protected $modelSql = [];
|
||||
// 最后插入ID
|
||||
protected $lastInsID = null;
|
||||
// 返回或者影响记录数
|
||||
protected $numRows = 0;
|
||||
// 返回字段数
|
||||
protected $numCols = 0;
|
||||
// 事务指令数
|
||||
protected $transTimes = 0;
|
||||
// 错误信息
|
||||
protected $error = '';
|
||||
// 数据库连接ID 支持多个连接
|
||||
protected $linkID = [];
|
||||
// 当前连接ID
|
||||
protected $_linkID = null;
|
||||
// 当前查询ID
|
||||
protected $queryID = null;
|
||||
// 数据库连接参数配置
|
||||
protected $config = [];
|
||||
|
||||
protected $queryTimes = 0;
|
||||
protected $executeTimes = 0;
|
||||
// PDO连接参数
|
||||
protected $options = [
|
||||
PDO::ATTR_CASE => PDO::CASE_LOWER,
|
||||
PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
|
||||
PDO::ATTR_ORACLE_NULLS => PDO::NULL_NATURAL,
|
||||
PDO::ATTR_STRINGIFY_FETCHES => false,
|
||||
];
|
||||
|
||||
/**
|
||||
* 架构函数 读取数据库配置信息
|
||||
* @access public
|
||||
* @param array $config 数据库配置数组
|
||||
*/
|
||||
public function __construct($config=''){
|
||||
if(!empty($config)) {
|
||||
$this->config = $config;
|
||||
if(empty($this->config['params'])) {
|
||||
$this->config['params'] = [];
|
||||
}
|
||||
$this->config['params'] = $this->options+$this->config['params'];
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接数据库方法
|
||||
* @access public
|
||||
*/
|
||||
public function connect($config='',$linkNum=0) {
|
||||
if ( !isset($this->linkID[$linkNum]) ) {
|
||||
if(empty($config)) $config = $this->config;
|
||||
try{
|
||||
if(empty($config['dsn'])) {
|
||||
$config['dsn'] = $config['dbms'].':dbname='.$config['database'].';host='.$config['hostname'];
|
||||
if(!empty($config['hostport'])) {
|
||||
$config['dsn'] .= ';port='.$config['hostport'];
|
||||
}elseif(!empty($config['unix_socket'])){
|
||||
$config['dsn'] .= ';unix_socket='.$config['unix_socket'];
|
||||
}
|
||||
}
|
||||
$this->linkID[$linkNum] = new PDO( $config['dsn'], $config['username'], $config['password'],$config['params']);
|
||||
}catch (\PDOException $e) {
|
||||
throw_exception($e->getMessage());
|
||||
}
|
||||
if(!empty($config['charset'])) {
|
||||
$this->linkID[$linkNum]->exec('SET NAMES '.$config['charset']);
|
||||
}
|
||||
// 注销数据库连接配置信息
|
||||
if(1 != $config['deploy']) $this->config = [];
|
||||
}
|
||||
return $this->linkID[$linkNum];
|
||||
}
|
||||
|
||||
/**
|
||||
* 释放查询结果
|
||||
* @access public
|
||||
*/
|
||||
public function free() {
|
||||
$this->PDOStatement = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行查询 返回数据集
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return mixed
|
||||
*/
|
||||
public function query($str,$bind=[]) {
|
||||
$this->initConnect(false);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( !empty($this->PDOStatement) ) $this->free();
|
||||
$this->queryTimes++;
|
||||
// 记录开始执行时间
|
||||
Debug::remark('queryStartTime','time');
|
||||
$this->PDOStatement = $this->_linkID->prepare($str);
|
||||
if(false === $this->PDOStatement)
|
||||
throw_exception($this->error());
|
||||
$result = $this->PDOStatement->execute($bind);
|
||||
$this->debug();
|
||||
if ( false === $result ) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
return $this->getResult();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行语句
|
||||
* @access public
|
||||
* @param string $str sql指令
|
||||
* @return integer
|
||||
*/
|
||||
public function execute($str,$bind=[]) {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
$this->queryStr = $str;
|
||||
//释放前次的查询结果
|
||||
if ( !empty($this->PDOStatement) ) $this->free();
|
||||
$this->executeTimes++;
|
||||
// 记录开始执行时间
|
||||
Debug::remark('queryStartTime','time');
|
||||
$this->PDOStatement = $this->_linkID->prepare($str);
|
||||
if(false === $this->PDOStatement) {
|
||||
throw_exception($this->error());
|
||||
}
|
||||
$result = $this->PDOStatement->execute($bind);
|
||||
$this->debug();
|
||||
if ( false === $result) {
|
||||
$this->error();
|
||||
return false;
|
||||
} else {
|
||||
$this->numRows = $this->PDOStatement->rowCount();
|
||||
if(preg_match("/^\s*(INSERT\s+INTO|REPLACE\s+INTO)\s+/i", $str)) {
|
||||
$this->lastInsID = $this->getLastInsertId();
|
||||
}
|
||||
return $this->numRows;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最后插入id
|
||||
* @access public
|
||||
* @return integer
|
||||
*/
|
||||
public function getLastInsertId() {
|
||||
return $this->_linkID->lastInsertId();
|
||||
}
|
||||
|
||||
/**
|
||||
* 启动事务
|
||||
* @access public
|
||||
* @return void
|
||||
*/
|
||||
public function startTrans() {
|
||||
$this->initConnect(true);
|
||||
if ( !$this->_linkID ) return false;
|
||||
//数据rollback 支持
|
||||
if ($this->transTimes == 0) {
|
||||
$this->_linkID->beginTransaction();
|
||||
}
|
||||
$this->transTimes++;
|
||||
return ;
|
||||
}
|
||||
|
||||
/**
|
||||
* 用于非自动提交状态下面的查询提交
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function commit() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = $this->_linkID->commit();
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 事务回滚
|
||||
* @access public
|
||||
* @return boolen
|
||||
*/
|
||||
public function rollback() {
|
||||
if ($this->transTimes > 0) {
|
||||
$result = $this->_linkID->rollback();
|
||||
$this->transTimes = 0;
|
||||
if(!$result){
|
||||
$this->error();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获得所有的查询数据
|
||||
* @access private
|
||||
* @return array
|
||||
*/
|
||||
private function getResult() {
|
||||
//返回数据集
|
||||
$result = $this->PDOStatement->fetchAll(PDO::FETCH_ASSOC);
|
||||
$this->numRows = count( $result );
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* 关闭数据库
|
||||
* @access public
|
||||
*/
|
||||
public function close() {
|
||||
$this->_linkID = null;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库错误信息
|
||||
* 并显示当前的SQL语句
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function error() {
|
||||
if($this->PDOStatement) {
|
||||
$error = $this->PDOStatement->errorInfo();
|
||||
$this->error = $error[2];
|
||||
}else{
|
||||
$this->error = '';
|
||||
}
|
||||
if('' != $this->queryStr){
|
||||
$this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
|
||||
}
|
||||
Log::record($this->error,'ERR');
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最近一次查询的sql语句
|
||||
* @param string $model 模型名
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getLastSql($model='') {
|
||||
return $model?$this->modelSql[$model]:$this->queryStr;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最近插入的ID
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getLastInsID() {
|
||||
return $this->lastInsID;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取最近的错误信息
|
||||
* @access public
|
||||
* @return string
|
||||
*/
|
||||
public function getError() {
|
||||
return $this->error;
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置当前操作模型
|
||||
* @access public
|
||||
* @param string $model 模型名
|
||||
* @return void
|
||||
*/
|
||||
public function setModel($model){
|
||||
$this->model = $model;
|
||||
}
|
||||
|
||||
/**
|
||||
* 数据库调试 记录当前SQL
|
||||
* @access protected
|
||||
*/
|
||||
protected function debug() {
|
||||
$this->modelSql[$this->model] = $this->queryStr;
|
||||
$this->model = '_think_';
|
||||
// 记录操作结束时间
|
||||
Debug::remark('queryEndTime','time');
|
||||
Log::record($this->queryStr.' [ RunTime:'.Debug::getUseTime('queryStartTime','queryEndTime').'s ]','SQL');
|
||||
}
|
||||
|
||||
/**
|
||||
* 初始化数据库连接
|
||||
* @access protected
|
||||
* @param boolean $master 主服务器
|
||||
* @return void
|
||||
*/
|
||||
protected function initConnect($master=true) {
|
||||
if(1 == $this->config['deploy'])
|
||||
// 采用分布式数据库
|
||||
$this->_linkID = $this->multiConnect($master);
|
||||
else
|
||||
// 默认单数据库
|
||||
if ( !$this->_linkID ) $this->_linkID = $this->connect();
|
||||
}
|
||||
|
||||
/**
|
||||
* 连接分布式服务器
|
||||
* @access protected
|
||||
* @param boolean $master 主服务器
|
||||
* @return void
|
||||
*/
|
||||
protected function multiConnect($master=false) {
|
||||
static $_config = [];
|
||||
if(empty($_config)) {
|
||||
// 缓存分布式数据库配置解析
|
||||
foreach ($this->config as $key=>$val){
|
||||
$_config[$key] = explode(',',$val);
|
||||
}
|
||||
}
|
||||
// 数据库读写是否分离
|
||||
if(Config::get('db_rw_separate')){
|
||||
// 主从式采用读写分离
|
||||
if($master)
|
||||
// 主服务器写入
|
||||
$r = floor(mt_rand(0,Config::get('db_master_num')-1));
|
||||
else{
|
||||
if(is_numeric(Config::get('db_slave_no'))) {// 指定服务器读
|
||||
$r = Config::get('db_slave_no');
|
||||
}else{
|
||||
// 读操作连接从服务器
|
||||
$r = floor(mt_rand(Config::get('db_master_num'),count($_config['hostname'])-1)); // 每次随机连接的数据库
|
||||
}
|
||||
}
|
||||
}else{
|
||||
// 读写操作不区分服务器
|
||||
$r = floor(mt_rand(0,count($_config['hostname'])-1)); // 每次随机连接的数据库
|
||||
}
|
||||
$db_config = [
|
||||
'username' => isset($_config['username'][$r])?$_config['username'][$r]:$_config['username'][0],
|
||||
'password' => isset($_config['password'][$r])?$_config['password'][$r]:$_config['password'][0],
|
||||
'hostname' => isset($_config['hostname'][$r])?$_config['hostname'][$r]:$_config['hostname'][0],
|
||||
'hostport' => isset($_config['hostport'][$r])?$_config['hostport'][$r]:$_config['hostport'][0],
|
||||
'database' => isset($_config['database'][$r])?$_config['database'][$r]:$_config['database'][0],
|
||||
'dsn' => isset($_config['dsn'][$r])?$_config['dsn'][$r]:$_config['dsn'][0],
|
||||
'params' => isset($_config['params'][$r])?$_config['params'][$r]:$_config['params'][0],
|
||||
];
|
||||
return $this->connect($db_config,$r);
|
||||
}
|
||||
|
||||
/**
|
||||
* 析构方法
|
||||
* @access public
|
||||
*/
|
||||
public function __destruct() {
|
||||
// 释放查询
|
||||
if ($this->queryID){
|
||||
$this->free();
|
||||
}
|
||||
// 关闭连接
|
||||
$this->close();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user