PHP 批量更新

mysql更新語句很簡單，更新一條數據的某個字段，通常這樣寫：

UPDATE mytable SET myfield = 'value' WHERE other_field = 'other_value';

若是更新同一字段爲同一個值，mysql也很簡單，修改下where便可：

UPDATE mytable SET myfield = 'value' WHERE other_field in ('other_values');

這裏注意，other_values是一個逗號，分隔的字符串，如：1,2,3

1 常規方案
那若是更新多條數據爲不一樣的值，可能不少人會這樣寫：

foreach ($display_order as $id => $ordinal) {
    $sql = "UPDATE categories SET display_order = $ordinal WHERE id = $id";
    mysql_query($sql);
}

便是循環一條一條的更新記錄。

一條記錄update一次，這樣性能不好，也很容易形成阻塞。

2 高效方案
那麼能不能一條sql語句實現批量更新呢？

2.1 CASE WHEN
mysql並無提供直接的方法來實現批量更新，可是能夠用點小技巧來實現。

UPDATE mytable SET
    myfield = CASE id
        WHEN 1 THEN 'value'
        WHEN 2 THEN 'value'
        WHEN 3 THEN 'value'
    END
WHERE id IN (1,2,3)

這裏使用了case when 這個小技巧來實現批量更新。

舉個例子：

UPDATE categories SET
    display_order = CASE id
        WHEN 1 THEN 3
        WHEN 2 THEN 4
        WHEN 3 THEN 5
    END
WHERE id IN (1,2,3)

這句sql的意思是，更新display_order 字段：

若是id=1 則display_order 的值爲3，
若是id=2 則 display_order 的值爲4，
若是id=3 則 display_order 的值爲5。
便是將條件語句寫在了一塊兒。

這裏的where部分不影響代碼的執行，可是會提升sql執行的效率。

確保sql語句僅執行須要修改的行數，這裏只有3條數據進行更新，而where子句確保只有3行數據執行。

3.2 更新多值
若是更新多個值的話，只須要稍加修改：

UPDATE categories SET
    display_order = CASE id
        WHEN 1 THEN 3
        WHEN 2 THEN 4
        WHEN 3 THEN 5
    END,
    title = CASE id
        WHEN 1 THEN 'New Title 1'
        WHEN 2 THEN 'New Title 2'
        WHEN 3 THEN 'New Title 3'
    END
WHERE id IN (1,2,3)

到這裏，已經完成一條mysql語句更新多條記錄了。

可是要在業務中運用，須要結合服務端語言。

3.3 封裝成PHP函數
在PHP中，咱們把這個功能封裝成函數，之後直接調用。

爲提升可用性，咱們考慮處理更全面的狀況。

以下時須要更新的數據，咱們要根據id和parent_id字段更新post表的內容。

其中，id的值會變，parent_id的值同樣。

$data = [
    ['id' => 1, 'parent_id' => 100, 'title' => 'A', 'sort' => 1],
    ['id' => 2, 'parent_id' => 100, 'title' => 'A', 'sort' => 3],
    ['id' => 3, 'parent_id' => 100, 'title' => 'A', 'sort' => 5],
    ['id' => 4, 'parent_id' => 100, 'title' => 'B', 'sort' => 7],
    ['id' => 5, 'parent_id' => 101, 'title' => 'A', 'sort' => 9],
];

例如，咱們想讓parent_id爲100、title爲A的記錄依據不一樣id批量更新：

echo batchUpdate($data, 'id', ['parent_id' => 100, 'title' => 'A']);

其中，batchUpdate()實現的PHP代碼以下：

/**
 * 批量更新函數
 * @param $data array 待更新的數據，二維數組格式
 * @param array $params array 值相同的條件，鍵值對應的一維數組
 * @param string $field string 值不一樣的條件，默認爲id
 * @return bool|string
 */
function batchUpdate($data, $field, $params = [])
{
   if (!is_array($data) || !$field || !is_array($params)) {
      return false;
   }

    $updates = parseUpdate($data, $field);
    $where = parseParams($params);

    // 獲取全部鍵名爲$field列的值，值兩邊加上單引號，保存在$fields數組中
    // array_column()函數須要PHP5.5.0+，若是小於這個版本，能夠本身實現，
    // 參考地址：http://php.net/manual/zh/function.array-column.php#118831
    $fields = array_column($data, $field);
    $fields = implode(',', array_map(function($value) {
        return "'".$value."'";
    }, $fields));

    $sql = sprintf("UPDATE `%s` SET %s WHERE `%s` IN (%s) %s", 'post', $updates, $field, $fields, $where);

   return $sql;
}

/**
 * 將二維數組轉換成CASE WHEN THEN的批量更新條件
 * @param $data array 二維數組
 * @param $field string 列名
 * @return string sql語句
 */
function parseUpdate($data, $field)
{
    $sql = '';
    $keys = array_keys(current($data));
    foreach ($keys as $column) {

        $sql .= sprintf("`%s` = CASE `%s` \n", $column, $field);
        foreach ($data as $line) {
            $sql .= sprintf("WHEN '%s' THEN '%s' \n", $line[$field], $line[$column]);
        }
        $sql .= "END,";
    }

    return rtrim($sql, ',');
}

/**
 * 解析where條件
 * @param $params
 * @return array|string
 */
function parseParams($params)
{
   $where = [];
   foreach ($params as $key => $value) {
      $where[] = sprintf("`%s` = '%s'", $key, $value);
   }
   
   return $where ? ' AND ' . implode(' AND ', $where) : '';
}

獲得這樣一個批量更新的SQL語句：

UPDATE `post` SET `id` = CASE `id` 
WHEN '1' THEN '1' 
WHEN '2' THEN '2' 
WHEN '3' THEN '3' 
WHEN '4' THEN '4' 
WHEN '5' THEN '5' 
END,`parent_id` = CASE `id` 
WHEN '1' THEN '100' 
WHEN '2' THEN '100' 
WHEN '3' THEN '100' 
WHEN '4' THEN '100' 
WHEN '5' THEN '101' 
END,`title` = CASE `id` 
WHEN '1' THEN 'A' 
WHEN '2' THEN 'A' 
WHEN '3' THEN 'A' 
WHEN '4' THEN 'B' 
WHEN '5' THEN 'A' 
END,`sort` = CASE `id` 
WHEN '1' THEN '1' 
WHEN '2' THEN '3' 
WHEN '3' THEN '5' 
WHEN '4' THEN '7' 
WHEN '5' THEN '9' 
END WHERE `id` IN ('1','2','3','4','5')  AND `parent_id` = '100' AND `title` = 'A'

生成的SQL把全部的狀況都列了出來。

不過由於有WHERE限定了條件，因此只有id爲一、二、3這幾條記錄被更新。

若是隻須要更新某一列，其餘條件不限，那麼傳入的$data能夠更簡單：

$data = [
    ['id' => 1, 'sort' => 1],
    ['id' => 2, 'sort' => 3],
    ['id' => 3, 'sort' => 5],
];
echo batchUpdate($data, 'id');

這樣的數據格式傳入，就能夠修改id從1~3的記錄，將sort分別改成一、三、5。

獲得SQL語句：

UPDATE `post` SET `id` = CASE `id` 
WHEN '1' THEN '1' 
WHEN '2' THEN '2' 
WHEN '3' THEN '3' 
END,`sort` = CASE `id` 
WHEN '1' THEN '1' 
WHEN '2' THEN '3' 
WHEN '3' THEN '5' 
END WHERE `id` IN ('1','2','3')

這種狀況更加簡單高效。