全新的PDO数据库操作类(仅适用Mysql)
1年前,也差不多刚开博那会,分享过一个pdo的数据库操作类(可参见:http://www.cnblogs.com/hooray/archive/2011/06/30/2094743.html),与其说是类,其实就只是几个封装好的函数,整体略显稚嫩,但也是这么个东西,在公司里也用了1年之久。如今公司规模变大了,产品也日益完善,曾经的那个数据库操作函数虽说使用上没出什么大问题,但为了更显专业,花了1天时间重写了这个,现在,它确实是个类了。
/** * 作者:胡睿 * 日期:2012/07/21 * 电邮:hooray0905@foxmail.com */ class HRDB{ protected $pdo; protected $res; protected $config; /*构造函数*/ function __construct($config){ $this->Config = $config; $this->connect(); } /*数据库连接*/ public function connect(){ $this->pdo = new PDO($this->Config['dsn'], $this->Config['name'], $this->Config['password']); $this->pdo->query('set names utf8;'); //把结果序列化成stdClass //$this->pdo->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_OBJ); //自己写代码捕获Exception $this->pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION); } /*数据库关闭*/ public function close(){ $this->pdo = null; } public function query($sql){ $res = $this->pdo->query($sql); if($res){ $this->res = $res; } } public function exec($sql){ $res = $this->pdo->exec($sql); if($res){ $this->res = $res; } } public function fetchAll(){ return $this->res->fetchAll(); } public function fetch(){ return $this->res->fetch(); } public function fetchColumn(){ return $this->res->fetchColumn(); } public function lastInsertId(){ return $this->res->lastInsertId(); } /** * 参数说明 * int $debug 是否开启调试,开启则输出sql语句 * 0 不开启 * 1 开启 * 2 开启并终止程序 * int $mode 返回类型 * 0 返回多条记录 * 1 返回单条记录 * 2 返回行数 * string/array $table 数据库表,两种传值模式 * 普通模式: * 'tb_member, tb_money' * 数组模式: * array('tb_member', 'tb_money') * string/array $fields 需要查询的数据库字段,允许为空,默认为查找全部,两种传值模式 * 普通模式: * 'username, password' * 数组模式: * array('username', 'password') * string/array $sqlwhere 查询条件,允许为空,两种传值模式 * 普通模式: * 'and type = 1 and username like "%os%"' * 数组模式: * array('type = 1', 'username like "%os%"') * string $orderby 排序,默认为id倒序 */ public function select($debug, $mode, $table, $fields="*", $sqlwhere="", $orderby="tbid desc"){ //参数处理 if(is_array($table)){ $table = implode(', ', $table); } if(is_array($fields)){ $fields = implode(', ', $fields); } if(is_array($sqlwhere)){ $sqlwhere = ' and '.implode(' and ', $sqlwhere); } //数据库操作 if($debug === 0){ if($mode === 2){ $this->query("select count(tbid) from $table where 1=1 $sqlwhere"); $return = $this->fetchColumn(); }else if($mode === 1){ $this->query("select $fields from $table where 1=1 $sqlwhere order by $orderby"); $return = $this->fetch(); }else{ $this->query("select $fields from $table where 1=1 $sqlwhere order by $orderby"); $return = $this->fetchAll(); } return $return; }else{ if($mode === 2){ echo "select count(tbid) from $table where 1=1 $sqlwhere"; }else if($mode === 1){ echo "select $fields from $table where 1=1 $sqlwhere order by $orderby"; } else{ echo "select $fields from $table where 1=1 $sqlwhere order by $orderby"; } if($debug === 2){ exit; } } } /** * 参数说明 * int $debug 是否开启调试,开启则输出sql语句 * 0 不开启 * 1 开启 * 2 开启并终止程序 * int $mode 返回类型 * 0 无返回信息 * 1 返回执行条目数 * 2 返回最后一次插入记录的id * string/array $table 数据库表,两种传值模式 * 普通模式: * 'tb_member, tb_money' * 数组模式: * array('tb_member', 'tb_money') * string/array $set 需要插入的字段及内容,两种传值模式 * 普通模式: * 'username = "test", type = 1, dt = now()' * 数组模式: * array('username = "test"', 'type = 1', 'dt = now()') */ public function insert($debug, $mode, $table, $set){ //参数处理 if(is_array($table)){ $table = implode(', ', $table); } if(is_array($set)){ $set = implode(', ', $set); } //数据库操作 if($debug === 0){ if($mode === 2){ $this->query("insert into $table set $set"); $return = $this->lastInsertId(); }else if($mode === 1){ $this->exec("insert into $table set $set"); $return = $this->res; }else{ $this->query("insert into $table set $set"); $return = NULL; } return $return; }else{ echo "insert into $table set $set"; if($debug === 2){ exit; } } } /** * 参数说明 * int $debug 是否开启调试,开启则输出sql语句 * 0 不开启 * 1 开启 * 2 开启并终止程序 * int $mode 返回类型 * 0 无返回信息 * 1 返回执行条目数 * string $table 数据库表,两种传值模式 * 普通模式: * 'tb_member, tb_money' * 数组模式: * array('tb_member', 'tb_money') * string/array $set 需要更新的字段及内容,两种传值模式 * 普通模式: * 'username = "test", type = 1, dt = now()' * 数组模式: * array('username = "test"', 'type = 1', 'dt = now()') * string/array $sqlwhere 修改条件,允许为空,两种传值模式 * 普通模式: * 'and type = 1 and username like "%os%"' * 数组模式: * array('type = 1', 'username like "%os%"') */ public function update($debug, $mode, $table, $set, $sqlwhere=""){ //参数处理 if(is_array($table)){ $table = implode(', ', $table); } if(is_array($set)){ $set = implode(', ', $set); } if(is_array($sqlwhere)){ $sqlwhere = ' and '.implode(' and ', $sqlwhere); } //数据库操作 if($debug === 0){ if($mode === 1){ $this->exec("update $table set $set where 1=1 $sqlwhere"); $return = $this->res; }else{ $this->query("update $table set $set where 1=1 $sqlwhere"); $return = NULL; } return $return; }else{ echo "update $table set $set where 1=1 $sqlwhere"; if($debug === 2){ exit; } } } /** * 参数说明 * int $debug 是否开启调试,开启则输出sql语句 * 0 不开启 * 1 开启 * 2 开启并终止程序 * int $mode 返回类型 * 0 无返回信息 * 1 返回执行条目数 * string $table 数据库表 * string/array $sqlwhere 删除条件,允许为空,两种传值模式 * 普通模式: * 'and type = 1 and username like "%os%"' * 数组模式: * array('type = 1', 'username like "%os%"') */ public function delete($debug, $mode, $table, $sqlwhere=""){ //参数处理 if(is_array($sqlwhere)){ $sqlwhere = ' and '.implode(' and ', $sqlwhere); } //数据库操作 if($debug === 0){ if($mode === 1){ $this->exec("delete from $table where 1=1 $sqlwhere"); $return = $this->res; }else{ $this->query("delete from $table where 1=1 $sqlwhere"); $return = NULL; } return $return; }else{ echo "delete from $table where 1=1 $sqlwhere"; if($debug === 2){ exit; } } } }
其实使用上,和之前的相差不大,目的就是为了方便移植。
本次重写着重处理了几个问题:
① insert语句太复杂,fields与values对应容易出现误差
我们看下最常见的一句sql插入语句
insert into tb_member (username, type, dt) values ('test', 1, now())
在传统模式下,fields和values参数是分开传入的,但却要保证两者参数传入的顺序一致。这很容易导致顺序错乱或者漏传某个参数。
这次已经把问题修改了,采用了mysql独有的insert语法,同样是上面那功能,就可以换成这样的写法
insert into tb_member set username = "test", type = 1, lastlogindt = now()
就像update一样,一目了然。
② 部分参数可以用数组代替
比如这样一句sql
delete from tb_member where 1=1 and tbid = 1 and username = "hooray"
在原先调用方法的时候,需要手动拼装好where条件,这样操作的成本很高,现在完全可以用这种形式
$where = array( 'tbid = 1', 'username = "hooray"' ); $db->delete(1, 0, 'tb_member', $where);
条件再多也不会打乱你的思路。同样,不仅仅是where参数,update里的set也可以以这种形式(具体可参见完整源码)
$set = array('username = "123"', 'type = 1', 'lastlogindt = now()'); $where = array('tbid = 1'); $db->update(1, 0, 'tb_member', $set, $where);
③ 可自定义sql语句
有时候,sql过于复杂,导致无法使用类里提供的方法去组装sql语句,这时候就需要一个功能,就是能直接传入我已经组装好的sql语句执行,并返回信息。现在,这功能也有了
$db->query('select username, password from tb_member'); $rs = $db->fetchAll();
是不是很像pdo原生态的写法?
④ 支持创建多数据库连接
原先的因为只是数据库操作方法,所以并不支持多数据库连接,在实现上需要复制出2个相同的文件,修改部分变量,操作实属复杂。现在这问题也解决了。
$db_hoorayos_config = array( 'dsn'=>'mysql:host=localhost;dbname=hoorayos', 'name'=>'root', 'password'=>'hooray' ); $db = new HRDB($db_hoorayos_config); $db_hoorayos_config2 = array( 'dsn'=>'mysql:host=localhost;dbname=hoorayos2', 'name'=>'root', 'password'=>'hooray' ); $db2 = new HRDB($db_hoorayos_config2);
这样就能同时创建2个数据库连接,方便处理数据库与数据库交互的情况。
大致新功能就是这么多了,整个代码并不多,欢迎阅读了解。下面是我在编写时写的测试代码,也一并提供上来,方便大家学习。
require_once('global.php'); require_once('inc/setting.inc.php'); $db = new HRDB($db_hoorayos_config); echo '<hr><b>select测试</b><hr>'; echo '普通模式,直接字符串传入<br>'; $rs = $db->select(1, 0, 'tb_member', 'username, password', 'and type = 1 and username like "%os%"'); echo '<br>数组模式,可传入数组<br>'; $fields = array('username', 'password'); $where = array('type = 1', 'username like "%os%"'); $rs = $db->select(1, 0, 'tb_member', $fields, $where); echo '<hr><b>insert测试</b><hr>'; echo '普通模式,直接字符串传入<br>'; $db->insert(1, 0, 'tb_member', 'username = "test", type = 1, lastlogindt = now()'); echo '<br>数组模式,可传入数组<br>'; $set = array('username = "test"', 'type = 1', 'lastlogindt = now()'); $db->insert(1, 0, 'tb_member', $set); echo '<hr><b>update测试</b><hr>'; echo '普通模式,直接字符串传入<br>'; $db->update(1, 0, 'tb_member', 'username = "123", type = 1, lastlogindt = now()', 'and tbid = 7'); echo '<br>数组模式,可传入数组<br>'; $set = array('username = "123"', 'type = 1', 'lastlogindt = now()'); $where = array('tbid = 1'); $db->update(1, 0, 'tb_member', $set, $where); echo '<hr><b>delete测试</b><hr>'; echo '普通模式,直接字符串传入<br>'; $db->delete(1, 0, 'tb_member', 'and tbid = 1 and username = "hooray"'); echo '<br>数组模式,可传入数组<br>'; $where = array( 'tbid = 1', 'username = "hooray"' ); $db->delete(1, 0, 'tb_member', $where); echo '<hr><b>自定义sql</b><hr>'; $db->query('select username, password from tb_member'); $rs = $db->fetchAll(); var_dump($rs); $db->close();
hooray.github.io