为Discuz 中增加一个数据插入方法支持on duplicate key update

737 阅读1分钟

ON DUPLICATE KEY UPDATE的介绍

基本用法:ON DUPLICATE KEY UPDATE 是一种MySQL独有的语法,它在插入新数据时,如果遇到唯一键冲突(即已存在相同的唯一键值),则会执行更新操作,而不是抛出异常或忽略该条数据。这个语法可以大大简化我们的代码,减少不必要的判断和查询操作(ON DUPLICATE KEY UPDATE网上很多,可参考blog.csdn.net/weixin_4911… ,这里就不再赘述)。

其用法总结

1:on duplicate key update 语句根据主键id或唯一键来判断当前插入是否已存在。 2:记录已存在时,只会更新on duplicate key update之后指定的字段。 3:如果同时传递了主键和唯一键,以主键为判断存在依据,该唯一键字段指定内容可以被修改

Disucz 中如何增加on duplicate key update 方法

找到文件:/souce/class/discuz/discuz_database.php,在50行左右,在insert方法前面 新增一个方法updateinsert,此方法与db::insert参数一致。

public static function updateinsert($table, $data ,$returnid = false, $replace = false, $silent = false) {
        $sql = $comma = '';
        foreach ($data as $k => $v) {
            $insKey[] = self::quote_field($k);
            $insValue[] =  self::quote($v);
                $sql .= $comma . self::quote_field($k) . '=' . self::quote($v);
                $comma = ' , ';
        }
        $Key = implode(",", $insKey);
        $Value = implode(",", $insValue);

        $table = self::table($table);
        $silent = $silent ? 'SILENT' : '';
        return self::query("INSERT INTO $table ($Key) values($Value) on duplicate key update $sql", null, $silent, !$returnid);
}
    

用法举例

创建一个测试表,首次设定信息后,后期除了class字段之外,不定期批量更新其它字段内容。

drop table if exists my_table;
create table my_table(
	id int primary key auto_increment comment '主键id',
	customer varchar(30) unique not null comment '客户姓名',
	total int default 2 comment '年龄',
	class varchar(10) comment '组别'
) comment '测试表';
insert into my_table(id,customer,total,class) values(1,'张三',20,'电销组'),(2,'李四',21,'上海组');

// 更新部分字段的数据
$data = array(
    'id' => 1,
    'customer' =>'张三',
    'total' =>321
);

//原来的replace into 方式插入,在更新时会把 class字段清零,
// $res = DB::insert('my_table', $data, true, true); 

// 新的方式会保留 class 字段值
$res = DB::updateinsert('my_table', $data, true, true); 

怎么样,Get到了么?希望对你的项目有帮助!