123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149 |
- <?php
- // +----------------------------------------------------------------------
- // | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
- // +----------------------------------------------------------------------
- // | Copyright (c) 2006-2014 http://thinkphp.cn All rights reserved.
- // +----------------------------------------------------------------------
- // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
- // +----------------------------------------------------------------------
- // | Author: liu21st <liu21st@gmail.com>
- // +----------------------------------------------------------------------
- namespace Think\Db;
- use Think\Config;
- use Think\Debug;
- use Think\Log;
- use PDO;
- abstract class Driver {
- // PDO操作实例
- protected $PDOStatement = null;
- // 当前操作所属的模型名
- protected $model = '_think_';
- // 当前SQL指令
- protected $queryStr = '';
- protected $modelSql = array();
- // 最后插入ID
- protected $lastInsID = null;
- // 返回或者影响记录数
- protected $numRows = 0;
- // 事务指令数
- protected $transTimes = 0;
- // 错误信息
- protected $error = '';
- // 数据库连接ID 支持多个连接
- protected $linkID = array();
- // 当前连接ID
- protected $_linkID = null;
- // 数据库连接参数配置
- protected $config = array(
- 'type' => '', // 数据库类型
- 'hostname' => '127.0.0.1', // 服务器地址
- 'database' => '', // 数据库名
- 'username' => '', // 用户名
- 'password' => '', // 密码
- 'hostport' => '', // 端口
- 'dsn' => '', //
- 'params' => array(), // 数据库连接参数
- 'charset' => 'utf8', // 数据库编码默认采用utf8
- 'prefix' => '', // 数据库表前缀
- 'debug' => false, // 数据库调试模式
- 'deploy' => 0, // 数据库部署方式:0 集中式(单一服务器),1 分布式(主从服务器)
- 'rw_separate' => false, // 数据库读写是否分离 主从式有效
- 'master_num' => 1, // 读写分离后 主服务器数量
- 'slave_no' => '', // 指定从服务器序号
- 'db_like_fields' => '',
- );
- // 数据库表达式
- protected $exp = array('eq'=>'=','neq'=>'<>','gt'=>'>','egt'=>'>=','lt'=>'<','elt'=>'<=','notlike'=>'NOT LIKE','like'=>'LIKE','in'=>'IN','notin'=>'NOT IN','not in'=>'NOT IN','between'=>'BETWEEN','not between'=>'NOT BETWEEN','notbetween'=>'NOT BETWEEN');
- // 查询表达式
- protected $selectSql = 'SELECT%DISTINCT% %FIELD% FROM %TABLE%%FORCE%%JOIN%%WHERE%%GROUP%%HAVING%%ORDER%%LIMIT% %UNION%%LOCK%%COMMENT%';
- // 查询次数
- protected $queryTimes = 0;
- // 执行次数
- protected $executeTimes = 0;
- // PDO连接参数
- protected $options = array(
- PDO::ATTR_CASE => PDO::CASE_LOWER,
- PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
- PDO::ATTR_ORACLE_NULLS => PDO::NULL_NATURAL,
- PDO::ATTR_STRINGIFY_FETCHES => false,
- );
- protected $bind = array(); // 参数绑定
- /**
- * 架构函数 读取数据库配置信息
- * @access public
- * @param array $config 数据库配置数组
- */
- public function __construct($config=''){
- if(!empty($config)) {
- $this->config = array_merge($this->config,$config);
- if(is_array($this->config['params'])){
- $this->options = $this->config['params'] + $this->options;
- }
- }
- }
- /**
- * 连接数据库方法
- * @access public
- */
- public function connect($config='',$linkNum=0,$autoConnection=false) {
- if ( !isset($this->linkID[$linkNum]) ) {
- if(empty($config)) $config = $this->config;
- try{
- if(empty($config['dsn'])) {
- $config['dsn'] = $this->parseDsn($config);
- }
- if(version_compare(PHP_VERSION,'5.3.6','<=')){
- // 禁用模拟预处理语句
- $this->options[PDO::ATTR_EMULATE_PREPARES] = false;
- }
- $this->linkID[$linkNum] = new PDO( $config['dsn'], $config['username'], $config['password'],$this->options);
- }catch (\PDOException $e) {
- if($autoConnection){
- trace($e->getMessage(),'','ERR');
- return $this->connect($autoConnection,$linkNum);
- }elseif($config['debug']){
- E($e->getMessage());
- }
- }
- }
- return $this->linkID[$linkNum];
- }
- /**
- * 解析pdo连接的dsn信息
- * @access public
- * @param array $config 连接信息
- * @return string
- */
- protected function parseDsn($config){}
- /**
- * 释放查询结果
- * @access public
- */
- public function free() {
- $this->PDOStatement = null;
- }
- /**
- * 执行查询 返回数据集
- * @access public
- * @param string $str sql指令
- * @param boolean $fetchSql 不执行只是获取SQL
- * @return mixed
- */
- public function query($str,$fetchSql=false) {
- $this->initConnect(false);
- if ( !$this->_linkID ) return false;
- $this->queryStr = $str;
- if(!empty($this->bind)){
- $that = $this;
- $this->queryStr = strtr($this->queryStr,array_map(function($val) use($that){ return '\''.$that->escapeString($val).'\''; },$this->bind));
- }
- if($fetchSql){
- return $this->queryStr;
- }
- //释放前次的查询结果
- if ( !empty($this->PDOStatement) ) $this->free();
- $this->queryTimes++;
- N('db_query',1); // 兼容代码
- // 调试开始
- $this->debug(true);
- $this->PDOStatement = $this->_linkID->prepare($str);
- if(false === $this->PDOStatement){
- $this->error();
- return false;
- }
- foreach ($this->bind as $key => $val) {
- if(is_array($val)){
- $this->PDOStatement->bindValue($key, $val[0], $val[1]);
- }else{
- $this->PDOStatement->bindValue($key, $val);
- }
- }
- $this->bind = array();
- try{
- $result = $this->PDOStatement->execute();
- // 调试结束
- $this->debug(false);
- if ( false === $result ) {
- $this->error();
- return false;
- } else {
- return $this->getResult();
- }
- }catch (\PDOException $e) {
- $this->error();
- return false;
- }
- }
- /**
- * 执行语句
- * @access public
- * @param string $str sql指令
- * @param boolean $fetchSql 不执行只是获取SQL
- * @return mixed
- */
- public function execute($str,$fetchSql=false) {
- $this->initConnect(true);
- if ( !$this->_linkID ) return false;
- $this->queryStr = $str;
- if(!empty($this->bind)){
- $that = $this;
- $this->queryStr = strtr($this->queryStr,array_map(function($val) use($that){ return '\''.$that->escapeString($val).'\''; },$this->bind));
- }
- if($fetchSql){
- return $this->queryStr;
- }
- //释放前次的查询结果
- if ( !empty($this->PDOStatement) ) $this->free();
- $this->executeTimes++;
- N('db_write',1); // 兼容代码
- // 记录开始执行时间
- $this->debug(true);
- $this->PDOStatement = $this->_linkID->prepare($str);
- if(false === $this->PDOStatement) {
- $this->error();
- return false;
- }
- foreach ($this->bind as $key => $val) {
- if(is_array($val)){
- $this->PDOStatement->bindValue($key, $val[0], $val[1]);
- }else{
- $this->PDOStatement->bindValue($key, $val);
- }
- }
- $this->bind = array();
- try{
- $result = $this->PDOStatement->execute();
- // 调试结束
- $this->debug(false);
- 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->_linkID->lastInsertId();
- }
- return $this->numRows;
- }
- }catch (\PDOException $e) {
- $this->error();
- return false;
- }
- }
- /**
- * 启动事务
- * @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 boolean
- */
- 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 boolean
- */
- 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
- * @param boolean $execute 是否包含所有查询
- * @return integer
- */
- public function getQueryTimes($execute=false){
- return $execute?$this->queryTimes+$this->executeTimes:$this->queryTimes;
- }
- /**
- * 获得执行次数
- * @access public
- * @return integer
- */
- public function getExecuteTimes(){
- return $this->executeTimes;
- }
- /**
- * 关闭数据库
- * @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[1].':'.$error[2];
- }else{
- $this->error = '';
- }
- if('' != $this->queryStr){
- $this->error .= "\n [ SQL语句 ] : ".$this->queryStr;
- }
- // 记录错误日志
- trace($this->error,'','ERR');
- if($this->config['debug']) {// 开启数据库调试模式
- E($this->error);
- }else{
- 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){
- if(is_array($val) && 'exp' == $val[0]){
- $set[] = $this->parseKey($key).'='.$val[1];
- }elseif(is_null($val)){
- $set[] = $this->parseKey($key).'=NULL';
- }elseif(is_scalar($val)) {// 过滤非标量数据
- if(0===strpos($val,':') && in_array($val,array_keys($this->bind)) ){
- $set[] = $this->parseKey($key).'='.$this->escapeString($val);
- }else{
- $name = count($this->bind);
- $set[] = $this->parseKey($key).'=:'.$name;
- $this->bindParam($name,$val);
- }
- }
- }
- return ' SET '.implode(',',$set);
- }
- /**
- * 参数绑定
- * @access protected
- * @param string $name 绑定参数名
- * @param mixed $value 绑定值
- * @return void
- */
- protected function bindParam($name,$value){
- $this->bind[':'.$name] = $value;
- }
- /**
- * 字段名分析
- * @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 = strpos($value,':') === 0 && in_array($value,array_keys($this->bind))? $this->escapeString($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) && '' !== $fields) {
- $fields = explode(',',$fields);
- }
- if(is_array($fields)) {
- // 完善数组方式传字段名的支持
- // 支持 'field1'=>'field2' 这样的字段别名定义
- $array = 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);
- }else{
- $fieldsStr = '*';
- }
- //TODO 如果是查询全部字段,并且是join的方式,那么就把要查的表加个别名,以免字段被覆盖
- return $fieldsStr;
- }
- /**
- * table分析
- * @access protected
- * @param mixed $table
- * @return string
- */
- protected function parseTable($tables) {
- if(is_array($tables)) {// 支持别名定义
- $array = array();
- foreach ($tables as $table=>$alias){
- if(!is_numeric($table))
- $array[] = $this->parseKey($table).' '.$this->parseKey($alias);
- else
- $array[] = $this->parseKey($alias);
- }
- $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,array('AND','OR','XOR'))){
- // 定义逻辑运算规则 例如 OR XOR AND NOT
- $operate = ' '.$operate.' ';
- unset($where['_logic']);
- }else{
- // 默认进行 AND 运算
- $operate = ' AND ';
- }
- foreach ($where as $key=>$val){
- if(is_numeric($key)){
- $key = '_complex';
- }
- if(0===strpos($key,'_')) {
- // 解析特殊条件表达式
- $whereStr .= $this->parseThinkWhere($key,$val);
- }else{
- // 查询字段的安全过滤
- // if(!preg_match('/^[A-Z_\|\&\-.a-z0-9\(\)\,]+$/',trim($key))){
- // E(L('_EXPRESS_ERROR_').':'.$key);
- // }
- // 多条件支持
- $multi = is_array($val) && isset($val['_multi']);
- $key = trim($key);
- if(strpos($key,'|')) { // 支持 name|title|nickname 方式定义查询字段
- $array = explode('|',$key);
- $str = array();
- 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 = array();
- 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])) {
- $exp = strtolower($val[0]);
- if(preg_match('/^(eq|neq|gt|egt|lt|elt)$/',$exp)) { // 比较运算
- $whereStr .= $key.' '.$this->exp[$exp].' '.$this->parseValue($val[1]);
- }elseif(preg_match('/^(notlike|like)$/',$exp)){// 模糊查找
- if(is_array($val[1])) {
- $likeLogic = isset($val[2])?strtoupper($val[2]):'OR';
- if(in_array($likeLogic,array('AND','OR','XOR'))){
- $like = array();
- foreach ($val[1] as $item){
- $like[] = $key.' '.$this->exp[$exp].' '.$this->parseValue($item);
- }
- $whereStr .= '('.implode(' '.$likeLogic.' ',$like).')';
- }
- }else{
- $whereStr .= $key.' '.$this->exp[$exp].' '.$this->parseValue($val[1]);
- }
- }elseif('bind' == $exp ){ // 使用表达式
- $whereStr .= $key.' = :'.$val[1];
- }elseif('exp' == $exp ){ // 使用表达式
- $whereStr .= $key.' '.$val[1];
- }elseif(preg_match('/^(notin|not in|in)$/',$exp)){ // IN 运算
- if(isset($val[2]) && 'exp'==$val[2]) {
- $whereStr .= $key.' '.$this->exp[$exp].' '.$val[1];
- }else{
- if(is_string($val[1])) {
- $val[1] = explode(',',$val[1]);
- }
- $zone = implode(',',$this->parseValue($val[1]));
- $whereStr .= $key.' '.$this->exp[$exp].' ('.$zone.')';
- }
- }elseif(preg_match('/^(notbetween|not between|between)$/',$exp)){ // BETWEEN运算
- $data = is_string($val[1])? explode(',',$val[1]):$val[1];
- $whereStr .= $key.' '.$this->exp[$exp].' '.$this->parseValue($data[0]).' AND '.$this->parseValue($data[1]);
- }else{
- E(L('_EXPRESS_ERROR_').':'.$val[0]);
- }
- }else {
- $count = count($val);
- $rule = isset($val[$count-1]) ? (is_array($val[$count-1]) ? strtoupper($val[$count-1][0]) : strtoupper($val[$count-1]) ) : '' ;
- if(in_array($rule,array('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{
- $whereStr .= $this->parseWhereItem($key,$val[$i]).' '.$rule.' ';
- }
- }
- $whereStr = '( '.substr($whereStr,0,-4).' )';
- }
- }else {
- //对字符串类型字段采用模糊匹配
- $likeFields = $this->config['db_like_fields'];
- if($likeFields && preg_match('/^('.$likeFields.')$/i',$key)) {
- $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 = 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)) {
- $joinStr = ' '.implode(' ',$join).' ';
- }
- return $joinStr;
- }
- /**
- * order分析
- * @access protected
- * @param mixed $order
- * @return string
- */
- protected function parseOrder($order) {
- if(is_array($order)) {
- $array = 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 protected
- * @param array $bind
- * @return array
- */
- protected function parseBind($bind){
- $this->bind = array_merge($this->bind,$bind);
- }
- /**
- * index分析,可在操作链中指定需要强制使用的索引
- * @access protected
- * @param mixed $index
- * @return string
- */
- protected function parseForce($index) {
- if(empty($index)) return '';
- if(is_array($index)) $index = join(",", $index);
- return sprintf(" FORCE INDEX ( %s ) ", $index);
- }
- /**
- * ON DUPLICATE KEY UPDATE 分析
- * @access protected
- * @param mixed $duplicate
- * @return string
- */
- protected function parseDuplicate($duplicate){
- return '';
- }
- /**
- * 插入记录
- * @access public
- * @param mixed $data 数据
- * @param array $options 参数表达式
- * @param boolean $replace 是否replace
- * @return false | integer
- */
- public function insert($data,$options=array(),$replace=false) {
- $values = $fields = array();
- $this->model = $options['model'];
- $this->parseBind(!empty($options['bind'])?$options['bind']:array());
- foreach ($data as $key=>$val){
- if(is_array($val) && 'exp' == $val[0]){
- $fields[] = $this->parseKey($key);
- $values[] = $val[1];
- }elseif(is_null($val)){
- $fields[] = $this->parseKey($key);
- $values[] = 'NULL';
- }elseif(is_scalar($val)) { // 过滤非标量数据
- $fields[] = $this->parseKey($key);
- if(0===strpos($val,':') && in_array($val,array_keys($this->bind))){
- $values[] = $this->parseValue($val);
- }else{
- $name = count($this->bind);
- $values[] = ':'.$name;
- $this->bindParam($name,$val);
- }
- }
- }
- // 兼容数字传入方式
- $replace= (is_numeric($replace) && $replace>0)?true:$replace;
- $sql = (true===$replace?'REPLACE':'INSERT').' INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') VALUES ('.implode(',', $values).')'.$this->parseDuplicate($replace);
- $sql .= $this->parseComment(!empty($options['comment'])?$options['comment']:'');
- return $this->execute($sql,!empty($options['fetch_sql']) ? true : false);
- }
- /**
- * 批量插入记录
- * @access public
- * @param mixed $dataSet 数据集
- * @param array $options 参数表达式
- * @param boolean $replace 是否replace
- * @return false | integer
- */
- public function insertAll($dataSet,$options=array(),$replace=false) {
- $values = array();
- $this->model = $options['model'];
- if(!is_array($dataSet[0])) return false;
- $this->parseBind(!empty($options['bind'])?$options['bind']:array());
- $fields = array_map(array($this,'parseKey'),array_keys($dataSet[0]));
- foreach ($dataSet as $data){
- $value = array();
- foreach ($data as $key=>$val){
- if(is_array($val) && 'exp' == $val[0]){
- $value[] = $val[1];
- }elseif(is_null($val)){
- $value[] = 'NULL';
- }elseif(is_scalar($val)){
- if(0===strpos($val,':') && in_array($val,array_keys($this->bind))){
- $value[] = $this->parseValue($val);
- }else{
- $name = count($this->bind);
- $value[] = ':'.$name;
- $this->bindParam($name,$val);
- }
- }
- }
- $values[] = 'SELECT '.implode(',', $value);
- }
- $sql = 'INSERT INTO '.$this->parseTable($options['table']).' ('.implode(',', $fields).') '.implode(' UNION ALL ',$values);
- $sql .= $this->parseComment(!empty($options['comment'])?$options['comment']:'');
- return $this->execute($sql,!empty($options['fetch_sql']) ? true : false);
- }
- /**
- * 通过Select方式插入记录
- * @access public
- * @param string $fields 要插入的数据表字段名
- * @param string $table 要插入的数据表名
- * @param array $option 查询数据参数
- * @return false | integer
- */
- public function selectInsert($fields,$table,$options=array()) {
- $this->model = $options['model'];
- $this->parseBind(!empty($options['bind'])?$options['bind']:array());
- if(is_string($fields)) $fields = explode(',',$fields);
- array_walk($fields, array($this, 'parseKey'));
- $sql = 'INSERT INTO '.$this->parseTable($table).' ('.implode(',', $fields).') ';
- $sql .= $this->buildSelectSql($options);
- return $this->execute($sql,!empty($options['fetch_sql']) ? true : false);
- }
- /**
- * 更新记录
- * @access public
- * @param mixed $data 数据
- * @param array $options 表达式
- * @return false | integer
- */
- public function update($data,$options) {
- $this->model = $options['model'];
- $this->parseBind(!empty($options['bind'])?$options['bind']:array());
- $table = $this->parseTable($options['table']);
- $sql = 'UPDATE ' . $table . $this->parseSet($data);
- if(strpos($table,',')){// 多表更新支持JOIN操作
- $sql .= $this->parseJoin(!empty($options['join'])?$options['join']:'');
- }
- $sql .= $this->parseWhere(!empty($options['where'])?$options['where']:'');
- if(!strpos($table,',')){
- // 单表更新支持order和lmit
- $sql .= $this->parseOrder(!empty($options['order'])?$options['order']:'')
- .$this->parseLimit(!empty($options['limit'])?$options['limit']:'');
- }
- $sql .= $this->parseComment(!empty($options['comment'])?$options['comment']:'');
- return $this->execute($sql,!empty($options['fetch_sql']) ? true : false);
- }
- /**
- * 删除记录
- * @access public
- * @param array $options 表达式
- * @return false | integer
- */
- public function delete($options=array()) {
- $this->model = $options['model'];
- $this->parseBind(!empty($options['bind'])?$options['bind']:array());
- $table = $this->parseTable($options['table']);
- $sql = 'DELETE FROM '.$table;
- if(strpos($table,',')){// 多表删除支持USING和JOIN操作
- if(!empty($options['using'])){
- $sql .= ' USING '.$this->parseTable($options['using']).' ';
- }
- $sql .= $this->parseJoin(!empty($options['join'])?$options['join']:'');
- }
- $sql .= $this->parseWhere(!empty($options['where'])?$options['where']:'');
- if(!strpos($table,',')){
- // 单表删除支持order和limit
- $sql .= $this->parseOrder(!empty($options['order'])?$options['order']:'')
- .$this->parseLimit(!empty($options['limit'])?$options['limit']:'');
- }
- $sql .= $this->parseComment(!empty($options['comment'])?$options['comment']:'');
- return $this->execute($sql,!empty($options['fetch_sql']) ? true : false);
- }
- /**
- * 查找记录
- * @access public
- * @param array $options 表达式
- * @return mixed
- */
- public function select($options=array()) {
- $this->model = $options['model'];
- $this->parseBind(!empty($options['bind'])?$options['bind']:array());
- $sql = $this->buildSelectSql($options);
- $result = $this->query($sql,!empty($options['fetch_sql']) ? true : false);
- return $result;
- }
- /**
- * 生成查询SQL
- * @access public
- * @param array $options 表达式
- * @return string
- */
- public function buildSelectSql($options=array()) {
- if(isset($options['page'])) {
- // 根据页数计算limit
- list($page,$listRows) = $options['page'];
- $page = $page>0 ? $page : 1;
- $listRows= $listRows>0 ? $listRows : (is_numeric($options['limit'])?$options['limit']:20);
- $offset = $listRows*($page-1);
- $options['limit'] = $offset.','.$listRows;
- }
- $sql = $this->parseSql($this->selectSql,$options);
- return $sql;
- }
- /**
- * 替换SQL语句中表达式
- * @access public
- * @param array $options 表达式
- * @return string
- */
- public function parseSql($sql,$options=array()){
- $sql = str_replace(
- array('%TABLE%','%DISTINCT%','%FIELD%','%JOIN%','%WHERE%','%GROUP%','%HAVING%','%ORDER%','%LIMIT%','%UNION%','%LOCK%','%COMMENT%','%FORCE%'),
- array(
- $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->parseLock(isset($options['lock'])?$options['lock']:false),
- $this->parseComment(!empty($options['comment'])?$options['comment']:''),
- $this->parseForce(!empty($options['force'])?$options['force']:'')
- ),$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
- * @param boolean $start 调试开始标记 true 开始 false 结束
- */
- protected function debug($start) {
- if($this->config['debug']) {// 开启数据库调试模式
- if($start) {
- G('queryStartTime');
- }else{
- $this->modelSql[$this->model] = $this->queryStr;
- //$this->model = '_think_';
- // 记录操作结束时间
- G('queryEndTime');
- trace($this->queryStr.' [ RunTime:'.G('queryStartTime','queryEndTime').'s ]','','SQL');
- }
- }
- }
- /**
- * 初始化数据库连接
- * @access protected
- * @param boolean $master 主服务器
- * @return void
- */
- protected function initConnect($master=true) {
- if(!empty($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) {
- // 分布式数据库配置解析
- $_config['username'] = explode(',',$this->config['username']);
- $_config['password'] = explode(',',$this->config['password']);
- $_config['hostname'] = explode(',',$this->config['hostname']);
- $_config['hostport'] = explode(',',$this->config['hostport']);
- $_config['database'] = explode(',',$this->config['database']);
- $_config['dsn'] = explode(',',$this->config['dsn']);
- $_config['charset'] = explode(',',$this->config['charset']);
- $m = floor(mt_rand(0,$this->config['master_num']-1));
- // 数据库读写是否分离
- if($this->config['rw_separate']){
- // 主从式采用读写分离
- if($master)
- // 主服务器写入
- $r = $m;
- else{
- if(is_numeric($this->config['slave_no'])) {// 指定服务器读
- $r = $this->config['slave_no'];
- }else{
- // 读操作连接从服务器
- $r = floor(mt_rand($this->config['master_num'],count($_config['hostname'])-1)); // 每次随机连接的数据库
- }
- }
- }else{
- // 读写操作不区分服务器
- $r = floor(mt_rand(0,count($_config['hostname'])-1)); // 每次随机连接的数据库
- }
-
- if($m != $r ){
- $db_master = array(
- 'username' => isset($_config['username'][$m])?$_config['username'][$m]:$_config['username'][0],
- 'password' => isset($_config['password'][$m])?$_config['password'][$m]:$_config['password'][0],
- 'hostname' => isset($_config['hostname'][$m])?$_config['hostname'][$m]:$_config['hostname'][0],
- 'hostport' => isset($_config['hostport'][$m])?$_config['hostport'][$m]:$_config['hostport'][0],
- 'database' => isset($_config['database'][$m])?$_config['database'][$m]:$_config['database'][0],
- 'dsn' => isset($_config['dsn'][$m])?$_config['dsn'][$m]:$_config['dsn'][0],
- 'charset' => isset($_config['charset'][$m])?$_config['charset'][$m]:$_config['charset'][0],
- );
- }
- $db_config = array(
- '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],
- 'charset' => isset($_config['charset'][$r])?$_config['charset'][$r]:$_config['charset'][0],
- );
- return $this->connect($db_config,$r,$r == $m ? false : $db_master);
- }
- /**
- * 析构方法
- * @access public
- */
- public function __destruct() {
- // 释放查询
- if ($this->PDOStatement){
- $this->free();
- }
- // 关闭连接
- $this->close();
- }
- }
|