前言

在写一个项目时用了easyswoole框架,需要用到批量更新方法,但是easyswoole没有去实现方法,只能自己封装了。

实现代码

找到ORM包的src/AbstractModel.php文件,在文件最后添加上下面代码就可以实现了。

    /**
     * 批量更新数据
     * 根据$field 字段 更新数据
     **/
    public function upDataInfo($data, $field = null, $where = [])
    {
        //主键没有定义自动寻找主键
        if (is_null($field)) {
            $field = $this->schemaInfo()->getPkFiledName();
            if (empty($field)) {
                throw new Exception('upDataInfo needs primaryKey for model ');
            }
        }
        //语句组装
        $updates = $this->parseUpdate($data, $field);
        $where = $this->parseParams($where);
        $fields_c = array_map(function ($v) use ($field) {
            return $v["$field"];
        }, $data);
        $fields = implode(',', array_map(function ($value) {
            return "'" . $value . "'";
        }, $fields_c));
        $sql = sprintf("UPDATE `%s` SET %s WHERE `%s` IN (%s) %s", $this->tableName, $updates, $field, $fields, $where);
        return $sql;
    }

    /**
     * Function parseUpdate
     * 将二维数组转换成CASE WHEN THEN的批量更新条件
     */
    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, ',');
    }

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

使用

$model = new UploadModel();
        echo $model->upDataInfo([
            ['id' => 1, 'name' => '教书先生', 'email' => '599928887@qq.com'],
            ['id' => 2, 'name' => '教书先生2', 'email' => '599928887@qq.com']
        ], 'id');

运行结果

UPDATE `upload_list` SET `id` = CASE `id` 
WHEN '1' THEN '1' 
WHEN '2' THEN '2' 
END,`name` = CASE `id` 
WHEN '1' THEN '教书先生' 
WHEN '2' THEN '教书先生2' 
END,`email` = CASE `id` 
WHEN '1' THEN '599928887@qq.com' 
WHEN '2' THEN '599928887@qq.com' 
END WHERE `id` IN ('1','2')

最后

还是希望官方下次更新可以实现下批量更新数据方法,每次拉框架还要修改文件。

最后修改:2022 年 04 月 20 日
如果觉得我的文章对你有用,请随意赞赏