封装一个PDO数据库操作类

本帖最后由 zhou_gjun 于 2009-11-30 21:55 编辑

<?php
/**
　* 数据库PDO操作
　*/

class MysqlPdo {

        public static $PDOStatement = null;
        /**
        * 数据库的连接参数配置
        * @var array
        * @access public
    */
        public static $config = array();

        /**
    * 是否使用永久连接
    * @var bool
        * @access public
    */
        public static $pconnect  = false;

        /**
    * 错误信息
    * @var string
        * @access public
    */
    public static $error = '';

        /**
        * 单件模式,保存Pdo类唯一实例,数据库的连接资源
        * @var object
        * @access public
    */
        protected static $link;

        /**
    * 是否已经连接数据库
    * @var bool
        * @access public
    */
        public static $connected = false;

        /**
    * 数据库版本
    * @var string
        * @access public
    */
    public static $dbVersion = null;

        /**
    * 当前SQL语句
    * @var string
        * @access public
    */
        public static $queryStr = '';

        /**
    * 最后插入记录的ID
    * @var integer
        * @access public
    */
    public static $lastInsertId = null;

        /**
    * 返回影响记录数
    * @var integer
        * @access public
    */
    public static $numRows = 0;

        // 事务指令数
    public static $transTimes = 0;

        /**
         * 构造函数，
         * @param $dbconfig 数据库连接相关信息，array('ServerName', 'UserName', 'Password', 'DefaultDb', 'DB_Port', 'DB_TYPE')
         */
    public function __construct($dbConfig=''){
        if (!class_exists('PDO')) throw_exception("不支持:PDO");
        //若没有传输任何参数，则使用默认的数据定义
                if (!is_array($dbConfig)) {
                        $dbConfig = array(
                                'hostname' => DB_HOST,
                                'username' => DB_USER,
                                'password' => DB_PWD,
                                'database' => DB_NAME,
                                'hostport' => DB_PORT,
                                'dbms'     => DB_TYPE,
                                'dsn'      => DB_TYPE.":host=".DB_HOST.";dbname=".DB_NAME
                        );
                }
        if(empty($dbConfig['hostname'])) throw_exception("没有定义数据库配置");
                self::$config = $dbConfig;
        if(empty(self::$config['params'])) self::$config['params']        = array();
               


                /*************************************华丽分隔线*******************************************/
                if (!isset(self::$link) ) {
                        $configs = self::$config;
                        if(self::$pconnect) {
                                $configs['params'][constant('PDO::ATTR_PERSISTENT')] = true;
                        }
                        try {
                                self::$link = new PDO( $configs['dsn'], $configs['username'], $configs['password'],$configs['params']);
                        } catch (PDOException $e) {
                           throw_exception($e->getMessage());
                           //exit('连接失败:'.$e->getMessage());   
                        }
            if(!self::$link) {
                throw_exception('PDO CONNECT ERROR');
                return false;
            }
                        self::$link->exec('SET NAMES '.DB_CHARSET);
            self::$dbVersion = self::$link->getAttribute(constant("PDO::ATTR_SERVER_INFO"));
            // 标记连接成功
                        self::$connected = true;
            // 注销数据库连接配置信息
            unset($configs);
        }
        return self::$link;



    }
    /**
    * 释放查询结果
    * @access function
    */
    static function free() {
                self::$PDOStatement = null;
    }

        /*********************************************************************************************************/
        /*                                             数据库操作                                                     */
        /*********************************************************************************************************/
        /**
    * 获得所有的查询数据
    * @access function
    * @return array
    */
    static function getAll($sql=null) {               
        self::query($sql);
        //返回数据集
                $result        = self::$PDOStatement->fetchAll(constant('PDO::FETCH_ASSOC'));
                return $result;
    }

        /**
    * 获得一条查询结果
    * @access function
    * @param string $sql  SQL指令
    * @param integer $seek 指针位置
    * @return array
    */
    static function getRow($sql=null) {               
                self::query($sql);
                // 返回数组集
                $result = self::$PDOStatement->fetch(constant('PDO::FETCH_ASSOC'),constant('PDO::FETCH_ORI_NEXT'));
                return $result;
    }

