如何使用 pymongo 获取 CosmosDB 查询中消耗的 RU/s？

Question

我在 Azure 门户中执行查询时遇到问题，它消耗 ~5 RU/s，但是当我在 pymongo 中执行相同的查询时，它会抛出“请求很大”，表明我没有有足够的（我们提供了 3000 RU/s）。

如何从 Python 中找出我的 pymongo 查询消耗了多少 RU/s？

编辑：有没有办法找到 Cosmos 使用的总 RU/s（例如其他用户？）

Answer 1

在 Pymongo 中使用runCommand 方法：

from pymongo import MongoClient

client = MongoClient()

client = MongoClient(
    'mongodb://***@***.documents.azure.com:10255/?ssl=true&replicaSet=globaldb')
db = client.testdb
coll = db.coll
result = coll.find_one()
print(coll.find_one())

result = db.command({"getLastRequestStatistics": 1})

print("Last RU Consume:", result["RequestCharge"])

输出：

从这个link得到这个解决方案，官方document没有提到。