首页 > php技术 > php文章> 详细内容
php文章 > 正文
全新的PDO数据库操作类php版(仅适用Mysql)
日期:2018-02-16 

/**
 * 作者:胡睿
* 日期: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();