zoukankan      html  css  js  c++  java
  • PHP数据库基于PDO操作类(mysql)

    这是网上找的关于Mysql的操作类,非常适合初学者使用

    <?php
    
    class Mysql {
    
            protected static $_dbh = null; //静态属性,所有数据库实例共用,避免重复连接数据库
            protected $_dbType = 'mysql';
            protected $_pconnect = true; //是否使用长连接
            protected $_host = 'localhost';
            protected $_port = 3306;
            protected $_user = 'root';
            protected $_pass = '';
            protected $_dbName = null; //数据库名
            protected $_sql = false; //最后一条sql语句
            protected $_where = '';
            protected $_order = '';
            protected $_limit = '';
            protected $_field = '*';
            protected $_clear = 0; //状态,0表示查询条件干净,1表示查询条件污染
            protected $_trans = 0; //事务指令数 
    
            /**
             * 初始化类
             * @param array $conf 数据库配置
             */
            public function __construct(array $conf) {
                class_exists('PDO') or die("PDO: class not exists.");
                $this->_host = $conf['host'];
                $this->_port = $conf['port'];
                $this->_user = $conf['user'];
                $this->_pass = $conf['password'];
                $this->_dbName = $conf['dbname'];
                //连接数据库
                if ( is_null(self::$_dbh) ) {
                    $this->_connect();
                }
            }
    
            /**
             * 连接数据库的方法
             */
            protected function _connect() {
                $dsn = $this->_dbType.':host='.$this->_host.';port='.$this->_port.';dbname='.$this->_dbName;
                //持久化连接
                $options = $this->_pconnect ? array(PDO::ATTR_PERSISTENT=>true) : array();
                try { 
                    $dbh = new PDO($dsn, $this->_user, $this->_pass, $options);
                    $dbh->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);  //设置如果sql语句执行错误则抛出异常,事务会自动回滚
                    $dbh->setAttribute(PDO::ATTR_EMULATE_PREPARES, false); //禁用prepared statements的仿真效果(防SQL注入)
                } catch (PDOException $e) { 
                    die('Connection failed: ' . $e->getMessage());
                }
                $dbh->exec('SET NAMES utf8');
                self::$_dbh = $dbh;
            }
    
            /** 
            * 字段和表名添加 `符号
            * 保证指令中使用关键字不出错 针对mysql 
            * @param string $value 
            * @return string 
            */
            protected function _addChar($value) { 
                if ('*'==$value || false!==strpos($value,'(') || false!==strpos($value,'.') || false!==strpos($value,'`')) { 
                    //如果包含* 或者 使用了sql方法 则不作处理 
                } elseif (false === strpos($value,'`') ) { 
                    $value = '`'.trim($value).'`';
                } 
                return $value; 
            }
    
            /** 
            * 取得数据表的字段信息 
            * @param string $tbName 表名
            * @return array 
            */
            protected function _tbFields($tbName) {
                $sql = 'SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME="'.$tbName.'" AND TABLE_SCHEMA="'.$this->_dbName.'"';
                $stmt = self::$_dbh->prepare($sql);
                $stmt->execute();
                $result = $stmt->fetchAll(PDO::FETCH_ASSOC);
                $ret = array();
                foreach ($result as $key=>$value) {
                    $ret[$value['COLUMN_NAME']] = 1;
                }
                return $ret;
            }
    
            /** 
            * 过滤并格式化数据表字段
            * @param string $tbName 数据表名 
            * @param array $data POST提交数据 
            * @return array $newdata 
            */
            protected function _dataFormat($tbName,$data) {
                if (!is_array($data)) return array();
                $table_column = $this->_tbFields($tbName);
                $ret=array();
                foreach ($data as $key=>$val) {
                    if (!is_scalar($val)) continue; //值不是标量则跳过
                    if (array_key_exists($key,$table_column)) {
                        $key = $this->_addChar($key);
                        if (is_int($val)) { 
                            $val = intval($val); 
                        } elseif (is_float($val)) { 
                            $val = floatval($val); 
                        } elseif (preg_match('/^(w*(+|-|*|/)?w*)$/i', $val)) {
                            // 支持在字段的值里面直接使用其它字段 ,例如 (score+1) (name) 必须包含括号
                            $val = $val;
                        } elseif (is_string($val)) { 
                            //将字符串中的单引号(')、双引号(")、反斜线()与 NUL(NULL 字符转义
                            $val = '"'.addslashes($val).'"';
                        }
                        $ret[$key] = $val;
                    }
                }
                return $ret;
            }
    
            /**
            * 执行查询 主要针对 SELECT, SHOW 等指令
            * @param string $sql sql指令 
            * @return mixed 
            */
            protected function _doQuery($sql='') {
                $this->_sql = $sql;
                $pdostmt = self::$_dbh->prepare($this->_sql); //prepare或者query 返回一个PDOStatement
                $pdostmt->execute();
                $result = $pdostmt->fetchAll(PDO::FETCH_ASSOC);
                return $result;
            }
    
            /** 
            * 执行语句 针对 INSERT, UPDATE 以及DELETE,exec结果返回受影响的行数
            * @param string $sql sql指令 
            * @return integer 
            */
            protected function _doExec($sql='') {
                $this->_sql = $sql;
                return self::$_dbh->exec($this->_sql);
            }
    
            /** 
            * 执行sql语句,自动判断进行查询或者执行操作 
            * @param string $sql SQL指令 
            * @return mixed 
            */
            public function doSql($sql='') {
                $queryIps = 'INSERT|UPDATE|DELETE|REPLACE|CREATE|DROP|LOAD DATA|SELECT .* INTO|COPY|ALTER|GRANT|REVOKE|LOCK|UNLOCK'; 
                if (preg_match('/^s*"?(' . $queryIps . ')s+/i', $sql)) { 
                    return $this->_doExec($sql);
                }
                else {
                    //查询操作
                    return $this->_doQuery($sql);
                }
            }
    
            /** 
            * 获取最近一次查询的sql语句 
            * @return String 执行的SQL 
            */
            public function getLastSql() { 
                return $this->_sql;
            }
    
            /**
             * 插入方法
             * @param string $tbName 操作的数据表名
             * @param array $data 字段-值的一维数组
             * @return int 受影响的行数
             */
            public function insert($tbName,array $data){
                $data = $this->_dataFormat($tbName,$data);
                if (!$data) return;
                $sql = "insert into ".$tbName."(".implode(',',array_keys($data)).") values(".implode(',',array_values($data)).")";
                return $this->_doExec($sql);
            }
    
            /**
             * 删除方法
             * @param string $tbName 操作的数据表名
             * @return int 受影响的行数
             */
            public function delete($tbName) {
                //安全考虑,阻止全表删除
                if (!trim($this->_where)) return false;
                $sql = "delete from ".$tbName." ".$this->_where;
                $this->_clear = 1;
                $this->_clear();
                return $this->_doExec($sql);
            }
    
            /**
             * 更新函数
             * @param string $tbName 操作的数据表名
             * @param array $data 参数数组
             * @return int 受影响的行数
             */
            public function update($tbName,array $data) {
                //安全考虑,阻止全表更新
                if (!trim($this->_where)) return false;
                $data = $this->_dataFormat($tbName,$data);
                if (!$data) return;
                $valArr = '';
                foreach($data as $k=>$v){
                    $valArr[] = $k.'='.$v;
                }
                $valStr = implode(',', $valArr);
                $sql = "update ".trim($tbName)." set ".trim($valStr)." ".trim($this->_where);
                return $this->_doExec($sql);
            }
    
            /**
             * 查询函数
             * @param string $tbName 操作的数据表名
             * @return array 结果集
             */
            public function select($tbName='') {
                $sql = "select ".trim($this->_field)." from ".$tbName." ".trim($this->_where)." ".trim($this->_order)." ".trim($this->_limit);
                //echo $sql;
                $this->_clear = 1;
                $this->_clear();
                return $this->_doQuery(trim($sql));
            }
    
            /**
             * @param mixed $option 组合条件的二维数组,例:$option['field1'] = array(1,'=>','or')
             * @return $this
             */
            public function where($option) {
                if ($this->_clear>0) $this->_clear();
                $this->_where = ' where ';
                $logic = 'and';
                if (is_string($option)) {
                    $this->_where .= $option;
                }
                elseif (is_array($option)) {
                    foreach($option as $k=>$v) {
                        if (is_array($v)) {
                            $relative = isset($v[1]) ? $v[1] : '=';
                            $logic    = isset($v[2]) ? $v[2] : 'and';
                            $condition = ' ('.$this->_addChar($k).' '.$relative.' '.$v[0].') ';
                        }
                        else {
                            $logic = 'and';
                            $condition = ' ('.$this->_addChar($k).'='.$v.') ';
                        }
                        $this->_where .= isset($mark) ? $logic.$condition : $condition;
                        $mark = 1;
                    }
                }
                return $this;
            }
    
            /**
             * 设置排序
             * @param mixed $option 排序条件数组 例:array('sort'=>'desc')
             * @return $this
             */
            public function order($option) {
                if ($this->_clear>0) $this->_clear();
                $this->_order = ' order by ';
                if (is_string($option)) {
                    $this->_order .= $option;
                }
                elseif (is_array($option)) {
                    foreach($option as $k=>$v){
                        $order = $this->_addChar($k).' '.$v;
                        $this->_order .= isset($mark) ? ','.$order : $order;
                        $mark = 1;
                    }
                }
                return $this;
            }
    
            /**
             * 设置查询行数及页数
             * @param int $page pageSize不为空时为页数,否则为行数
             * @param int $pageSize 为空则函数设定取出行数,不为空则设定取出行数及页数
             * @return $this
             */
            public function limit($page,$pageSize=null) {
                if ($this->_clear>0) $this->_clear();
                if ($pageSize===null) {
                    $this->_limit = "limit ".$page;
                }
                else {
                    $pageval = intval( ($page - 1) * $pageSize);
                    $this->_limit = "limit ".$pageval.",".$pageSize;
                }
                return $this;
            }
    
            /**
             * 设置查询字段
             * @param mixed $field 字段数组
             * @return $this
             */
            public function field($field){
                if ($this->_clear>0) $this->_clear();
                if (is_string($field)) {
                    $field = explode(',', $field);
                }
                $nField = array_map(array($this,'_addChar'), $field);
                $this->_field = implode(',', $nField);
                return $this;
            }
    
            /**
             * 清理标记函数
             */
            protected function _clear() {
                $this->_where = '';
                $this->_order = '';
                $this->_limit = '';
                $this->_field = '*';
                $this->_clear = 0;
            }
    
            /**
             * 手动清理标记
             * @return $this
             */
            public function clearKey() {
                $this->_clear();
                return $this;
            }
    
            /**
            * 启动事务 
            * @return void 
            */
            public function startTrans() { 
                //数据rollback 支持 
                if ($this->_trans==0) self::$_dbh->beginTransaction();
                $this->_trans++; 
                return; 
            }
    
            /** 
            * 用于非自动提交状态下面的查询提交 
            * @return boolen 
            */
            public function commit() {
                $result = true;
                if ($this->_trans>0) { 
                    $result = self::$_dbh->commit(); 
                    $this->_trans = 0;
                } 
                return $result;
            }
    
            /** 
            * 事务回滚 
            * @return boolen 
            */
            public function rollback() {
                $result = true;
                if ($this->_trans>0) {
                    $result = self::$_dbh->rollback();
                    $this->_trans = 0;
                }
                return $result;
            }
    
            /**
            * 关闭连接
            * PHP 在脚本结束时会自动关闭连接。
            */
            public function close() {
                if (!is_null(self::$_dbh)) self::$_dbh = null;
            }
    
    }
  • 相关阅读:
    第十八章 大浏览量系统的静态化架构设计(待续)
    第十七章 Velocity优化实践(待续)
    第十六章 Velocity工作原理解析(待续)
    第十五章 深入分析iBatis框架之系统架构与映射原理(待续)
    第十四章 Spring MVC的工作机制与设计模式(待续)
    第十三章 Spring框架的设计理念与设计模式分析(待续)
    第十二章 Jetty的工作原理解析(待续)
    第十一章 Tomcat的系统架构与设计模式(待续)
    Luogu P1137 旅行计划 【拓扑排序+Dp】By cellur925
    [USACO4.1]麦香牛块Beef McNuggets By cellur925
  • 原文地址:https://www.cnblogs.com/xiaoliwang/p/7963471.html
Copyright © 2011-2022 走看看