Azure Function（Node.js）：POST请求后获取DocumentDB生成的ID

我明白你现在的困扰：通过Azure Function的HTTP触发器发起POST请求，用Cosmos DB输出绑定插入文档后，想把带有Cosmos自动生成的_id字段的新文档返回给客户端，但目前的代码实现做不到这一点。这是因为默认的documentDB输出绑定是单向的——它只负责把你传入的文档写入数据库，不会将写入后带ID的文档回传给函数上下文。下面给你两种解决方案，推荐第一种：

方案1：使用Cosmos DB JavaScript SDK手动插入（推荐） #

这种方式能直接获取到插入后带自动生成ID的完整文档，逻辑清晰且可靠。

步骤1：修改function.json #

移除原来的documentDB输出绑定，因为我们要手动用SDK处理：

{
  "bindings": [
    {
      "authLevel": "function",
      "type": "httpTrigger",
      "direction": "in",
      "name": "req",
      "methods": ["post"],
      "route": "blocks"
    },
    {
      "type": "http",
      "direction": "out",
      "name": "res"
    }
  ],
  "disabled": false
}

步骤2：更新index.js代码 #

用SDK初始化客户端，插入文档后直接返回带ID的结果：

const { CosmosClient } = require("@azure/cosmos");

module.exports = async function (context, req) {
    context.log('Add block to CosmosDB.');
    
    // 校验请求体是否存在
    if (!req.body) {
        context.res = {
            status: 400,
            body: "请在请求体中提供block对象。"
        };
        return;
    }

    try {
        // 初始化Cosmos客户端，使用配置好的连接字符串
        const client = new CosmosClient(process.env.CosmosDBConnection);
        const database = client.database("digitalnetworkacquisition");
        const container = database.container("blocks");

        // 插入文档，Cosmos会自动生成_id字段
        const { resource: createdBlock } = await container.items.create(req.body);

        // 返回带自动生成ID的文档给客户端
        context.res = {
            status: 201, // 201表示资源创建成功
            headers: {
                "Content-Type": "application/json"
            },
            body: createdBlock
        };
    } catch (error) {
        context.log.error("插入block时出错:", error);
        context.res = {
            status: 500,
            body: "无法在Cosmos DB中创建block。"
        };
    }
};

步骤3：安装Cosmos DB SDK包 #

在你的Function项目目录下运行命令安装依赖：

npm install @azure/cosmos

方案2：继续使用输出绑定（不推荐） #

如果你坚持想用输出绑定，只能在插入后通过查询获取文档，但这种方式有明显局限性：需要文档有唯一标识字段（比如自定义的唯一键），而且受Cosmos DB一致性级别的影响，可能出现短暂的查询不到的情况。代码示例如下（仅作参考）：

const { CosmosClient } = require("@azure/cosmos");

module.exports = async function (context, req) {
    context.log('Add block to CosmosDB.');
    
    if (!req.body) {
        context.res = {
            status: 400,
            body: "请在请求体中提供block对象。"
        };
        return;
    }

    // 用输出绑定插入文档
    context.bindings.block = req.body;

    // 初始化客户端查询文档（假设你的block有一个uniqueKey字段作为唯一标识）
    const client = new CosmosClient(process.env.CosmosDBConnection);
    const database = client.database("digitalnetworkacquisition");
    const container = database.container("blocks");

    const query = `SELECT * FROM c WHERE c.uniqueKey = @uniqueKey`;
    const { resources: [createdBlock] } = await container.items.query(query, { 
        parameters: [{ name: "@uniqueKey", value: req.body.uniqueKey }] 
    }).fetchAll();

    if (createdBlock) {
        context.res = {
            status: 201,
            body: createdBlock
        };
    } else {
        context.res = {
            status: 500,
            body: "创建block后无法查询到结果。"
        };
    }
};

关键说明 #

为什么输出绑定不能直接返回带ID的文档？因为documentDB输出绑定的设计是单向写入：它接收你传入的对象并写入数据库，但不会将写入后带自动生成_id的文档回传给函数上下文。所以你设置的context.bindings.block始终是原来的请求体对象，没有新增的ID字段。

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