Firestore add () в цикле forEach не работает

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.`);
});