如何确保云功能中的所有操作均已成功完成?

时间:2018-09-09 10:33:51

标签: javascript firebase asynchronous google-cloud-firestore google-cloud-functions

我正在使用Firebase Cloud Functions,该功能由在Firestore中创建文档触发。创建对象时,我需要并行进行两个不同的操作:

  1. 更新特定文档中的字段值(而不是创建并触发云功能的文档)
  2. 在另一个文档上进行交易

所以我的问题是:

  1. 在结束云功能本身之前,如何确保我的两项操作均已成功完成
  2. 如何为两个操作中的每一个实现单独的重试机制(因为我不希望为整个功能使用通用的重试机制,因为它可以重做事务操作,即使它是另一个操作失败)?

这是我当前的代码:

exports.onCityCreated = functions.firestore
    .document('Cities/{cityId}')
    .onCreate((snap, context) => {
        const db = admin.firestore(); 
        const newCity = snap.data();
        const mayorId = newEvent.mayorID;
        const mayorRef = db.doc('Users/'+ mayorId);

        const timestamp = admin.firestore.FieldValue.serverTimestamp();
        db.doc('Utils/lastPost').update({timestamp: timestamp});    //First Operation - Timestamp Update

        return db.runTransaction(t => {    //Second Operation - Transaction
                return t.get(mayorRef).then(snapshot => {
                    var new_budget = snapshot.data().b - 100;
                    return t.update(mayorRef, {b: new_budget});
                })
            .then(result => {
                return console.log('Transaction success!');
            })
            .catch(err => {
                console.log('Transaction failure:', err);
            });
        });
});

1 个答案:

答案 0 :(得分:2)

只要您有这样的多种操作,解决方案就是使用Promise.all()。这需要一组诺言,然后返回一个诺言,当您传递的所有诺言都得到解决时,该诺言就会解决。

exports.onCityCreated = functions.firestore
    .document('Cities/{cityId}')
    .onCreate((snap, context) => {
        const db = admin.firestore(); 
        const newCity = snap.data();
        const mayorId = newEvent.mayorID;
        const mayorRef = db.doc('Users/'+ mayorId);

        const timestamp = admin.firestore.FieldValue.serverTimestamp();
        var p1 = db.doc('Utils/lastPost').update({timestamp: timestamp});

        var p1 = db.runTransaction(t => {
                return t.get(mayorRef).then(snapshot => {
                    var new_budget = snapshot.data().b - 100;
                    return t.update(mayorRef, {b: new_budget});
                })
        });
        return Promise.all([p1, p2]);
});