        /**
    * 执行sql语句，自动判断进行查询或者执行操作
    * @access function
    * @param string  $sql SQL指令
    * @return mixed
    */
    static function doSql($sql='') {
        if(self::isMainIps($sql)) {
                return self::execute($sql);
        }else {
                        return self::getAll($sql);
        }
    }
    /**
    * 根据指定ID查找表中记录(仅用于单表操作)
    * @access function
    * @param integer $priId  主键ID
    * @param string $tables  数据表名
    * @param string $fields  字段名
    * @return ArrayObject 表记录
    */
        static function findById($tabName,$priId,$fields='*'){
        $sql = 'SELECT %s FROM %s WHERE id=%d';
        return self::getRow(sprintf($sql, self::parseFields($fields), $tabName, $priId));
    }

    /**
    * 查找记录
    * @access function
    * @param string  $tables  数据表名
    * @param mixed   $where   查询条件
    * @param string  $fields  字段名
    * @param string  $order   排序
    * @param string  $limit   取多少条数据
    * @param string  $group   分组
    * @param string  $having
    * @param boolean $lock    是否加锁
    * @return ArrayObject
    */
    static function find($tables,$where="",$fields='*',$order=null,$limit=null,$group=null,$having=null) {
                $sql = 'SELECT '.self::parseFields($fields)
                                                .' FROM '.$tables
                                                .self::parseWhere($where)
                                                .self::parseGroup($group)
                                                .self::parseHaving($having)
                                                .self::parseOrder($order)
                                                .self::parseLimit($limit);
               
                $dataAll = self::getAll($sql);
                if(count($dataAll)==1){$rlt=$dataAll[0];}else{$rlt=$dataAll;}
        return $rlt;
    }

    /**
     * 插入（单条）记录
     * @access function
     * @param mixed  $data   数据
     * @param string $table 数据表名
     * @return false | integer
     */
    static function add($data,$table) {
                //过滤提交数据
                $data=self::filterPost($table,$data);
                foreach ($data as $key=>$val){
                        if(is_array($val) && strtolower($val[0]) == 'exp') {
                                $val = $val[1];        // 使用表达式   ???
                        }elseif (is_scalar($val)){
                                $val = self::fieldFormat($val);
                        }else{
                                // 去掉复合对象
                                continue;
                        }
                        $data[$key]        = $val;
                }
        $fields = array_keys($data);
        array_walk($fields, array($this, 'addSpecialChar'));
        $fieldsStr = implode(',', $fields);
        $values = array_values($data);
        $valuesStr = implode(',', $values);
        $sql = 'INSERT INTO '.$table.' ('.$fieldsStr.') VALUES ('.$valuesStr.')';
                return self::execute($sql);
    }

        /**
    * 更新记录
    * @access function
    * @param mixed  $sets 数据
    * @param string $table  数据表名
    * @param string $where  更新条件
    * @param string $limit
    * @param string $order
    * @return false | integer
    */
    static function update($sets,$table,$where,$limit=0,$order='') {
                $sets = self::filterPost($table,$sets);
        $sql = 'UPDATE '.$table.' SET '.self::parseSets($sets).self::parseWhere($where).self::parseOrder($order).self::parseLimit($limit);
        return self::execute($sql);
    }
        
        /**
    * 保存某个字段的值
    * @access function
    * @param string $field 要保存的字段名
    * @param string $value  字段值
    * @param string $table  数据表
    * @param string $where 保存条件  
    * @param boolean $asString 字段值是否为字符串
    * @return void
    */
        static function setField($field, $value, $table, $condition="", $asString=false) {
        // 如果有'(' 视为 SQL指令更新 否则 更新字段内容为纯字符串
        if(false === strpos($value,'(') || $asString)  $value = '"'.$value.'"';
                $sql        = 'UPDATE '.$table.' SET '.$field.'='.$value.self::parseWhere($condition);
        return self::execute($sql);
        }

    /**
    * 删除记录
    * @access function
    * @param mixed  $where 为条件Map、Array或者String
    * @param string $table  数据表名
    * @param string $limit
    * @param string $order
    * @return false | integer
    */
    static function remove($where,$table,$limit='',$order='') {
        $sql = 'DELETE FROM '.$table.self::parseWhere($where).self::parseOrder($order).self::parseLimit($limit);
        return self::execute($sql);
    }

