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