CosmosDB存储过程 - 承诺而不是回调

时间:2017-10-24 12:18:21

标签: javascript asynchronous stored-procedures azure-cosmosdb

是否可以在CosmosDB(DocumentDB)存储过程API调用中使用javascript promises而不是常规回调?用法是实施分页。

例如

token = getToken();
doSomething(token);    

//....
function getToken(....) {   

  //...
  collection.queryDocuments(link, query, queryOptions, function(error, documents, responseOptions) {
    return responseOptions.continuation;
  });
}

将无效,因为令牌在回调中返回,并继续执行。您能举例说明如何实现这个目标吗?

3 个答案:

答案 0 :(得分:1)

Cosmos DB docs中引用的ECMAScript版本支持async / await和Promises。我可以在我的存储过程中使用这两个。

这是一个函数,它返回一个生成参数化文档查询的promise:

function queryDocumentsAsync(sql, parameters, options) {
    const querySpec = {
        query: sql,
        parameters: parameters
    };

    return new Promise((resolve, reject)=>{
        let isAccepted = __.queryDocuments(__.getSelfLink(), querySpec, options || {}, (err, feed, options) => {
            if(err) reject(err);
            resolve({
                feed: feed,
                options: options
            });
        });
        if(!isAccepted) throw "Query was not accepted.";
    });
}

我看到了使用这种方法强制回滚的一些限制。如果你抛出一个错误,它会被承诺链吞没而永远不会离开。

答案 1 :(得分:0)

这是一个有关如何使用异步等待进行查询和替换方案的示例。

function async_sample() {
    const ERROR_CODE = {
        NotAccepted: 429
    };

    const asyncHelper = {
        queryDocuments(sqlQuery, options) {
            return new Promise((resolve, reject) => {
                const isAccepted = __.queryDocuments(__.getSelfLink(), sqlQuery, options, (err, feed, options) => {
                    if (err) reject(err);
                    resolve({ feed, options });
                });
                if (!isAccepted) reject(new Error(ERROR_CODE.NotAccepted, "replaceDocument was not accepted."));
            });
        },

        replaceDocument(doc) {
            return new Promise((resolve, reject) => {
                const isAccepted = __.replaceDocument(doc._self, doc, (err, result, options) => {
                    if (err) reject(err);
                    resolve({ result, options });
                });
                if (!isAccepted) reject(new Error(ERROR_CODE.NotAccepted, "replaceDocument was not accepted."));
            });
        }
    };

    async function main() {
        let continuation;
        do {
            let { feed, options } = await asyncHelper.queryDocuments("SELECT * from c", { continuation });

            for (let doc of feed) {
                doc.newProp = 1;
                await asyncHelper.replaceDocument(doc);
            }

            continuation = options.continuation;
        } while (continuation);
    }

    main().catch(err => getContext().abort(err));
}

答案 2 :(得分:0)

借助一些技巧,您可以使用webpack内联节点依赖项,包括promisify,它可以使您执行此操作:

https://github.com/Oblarg/cosmosdb-storedprocs-ts/blob/master/BuildStoredProcs.js