来源： Swoole整合ThinkPHP3.2系列教程四_一个不靠谱的程序员-CSDN博客
在cli环境下，PHP程序需要长时间运行，客户端与MySQL服务器之间的TCP连接是不稳定的。
MySQL-Server会在一定时间内自动切断连接
PHP程序遇到空闲期时长时间没有MySQL查询，MySQL-Server也会切断连接回收资源
其他情况，在MySQL服务器中执行kill process杀掉某个连接，MySQL服务器重启
所以我们要考虑数据库断线重连的问题，但是ThinkPHP3.2里DB驱动类里并没有断线重连的例子，好消息是TP5里有，于是我照着TP5里的源码，改了TP3.2里的DB驱动类，来达到数据库断线自动重连。
附上代码
完整的DB.class.php文件 保证可用。
<?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>
// +———————————————————————-
// 2017-8-24 09:26:23 增加了断线重连功能
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(‘query’);
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(‘query’);
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(‘execute’);
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(‘execute’);
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;
$this->linkID = [];
return $this;
}
/**
* 数据库错误信息
* 并显示当前的SQL语句
* @access public
* @return string
*/
public function error($method=”) {
if($this->PDOStatement) {
$error = $this->PDOStatement->errorInfo();
$this->error = $error[1].’:’.$error[2];
}else{
$this->error = ”;
}
//2017-8-24 09:18:45 判断是否断线 增加断线重连功能
if($this->isBreak($error[2]) && $method){
return $this->close()->{$method}($this->queryStr);
}
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);
}
/**
* 是否断线
* 2017-8-24 09:12:40 摘自TP5 DB/Connection.php
* 修改传入的参数 来适配TP3.2
* @access protected
* @param String $error 返回的错误信息
* @return bool
*/
protected function isBreak($error)
{
$info = [
‘server has gone away’,
‘no connection to the server’,
‘Lost connection’,
‘is dead or not enabled’,
‘Error while sending’,
‘decryption failed or bad record mac’,
‘server closed the connection unexpectedly’,
‘SSL connection has been closed unexpectedly’,
‘Error writing data to the connection’,
‘Resource deadlock avoided’,
];
foreach ($info as $msg) {
if (false !== stripos($error, $msg)) {
return true;
}
}
return false;
}
/**
* 析构方法
* @access public
*/
public function __destruct() {
// 释放查询
if ($this->PDOStatement){
$this->free();
}
// 关闭连接
$this->close();
}
}
————————————————
版权声明：本文为CSDN博主「一个不靠谱的程序员」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/u013705066/article/details/77679827
Mikel