如何用Google Datastore查询实体总数？Python代码num_results返回0异常

这个问题我之前也碰到过，核心原因是Google Cloud Datastore Python客户端返回的QueryResultIterator（也就是你代码里的query_iter）的num_results属性工作逻辑和你预期的不一样。

为什么num_results返回0？ #

当你调用query.fetch()时，返回的迭代器是懒加载的——它不会立刻向Datastore请求所有结果，而是等你开始遍历它的时候才会逐步拉取数据。在你还没开始遍历之前，客户端根本没有从Datastore获取到结果总数的信息，所以此时num_results的初始值就是0。你代码里先打印num_results，之后才遍历迭代器，自然会输出0。

解决方法 #

这里提供几种可行的方案，你可以根据自己的场景选择：

方法一：将迭代器转为列表（最常用）

把迭代器转换成列表会一次性拉取所有结果到内存中，之后用len()就能直接获取总数，同时也能正常遍历实体：

def implicit():
    query = client.query(kind='task')
    query.add_filter('status', '=', True)
    # 把迭代器转为列表，触发结果拉取
    results = list(query.fetch())
    print(len(results))  # 此时会输出预期的5
    for entity in results:
        print(entity['title'])

⚠️ 注意：这种方法适合结果集不大的场景，避免内存占用过高。

方法二：先遍历迭代器再获取num_results

如果你不想一次性加载所有结果，可以先遍历迭代器（此时客户端会拉取数据并填充num_results），之后再访问该属性。不过要注意迭代器只能遍历一次，所以需要提前保存实体：

def implicit():
    query = client.query(kind='task')
    query.add_filter('status', '=', True)
    query_iter = query.fetch()
    entities = []
    for entity in query_iter:
        entities.append(entity)
        print(entity['title'])
    # 遍历完成后，num_results会被正确填充
    print(query_iter.num_results)  # 这里会输出5

方法三：单独调用count()获取总数

如果只需要计数，或者想在遍历前就知道总数，可以用query.count()方法。不过这会发起一个单独的查询，所以如果在计数和拉取结果之间数据有变化，两者可能不一致：

def implicit():
    query = client.query(kind='task')
    query.add_filter('status', '=', True)
    # 单独发起计数查询
    total_count = query.count()
    print(total_count)  # 输出5
    # 再拉取结果遍历
    query_iter = query.fetch()
    for entity in query_iter:
        print(entity['title'])

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