如何在Mongoose的findByIdAndUpdate中结合$set使用自定义JS函数

你的代码写法无法实现预期效果，原因是MongoDB的更新操作符（如$set）中无法直接调用客户端的JS函数，'$savedResponses'在这里只是普通字符串，不会被解析为文档的字段值，你的customJSFunction实际接收的是字符串"$savedResponses"，而非数据库中存储的真实数据。

要解决并发更新问题，同时实现自定义逻辑的原子更新，推荐使用MongoDB的聚合管道更新（4.2+版本支持），以下是两种可行方案：

方案1：用内置聚合操作符实现逻辑（优先推荐） #

如果你的customJSFunction逻辑可以通过MongoDB内置的聚合操作符（如$map、$filter、$push等）实现，直接使用这些操作符性能更优，示例如下：

// 假设customJSFunction是给savedResponses数组添加一个新元素
let test = await testModel.findByIdAndUpdate(
  'someId',
  [
    {
      $set: {
        savedResponses: { $concatArrays: ['$savedResponses', ['newResponse']] }
      }
    }
  ],
  { new: true }
);

方案2：使用服务器端自定义JS函数（$function） #

如果逻辑复杂无法用内置操作符实现，MongoDB 4.4+支持$function在服务器端执行自定义JS逻辑，该操作是原子性的，能避免并发问题：

let test = await testModel.findByIdAndUpdate(
  'someId',
  [
    {
      $set: {
        savedResponses: {
          $function: {
            body: function(savedResponses) {
              // 这里写你customJSFunction中的具体逻辑
              // 注意：该函数在服务器端执行，无法访问客户端变量
              // do something with savedResponses
              return updatedSavedResponses;
            },
            args: ['$savedResponses'], // 传入文档中的savedResponses字段值
            lang: 'js'
          }
        }
      }
    }
  ],
  { new: true }
);

关键注意事项 #

• 服务器端JS函数无法访问客户端的变量、环境或其他外部资源，逻辑必须自包含
• 内置聚合操作符的执行性能远高于自定义JS函数，能用内置操作符实现的逻辑优先选择
• 聚合管道更新是原子操作，全程在数据库端完成，避免了"查询-修改-保存"模式的并发冲突问题

内容的提问来源于stack exchange，提问作者Muhammad Zain Ejaz