如何通过单次数据库调用按条件更新不同字段

现有如下Job实体类：

public class Job
{
    public Guid Id { get; set; }

    public bool IsPending { get; set; }

    public bool IsComplete { get; set; }
}

我拥有两个Guid类型的ID列表List<Guid> A和List<Guid> B。需要对列表A中的ID对应的Job设置Update.Set("IsPending", true)，对列表B中的ID对应的Job设置Update.Set("IsComplete", true)。目前通过两次数据库调用实现：

this.UpdateManyAsync(Builders<Job>.Filter.In(f => f.Id, A), Builders<Job>.Update.Set(x => x.IsPending, true), cancellationToken: cancellationToken)

以及第二次调用：

this.UpdateManyAsync(Builders<Job>.Filter.In(f => f.Id, B), Builders<Job>.Update.Set(x => x.IsComplete, true), cancellationToken: cancellationToken)

答案：可以通过单次调用完成 #

你可以利用MongoDB的条件更新能力，在单次UpdateManyAsync调用中完成两个字段的针对性更新，具体实现代码如下：

// 构建更新规则：根据ID所在列表设置对应字段
var updateDefinition = Builders<Job>.Update
    // 当ID在列表A中时，IsPending设为true
    .Set(x => x.IsPending, new BsonDocument("$in", new BsonArray { "$_id", new BsonArray(A) }))
    // 当ID在列表B中时，IsComplete设为true
    .Set(x => x.IsComplete, new BsonDocument("$in", new BsonArray { "$_id", new BsonArray(B) }));

// 构建过滤条件：仅更新ID在A或B中的文档
var filter = Builders<Job>.Filter.Or(
    Builders<Job>.Filter.In(f => f.Id, A),
    Builders<Job>.Filter.In(f => f.Id, B)
);

// 单次调用完成批量更新
await this.UpdateManyAsync(filter, updateDefinition, cancellationToken: cancellationToken);

逻辑说明 #

• 这里使用MongoDB的$in操作符作为条件判断：如果文档的_id在目标列表中，对应字段就会被设为true；不在则保持原有值。
• 过滤条件限定只处理ID在A或B中的文档，避免更新无关数据。
• 如果某个ID同时存在于A和B列表中，两个字段都会被更新为true，效果和两次调用完全一致。

注意事项 #

• 确保你的MongoDB服务器版本支持条件更新（MongoDB 3.2及以上版本均支持该语法）。
• 如果A和B列表存在大量重复ID，这种方式比两次调用更高效，减少了一次网络往返开销。

内容的提问来源于stack exchange，提问作者Mohammad Sarfaraz