如何在Azure Cosmos DB上获得批量插入的延续令牌？

Question

我想上传一个CSV文件，该文件表示要以快速和原子的方式添加到我的Cosmos DB集合中的10k文档。我有一个类似以下伪代码的存储过程：

function createDocsFromCSV(csv_text) {
    function parse(txt) { // ... parsing code here ... }

    var collection = getContext().getCollection();
    var response = getContext().getResponse();

    var docs_to_create = parse(csv_text);
    for(var ii=0; ii<docs_to_create.length; ii++) {
        var accepted = collection.createDocument(collection.getSelfLink(),
                                                    docs_to_create[ii],
                                                    function(err, doc_created) {
                                                        if(err) throw new Error('Error' + err.message);
                                                    });
        if(!accepted) {
            throw new Error('Timed out creating document ' + ii);
        }
    }
}

当我运行它时，存储过程在超时之前创建了大约1200个文档（因此回滚并不创建任何文档）。

以前，我使用延续标记在存储过程中成功更新（而不是创建）数千个文档，此答案作为指导：https://stackoverflow.com/a/34761098/277504。但是在搜索文档（例如 https://azure.github.io/azure-documentdb-js-server/Collection.html）之后，我没有看到像查询文档那样通过创建文档来获取延续令牌的方法。

有没有办法利用存储过程来创建批量文档？

Answer 1

值得注意的是，存储过程有执行限制，所有操作必须在服务器指定的请求超时期限内完成。如果操作没有在该时间限制内完成，事务将自动回滚。

为了简化开发以处理时间限制，所有CRUD（创建、读取、更新和删除）操作都返回一个布尔值，表示该操作是否会完成。此布尔值可用于结束执行的信号和实现基于延续的模型以恢复执行（这在我们下面的代码示例中说明）。更多详细信息，请参阅文档。

上面提供的批量插入存储过程通过返回成功创建的文档数量来实现延续模型。

伪码：

function createDocsFromCSV(csv_text,count) {
    function parse(txt) { // ... parsing code here ... }

    var collection = getContext().getCollection();
    var response = getContext().getResponse();

    var docs_to_create = parse(csv_text);
    for(var ii=count; ii<docs_to_create.length; ii++) {
        var accepted = collection.createDocument(collection.getSelfLink(),
                                                    docs_to_create[ii],
                                                    function(err, doc_created) {
                                                        if(err) throw new Error('Error' + err.message);
                                                    });
        if(!accepted) {
            getContext().getResponse().setBody(count);
        }
    }
}

然后，您可以检查客户端的输出文档计数，并使用count参数重新运行存储过程，以创建剩余的文档集，直到计数大于csv_text的长度。

希望对你有帮助。