批量获取 DocumentReferences？

debugcn 发表于 Dev

药物

我正在尝试改进 firestore get 功能，我有类似的东西：

    return admin.firestore().collection("submissions").get().then(
        async (x) => {
            var toRet: any = [];
            for (var i = 0; i < 10; i++) {
                try {
                    var hasMedia = x.docs[i].data()['mediaRef'];
                    if (hasMedia != null) {
                        var docData = (await x.docs[i].data()) as MediaSubmission;

                        let submission: MediaSubmission = new MediaSubmission();
                        submission.author = x.docs[i].data()['author'];
                        submission.description = x.docs[i].data()['description'];

                        var mediaRef = await admin.firestore().doc(docData.mediaRef).get();
                        submission.media = mediaRef.data() as MediaData;
                        toRet.push(submission);
                    }
                }
                catch (e) {
                    console.log("ERROR GETTIGN MEDIA: " + e);
                }
            }
            return res.status(200).send(toRet);
        });

第一次得到很好，但性能最差就行了：

var mediaRef = await admin.firestore().doc(docData.mediaRef).get();

我认为这是因为呼叫没有分批。

是否可以对一组 mediaRef 进行批量获取以提高性能？

本质上，我有一组文档，这些文档的外部引用由指向单独集合中路径的字符串存储，并且已证明获取这些引用很慢。

奇图斯

那这个呢？我做了一些重构以使用更多的等待/异步代码，希望我的评论有帮助。

主要思想是使用Promise.all并等待所有 mediaRefs 检索

async function test(req, res) {
  // get all docs
  const { docs } = await admin
    .firestore()
    .collection('submissions')
    .get();

  // get data property only of docs with mediaRef
  const datas = await Promise.all(
    docs.map(doc => doc.data()).filter(data => data.mediaRef),
  );

  // get all media in one batch - this is the important change
  const mediaRefs = await Promise.all(
    datas.map(({ mediaRef }) =>
      admin
        .firestore()
        .doc(mediaRef)
        .get(),
    ),
  );

  // create return object
  const toRet = datas.map((data: MediaSubmission, i) => {
    const submission = new MediaSubmission();
    submission.author = data.author;
    submission.description = data.description;

    submission.media = mediaRefs[i].data() as MediaData;

    return submission;
  });
  return res.status(200).send(toRet);
}

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。