前言
在写一个项目时用了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')
最后
还是希望官方下次更新可以实现下批量更新数据方法,每次拉框架还要修改文件。