查詢語句 (DQL) 的構(gòu)造功能開發(fā)完畢偎快，我們?cè)俳o查詢構(gòu)造器增加一些對(duì) DML (Data Manipulation Language) 語句的支持冯乘，如簡單的 insert、update晒夹、delete 操作裆馒。

insert

我們先回顧下 PDO 原生的 insert 操作怎么進(jìn)行：

// 預(yù)編譯
$pdoSt = $pdo->prepare("INSERT INTO test_table ('username', 'age') VALUES (:username, :age);");
// 綁定參數(shù)
$pdoSt->bindValue(':username', 'Jack', PDO::PARAM_STR)
$pdoSt->bindValue(':age', 18, PDO::PARAM_INT)
// 執(zhí)行
$pdoSt->execute(); 
// 獲取執(zhí)行數(shù)據(jù)
$count = $pdoSt->rowCount(); // 返回被影響的行數(shù)
$lastId = $pdo->lastInsertId(); // 獲取最后插入行的 ID

數(shù)據(jù)插入

和查詢語句的執(zhí)行過程并沒有太大差別，只是語法不同丐怯∨绾茫回想第二篇，我們新建了 _buildQuery() 方法去構(gòu)造最終的 SQL读跷，對(duì)于 insert梗搅，我們?cè)诨愋陆?_buildInsert() 方法：

protected function _buildInsert()
{
    $this->_prepare_sql = 'INSERT INTO '.$this->_table.$this->_insert_str;
}

基類添加 _insert_str 屬性：

protected $_insert_str = '';

修改 _reset() 函數(shù)，將 _insert_str 屬性的初始化過程添加進(jìn)去：

protected function _reset()
{
    $this->_table = '';
    $this->_prepare_sql = '';
    $this->_cols_str = ' * ';
    $this->_where_str = '';
    $this->_orderby_str = '';
    $this->_groupby_str = '';
    $this->_having_str = '';
    $this->_join_str = '';
    $this->_limit_str = '';
    $this->_insert_str = ''; // 重置 insert 語句
    $this->_bind_params = [];
}

基類中添加 insert() 方法：

public function insert(array $data)
{
    // 構(gòu)建字符串
    $field_str = '';
    $value_str = '';
    foreach ($data as $key => $value) {
        $field_str .= ' '.self::_wrapRow($key).',';
        $plh = self::_getPlh(); // 生產(chǎn)占位符
        $this->_bind_params[$plh] = $value; //保存綁定數(shù)據(jù)
        $value_str .= ' '.$plh.',';
    }
    // 清除右側(cè)多余的逗號(hào)
    $field_str = rtrim($field_str, ',');
    $value_str = rtrim($value_str, ',');

    $this->_insert_str = ' ('.$field_str.') VALUES ('.$value_str.') ';
    // 構(gòu)造 insert 語句
    $this->_buildInsert();
    // 執(zhí)行
    $this->_execute();
    // 獲取影響的行數(shù)
    return $this->_pdoSt->rowCount();
}

對(duì)上述代碼效览，我們申明了 insert() 方法的參數(shù)是一個(gè)鍵值數(shù)組无切，用來傳入要插入的字段、值映射丐枉。默認(rèn)返回被影響的行數(shù) (比較通用)哆键。

測(cè)試

試著插入一條數(shù)據(jù)吧：

$insert_data = [
    'username' => 'jack',
    'age'      => 18,
];

$results = $driver->table('test_table')->insert($insert_data);

獲取最后插入行的 ID

當(dāng)一個(gè)表中有自增 id 且為主鍵時(shí)，這個(gè) id 可以被看作區(qū)分?jǐn)?shù)據(jù)的唯一標(biāo)識(shí)瘦锹。而在插入一條數(shù)據(jù)后獲取這條新增數(shù)據(jù)的 id 也是常見的業(yè)務(wù)需求籍嘹。

PDO 提供了一個(gè)簡單的獲取最后插入行的 ID 的方法 PDO::lastInsertId() 供我們使用。

