/**
* 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;
}
}
}
?>