    /**
    +----------------------------------------------------------
    * 修改或保存数据(仅用于单表操作)
    * 有主键ID则为修改，无主键ID则为增加
    * 修改记录：
    +----------------------------------------------------------
    * @access function
    +----------------------------------------------------------
    * @param $tabName  表名
    * @param $aPost    提交表单的 $_POST
    * @param $priId    主键ID
    * @param $aNot     要排除的一个字段或数组
    * @param $aCustom  自定义的一个数组，附加到数据库中保存
    * @param $isExits  是否已经存在 存在：true, 不存在：false
    +----------------------------------------------------------
    * @return Boolean 修改或保存是否成功
    +----------------------------------------------------------
    */
        static function saveOrUpdate($tabName, $aPost, $priId="", $aNot="", $aCustom="", $isExits=false) {
                if(empty($tabName) || !is_array($aPost) || is_int($aNot)) return false;
                if(is_string($aNot) && !empty($aNot)) $aNot = array($aNot);
                if(is_array($aNot) && is_int(key($aNot))) $aPost = array_diff_key($aPost, array_flip($aNot));
                if(is_array($aCustom) && is_string(key($aCustom))) $aPost = array_merge($aPost,$aCustom);

                if (empty($priId) && !$isExits) { //新增
                        $aPost = array_filter($aPost, array($this, 'removeEmpty'));
            return self::add($aPost, $tabName);
                } else { //修改
            return self::update($aPost, $tabName, "id=".$priId);
                }
        }

    /**
    * 获取最近一次查询的sql语句
    * @access function
    * @param
    * @return String 执行的SQL
    */
        static function getLastSql() {
                $link = self::$link;               
        if ( !$link ) return false;
                return self::$queryStr;
        }

    /**
    * 获取最后插入的ID
    * @access function
    * @param
    * @return integer 最后插入时的数据ID
    */
        static function getLastInsId(){
                $link = self::$link;               
        if ( !$link ) return false;
                return self::$lastInsertId;
        }
        
        /**
    * 获取DB版本
    * @access function
    * @param
    * @return string
    */
    static function getDbVersion(){
        $link = self::$link;               
        if ( !$link ) return false;
                return self::$dbVersion;
    }

    /**
    * 取得数据库的表信息
    * @access function
    * @return array
    */
    static function getTables() {
        $info = array();
        if(self::query("SHOW TABLES")) {
            $result = self::getAll();
            foreach ($result as $key => $val) {
                $info[$key] = current($val);
            }
        }
        return $info;
    }

    /**
    * 取得数据表的字段信息
    * @access function
    * @return array
    */
    static function getFields($tableName) {
                // 获取数据库联接
                $link = self::$link;
        $sql = "SELECT
                           ORDINAL_POSITION ,COLUMN_NAME, COLUMN_TYPE, DATA_TYPE,
               IF(ISNULL(CHARACTER_MAXIMUM_LENGTH), (NUMERIC_PRECISION +  NUMERIC_SCALE), CHARACTER_MAXIMUM_LENGTH) AS MAXCHAR,
               IS_NULLABLE, COLUMN_DEFAULT, COLUMN_KEY, EXTRA, COLUMN_COMMENT
                        FROM
                           INFORMATION_SCHEMA.COLUMNS
                        WHERE
                          TABLE_NAME = :tabName AND TABLE_SCHEMA='".DB_NAME."'";
                self::$queryStr = sprintf($sql, $tableName);
        $sth = $link->prepare($sql);
        $sth->bindParam(':tabName', $tableName);
                $sth->execute();
                $result = $sth->fetchAll(constant('PDO::FETCH_ASSOC'));
        $info = array();
        foreach ($result as $key => $val) {
            $info[$val['COLUMN_NAME']] = array(
                'postion' => $val['ORDINAL_POSITION'],
                'name'    => $val['COLUMN_NAME'],
                'type'    => $val['COLUMN_TYPE'],
                'd_type'  => $val['DATA_TYPE'],
                'length'  => $val['MAXCHAR'],
                'notnull' => (strtolower($val['IS_NULLABLE']) == "no"),
                'default' => $val['COLUMN_DEFAULT'],
                'primary' => (strtolower($val['COLUMN_KEY']) == 'pri'),
                'autoInc' => (strtolower($val['EXTRA']) == 'auto_increment'),
                'comment' => $val['COLUMN_COMMENT']
            );
        }

        // 有错误则抛出异常
        self::haveErrorThrowException();

        return $info;
    }



    /**
    * 关闭数据库
    * @access function
    */
    static  function close() {
        self::$link = null;
    }

    /**
    * SQL指令安全过滤
    * @access function
    * @param string $str  SQL指令
    * @return string
    */
    static  function escape_string($str) {
                return addslashes($str);
    }

