如何在Azure Functions中轻松追踪CosmosDB的RU/s消耗？

嘿，这个问题我之前帮不少开发者处理过，刚好有几个实用方案可以分享给你：

其实Azure本身就提供了不用自己写代码的RU追踪方式，完美匹配你想要的“简便”需求：

• Azure Monitor 内置指标关联：CosmosDB账户自带Total Request Units等核心指标，你只需要在Azure Functions的应用设置里开启APPINSIGHTS_INSTRUMENTATIONKEY，把函数的执行日志和Application Insights绑定。之后在Azure Monitor面板，就能通过关联ID把某个函数调用对应的所有CosmosDB请求RU汇总起来，不管你访问多少个容器都不会遗漏。
• 诊断日志 + Kusto精准查询：开启CosmosDB的DataPlaneRequests诊断日志，然后在Log Analytics里用Kusto查询关联Functions的执行日志。比如下面这个查询就能直接统计指定函数每次执行的总RU消耗：

  AppRequests
  | where OperationName == "Function.Execute" and Name == "YourTargetFunctionName"
  | join kind=inner (
      AzureDiagnostics
      | where ResourceType == "COSMOSDB" and Category == "DataPlaneRequests"
  ) on $left.CorrelationId == $right.CorrelationId
  | summarize TotalRUConsumed = sum(todouble(RequestCharge_s)) by OperationId, Name
  

如果现成集成满足不了你的需求（比如要在代码里实时计算RU并做业务逻辑处理），那你得注意Azure Functions的无服务器特性：每个函数执行都是独立的执行上下文，普通静态变量会被并发执行覆盖，这里给你一个靠谱的解决思路：

• 用异步上下文隔离变量：可以用AsyncLocal<T>（C#）、Python的contextvars或者JavaScript的AsyncLocalStorage来存储当前执行上下文的RU计数器。这类工具会自动隔离不同的异步执行流，完全不用担心并发执行互相干扰的问题。
• 举个C#的代码示例：
  先定义一个静态的RU追踪类：

  public static class RuConsumptionTracker
  {
      private static readonly AsyncLocal<decimal> _currentTotalRu = new AsyncLocal<decimal>();
  
      public static decimal CurrentTotalRu => _currentTotalRu.Value;
  
      public static void AddRequestCharge(decimal requestCharge)
      {
          _currentTotalRu.Value += requestCharge;
      }
  }
  

  然后在所有CosmosDB操作的地方，把返回的RequestCharge累加进去：

  // 访问容器A的操作
  var containerAResponse = await containerA.ReadItemAsync<MyData>(itemId, new PartitionKey(partitionKey));
  RuConsumptionTracker.AddRequestCharge(containerAResponse.RequestCharge);
  
  // 访问容器B的聚合操作
  var containerBQueryResponse = containerB.GetItemQueryIterator<MyData>(queryDefinition);
  while (containerBQueryResponse.HasMoreResults)
  {
      var response = await containerBQueryResponse.ReadNextAsync();
      RuConsumptionTracker.AddRequestCharge(response.RequestCharge);
  }
  

  最后在函数执行结束时，就能拿到当前执行的总RU：

  [FunctionName("DataAggregationFunction")]
  public static async Task<IActionResult> Run(
      [HttpTrigger(AuthorizationLevel.Function, "get", Route = null)] HttpRequest req,
      ExecutionContext context,
      ILogger log)
  {
      // 执行各种跨容器的CosmosDB操作...
  
      var totalRu = RuConsumptionTracker.CurrentTotalRu;
      log.LogInformation($"Function {context.FunctionName} completed with total RU: {totalRu}");
      return new OkObjectResult($"Total RU consumed: {totalRu}");
  }
  

  这个方案的核心是AsyncLocal会和当前的异步执行流绑定，每个函数执行的RU统计都是独立的，完全不会被其他并发执行的函数覆盖。

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