MongoDB处理批量update违反唯一约束的key

最新推荐文章于 2023-01-11 18:02:04 发布

cuanjian2673

最新推荐文章于 2023-01-11 18:02:04 发布

阅读量397

点赞数

文章标签：数据库

MongoDB批量update一般使用upsert即可，但是这样更新可能会违反唯一建约束导致全部都没有更新（关系型数据库亦是如此），针对这个情况可以一条条更新或者加判断进行更新来保证：不违反唯一约束的行都能得到更新。下面是处理方式，供有需求的人进行参考
【背景说明】

集合名：kk_device_likes_game
唯一索引:{"device_id" : 1,

"target_type" : 1,

"target_id" : 1

}

更新内容为：db.kk_device_likes_game.update({"target_type":5,"target_id":NumberLong(1030)},{"$set":{"target_id":NumberLong(1031)}},true,true)

将满足条件为"target_type":5,"target_id":NumberLong(1030)的target_id键值的NumberLong(1030)改为NumberLong(1031)

更新会失败，因为这个集合更新字段上面有复合唯一索引，使得更新后的键值与已有的其他键值冲突。这样导致能被更新的键值却不能更新。报错如下：

【问题解决】

针对上面情况：有两条思路

方式一：遍历满足条件集合，逐条更新

思路：获取满足条件的键值，遍历其每一个_id进行逐条更新，类似sql为update … where target_type=5 and target_id=1030 and _id =…

语法如下：

var cursor = db.kk_device_likes_game.find({"target_type":5,"target_id":NumberLong(1030)})

while (cursor.hasNext()) {

doc=cursor.next();

a=db.kk_device_likes_game.update({_id:doc._id,"target_type":5,"target_id":NumberLong(1030)},{"$set":{"target_id":NumberLong(1031)}})

print(a)

}

方式二：判断如果会违反约束就不更新该行

思路：获取满足条件的键值，根据复合唯一索引{"device_id" : 1,"target_type" : 1,"target_id" : 1}，将device_id值跟条件为 target_type=5 and target_id=1031的 device_id进行比对，如果有一样的device_id则该行不更新。类似sql为update …where target_type=5 and target_id=1030 and device_id not in (select device_id from where target_type=5 and target_id=1031 )

语法如下：

var cursor = db.kk_device_likes_game.find({"target_type":5,"target_id":NumberLong(1030)})

while (cursor.hasNext()) {

doc=cursor.next();

a=db.kk_device_likes_game.find({device_id:doc.device_id,"target_type":5,"target_id":NumberLong(1031)}).count()

if ( a == 0 ){

db.kk_device_likes_game.update({_id:doc._id,"target_type":5,"target_id":NumberLong(1030)},{"$set":{"target_id":NumberLong(1031)}});