<?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();

        $result =   $this->PDOStatement->execute();

        // 调试结束

        $this->debug(false);

        if ( false === $result ) {

            $this->error();

            return false;

        } else {

            return $this->getResult();

        }

    }



    /**

     * 执行语句

     * @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();

        $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;

        }

    }



    /**

     * 启动事务

     * @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_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_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_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();

    }

}

