使用DaprClient向CosmosDB插入数据时分区键不匹配问题求助

我尝试用DaprClient向CosmosDB插入数据，代码如下：

[HttpPost("saveData")]
public async Task<ActionResult<string>> SaveData([FromBody] InputMessage message, [FromServices] DaprClient daprClient)
{
    Dictionary<string, string> data = new Dictionary<string, string> { { "partitionKey", "ServiceName" } };

    if (message is null)
        return new BadRequestObjectResult(new { error = $"The client is null" });

    var state = await daprClient.GetStateEntryAsync<TestModel>(_cosmosStoreName, "ServiceName", metadata: data);

    if (state.Value != null)
    {
        return new BadRequestObjectResult(new { error = $"The client with id {message.id} already exists" });
    };

    state.Value = new TestModel
    {
        ServiceName = "ServiceName"
    };

    await state.SaveAsync();
    return "Good";
}

我的CosmosDB容器分区键为ServiceName，运行Dapr CLI并发起请求时收到以下错误：

{"code":"BadRequest","message":"Message: {\"Errors\":[\"PartitionKey extracted from document doesn't match the one specified in the header. Learn more: https:\\/\\/aka.ms\\/CosmosDB\\/sql\\/errors\\/wrong-pk-value\"]}\\r\\nActivityId: 6ca949f9-da2a-4fe9-a346-a67196817670

通过ActivityId在CosmosDB的AzureDiagnostics中追踪日志，发现PartitionKeyRUConsumption类别下的partitionKey_s为fanglava-coyote||ServiceName，怀疑是Dapr自动生成了这种格式的分区键，添加metadata参数后问题依旧。

错误原因 #

Dapr的CosmosDB状态存储组件默认会将app-id（你的Dapr应用ID，此处为fanglava-coyote）与指定的分区键值拼接成app-id||partitionKey格式作为实际分区键。而你未在组件配置中明确指定分区键的路径，导致Dapr无法正确识别文档中的ServiceName字段作为分区键，进而出现提取的分区键与请求头中指定值不匹配的错误。

正确实现步骤 #

1. 修改Dapr CosmosDB组件配置
   在你的Dapr组件配置文件（通常为components/cosmosdb-state.yaml）中添加partitionKey配置项，指定分区键的路径（对应文档中的ServiceName字段）：

   apiVersion: dapr.io/v1alpha1
   kind: Component
   metadata:
     name: <你的组件名称，对应代码中的_cosmosStoreName>
   spec:
     type: state.azure.cosmosdb
     version: v1
     metadata:
     - name: url
       value: <CosmosDB账户URL>
     - name: masterKey
       value: <CosmosDB主键>
     - name: database
       value: <数据库名称>
     - name: collection
       value: <容器名称>
     - name: partitionKey
       value: /ServiceName # 指定分区键字段的路径
   

2. 调整代码逻辑
   配置好分区键路径后，Dapr会自动从文档的ServiceName字段提取分区键，无需手动在metadata中指定。调整后的代码如下：

   [HttpPost("saveData")]
   public async Task<ActionResult<string>> SaveData([FromBody] InputMessage message, [FromServices] DaprClient daprClient)
   {
       if (message is null)
           return new BadRequestObjectResult(new { error = "请求消息为空" });
   
       // 无需传递metadata，Dapr会自动从TestModel的ServiceName字段提取分区键
       var state = await daprClient.GetStateEntryAsync<TestModel>(_cosmosStoreName, "ServiceName");
   
       if (state.Value != null)
       {
           return new BadRequestObjectResult(new { error = $"ID为{message.id}的记录已存在" });
       }
   
       state.Value = new TestModel
       {
           ServiceName = "ServiceName" // 确保该值与容器分区键的取值一致
       };
   
       await state.SaveAsync();
       return "操作成功";
   }
   

3. 验证TestModel定义
   确保你的TestModel类包含ServiceName字段，且字段名与配置的分区键路径一致：

   public class TestModel
   {
       public string ServiceName { get; set; }
       // 其他业务字段...
   }
   

补充说明 #

如果需要手动指定分区键值（而非从文档提取），可以在调用GetStateEntryAsync或SaveAsync时传递metadata，但必须保证该值与文档中的ServiceName字段值完全一致，同时组件配置中的partitionKey路径也要正确设置。

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