Cosmos DB存储过程并发调用下文档唯一性创建的可行性咨询

首先直接给结论：你当前的存储过程无法确保同一文档仅被创建一次，在多客户端并发调用的场景下，大概率会出现重复创建的情况，原因和解决方案如下：

为什么当前代码会出问题？ #

核心原因是竞态条件（Race Condition）：

• 当多个客户端几乎同时调用这个存储过程时，它们都会先执行查询操作，此时目标文档还未被任何一个客户端创建出来，所以所有请求都会进入「文档不存在，创建新文档」的分支。
• 虽然Cosmos DB的存储过程是在单个分区内原子执行的，但如果你的查询没有限定在单个分区（比如Param1/Param2不是集合的分区键），查询会跨分区执行，此时存储过程的事务无法覆盖整个查询+创建的流程，进一步放大了重复创建的可能性。
• 另外，你的代码存在SQL注入风险，直接将参数拼接到查询字符串中，如果参数包含特殊字符（比如"），会导致查询报错甚至被恶意利用。

如何确保文档仅被创建一次？ #

要彻底解决这个问题，需要结合Cosmos DB的唯一键约束和存储过程的事务特性，以下是具体方案：

1. 给集合添加唯一键约束 #

这是最可靠的底层保障：

• 登录Azure门户，找到你的Cosmos DB集合，在「设置」→「唯一键」中添加/Param1,/Param2的组合约束。这样Cosmos DB会自动拒绝任何重复插入Param1+Param2组合的请求，抛出HTTP 409（冲突）错误。

2. 修改存储过程处理冲突并修复SQL注入问题 #

修改后的存储过程会捕获唯一键冲突错误，重新查询并返回已存在的文档，同时使用参数化查询避免注入风险：

function createIfNotExists(param1, param2) {
    var collection = getContext().getCollection();
    var response = getContext().getResponse();
    
    // 使用参数化查询避免SQL注入
    var querySpec = {
        query: 'SELECT * FROM c WHERE c.Param1 = @param1 AND c.Param2 = @param2',
        parameters: [
            { name: '@param1', value: param1 },
            { name: '@param2', value: param2 }
        ]
    };

    var isAccepted = collection.queryDocuments(
        collection.getSelfLink(),
        querySpec,
        function (err, items, options) {
            if (err) throw err;
            
            if (!items || items.length === 0) {
                // 尝试创建文档
                var createAccepted = collection.createDocument(
                    collection.getSelfLink(),
                    { Param1: param1, Param2: param2 },
                    function (createErr, itemCreated) {
                        if (createErr) {
                            // 捕获唯一键冲突错误（HTTP 409）
                            if (createErr.code === 409) {
                                // 冲突发生，重新查询并返回已存在的文档
                                var retryQueryAccepted = collection.queryDocuments(
                                    collection.getSelfLink(),
                                    querySpec,
                                    function (retryErr, retryItems) {
                                        if (retryErr) throw retryErr;
                                        response.setBody(retryItems[0]);
                                    }
                                );
                                if (!retryQueryAccepted) throw new Error("Retry query failed");
                            } else {
                                throw createErr;
                            }
                        } else {
                            response.setBody(itemCreated);
                        }
                    }
                );
                if (!createAccepted) throw new Error("Could not create document");
            } else {
                response.setBody(items[0]);
            }
        }
    );
    if (!isAccepted) throw new Error("The query was not accepted by the server.");
}

3. 额外优化：绑定存储过程到单个分区 #

如果Param1（或Param2）可以作为集合的分区键，那么在调用存储过程时指定分区键值，这样存储过程的查询和创建操作都会在单个分区的事务上下文中执行，进一步降低竞态发生的概率（不过唯一键约束仍然是必需的，因为它能阻止任何途径的重复插入）。

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