MongoTemplate：更新数组首元素，数组未初始化时插入为列表

需求与问题 #

需要完成两个更新操作：

• 将指定文档的age字段设为old
• 将hobby数组首元素的velocity设为slow（其余元素保持fast）

现有代码在文档已存在hobby数组时执行正常，但当文档未初始化hobby字段时，执行后hobby会变成Map类型（键为0的嵌套对象），不符合期望的数组格式。

现有代码 #

mongoTemplate.updateFirst(
  Query.query(where("_id").is("1")),
    Update.update("age", "old")
      .set("hobby.0.velocity", "slow"), "collection name")

问题原因 #

MongoDB在更新不存在的字段时，使用点语法hobby.0.velocity会默认创建嵌套对象而非数组——因为它无法自动推断目标结构为数组类型。

解决方案 #

通过MongoDB聚合表达式$cond判断hobby字段是否存在，分情况处理更新逻辑：

• 若hobby已存在：更新数组首元素的velocity，保留其余元素
• 若hobby不存在：直接创建包含{velocity: "slow"}元素的数组

具体代码实现：

Update update = Update.update("age", "old")
    .set("hobby", AggregationExpression.from(MongoExpression.create(
        """
        {
            $cond: {
                if: { $exists: ["$hobby", true] },
                then: { $concatArrays: [
                    [{ $mergeObjects: [ { $arrayElemAt: ["$hobby", 0] }, { velocity: "slow" } ] }],
                    { $slice: ["$hobby", 1, { $size: "$hobby" }] }
                ] },
                else: [{ velocity: "slow" }]
            }
        }
        """
    )));

mongoTemplate.updateFirst(
    Query.query(where("_id").is("1")),
    update,
    "collection name"
);

逻辑说明 #

1. 判断hobby是否存在：用$exists检查字段状态
2. 存在时的处理：
   • 用$arrayElemAt取出数组首元素，通过$mergeObjects覆盖velocity为slow
   • 用$slice截取数组从索引1开始的剩余元素
   • 用$concatArrays拼接更新后的首元素和剩余元素，组成新数组
3. 不存在时的处理：直接返回[{velocity: "slow"}]作为新的hobby数组

验证结果 #

• 已有hobby数组的文档：更新后age为old，首元素velocity变为slow，其余元素保持不变，符合预期
• 无hobby字段的文档：更新后age为old，hobby生成包含{velocity: "slow"}的数组，而非Map类型

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