如何在CosmosDB中删除数组元素或获取数组元素索引？

一、直接通过属性匹配删除数组元素的方法 #

CosmosDB的remove类型Patch操作不支持直接通过属性匹配删除数组元素，必须指定元素的索引路径。但可以通过set类型Patch结合系统函数ARRAY_FILTER，直接替换数组为过滤后的结果，无需提前获取索引：

示例Patch操作 #

const patchOperations = [
  // 过滤掉userId匹配的元素，替换原数组
  {
    op: 'set',
    path: '/following/followingUsers',
    value: 'ARRAY_FILTER(/following/followingUsers, u => u.userId != @targetUserId)'
  },
  // 同步更新关注数量为新数组的长度
  {
    op: 'set',
    path: '/following/followingAmount',
    value: 'ARRAY_LENGTH(/following/followingUsers)'
  }
];

// 执行Patch时传入参数
const params = { targetUserId: 'foo' };

这个操作会一次性删除所有userId等于@targetUserId的元素，并自动更新followingAmount为最新的数组长度，无需额外查询索引。

二、获取元素索引的方法（若仍需使用remove操作） #

如果必须使用remove操作（比如只删除第一个匹配元素），可以通过以下两种方式获取索引，无需拉取整个数组：

方法1：使用用户定义函数（UDF） #

1. 创建UDF用于查找匹配userId的元素索引：

function findUserIndex(arr, targetUserId) {
  for (let i = 0; i < arr.length; i++) {
    if (arr[i].userId === targetUserId) {
      return i;
    }
  }
  return -1; // 未找到返回-1
}

2. 查询时调用UDF获取索引：

SELECT
  p.id,
  udf.findUserIndex(p.following.followingUsers, @targetUserId) AS followingIndex
FROM p
WHERE p.id = @currentUserId

方法2：使用SQL子查询生成索引 #

通过JOIN遍历数组元素并关联索引值，筛选匹配的结果：

SELECT
  p.id,
  matched.idx AS followingIndex
FROM p
JOIN (
  SELECT VALUE {
    "idx": i,
    "userId": u.userId
  }
  FROM u IN p.following.followingUsers
  JOIN i IN ARRAY(SELECT VALUE idx FROM idx IN [0,1,2,3,4,5,6,7,8,9,10] WHERE idx < ARRAY_LENGTH(p.following.followingUsers))
  WHERE u.userId = @targetUserId
) AS matched
WHERE p.id = @currentUserId

注意：如果数组长度可能超过示例中的10，需要调整[0,1,...]的范围，或者使用动态生成索引的逻辑（UDF方式更灵活）。

三、注意事项 #

• 使用ARRAY_FILTER的Patch方法更高效，一步完成删除和数量更新，推荐优先使用。
• 如果数组中存在多个相同userId的元素，remove操作只会删除指定索引的单个元素，而ARRAY_FILTER会删除所有匹配元素，根据需求选择。
• 执行Patch操作时，确保参数绑定正确，避免SQL注入风险。

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