PHP PDO操作MYSQL封装类

php代码

<?php
/**
 * 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=&#39;r&#39;,$persistent=false)
    {
        if(!isset($flag)){
            $flag = &#39;r&#39;;
        }

        if (!class_exists(&#39;PDO&#39;))
        {
            throw new Exception(&#39;not found PDO&#39;);
            return false;
        }
        $mysql_server = Yaf_Registry::get(&#39;mysql&#39;);
        if(!isset($mysql_server[$flag])){
            return false;
        }

        $options_arr = array(PDO::MYSQL_ATTR_INIT_COMMAND => &#39;SET NAMES &#39;.$mysql_server[$flag][&#39;charset&#39;],PDO::ATTR_DEFAULT_FETCH_MODE=>PDO::FETCH_ASSOC);
        if($persistent === true){
            $options_arr[PDO::ATTR_PERSISTENT] = true;
        }

        try {
            $pdo = new PDO($mysql_server[$flag][&#39;connectionString&#39;],$mysql_server[$flag][&#39;username&#39;],$mysql_server[$flag][&#39;password&#39;],$options_arr);
        } catch (PDOException $e) {
            throw new Exception($e->getMessage());
            //exit(&#39;连接失败:&#39;.$e->getMessage());
            return false;
        }

        if(!$pdo) {
            throw new Exception(&#39;PDO CONNECT ERROR&#39;);
            return false;
        }

        return $pdo;
    }

    /**
     * 得到操作数据库对象
     * @param string $dbname 对应的数据库是谁
     * @param bool $attr  是否长连接
     * return false说明给定的数据库不存在
     */
    public static function getInstance($dbname = &#39;r&#39;,$attr = false)
    {
        $mysql_server = Yaf_Registry::get(&#39;mysql&#39;);
        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  查询的条件  格式为[&#39;:id&#39;=>$id,&#39;:name&#39;=>$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对应的条件  格式为[[&#39;:id&#39;=>$id,&#39;:name&#39;=>$name],[&#39;:id&#39;=>$id,&#39;:name&#39;=>$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(&#39;Error where queryes sql:&#39;.$val.&#39; where:&#39;.$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 .= &#39; LIMIT &#39; . $pageSize;
            if ($page > 0) {
                $start_limit = ($page - 1) * $pageSize;
                $sql .= &#39; OFFSET &#39; . $start_limit;
            }
        }
        return $this->query($sql, $data);
    }

    /**
     * 这个是用来进行添加 删除  修改操作  使用事务操作
     * @param string $sql   执行查询的sql语句
     * @param array $data  查询的条件  格式为[&#39;:id&#39;=>$id,&#39;:name&#39;=>$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(&#39;Error DDLExecute <=====>&#39;.$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(&#39;Error where DDLExecutees sql:&#39;.$val.&#39; where:&#39;.$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(&#39;Error DDLExecutees array_sql:&#39;.json_encode($arr_sql).&#39; <=====>&#39;.$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(&#39;Error where CountRowses sql:&#39;.$val.&#39; where:&#39;.$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(&#39;Error execSq<=====>&#39;.$e->getMessage());
            return false;
        } finally {
            if (!empty($this->statement))
            {
                $this->statement->closeCursor();
                unset($this->statement);
            }
        }
    }

    /**
     * 执行查询的方法  它需要传一个连接数据库对象
     * @param string $sql   执行查询的sql语句
     * @param array $data  查询的条件  格式为[&#39;:id&#39;=>$id,&#39;:name&#39;=>$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 .= &#39; LIMIT &#39; . $pageSize;
            if ($page > 0) {
                $start_limit = ($page - 1) * $pageSize;
                $sql .= &#39; OFFSET &#39; . $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(&#39;sql:&#39;.$sql.&#39;<====>where:&#39;.json_encode($data).&#39;<====>error:&#39;.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;
        }
    }
}
?>