基類添加 insertGetLastId() 方法：

public function insertGetLastId(array $data)
{
    $this->insert($data);

    return $this->_pdo->lastInsertId();
}

測(cè)試：

$insert_data = [
    'username' => 'jack',
    'age'      => 18,
];

$lastId = $driver->table('test_table')->insertGetLastId($insert_data);

個(gè)體差異

然而沼本，上述的 insertGetLastId() 方法在 PostgreSql 中并不奏效噩峦。PostgreSql 中，使用 PDO::lastInsertId() 獲取結(jié)果需要傳入正確的自增序列名 (PostgreSQL 中創(chuàng)建表時(shí)抽兆，如果使用 serial 類型识补，默認(rèn)生成的自增序列名為：表名 + _ + 字段名 + _ + seq)”韬欤【1】

但是這個(gè)方式并不好用凭涂，因?yàn)樵L問 insertGetLastId() 方法時(shí)必須手動(dòng)傳入這個(gè)序列名稱，這樣 insertGetLastId() 方法對(duì)底層的依賴嚴(yán)重贴妻，比如當(dāng)?shù)讓域?qū)動(dòng)從 postgresql 換到 mysql 時(shí)切油，需要更改上層應(yīng)用。而我們希望無論是 mysql 還是 postgresql名惩，上層應(yīng)用調(diào)用 insertGetLastId() 方法時(shí)是無差別的澎胡，即底層對(duì)上層透明。

為了解決這個(gè)問題，就需要用到 postgresql 的 returning 語法了攻谁。postgresql 中 insert稚伍、update 和 delete 操作都有一個(gè)可選的 returning 子句，可以指定最后執(zhí)行的字段進(jìn)行返回戚宦，返回的數(shù)據(jù)可以像 select 一樣取結(jié)果个曙。【2】

對(duì)于我們返回最后插入行的 ID 的需求受楼，只需 returning id 就好垦搬。

當(dāng)然，基類的 insertGetLastId() 方法對(duì)于 postgresql 而言已經(jīng)無效了艳汽，我們?cè)?Pgsql 驅(qū)動(dòng)類中重寫 insertGetLastId() 方法：

public function insertGetLastId(array $data)
{
    // 構(gòu)建語句字符串猴贰、綁定數(shù)據(jù)
    $field_str = '';
    $value_str = '';
    foreach ($data as $key => $value) {
        $field_str .= ' '.self::_wrapRow($key).',';
        $plh = self::_getPlh();
        $this->_bind_params[$plh] = $value;
        $value_str .= ' '.$plh.',';
    }

    $field_str = rtrim($field_str, ',');
    $value_str = rtrim($value_str, ',');
    // 使用 returning 子句返回 id
    $this->_insert_str = ' ('.$field_str.') VALUES ('.$value_str.') RETURNING id ';
    // execute
    $this->_buildInsert();
    $this->_execute();
    // 使用 returning 子句后，可以像使用 SELECT 一樣獲取一個(gè) returning 指定字段的結(jié)果集骚灸。 
    $result = $this->_pdoSt->fetch(PDO::FETCH_ASSOC);
    // 返回 id
    return $result['id'];
}

OK糟趾，我們?cè)賮頊y(cè)試看看，是不是就好用了呢甚牲？

update

做完 insert义郑，update 就很簡單了，不同的是為了防止全局更新的失誤發(fā)生丈钙，update 構(gòu)造時(shí)強(qiáng)行要求使用 where 子句非驮。

同樣的，添加 _update_str 屬性雏赦，修改 _reset() 函數(shù)：

protected $_update_str = '';

...

protected function _reset()
{
    $this->_table = '';
    $this->_prepare_sql = '';
    $this->_cols_str = ' * ';
    $this->_where_str = '';
    $this->_orderby_str = '';
    $this->_groupby_str = '';
    $this->_having_str = '';
    $this->_join_str = '';
    $this->_limit_str = '';
    $this->_insert_str = '';
    $this->_update_str = '';
    $this->_bind_params = [];
}

