/** - * 数据库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')) self::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'])) self::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) {
- self::throw_exception($e->getMessage());
- }
- if(!self::$link) {
- self::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) {
- if($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) {
- if($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
- */
- 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 修改或保存是否成功
- +----------------------------------------------------------
- */
- 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($arrError[0] !== '00000') { // 有错误信息
- self::$error = $arrError[0]."|".$arrError[2]. "
[ SQL ] : ".self::$queryStr." "; - self::throw_exception(self::$error);
- return false;
- }
- //主要针对execute()方法抛出异常
- if(self::$queryStr=='')self::throw_exception('Query was empty
[ 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
- */
- 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){
- self::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){
- self::throw_exception(self::$error());
- return false;
- }
- }
- return true;
- }
/** - * 错误处理
- * @access function
- * @return void
- */
- static function throw_exception($err){
- echo 'ERROR:'.$err.'';
- }
- }
|