使用Golang获取Azure CosmosDB for NoSQL全量文档时返回null问题求助

核心问题分析 #

你当前代码的关键错误：

• 调用azcosmos.NewPartitionKeyString("assetType")时，传入的是分区键的字段名，而非某个具体的分区键值，这会导致Cosmos DB只查询分区键值等于"assetType"的分区（大概率不存在该分区），因此返回空结果。
• 要实现类似fetchAll的全集合查询，必须开启跨分区查询，否则只能查询单个分区的数据。

修正后的代码 #

func GetAssets(w http.ResponseWriter, r *http.Request) {
    query := "SELECT * FROM c"
    containerClient, err := getContainerClient(database, container)
    if err != nil {
        http.Error(w, "Failed to get container client", http.StatusInternalServerError)
        return
    }

    // 开启跨分区查询，这是全集合查询的核心配置
    queryOptions := &azcosmos.QueryOptions{
        EnableCrossPartitionQuery: true,
    }
    // 分区键传nil，表明要查询所有分区
    pager := containerClient.NewQueryItemsPager(query, nil, queryOptions)

    var assets []models.Asset
    for pager.More() {
        // 优先使用请求自带的Context，而非context.TODO()，支持请求取消时终止查询
        page, err := pager.NextPage(r.Context())

        if err != nil {
            log.Printf("Failed to execute query: %v", err)
            http.Error(w, "Failed to execute query", http.StatusInternalServerError)
            return
        }

        for _, item := range page.Items {
            var asset models.Asset
            if err := json.Unmarshal(item, &asset); err != nil {
                log.Printf("Failed to unmarshal item: %v", err)
                http.Error(w, "Failed to unmarshal item", http.StatusInternalServerError)
                return
            }
            assets = append(assets, asset)
        }
    }

    w.Header().Set("Content-Type", "application/json")
    if err := json.NewEncoder(w).Encode(assets); err != nil {
        log.Printf("Failed to encode response: %v", err)
        http.Error(w, "Failed to encode response", http.StatusInternalServerError)
    }
}

针对Golang新手的额外建议 #

• Context规范：尽量使用请求自带的r.Context()而非context.TODO()，这样在请求被用户取消时，能自动终止Cosmos DB的查询操作，避免不必要的资源消耗。
• 结构体字段标签：确保models.Asset的字段添加了正确的json映射标签，比如type Asset struct { AssetType string json:"assetType" }，否则json.Unmarshal无法正确匹配Cosmos DB文档中的字段，会导致返回对象的字段为空。
• 空结果友好处理：如果最终assets切片为空，可以主动返回[]而非默认的null，示例代码：

  if len(assets) == 0 {
      w.WriteHeader(http.StatusOK)
      w.Write([]byte("[]"))
      return
  }
  

• 日志细化：可以给不同阶段的错误日志添加前缀，比如[QUERY_ERROR]、[UNMARSHAL_ERROR]，方便快速定位问题。

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