全新的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();