mysql实现upsert

  • upsert(update or insert), 即更新或写入。
  • MySQL中实现upsert操作方式:
    思路:通过判断插入的记录里是否存在主键索引或唯一索引冲突,来决定是插入还是更新。当出现主键索引或唯一索引冲突时则进行update操作,否则进行insert操作。
    实现:使用 ON DUPLICATE KEY UPDATE

来看看下面具体实现过程。

一、准备数据表
CREATE TABLE `demo` (
  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,
  `a` tinyint(1) unsigned NOT NULL DEFAULT '0',
  `b` tinyint(1) unsigned NOT NULL DEFAULT '0',
  `c` tinyint(1) unsigned NOT NULL DEFAULT '0',
  `d` tinyint(1) unsigned NOT NULL DEFAULT '0',
  `e` tinyint(1) unsigned NOT NULL DEFAULT '0',
  `f` tinyint(1) unsigned NOT NULL DEFAULT '0',
  PRIMARY KEY (`id`),
  UNIQUE KEY `unq_a_b_c` (`a`,`b`,`c`)
) ENGINE=InnoDB AUTO_INCREMENT=1 DEFAULT CHARSET=latin1;

注意:表中存在两处索引,id为主键索引,a,b,c为联合唯一索引。

二、写入初始数据
insert into test.demo(a,b,c,d,e,f) values(1,1,1,1,1,1);

此时存在由abc散列组成唯一索引数据:1,1,1。

三、进一步实现
insert into demo(a,b,c,d,e,f) values(1,1,1,2,2,2) ON DUPLICATE KEY UPDATE a=2,b=2,c=3,d=4,e=5,f=6;
因为已经存在由abc三列组成唯一索引数据:1,1,1,本次又写入demo(a,b,c,d,e,f) values(1,1,1,2,2,2),会造成唯一索引冲突。
因此,会触发ON DUPLICATE KEY 后面的 UPDATE a=2,b=2,c=3,d=4,e=5,f=6操作。

至此,已经实现upsert功能。请记住 ON DUPLICATE KEY UPDATE的用法。

实现mysql的批量更新

 insert into statistic_customer(customer_id,current_period,period_number,client_upload_bill,update_time) values
 (1,201604,100,100,1540470829),
 (314,201604,100,100,1540470829),
 (315,201604,100,100,1540470829),
 (316,201611,100,100,1540470829)
 ON DUPLICATE KEY UPDATE
 customer_id=values(customer_id),
 current_period=values(current_period),
 period_number=values(period_number),
 client_upload_bill=values(client_upload_bill),
 update_time=values(update_time)
public static function test()
    {
        $updateData = [
            [
                'customer_id' => 1,
                'current_period' => 201604,
                'period_number' => 100,
                'client_upload_bill' => 100,
                'update_time' => time(),
            ],
            [
                'customer_id' => 314,
                'current_period' => 201604,
                'period_number' => 100,
                'client_upload_bill' => 100,
                'update_time' => time(),
            ],
            [
                'customer_id' => 315,
                'current_period' => 201604,
                'period_number' => 100,
                'client_upload_bill' => 100,
                'update_time' => time(),
            ],
            ['customer_id' => 316,
                'current_period' => 201611,
                'period_number' => 100,
                'client_upload_bill' => 100,
                'update_time' => time(),
            ],

        ];

        $sql = self::buildSQL('statistic_customer', $updateData);
        DB::insert($sql);
    }

    private static function buildSQL(string $tableName, array $updateData): string
    {
        $sql = "insert into {$tableName}(";
        $keys = array_keys($updateData[0]);
        $sql .= implode(',', $keys);
        $sql .= ') values';

        $valuesStr = '';
        foreach ($updateData as $value) {
            $valuesStr .= '(' . implode(',', array_values($value)) . '),';
        }
        $sql .= rtrim($valuesStr, ',');
        $sql .= ' ON DUPLICATE KEY UPDATE ';
        $updateStr = '';
        foreach ($keys as $key) {
            $updateStr .= "{$key}=values($key),";
        }
        $sql .= rtrim($updateStr, ',');
        return $sql;
    }

原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/193681.html

(0)
上一篇 2021年11月15日
下一篇 2021年11月15日

相关推荐

发表回复

登录后才能评论