    /*********************************************************************************************************/
    /*                                             内部操作方法                                              */
    /*********************************************************************************************************/
    /**
    * 有出错抛出异常
    * @access function
    * @return
    */
    static  function haveErrorThrowException() {
                $obj = empty(self::$PDOStatement) ? self::$link : self::$PDOStatement;
        $arrError = $obj->errorInfo();
        if(count($arrError) > 1) { // 有错误信息
                        //$this->rollback();
            self::$error = $arrError[2]. "<br/><br/> [ SQL语句 ] : ".self::$queryStr;
            //throw_exception($this->error);
            throw_exception(self::$error);
            return false;
        }
                //主要针对execute()方法抛出异常
                if(self::$queryStr=='')throw_exception('Query was empty<br/><br/>[ SQL语句 ] :');

    }
   
    /**
    * where分析
    * @access function
    * @param mixed $where 查询条件
    * @return string
    */
        static  function parseWhere($where) {
                $whereStr = '';
                if(is_string($where) || is_null($where)) {
            $whereStr = $where;
        }
        return empty($whereStr)?'':' WHERE '.$whereStr;
        }

    /**
    * order分析
    * @access function
    * @param mixed $order 排序
    * @return string
    */
    static function parseOrder($order) {
        $orderStr = '';
        if(is_array($order))
            $orderStr .= ' ORDER BY '.implode(',', $order);
        else if(is_string($order) && !empty($order))
            $orderStr .= ' ORDER BY '.$order;
        return $orderStr;
        }


    /**
    * limit分析
    * @access function
    * @param string $limit
    * @return string
    */
    static function parseLimit($limit) {
        $limitStr = '';
        if(is_array($limit)) {
            if(count($limit)>1)
                $limitStr .= ' LIMIT '.$limit[0].' , '.$limit[1].' ';
            else
                $limitStr .= ' LIMIT '.$limit[0].' ';
        } else if(is_string($limit) && !empty($limit))  {
            $limitStr .= ' LIMIT '.$limit.' ';
        }
        return $limitStr;
        }

    /**
    * group分析
    * @access function
    * @param mixed $group
    * @return string
    */
    static function parseGroup($group) {
        $groupStr = '';
        if(is_array($group))
            $groupStr .= ' GROUP BY '.implode(',', $group);
        else if(is_string($group) && !empty($group))
            $groupStr .= ' GROUP BY '.$group;
        return empty($groupStr)?'':$groupStr;
        }

    /**
    * having分析
    * @access function
    * @param string $having
    * @return string
    */
    static function parseHaving($having)  {
        $havingStr = '';
        if(is_string($having) && !empty($having))
            $havingStr .= ' HAVING '.$having;
        return $havingStr;
    }

    /**
    * fields分析
    * @access function
    * @param mixed $fields
    * @return string
    */
    static function parseFields($fields) {               
            if(is_array($fields)) {
            array_walk($fields, array($this, 'addSpecialChar'));
            $fieldsStr = implode(',', $fields);
        }else if(is_string($fields) && !empty($fields)) {
            if( false === strpos($fields,'`') ) {
                $fields = explode(',',$fields);
                array_walk($fields, array($this, 'addSpecialChar'));
                $fieldsStr = implode(',', $fields);
            }else {
                    $fieldsStr = $fields;
            }
        }else $fieldsStr = '*';
        return $fieldsStr;
        }

    /**
    * sets分析,在更新数据时调用
    * @access function
    * @param mixed $values
    * @return string
    */
    private function parseSets($sets) {
        $setsStr  = '';
        if(is_array($sets)){
            foreach ($sets as $key=>$val){
                                $key = self::addSpecialChar($key);
                                $val = self::fieldFormat($val);
                                $setsStr .= "$key = ".$val.",";
            }
            $setsStr = substr($setsStr,0,-1);
        }else if(is_string($sets)) {
            $setsStr = $sets;
        }
        return $setsStr;
        }

    /**
    * 字段格式化
    * @access function
    * @param mixed $value
    * @return mixed
    */
    static function fieldFormat(&$value) {
        if(is_int($value)) {
            $value = intval($value);
        } else if(is_float($value)) {
            $value = floatval($value);
        } elseif(preg_match('/^\(\w*(\+|\-|\*|\/)?\w*\)$/i',$value)){
                        // 支持在字段的值里面直接使用其它字段
                        // 例如 (score+1) (name) 必须包含括号
                        $value = $value;
                }else if(is_string($value)) {
            $value = '\''.self::escape_string($value).'\'';
        }
        return $value;
        }

    /**
    * 字段和表名添加` 符合
    * 保证指令中使用关键字不出错 针对mysql
    * @access function
    * @param mixed $value
    * @return mixed
    */
    static function addSpecialChar(&$value) {
        if( '*' == $value ||  false !== strpos($value,'(') || false !== strpos($value,'.') || false !== strpos($value,'`')) {
            //如果包含* 或者 使用了sql方法 则不作处理
        } elseif(false === strpos($value,'`') ) {
            $value = '`'.trim($value).'`';
        }
        return $value;
    }

