带有 Firebase Firestore 的 GraphQL 数据加载器

Question

我正在为这个项目使用 Firebase Firestore，我正在尝试实现数据加载器以减少对服务器的请求。

在这种特殊情况下，我有一个结构如下的 Category 集合： （我将使用 typescript 来更好地定义我的代码）

type Category = {
    name: String
    parent?: Category
}

当我查询 graphql 端点时，我会在解析器中验证该类别是否有父类，如果存在，我会查询数据库以获取详细信息。数据加载器来了。

这是我的 Firestore 的 Dataloader 实现：

 private dataLoader = new DataLoader(async (ids: readonly string[]) => {
    // firestore limits batches to 10
    let batches: Category[] = [];
    const tmp = [...ids];
    while (tmp.length) {
        const batch = tmp.splice(0, 10);
        const result = await this.collection.where(FieldPath.documentId(), 'in', [...batch]).get();
        batches = [...batches, ...result.docs.map(d => ({ id: d.id, ...d.data() } as Category))];
    }

    return batches;
});

解析器本身调用此方法

public async getById(id: string): Promise<Category> {
    const foo = await this.dataLoader.load(id);
    console.log('required', id, 'received', foo.id);
    return foo;
}

但正如您在此示例输出中看到的那样，它搞砸了很多时间：

required ztdOcVv9JdNp6U54ySed received 4jXbScd1qTwPStXocaoc
required Qnb86GsOy0gSgGxmMqYR received NeuCCRrJyurQTxvlAVhs
required NeuCCRrJyurQTxvlAVhs received Qnb86GsOy0gSgGxmMqYR
required 4jXbScd1qTwPStXocaoc received ztdOcVv9JdNp6U54ySed

我的理解是某个地方的异步会导致一些麻烦，但不知道在哪里或如何解决这个问题。这就是我在这里寻求帮助的原因。

提前致谢

Answer 1

您需要在 DataLoader 函数中返回 ids 的循环，将获取的对象（在本例中为类别）映射到相应的 id。

这是我如何实现类似的东西：

async function getItemsByIds(ids: string[]) {
  // where in supports max 10 values
  // https://firebase.google.com/docs/firestore/query-data/queries
  const chunks = _.chunk(ids, 10);
  const snapshots = await Promise.all(
    chunks.map((chunk) =>
      admin
        .firestore()
        .collection('items')
        .where(admin.firestore.FieldPath.documentId(), 'in', chunk)
        .get()
    )
  );
  return snapshots.flatMap((snapshot) => {
    return snapshot.docs.map((doc) => ({
      id: doc.id,
      ...doc.data(),
    }));
  });
}

const itemDataLoader = new DataLoader(async (ids) => {
  const items = await getItemsByIds(ids);
  return ids.map((id) => {
    const item = items.find((item) => item.id === id);
    if (!item) {
      return new Error('Item not found');
    }
    return item;
  });
})

注意：我使用 lodash 帮助每个请求分块 10 个项目并取消分块数组。