自寫的使用PDO對mysql資料庫的增刪改查操作類

志遠電子發表於2019-02-16

最近在做專案時用到了PDO操作mysql資料庫,於是自己寫了一個類檔案,命名為mysql_class.php檔案程式碼如下:

<?php
class mysql{
    //常量宣告
    const DSN = "mysql:host=[資料庫地址];dbname=[資料庫名];charset=utf8";//資料庫地址與資料庫名及編碼
    const USERNAME = "[資料庫使用者名稱]";//使用者名稱
    const PASSWD = "[資料庫密碼]";//密碼
    
    //私有變數宣告
    private $sql = NULL;//sql語句快取
    private $link = NULL;//資料庫連線
    private $result = NULL;//結果
    
    /*******************************************************************************
     * @ 名稱:建立連線
     * @ 屬性:私有
    *******************************************************************************/
    private function connect(){
        try {
            $this->link = new PDO(self::DSN, self::USERNAME, self::PASSWD, array(PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION));//建立連線
        }
        catch (PDOException $e) {
            die ("Error!:".$e->getMessage()."<hr/>");
        }
    }
    
    /*******************************************************************************
     * @ 名稱:執行sql語句
     * @ 屬性:私有
     * @ 入口引數:無
     * @ 出口引數:執行成功返回真,否則返回假,查詢語句儲存結果集陣列
    *******************************************************************************/
    private function sql(){
        try {
            $this->connect();
            $this->link->beginTransaction();//開啟一個事務
            $prepare = $this->link->prepare($this->sql);//準備查詢語句
            $prepare->execute();//執行查詢語句並返回結果集
            $cmd = strtolower(substr(trim($this->sql),0,6));//擷取命令字元
            if($cmd == "select"){
                $array = $prepare->fetch(PDO::FETCH_ASSOC);//獲取結果集中的所有資料
                if(count($array)){
                    $this->result = NULL;
                    $this->result = $array;//儲存結果集
                    return true;//查詢到結果返回真
                }else{
                    return false;//否則返回假
                }
            }else if($cmd == "insert" || $cmd == "delete" || $cmd == "update"){
                if($prepare){
                    return true;//執行成功返回真
                }else{
                    return false;//否則返回假
                }
            }
            $this->link->commit(); //如果正確執行完成那麼確認commit
        } catch (PDOException $e) {
            $this->link->rollBack();//如果執行中有錯誤的情況下回滾
            die ("Error!:".$e->getMessage()."<hr/>");
        }
    }
    
    /*******************************************************************************
     * @ 名稱:sql語句處理
     * @ 屬性:公有
     * @ 入口引數:cmd增刪改查字元命令;dsname資料表名;first第一個引數;second第二個引數;
     * @ 出口引數:執行成功返回真,否則返回假,查詢操作返回結果集陣列
     * @ 使用示例:
        $mysql->handle("insert","abc","openid,nickname","`123`,`abc`");//增加
        $mysql->handle("delete","abc","openid=`123`");//刪除
        $mysql->handle("update","abc","nickname=`def`","openid=`123`");//更新
        $res = $mysql->handle("select","abc","*","openid=`123`");//查詢
        if(is_array($res) == true){
            foreach($res as $key=>$val){
                echo $key."=".$val."<hr>";
            }
        }//遍歷查詢結果陣列
    *******************************************************************************/
    public function handle($cmd,$dsname,$first,$second=NULL){
        switch($cmd){
            case `insert`://插入
                $this->sql = "insert into $dsname ($first) values ($second)";
                break;
            case `delete`://刪除
                $this->sql = "delete from $dsname where $first";
                break;
            case `update`://更新
                $this->sql = "update $dsname set $first where $second";
                break;
            case `select`://查詢
                $this->sql = "select $first from $dsname where $second";
                break;
            default:
                die ("Syntax Error!");//提示語法錯誤
                break;
        }
        
        $res = $this->sql();//執行sql語句
        if($res){
            if($cmd == `select`){
                return $this->result;//返回查詢結果
            }else{
                return true;//執行成功返回真
            }
        }else{
            return false;//否則返回假
        }
        $this->link=NULL;;//關閉資料庫
    }
}

$mysql = new mysql;//資料庫類的例項化

?>

相關文章