- /**
- * auther soulence
- * 调用数据类文件
- * modify 2015/06/12
- */
- class DBConnect
- {
- private $dbname = null;
- private $pdo = null;
- private $persistent = false;
- private $statement = null;
- private $lastInsID = null;
- private static $_instance = [];
-
- private function __construct($dbname,$attr)
- {
- $this->dbname = $dbname;
- $this->persistent = $attr;
- }
-
- public static function db($flag='r',$persistent=false)
- {
- if(!isset($flag)){
- $flag = 'r';
- }
-
- if (!class_exists('PDO'))
- {
- throw new Exception('not found PDO');
- return false;
- }
- $mysql_server = Yaf_Registry::get('mysql');
- if(!isset($mysql_server[$flag])){
- return false;
- }
-
- $options_arr = array(PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES '.$mysql_server[$flag]['charset'],PDO::ATTR_DEFAULT_FETCH_MODE=>PDO::FETCH_ASSOC);
- if($persistent === true){
- $options_arr[PDO::ATTR_PERSISTENT] = true;
- }
-
- try {
- $pdo = new PDO($mysql_server[$flag]['connectionString'],$mysql_server[$flag]['username'],$mysql_server[$flag]['password'],$options_arr);
- } catch (PDOException $e) {
- throw new Exception($e->getMessage());
- //exit('连接失败:'.$e->getMessage());
- return false;
- }
-
- if(!$pdo) {
- throw new Exception('PDO CONNECT ERROR');
- return false;
- }
-
- return $pdo;
- }
-
- /**
- * 得到操作数据库对象
- * @param string $dbname 对应的数据库是谁
- * @param bool $attr 是否长连接
- * return false说明给定的数据库不存在
- */
- public static function getInstance($dbname = 'r',$attr = false)
- {
- $mysql_server = Yaf_Registry::get('mysql');
- if(!isset($mysql_server[$dbname])){
- return false;
- }
- $key = md5(md5($dbname.$attr,true));
- if (!isset(self::$_instance[$key]) || !is_object(self::$_instance[$key]))
- self::$_instance[$key] = new self($dbname,$attr);
- return self::$_instance[$key];
- }
-
- private function getConnect(){
- $this->pdo = self::db($this->dbname,$this->persistent);
- }
-
- /**
- * 查询操作
- * @param string $sql 执行查询的sql语句
- * @param array $data 查询的条件 格式为[':id'=>$id,':name'=>$name](推荐)或者为[1=>$id,2=>$name]
- * @param bool $one 是否返回一条内容 默认为否
- */
- public function query($sql, $data = [], $one = false)
- {
- if (!is_array($data) || empty($sql) || !is_string($sql))
- return false;
-
- $this->free();
-
- return $this->queryCommon($data,$sql,$one);
- }
-
- /**
- * 内部查询的共用方法
- */
- private function queryCommon($data,$sql,$one)
- {
- $this->pdoExec($data,$sql);
-
- if ($one){
- return $this->statement->fetch(PDO::FETCH_ASSOC);
- }else{
- return $this->statement->fetchAll(PDO::FETCH_ASSOC);
- }
- }
-
- /**
- * 多条SQL语句的查询操作
- * @param array $arr_sql 执行查询的sql语句数组 格式为[$sql1,$sql2]
- * @param array $arr_data 查询与$arr_sql对应的条件 格式为[[':id'=>$id,':name'=>$name],[':id'=>$id,':name'=>$name]](推荐)或者为[[1=>$id,2=>$name],[1=>$id,2=>$name]]
- * @param bool $one 是否返回一条内容 默认为否 这里如果设置为true那么每一条sql都只返回一条数据
- */
- public function queryes($arr_sql, $arr_data = [], $one = false)
- {
- if(!is_array($arr_sql) || empty($arr_sql) || !is_array($arr_data))
- return false;
-
- $this->free();
-
- $res = [];$i = 0;
- foreach ($arr_sql as $val) {
- if(!isset($arr_data[$i]))
- $arr_data[$i] = [];
- elseif(!is_array($arr_data[$i]))
- throw new Exception('Error where queryes sql:'.$val.' where:'.$arr_data[$i]);
-
- $res[] = $this->queryCommon($arr_data[$i],$val,$one);
- $i++;
- }
- return $res;
- }
-
- /**
- * 分页封装
- *
- * @param string $sql
- * @param int $page 表示从第几页开始取
- * @param int $pageSize 表示每页多少条
- * @param array $data 查询的条件
- */
- public function limitQuery($sql, $page=0, $pageSize=20, $data = [])
- {
- $page = intval($page);
- if ($page < 0) {
- return [];
- }
- $pageSize = intval($pageSize);
- if ($pageSize > 0) { // pageSize 为0时表示取所有数据
- $sql .= ' LIMIT ' . $pageSize;
- if ($page > 0) {
- $start_limit = ($page - 1) * $pageSize;
- $sql .= ' OFFSET ' . $start_limit;
- }
- }
- return $this->query($sql, $data);
- }
-
- /**
- * 这个是用来进行添加 删除 修改操作 使用事务操作
- * @param string $sql 执行查询的sql语句
- * @param array $data 查询的条件 格式为[':id'=>$id,':name'=>$name](推荐)或者为[1=>$id,2=>$name]
- * @param bool $Transaction 是否事务操作 默认为否
- */
- public function executeDDL($sql, $data = [],$Transaction = false){
- if (!is_array($data) || !is_string($sql))
- return false;
-
- $this->free();
-
- if($Transaction)
- $this->pdo->beginTransaction();//开启事务
- try{
- $this->execRes($data,$sql);
- if($Transaction)
- $this->pdo->commit();//事务提交
- return $this->lastInsID;
- } catch (Exception $e) {
- if($Transaction)
- $this->pdo->rollBack();//事务回滚
- throw new Exception('Error DDLExecute <=====>'.$e->getMessage());
- return false;
- }
- }
-
- /**
- * 这个是用来进行添加 删除 修改操作 使用事务操作
- * 它是执行多条的
- * @param array $arr_sql 需要执行操作的SQL语句数组
- * @param array $arr_data 与数组对应SQL语句的条件
- * @param bool $Transaction 是否事务操作 默认为否
- */
- public function executeDDLes($arr_sql, $arr_data = [],$Transaction = false){
-
- if(!is_array($arr_sql) || empty($arr_sql) || !is_array($arr_data))
- return false;
-
- $res = [];
-
- $this->free();
-
- if($Transaction)
- $this->pdo->beginTransaction();//开启事务
- try{
- $i = 0;
- foreach($arr_sql as $val){
- if(!isset($arr_data[$i]))
- $arr_data[$i] = [];
- elseif(!is_array($arr_data[$i])){
- if($Transaction)
- $this->pdo->rollBack();//事务回滚
- throw new Exception('Error where DDLExecutees sql:'.$val.' where:'.$arr_data[$i]);
- }
-
- $this->execRes($arr_data[$i],$val);
- $res[] = $this->lastInsID;
- $i++;
- }
-
- if($Transaction)
- $this->pdo->commit();//事务提交
-
- return $res;
- } catch (Exception $e) {
- if($Transaction)
- $this->pdo->rollBack();//事务回滚
- throw new Exception('Error DDLExecutees array_sql:'.json_encode($arr_sql).' <=====>'.$e->getMessage());
- return false;
- }
- return $res;
- }
-
- /**
- * 此方法是用来计算查询返回的条数 注意 它只支持SELECT COUNT(*) FROM TABLE...或者SELECT COUNT(0) FROM TABLE...方式
- * @param string $sql 查询的sql语句
- * @param array $data SQL语句的条件
- */
- public function countRows($sql,$data = []){
- if (!is_array($data) || empty($sql) || !is_string($sql))
- return false;
- $this->free();
-
- $res = $this->pdoExec($data,$sql);
-
- if($res == false)
- return false;
-
- return $this->statement->fetchColumn();
- }
-
- /**
- * 此方法是用来计算查询返回的条数 它是执行多条SQL
- * @param string $sql 查询的sql语句
- * @param array $data SQL语句的条件
- */
- public function countRowses($arr_sql,$arr_data = []){
-
- if(!is_array($arr_sql) || empty($arr_sql) || !is_array($arr_data))
- return false;
-
- $res = [];
-
- $this->free();
- $i = 0;
- foreach ($arr_sql as $val) {
- if(!isset($arr_data[$i]))
- $arr_data[$i] = [];
- elseif(!is_array($arr_data[$i]))
- throw new Exception('Error where CountRowses sql:'.$val.' where:'.$arr_data[$i]);
-
- $res1 = $this->pdoExec($arr_data[$i],$val);
-
- if($res1 == false)
- $res[] = false;
- else
- $res[] = $this->statement->fetchColumn();
- }
-
- return $res;
- }
-
- /**
- * 这里再提供一个方法 由于项目中会有很多需要提供开启事务 然后再进行操作 最后提交
- * @param bool $Transaction 是否事务操作 默认为否
- */
- public function getDB($Transaction=false)
- {
- $this->Transaction = $Transaction;
- $this->getConnect();
- if($Transaction === true)
- $this->pdo->beginTransaction();//开启事务
- return $this;
- }
-
- /**
- * 此方法可以执行多次 它是执行DDL语句的
- * 注意 它是需要配合getDB和sQCommit一起使用 不能单独使用哦
- * 如果没有开启事务 sQCommit方法可以不调用
- * @param string $sql 查询的sql语句
- * @param array $data SQL语句的条件
- */
- public function execSq($sql,$data = [])
- {
- if($this->checkParams($sql,$data) === false)
- return false;
-
- try{
- $this->execRes($data,$sql);
- return $this->lastInsID;
- } catch (Exception $e) {
- if(isset($this->Transaction) && $this->Transaction === true)
- $this->pdo->rollBack();//事务回滚
- throw new Exception('Error execSq<=====>'.$e->getMessage());
- return false;
- } finally {
- if (!empty($this->statement))
- {
- $this->statement->closeCursor();
- unset($this->statement);
- }
- }
- }
-
- /**
- * 执行查询的方法 它需要传一个连接数据库对象
- * @param string $sql 执行查询的sql语句
- * @param array $data 查询的条件 格式为[':id'=>$id,':name'=>$name](推荐)或者为[1=>$id,2=>$name]
- * @param bool $one 是否返回一条内容 默认为否
- */
- public function querySq($sql,$data = [],$one = false)
- {
- if($this->checkParams($sql,$data) === false)
- return false;
-
- return $this->pdoExecSq($sql,$data,[1,$one]);
- }
-
- /**
- * 分页封装
- *
- * @param string $sql
- * @param int $page 表示从第几页开始取
- * @param int $pageSize 表示每页多少条
- * @param array $data 查询的条件
- */
- public function limitQuerySq($sql, $page=0, $pageSize=20, $data = [])
- {
- $page = intval($page);
- if ($page < 0) {
- return [];
- }
- $pageSize = intval($pageSize);
- if ($pageSize > 0) { // pageSize 为0时表示取所有数据
- $sql .= ' LIMIT ' . $pageSize;
- if ($page > 0) {
- $start_limit = ($page - 1) * $pageSize;
- $sql .= ' OFFSET ' . $start_limit;
- }
- }
- return $this->querySq($sql, $data);
- }
-
- /**
- * 此方法是用来计算查询返回的条数 注意 它只支持SELECT COUNT(*) FROM TABLE...或者SELECT COUNT(0) FROM TABLE...方式
- * @param string $sql 查询的sql语句
- * @param array $data SQL语句的条件
- */
- public function countRowsSq($sql,$data = []){
- if($this->checkParams($sql,$data) === false)
- return false;
- return $this->pdoExecSq($sql,$data,[2]);
- }
-
- /**
- * 这里再提供一个方法 这是最后提交操作 如果没有开启事务 此方法最后可以不调用的
- */
- public function sQCommit()
- {
- if(empty($this->pdo) || !is_object($this->pdo))
- return false;
- if(isset($this->Transaction) && $this->Transaction === true)
- $this->pdo->commit();//提交事务
- unset($this->pdo);
- }
-
- /**
- * 内部调用方法
- */
- public function checkParams($sql,$data)
- {
- if (empty($this->pdo) || !is_object($this->pdo) || !is_array($data) || empty($sql) || !is_string($sql))
- return false;
-
- return true;
- }
-
- /**
- * 内部调用方法
- */
- private function pdoExecSq($sql,$data,$select = []){
- try{
- $res = $this->pdoExec($data,$sql);
- if(empty($select))
- return $res;
- else{
- if($select[0] === 1){
- if($select[1] === true)
- return $this->statement->fetch(PDO::FETCH_ASSOC);
- else
- return $this->statement->fetchAll(PDO::FETCH_ASSOC);
- }elseif($select[0] === 2)
- return $this->statement->fetchColumn();
- else
- return false;
- }
- } catch (Exception $e) {
- throw new Exception($e->getMessage());
- return false;
- } finally {
- if (!empty($this->statement))
- {
- $this->statement->closeCursor();
- unset($this->statement);
- }
- }
- }
-
- /**
- * 内部调用方法
- */
- private function execRes($data,$sql){
-
- $res = $this->pdoExec($data,$sql);
-
- $in_id = $this->pdo->lastInsertId();
-
- if (preg_match("/^\s*(INSERT\s+INTO|REPLACE\s+INTO)\s+/i", $sql) && !empty($in_id))
- $this->lastInsID = $in_id;
- else
- $this->lastInsID = $res;
- }
-
- /**
- * 内部调用方法 用来直接执行SQL语句的方法
- */
- private function pdoExec($data,$sql){
- $this->statement = $this->pdo->prepare($sql);
- if (false === $this->statement)
- return false;
- if (!empty($data))
- {
- foreach ($data as $k => $v)
- {
- $this->statement->bindValue($k, $v);
- }
- }
- $res = $this->statement->execute();
- if (!$res)
- {
- throw new Exception('sql:'.$sql.'<====>where:'.json_encode($data).'<====>error:'.json_encode($this->statement->errorInfo()));
- }else{
- return $res;
- }
- }
-
- /**
- * 内部调用方法 用来释放的
- */
- private function free()
- {
- if (is_null($this->pdo))
- $this->getConnect();
-
- if (!empty($this->statement))
- {
- $this->statement->closeCursor();
- $this->statement = null;
- }
- }
- }
- ?>
|