如何在Cosmos DB中获取已插入文档ID及插入时一次性获取ID？

1. 如何获取已添加至Cosmos DB集合中的文档的ID？ #

根据不同场景，有几种简单可行的方式：

• 若文档有唯一业务标识字段：比如你给文档设置了userPhone、orderNumber这类唯一属性，直接用SQL查询就能拿到对应的ID（包括你自定义的ID和Cosmos内置的id）。示例查询：

  SELECT c.id, c.myCustomId FROM c WHERE c.orderNumber = 'ORD-2024-001'
  

  执行后结果会同时返回文档的内置ID和你的自定义ID字段。

• 若文档是刚插入的：其实在插入操作完成时，Cosmos的SDK会直接返回插入后的完整文档，里面包含所有ID信息。比如用C# SDK实现：

  var insertResponse = await myContainer.CreateItemAsync(myNewDocument);
  string cosmosBuiltInId = insertResponse.Resource.Id; // 内置ID
  string myCustomId = insertResponse.Resource.MyCustomId; // 自定义ID
  

  完全不需要额外发起查询，省掉一次数据库往返请求。

• 若无唯一标识字段：这种情况比较少见，但如果遇到，可通过变更Feed追踪插入的文档，或遍历集合（不建议在大集合中使用）。不过更推荐的方式是给文档添加唯一业务字段，方便后续查询管理。

2. 单集合存储多类型文档，自定义了Id字段，如何一次操作完成插入并获取ID？ #

这个问题很典型——你完全不需要额外执行查询，插入操作的响应里就包含了你需要的所有ID信息，不管是自定义的还是Cosmos内置的：

场景一：自行指定自定义ID和Cosmos内置ID #

如果你在插入前就给文档设置了myCustomId和Cosmos的id字段（比如把业务ID设为内置id），那你其实已经知晓这些ID，但SDK仍会返回完整文档以确认插入成功：

// 待插入的文档
{
  "id": "user_001", // 直接指定Cosmos内置ID
  "type": "user",
  "myCustomId": "user_001",
  "name": "张三"
}

插入后从响应中即可获取这些ID，比如用JavaScript SDK：

const { resource: savedDoc } = await container.items.create(yourDocument);
console.log("内置ID：", savedDoc.id);
console.log("自定义ID：", savedDoc.myCustomId);

场景二：让Cosmos自动生成内置ID #

如果你未指定内置id字段，Cosmos会自动生成一个唯一字符串作为ID，这个ID会直接包含在插入响应里。比如用Python SDK实现：

new_product = {
  "type": "product",
  "name": "无线耳机",
  "price": 199.99
}

insert_result = container.create_item(body=new_product)
# 获取Cosmos自动生成的内置ID
generated_id = insert_result['id']
# 同时可获取你设置的type字段及其他属性
product_type = insert_result['type']

核心要点： #

插入操作本身会返回完整的持久化文档，所有ID信息都包含其中，无需额外发起查询。这种方式既高效又能保证数据一致性，毕竟刚插入的文档是最新状态。

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