node.js - forEach 循环中的 Firestore add() 不起作用

Question

我正在尝试创建一个 Firebase 函数，将子集合从一个用户克隆到另一个用户。

但是，我在最后一步中遇到了问题:向新用户添加子集合文档。

这是我的功能:

import { Firestore } from '@google-cloud/firestore';

exports.handler = async function(req: any, res: any, db: Firestore) {
    const fromUid = req.body.from;
    const toUid = req.body.to;
    let transactionCount = 0;

    const userToRef = await db
        .collection('users')
        .doc(toUid)
        .collection('transactions');
    await userToRef.add({ ...{ data: 'some-dummy-data' } }); // At this point I get write to the collection
    const transactions = await db
        .collection(`users/${fromUid}/transactions`)
        .get();

    const transactionsList: any[] = [];
    transactions.forEach((doc: any) => {
        const transaction = doc.data();
        transaction.uid = toUid;
        transactionsList.push(transaction);
        userToRef.add({
            // Here write to the collection doesnt work
            ...transaction
        });
        transactionCount++;
    });
    console.log(transactionsList);

    res.send(
        `Successfully cloned ${transactionCount} transactions from ${fromUid} to ${toUid}.`
    );
    return '';
};

在上面的代码示例中，我对用户子集合进行了两次写入(请参阅代码注释)。第一个工作正常，第二个则不行。

知道为什么我不能在 forEach 循环中进行写入吗？

Answer 1

您应该使用Promise.all() ，以便等待对 userToRef 集合的所有写入完成，然后再通过 res 对象将响应发送回客户端。

以下改编代码应该可以工作(未经测试)。

但是我不清楚你最后应该做什么，因为我们不知道如何调用这个函数(你可能从云函数本身调用它？)。您可能不需要执行 return ''; 因为通过执行 res.send() 您实际上正在终止该函数。为了供您引用，我在下面添加了经过测试的 HTTPS 云函数的完整代码，该函数执行一组类似的集合写入。

exports.handler = async function (req: any, res: any, db: Firestore) {
    const fromUid = req.body.from;
    const toUid = req.body.to;
    let transactionCount = 0;

    const userToRef = await db
        .collection('users')
        .doc(toUid)
        .collection('transactions');
    await userToRef.add({ ...{ data: 'some-dummy-data' } }); // At this point I get write to the collection
    const transactions = await db
        .collection(`users/${fromUid}/transactions`)
        .get();

    //const transactionsList: any[] = [];

    const promises: Promise<DocumentReference>[] = [];

    transactions.forEach((doc: any) => {
        const transaction = doc.data();
        transaction.uid = toUid;
        //transactionsList.push(transaction);
        promises.push(
            userToRef.add({
                // Here write to the collection doesnt work
                ...transaction
            }));
        transactionCount++;
    });
    //console.log(transactionsList);

    await Promise.all(promises);
    res.send(
        `Successfully cloned ${transactionCount} transactions from ${fromUid} to ${toUid}.`
    );
    //return '';
};

<小时/>

以下是实现一组类似集合写入的 HTTPS 云函数的代码:

import * as functions from 'firebase-functions';
import * as admin from 'firebase-admin';
import { DocumentReference } from '@google-cloud/firestore';

admin.initializeApp();

export const helloWorld = functions.https.onRequest(async (req, res) => {
  const db = admin.firestore();

  let writeCount = 0;

  const theArray: string[] = ['AA', 'BB', 'CC'];

  const promises: Promise<DocumentReference>[] = [];

  theArray.forEach((value: string) => {
    promises.push(db.collection('testCollection').add({ val: value }));
    writeCount++;
  });

  await Promise.all(promises);

  res.send(`Successfully wrote ${writeCount} documents.`);
});