    /**
    +----------------------------------------------------------
    * 去掉空元素
    +----------------------------------------------------------
    * @access function
    +----------------------------------------------------------
    * @param mixed $value
    +----------------------------------------------------------
    * @return mixed
    +----------------------------------------------------------
    */
    static function removeEmpty($value){
        return !empty($value);
    }

        /**
    * 执行查询 主要针对 SELECT, SHOW 等指令
    * @access function
    * @param string $sql sql指令
    * @return mixed
    */
    static  function query($sql='') {
                // 获取数据库联接
                $link = self::$link;               
        if ( !$link ) return false;
        
        self::$queryStr = $sql;
        //释放前次的查询结果
        if ( !empty(self::$PDOStatement) ) self::free();
        self::$PDOStatement = $link->prepare(self::$queryStr);
        $bol = self::$PDOStatement->execute();
        // 有错误则抛出异常
        self::haveErrorThrowException();
        return $bol;
    }
    /**
    * 数据库操作方法
    * @access function
    * @param string  $sql  执行语句
    * @param boolean $lock  是否锁定(默认不锁定)
    * @return void
   
    public function execute($sql='',$lock=false) {
        if(empty($sql)) $sql  = $this->queryStr;
        return $this->_execute($sql);
    }*/
   
        /**
    * 执行语句 针对 INSERT, UPDATE 以及DELETE
    * @access function
    * @param string $sql  sql指令
    * @return integer
    */
    static function execute($sql='') {
                // 获取数据库联接
                $link = self::$link;
        if ( !$link ) return false;
        self::$queryStr = $sql;
        //释放前次的查询结果
        if ( !empty(self::$PDOStatement) ) self::free();
                $result = $link->exec(self::$queryStr);
        // 有错误则抛出异常
        self::haveErrorThrowException();
        if ( false === $result) {
            return false;
        } else {
                        self::$numRows = $result;
            self::$lastInsertId = $link->lastInsertId();
            return self::$numRows;
        }
    }

        /**
    * 是否为数据库更改操作
    * @access private
    * @param string $query  SQL指令
    * @return boolen 如果是查询操作返回false
    */
    static function isMainIps($query) {
        $queryIps = 'INSERT|UPDATE|DELETE|REPLACE|CREATE|DROP|LOAD DATA|SELECT .* INTO|COPY|ALTER|GRANT|REVOKE|LOCK|UNLOCK';
        if (preg_match('/^\s*"?(' . $queryIps . ')\s+/i', $query)) {
            return true;
        }
        return false;
    }

        /**
     * 过滤POST提交数据
     * @access private
     * @param mixed  $data   POST提交数据
     * @param string $table 数据表名
     * @return  mixed $newdata
     */
     static function filterPost($table,$data) {
                $table_column = self::getFields($table);
                $newdata=array();
                foreach ($table_column as $key=>$val){
                        if(array_key_exists($key,$data) && ($data[$key])!==''){
                                $newdata[$key] = $data[$key];
                        }
                }
                return $newdata;
     }

        /**
    * 启动事务
    * @access function
    * @return void
    */
        static function startTrans() {
               
        //数据rollback 支持
                $link = self::$link;
        if ( !$link ) return false;
                if (self::$transTimes == 0) {
                        $link->beginTransaction();
                }
                self::$transTimes++;
                return ;
        }

    /**
    * 用于非自动提交状态下面的查询提交
    * @access function
    * @return boolen
    */
    static function commit() {
               
                $link = self::$link;
        if ( !$link ) return false;

        if (self::$transTimes > 0) {
            $result = $link->commit();
            self::$transTimes = 0;
            if(!$result){
                throw_exception(self::$error());
                return false;
            }
        }
        return true;
    }

    /**
    * 事务回滚
    * @access function
    * @return boolen
    */
    public function rollback() {

                $link = self::$link;
        if ( !$link ) return false;

        if (self::$transTimes > 0) {
            $result = $link->rollback();
            self::$transTimes = 0;
            if(!$result){
                throw_exception(self::$error());
                return false;
            }
        }
        return true;
    }

}
?>

这个类不是可以下载得到吗

这个类好长哦

请问可以在哪里下载得到.

还行

有点意思，还有不少地方可以优化

!!

有地方可以下载没？？楼主。提供源码，谢谢啦，这个复制 过去有问题啦!!

这东西怎么用啊，这么长，我晕死!