sql - Cosmos 批量读取项查询

Question

是否可以在cosmos db中进行批量查询？

我有一个 item_ids 列表，是否可以在一个查询中获取输出？目前我正在使用循环，这需要很多时间，因为我在一个列表中有 5K 到 20K ID，为了加快处理时间，我们可以使用任何其他解决方案来代替使用线程吗？

使用循环的示例代码:

def read_item(self):
         for item_id in self.ids():
            response = self.container.read_item(item=item_id, partition_key=item_id)

云:Azure数据库:宇宙数据库应用程序编程接口(interface):SQL语言:Python 3.x

Answer 1

您可以使用.query_items()方法返回具有指定ID的所有项目:

import json

def read_item(self):
    id_list_str = json.dumps(self.ids())
    response = self.container.query_items(
        f"SELECT * FROM c WHERE ARRAY_CONTAINS({id_list_str}, c.id)",
        enable_cross_partition_query=True
    )