Firebase 函数在同一个 onCall 函数中读取和添加数据

Question

我有一个功能，当我单击按钮时会触发该功能。根据 docId，我可以在 Firestore 中检索正确的下载 URL。但我也想更新我的“客户”集合中的特定字段（availableDownloads）。我无法让它工作。

此代码运行良好。它返回下载地址。

exports.getDownloadUrl = functions.https.onCall(async(data, context) => {
    var docRef= await db.collection('projects').doc(data.docId);
    return docRef.get().then(function(doc){
        const downloadURL = doc.data().downloadURL;
        return downloadURL;
    }).catch(function(error) {
        // Handle error
    });
});

然而事实并非如此。它返回 null

exports.getDownloadUrl = functions.https.onCall(async(data, context) => {
    var docRef= await db.collection('projects').doc(data.docId);
    return docRef.get()
    .then(async function(doc){
        const downloadURL = doc.data().downloadURL;
        const userRef = await db.collection('customers').doc(context.auth.uid);
        return userRef.update({
            availableDownloads: admin.firestore.FieldValue.increment(-1)
        }).then(()=> {
            return downloadURL;
        }).catch((error)=> {
            
        })
        
    }).catch(function(error) {
        // Handle error
    });
});

Answer 1

我首先想到的是.update() is an asynchronous function，所以你需要在userRef.update(...)前面加上一个await。

您还应该考虑坚持使用箭头函数 (() => {}) 或匿名函数 (function () {})，但这只是一种风格说明。 :)

ETA：正如 Frank van Puffelen 指出的那样，您不需要只需awaits 即可简单地声明 DocumentReference，因为这些不是异步函数。 （看var docRef= await db.collection('projects').doc(data.docId);这一行）。

Answer 2

除了async/await 与then/catch 混合使用的潜在问题外，您的主要问题是由于 variable scoping。 downloadURL 变量是在第一个 then() 块中声明的局部变量，因此在第二个 then() 块中不可访问。

用下面的代码可以很容易看出问题：

  function later(delay, value) {
    return new Promise((resolve) => setTimeout(resolve, delay, value));
  }

  later(500, 'value1')
    .then((value) => {
      console.log('First then block: ' + value);
      const downloadURL = value;
      return later(1000, 'value2');
    })
    .then((value) => {
      console.log('Second then block: ' + value);
      console.log('Second then block: ' + downloadURL); // Here it is going to throw an error
    })
    .catch((error) => {
      console.log(error);
    });

---

需要在Cloud Function的作用域内声明downloadURL变量，如下：

exports.getDownloadUrl = functions.https.onCall((data, context) => {
    let downloadURL;
    const docRef = db.collection('projects').doc(data.docId);
    return docRef.get()
        .then((doc) => {
            downloadURL = doc.data().downloadURL;
            const userRef = db.collection('customers').doc(context.auth.uid);
            return userRef.update({
                availableDownloads: admin.firestore.FieldValue.increment(-1)
            });
        })
        .then(() => {
            return downloadURL;
        }).catch((error) => {
            // Handle error
        })
});

---

请注意，上面的代码没有使用 async/await。如果要使用 async/await，请执行以下操作：

exports.getDownloadUrl = functions.https.onCall(async (data, context) => {

    try {
        const docRef = db.collection('projects').doc(data.docId);

        const doc = await docRef.get();
        const downloadURL = doc.data().downloadURL;

        const userRef = db.collection('customers').doc(context.auth.uid);
        await userRef.update({
            availableDownloads: admin.firestore.FieldValue.increment(-1)
        })

        return downloadURL;
    } catch (error) {
        // Handle error
    }

});

您可以看到，使用此代码，由于使用 async/await 对其进行了简化，因此不会因为变量作用域而变得更加复杂。