Azure Functions中用Python递增Cosmos DB NoSQL文档值的问题

错误原因分析 #

你的代码存在以下几个关键问题导致500错误：

• 变量名冲突：最初从路由参数获取的name变量，被后续for name in inputDocument的循环覆盖，导致后续逻辑中name变成文档对象而非预期的路由参数值，引发类型错误。
• 变量作用域问题：user_json在for user in inputDocument循环内部定义，若inputDocument为空（未找到目标文档），循环不会执行，user_json未被初始化，后续访问时会抛出NameError。
• 硬编码文档标识：仍在硬编码id和testpartitionkey，没有从输入文档中动态获取，不符合"动态获取文档ID和分区键"的需求。
• 未处理文档不存在的情况：当inputDocument为空时，没有返回明确的错误响应，直接执行后续逻辑导致崩溃。

修正后的代码 #

import azure.functions as func
import json
import logging

def main(req: func.HttpRequest, inputDocument: func.DocumentList, outputDocument: func.Out[func.DocumentList]) -> func.HttpResponse:
    logging.info('Python HTTP trigger function processed a request.')

    # 避免变量名冲突，用doc_id存储路由参数的文档ID
    doc_id = req.route_params.get('name')
    if not doc_id:
        return func.HttpResponse("缺少文档ID参数", status_code=400)

    # 检查是否获取到目标文档
    if not inputDocument:
        return func.HttpResponse(f"未找到ID为{doc_id}的文档", status_code=404)

    # 假设inputDocument只返回一个匹配的文档（按ID和分区键查询）
    target_doc = inputDocument[0]
    doc_json = target_doc.to_json()
    logging.info(f"获取到目标文档: {doc_json}")
    
    # 将文档转换为字典，动态保留所有原有字段（包括ID和分区键）
    updated_doc = json.loads(doc_json)
    increment_key = "visitor"
    
    # 处理字段不存在的情况，初始化为1
    if increment_key in updated_doc:
        updated_doc[increment_key] += 1
    else:
        updated_doc[increment_key] = 1

    # 准备输出文档并更新
    new_docs = func.DocumentList()
    new_docs.append(func.Document.from_json(json.dumps(updated_doc)))
    outputDocument.set(new_docs)

    return func.HttpResponse(f"更新成功，{increment_key}的新值为: {updated_doc[increment_key]}", status_code=200)

关键改进点 #

• 变量名规范：使用doc_id存储路由参数的文档ID，避免与循环迭代变量冲突。
• 空值处理：增加对路由参数缺失、目标文档不存在的判断，返回明确的HTTP错误码和信息。
• 动态字段保留：通过target_doc.to_json()和json.loads()转换文档，自动保留所有原有字段（包括ID和分区键），无需硬编码。
• 容错处理：如果visitor字段不存在，自动初始化为1，避免键不存在的错误。

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