構(gòu)造 update 語句的方法：

protected function _buildUpdate()
{
    $this->_prepare_sql = 'UPDATE '.$this->_table.$this->_update_str.$this->_where_str;
}

基類中添加 update() 方法：

public function update(array $data)
{
    // 檢測(cè)有沒有設(shè)置 where 子句
    if(empty($this->_where_str)) {
        throw new \InvalidArgumentException("Need where condition");
    }
    // 構(gòu)建語句劫笙、參數(shù)綁定
    $this->_update_str = ' SET ';
    foreach ($data as $key => $value) {
        $plh = self::_getPlh();
        $this->_bind_params[$plh] = $value;
        $this->_update_str .= ' '.self::_wrapRow($key).' = '.$plh.',';
    }

    $this->_update_str = rtrim($this->_update_str, ',');

    $this->_buildUpdate();
    $this->_execute();
    // 返回影響的行數(shù)
    return $this->_pdoSt->rowCount();
}

更新數(shù)據(jù)示例：

$update_data = [
    'username' => 'lucy',
    'age'      => 22,
];

$results = $driver->table('test_table')
            ->where('username', 'jack')
            ->update($update_data);

delete

相比 insert、update星岗，delete 語句更為簡單填大，只需 where 子句即可。和 update 一樣俏橘，需要防止誤操作刪除所有數(shù)據(jù)允华。

構(gòu)造 delete 語句的方法：

protected function _buildDelete()
{
    $this->_prepare_sql = 'DELETE FROM '.$this->_table.$this->_where_str;
}

基類中添加 delete() 方法：

public function delete()
{
    // 檢測(cè)有沒有設(shè)置 where 子句
    if(empty($this->_where_str)) {
        throw new \InvalidArgumentException("Need where condition");
    }

    $this->_buildDelete();
    $this->_execute();

    return $this->_pdoSt->rowCount();
}

刪除數(shù)據(jù)示例：

$results = $driver->table('test_table')
            ->where('age', 18)
            ->delete();

事務(wù)

既然有了 DML 操作，那么就少不了事務(wù)寥掐。對(duì)于事務(wù)靴寂，我們可以直接使用 PDO 提供的 PDO::beginTransaction()、PDO::commit()召耘、PDO::rollBack()百炬、PDO::inTransaction() 方法來實(shí)現(xiàn)。

基類添加 beginTrans() 方法：

// 開始事務(wù)
public function beginTrans()
{
    try {
        return $this->_pdo->beginTransaction();
    } catch (PDOException $e) {
        // 斷線重連
        if ($this->_isTimeout($e)) {

            $this->_closeConnection();
            $this->_connect();

            try {
                return $this->_pdo->beginTransaction();
            } catch (PDOException $e) {
                throw $e;
            }

        } else {
            throw $e;
        }
    }
}

注：因?yàn)?PDO::beginTransaction() 也是和 PDO::prepare() 一樣會(huì)連接數(shù)據(jù)庫的方法污它，所以需要做斷線重連的操作剖踊。

commitTrans() 方法：

// 提交事務(wù)
public function commitTrans()
{
    return $this->_pdo->commit(); 
}

rollBackTrans() 方法：

// 回滾事務(wù)
public function rollBackTrans()
{
    if ($this->_pdo->inTransaction()) {
        // 如果已經(jīng)開始了事務(wù)庶弃，則運(yùn)行回滾操作
        return $this->_pdo->rollBack();
    }
}

事務(wù)使用示例：

// 注冊(cè)事務(wù)
$driver->beginTrans();
$results = $driver->table('test_table')
            ->where('age', 18)
            ->delete();
$driver->commitTrans(); // 確認(rèn)刪除

// 回滾事務(wù)
$driver->beginTrans();
$results = $driver->table('test_table')
            ->where('age', 18)
            ->delete();
$driver->rollBackTrans(); // 撤銷刪除

參考

【1】PHP Manual - PDO::lastInsertId

【2】PostgreSQL Documentation - Returning Data From Modified Rows