📡

Metaに学ぶ、大規模開発のデータフェッチ設計と最適化

に公開

データフェッチの設計は、保守性とパフォーマンスに強く影響します。これらはよくトレードオフに扱われ、パフォーマンスを優先すると保守性が犠牲に、保守性を優先するとパフォーマンスが犠牲になりがちです。

Metaでは大規模開発における保守性とパフォーマンスの両立を目指し、研究が行われてきました。本稿では、Metaが重視している自律分散型のデータフェッチ設計と、それを支えるバッチングと短命のキャッシュについて解説します。

要約

  • 大規模開発における保守性には、自律分散型の設計思想が重要
  • 無策にデータフェッチコロケーションすると、パフォーマンスがトレードオフになる
  • Metaではバッチングと短命のキャッシュによって、保守性とパフォーマンスがトレードオフにならないようにしてる
  • DataLoaderReact Cacheは、これらを容易に実現する手段

前提

本稿では、データフェッチ設計を考える上で以下の考え方を前提とします。

APIの粒度

REST APIにおいて、責務が大きすぎる粗粒度なAPIはGod APIと呼ばれ、通信回数を抑えられる反面、変更容易性やAPI自体のパフォーマンス問題が起きやすい傾向にあります。一方、責務が小さく設計された細粒度なAPIはChatty API(おしゃべりなAPI)と呼ばれ、データフェッチをコロケーション[1]できるなどのメリットが得られる一方、通信回数が増えたりデータフェッチのウォーターフォールが発生しやすいため、アプリケーションのパフォーマンス劣化要因になりえます。

これらはそれぞれアンチパターンとされることがありますが、実際には観点次第で最適解が異なるので、一概にアンチパターンなのではなくそれぞれトレードオフが伴うと捉えるべきです。

リソース単位の粒度 設計観点 パフォーマンス(低速な通信) パフォーマンス(高速な通信)
細粒度
粗粒度

本稿で扱うデータフェッチは高速なサーバー間通信を前提にしているため、バックエンドは細粒度なREST APIで設計することが最適だと考えます。

データフェッチの設計パターン

筆者の考えでは、データフェッチの設計は大きく2パターンに分けられます。データフェッチ層を設けるなどするような中央集権型の設計と、データフェッチコロケーションに代表される自律分散型の設計です。

  • 中央集権型: 責務を集約し、一元管理を重視する
  • 自律分散型: 責務を末端に分散し、自律性を重視する

MetaやReactにおける自律分散型の設計の歴史については、筆者の前回の記事で詳細に解説しています。興味のある方はご参照ください。

https://zenn.dev/akfm/articles/react-team-vision

解説

冒頭で触れたように、Metaでは自律分散型の設計が重視されており、特に大規模開発の保守性において重要だと考えられています。データフェッチ層を設けるような中央集権型の設計はなぜ好まれないのでしょう?

実装例を元に問題点を考察してみます。

中央集権型の弊害

例として、Next.jsでブログ記事一覧ページを実装することを考えてみます。

APIのエンドポイントごとに以下のような関数がすでに定義されているものとします。fetchPosts()で得られるPost[]は著者情報、コメント数、閲覧数の詳細を含みません

  • fetchPosts(options: { page: number }): 記事一覧の取得
  • fetchAuthors(authorIds: string[]): 複数著者情報の取得
  • fetchCommentCountsForPosts(postIds: string[]): 複数記事のコメント数取得
  • fetchCommentCount(postId: string): 単一記事のコメント数取得
  • fetchViewCountsForPosts(postIds: string[]): 複数記事の閲覧数取得
  • fetchViewCount(postId: string): 単一記事の閲覧数取得
より詳細な定義
type Post = {
  id: string;
  title: string;
  authorIds: string[];
  summary: string;
};

async function fetchPosts({ page }: { page: number }) {
  const res = await fetch(`${API_URL}/posts?page=${page}`);
  const posts: Post[] = await res.json();
  return posts;
}

type Author = {
  id: string;
  name: string;
};

async function fetchAuthors(authorIds: string[]) {
  const res = await fetch(
    `${API_URL}/authors?${authorIds.map((id) => `id=${id}`).join("&")}`,
  );
  const authors: Author[] = await res.json();
  return authors;
}

type CommentCount = {
  postId: string;
  count: number;
};

async function fetchCommentCountsForPosts(postIds: string[]) {
  const res = await fetch(
    `${API_URL}/posts/comments_counts?${postIds.map((id) => `postId=${id}`).join("&")}`,
  );
  const commentCounts: CommentCount[] = await res.json();
  return commentCounts;
}

async function fetchCommentCount(postId: string) {
  const res = await fetch(`${API_URL}/posts/${postId}/comments_count`);
  const commentCount: CommentCount = await res.json();
  return commentCount;
}

type ViewCount = {
  postId: string;
  count: number;
};

async function fetchViewCountsForPosts(postIds: string[]) {
  const res = await fetch(
    `${API_URL}/posts/view_counts?${postIds.map((id) => `postId=${id}`).join("&")}`,
  );
  const viewCounts: ViewCount[] = await res.json();
  return viewCounts;
}

async function fetchViewCount(postId: string) {
  const res = await fetch(`${API_URL}/posts/${postId}/view_count`);
  const viewCount: ViewCount = await res.json();
  return viewCount;
}

以下は<Page>コンポーネントを中央集権的なデータフェッチ層として扱う実装例です。記事のタイトルとサマリーのみが表示されます。

export async function Page(props: {
  searchParams: Promise<{ page?: string }>;
}) {
  const searchParams = await props.searchParams;
  const page = searchParams.page ? Number(searchParams.page) : 1;
  const posts = await fetchPosts({
    page,
  }); // Post[]

  // ...`posts`を参照
}

type Post = {
  id: string;
  title: string;
  authorIds: string[];
  summary: string;
};

これは非常にシンプルでわかりやすい例です。このままでも特に問題ないでしょう。

しかし、このままではPostに含まれている情報が少なく、ブログ一覧として出せる情報も少なすぎるので、以下の情報を追加で表示する改修をするとします。

  • 著者情報
  • コメント数
  • 閲覧数

以下は改修後の実装例です。

export async function Page(props: {
  searchParams: Promise<{ page?: string }>;
}) {
  const searchParams = await props.searchParams;

  // ベースとなるブログ一覧を取得
  const page = searchParams.page ? Number(searchParams.page) : 1;
  const posts = await fetchPosts({
    page,
  });

  // ブログ一覧を補強する情報を一括で取得
  const postIds = posts.map((post) => post.id);
  const uniqueAuthorIds = Array.from(
    new Set(posts.flatMap((post) => post.authorIds)),
  );
  const [allAuthors, commentCounts, viewCounts] = await Promise.all([
    fetchAuthors(uniqueAuthorIds),
    fetchCommentCountsForPosts(postIds),
    fetchViewCountsForPosts(postIds),
  ]);

  // `richPosts: RichPost[]`を組み立て
  const authorsMap = new Map(allAuthors.map((author) => [author.id, author]));
  const commentCountsMap = new Map(
    commentCounts.map((item) => [item.postId, item.count]),
  );
  const viewCountsMap = new Map(
    viewCounts.map((item) => [item.postId, item.count]),
  );
  const richPosts = posts.map((post) => {
    const authors = post.authorIds
      .map((id) => authorsMap.get(id))
      .filter((author) => author !== undefined);
    const comments = commentCountsMap.get(post.id) ?? 0;
    const viewCount = viewCountsMap.get(post.id) ?? 0;

    return {
      ...post,
      authors,
      comments,
      viewCount,
    };
  });

  // ...`richPosts`を参照
}

データフェッチは計4回、うち3つはPromise.all()によって並行化することでデータフェッチは2段階に整理されており、God APIを避けつつある程度最適化されたデータフェッチ設計になっています。

一方、保守性の観点で言うとどうでしょうか?おそらく人によって様々だと思うのですが、筆者は依存関係が複雑で読みづらいと感じます。この程度なら許容範囲内という人でも、組み合わせる配列の数が更に増えていくと保守性に乏しいと感じるのではないでしょうか。

関数に分離するアプローチについて

「複雑に感じるなら関数に分離すればいい」という考え方もあるかもしれませんが、中央集権的な設計は集権された層で管理できることに価値があるため、関数分離は良い解決策にはならないと筆者は考えます。

例えば複数のデータフェッチを1つの関数に集約すると、データフェッチ層でどれだけデータフェッチを行っているか分かりづらくなります。

export async function Page(props: {
  searchParams: Promise<{ page?: string }>;
}) {
  const searchParams = await props.searchParams;
  const richPosts = await fetchAllRichPosts();

  // ...`richPosts`を参照
}

この場合、fetchAllRichPosts()でどれだけリクエストが走ったのか不透明です。fetchAllRichPosts()には含めたくないが、共有したいデータフェッチが増えるとしたら、修正は非常に面倒になるでしょう。

データフェッチ以外を関数抽出するアプローチも考えられます。

export async function Page(props: {
  searchParams: Promise<{ page?: string }>;
}) {
  const searchParams = await props.searchParams;
  // ベースとなるブログ一覧を取得
  const page = searchParams.page ? Number(searchParams.page) : 1;
  const posts = await fetchPosts({
    page,
  });

  // ブログ一覧を補強する情報を一括で取得
  const postIds = posts.map((post) => post.id);
  const uniqueAuthorIds = Array.from(
    new Set(posts.flatMap((post) => post.authorIds)),
  );
  const [allAuthors, commentCounts, viewCounts] = await Promise.all([
    fetchAuthors(uniqueAuthorIds),
    fetchCommentCountsForPosts(postIds),
    fetchViewCountsForPosts(postIds),
  ]);

  // `RichPost`を組み立て
  const richPosts = mergePosts({
    posts,
    allAuthors,
    commentCounts,
    viewCounts,
  });

  // ...`richPosts`を参照
}

元の実装と比べ、あまり短くなりませんでした。また、UIが参照しているデータの由来を調べる際には、mergePosts()の実装を確認する必要があります。

このように、関数に抽出するだけではおおよそ本質的な保守性の改善は見込めません。また、これらのアプローチはシンプルなルールにしづらいため、一貫性の欠如にも繋がりやすいと筆者は考えます。

自律分散型の弊害

一方、自律分散型の設計を採用し、必要なデータを必要な時に取得するデータフェッチコロケーションを適用すると、コードの見通しがとても良くなります。

具体的には、<Page>では必要なデータを全て揃えるようにするのではなく、記事一覧をループするのに最低限必要となる記事情報の取得のみを行います。著者情報・コメント数・閲覧数といった付加情報の取得は、実際にこれらが必要になる記事単位のコンポーネント<PostCassette>などで行います。

// page.tsx
export async function Page(props: {
  searchParams: Promise<{ page?: string }>;
}) {
  const searchParams = await props.searchParams;
  const page = searchParams.page ? Number(searchParams.page) : 1;
  const posts = await fetchPosts({
    page,
  });

  // `posts`をループして`<PostCassette>`を組み立てる
}

// post-cassette.tsx
export async function PostCassette({ post }: { post: Post }) {
  const [authors, comments, viewCount] = await Promise.all([
    fetchAuthors(post.authorIds),
    fetchCommentCount(post.id),
    fetchViewCount(post.id),
  ]);

  // ...`authors`, `comments`, `viewCount`を参照
}

修正前と比べて非常に読みやすく、シンプルになりました。データフェッチが参照単位に分割されたため可読性が高く、修正時のデグレリスクも低いと考えられます。このように自律分散的な設計は、読み手に必要なコンテキストを小さく留めることができます。大規模な開発では予測性は非常に重要なため、Metaは自律分散的な設計を重視しています。

しかし一方で、パフォーマンス観点では非常に大きな問題が発生します。<Page>では以下のように一括でデータフェッチを行なっていました。

const [allAuthors, commentCounts, viewCounts] = await Promise.all([
  fetchAuthors(uniqueAuthorIds),
  fetchCommentCountsForPosts(postIds),
  fetchViewCountsForPosts(postIds),
]);

修正後のコードでは、<PostCassette>内でpostの情報を元にデータフェッチをしています。

const [authors, comments, viewCount] = await Promise.all([
  fetchAuthors(post.authorIds),
  fetchCommentCount(post.id),
  fetchViewCount(post.id),
]);

<PostCassette>はループでレンダリングされるので、修正前は4回だったデータフェッチがpostsの取得1回+postsの取得分×3回分に増えており、典型的なN+1を引き起こしています。もしpostが100件だった場合、301回分のデータフェッチが発生します。

データフェッチはパフォーマンス観点でボトルネックになりやすい部分です。データフェッチの極端な増加は、無視できない非常に大きな問題です。

データフェッチのバッチング

ここまでの話を整理してみます。中央集権型の設計より自律分散型の設計の方が、保守性には優れています。しかし、自律分散型の設計では無視できないパフォーマンス問題を引き起こす可能性が高いと言えます。この場合、保守性とパフォーマンスはトレードオフするしかないのでしょうか?

Metaではこの問題をバッチングによって解決しています。バッチングとは、複数のデータフェッチを1つにまとめて、効率的に処理する機構です。具体的な仕組み[2]としては、Node.jsのprocess.nextTick()やブラウザ側のsetImmediate()などを利用して、データフェッチするタイミングを「少し待つ」ことで、バッチングを実現します。

これを容易に実現するため、MetaはDataLoaderというライブラリをOSSで提供しています。以下はDataLoaderを用いて著者情報のデータフェッチングをバッチングする例です。

async function authorsBatch(authorIds: readonly string[]) {
  const res = await fetch(
    `${API_URL}/authors?${authorIds.map((id) => `id=${id}`).join("&")}`,
  );
  if (!res.ok) {
    console.error("Failed to fetch authors in batch:", await res.text());
    return authorIds.map(() => null);
  }
  const allAuthors: Author[] = await res.json();

  return authorIds.map(
    (authorId) => allAuthors.find((author) => author.id === authorId) ?? null,
  );
}

// const authorLoader = new DataLoader(authorsBatch);
// authorLoader.load("1");
// authorLoader.load("2");
// 呼び出しはDataLoaderによってまとめられ、`authorsBatch(["1", "2"])`が呼び出される

これにより、前述のようなN+1問題を解決することができます。

// 予期せぬキャッシュ共有をしないよう、`React.cache()`でリクエスト単位のインスタンス生成
const getAuthorLoader = React.cache(() => new DataLoader(authorsBatch));

export async function fetchAuthors(authorIds: string[]) {
  const authorLoader = getAuthorLoader();
  return authorLoader.loadMany(authorIds);
}

このようにfetchAuthors()を実装すれば、データフェッチコロケーションしつつバッチングによりN+1が防げます。驚くべきことに、fetchAuthors()の使い方は何一つ変わりません。

// post-cassette.tsx
export async function PostCassette({ post }: { post: Post }) {
  const [authors, comments, viewCount] = await Promise.all([
    fetchAuthors(post.authorIds),
    fetchCommentCount(post.id),
    fetchViewCount(post.id),
  ]);

  // ...`authors`, `comments`, `viewCount`を参照
}

fetchCommentCount()fetchViewCount()も同様にDataLoaderによってバッチングすれば、上記のような実装のままN+1を解決し、データフェッチを元の通り4回に抑えることができます。

`fetchCommentCount()`と`fetchViewCount()`の修正
async function commentCountBatch(postIds: readonly string[]) {
  const res = await fetch(
    `${API_URL}/posts/comments_counts?${postIds.map((id) => `postId=${id}`).join("&")}`,
  );
  if (!res.ok) {
    console.error("Failed to fetch commentCount in batch:", await res.text());
    return postIds.map(() => null);
  }
  const commentCounts: CommentCount[] = await res.json();

  return postIds.map(
    (postId) =>
      commentCounts.find((commentCount) => commentCount.postId === postId) ??
      null,
  );
}
const getCommentCountLoader = React.cache(
  () => new DataLoader(commentCountBatch),
);

export async function fetchCommentCount(postId: string) {
  const commentCountLoader = getCommentCountLoader();
  return commentCountLoader.load(postId);
}

async function viewCountBatch(postIds: readonly string[]) {
  const res = await fetch(
    `${API_URL}/posts/view_counts?${postIds.map((id) => `postId=${id}`).join("&")}`,
  );
  if (!res.ok) {
    console.error("Failed to fetch viewCount in batch:", await res.text());
    return postIds.map(() => null);
  }
  const viewCounts: ViewCount[] = await res.json();

  return postIds.map(
    (postId) =>
      viewCounts.find((viewCount) => viewCount.postId === postId) ?? null,
  );
}
const getViewCountLoader = React.cache(() => new DataLoader(viewCountBatch));

export async function fetchViewCount(postId: string) {
  const viewCountLoader = getViewCountLoader();
  return viewCountLoader.load(postId);
}

短命のキャッシュ

データフェッチコロケーションでよく発生する問題がもう1つあります。同一リクエストの重複実行です。

現在ログインしているユーザー情報を取得するfetchCurrentUser()を用いて、ログイン時ヘッダーにアイコンを表示するとします。

export async function UserIcon() {
  const user = await fetchCurrentUser();

  // ...`user`を参照
}

同様に、記事詳細ページでコメントを追加するにはログイン状態を参照する必要があり、fetchCurrentUser()を実行する必要があるとします。

export async function AddComment() {
  const user = await fetchCurrentUser();

  // ...`user`を参照
}

これらのコンポーネントは離れているため、<Suspense>境界によってレンダリングタイミングが異なる可能性があり、バッチングできるとは限りません。また、バッチングできるとしても、引数がないのでDataLoaderが想定してる利用ケースではありません。

このような問題を容易に解決するためにReactが提供しているのが、React Cacheです。React Cacheはサーバーへのリクエストごとに作成される、短命のキャッシュです。すでにDataLoaderの実装例でも、DataLoaderのインスタンス保持のために利用していました。React Cacheはリクエストごとに破棄されるためインフラ側でキャッシュストレージを用意するなどの作業は不要で、また、メモリリークや予期せぬキャッシュ共有なども防ぐことができます。

React Cacheを用いたfetchCurrentUser()の実装例は以下です。

export const fetchCurrentUser = React.cache(async () => {
  const cookieStore = await cookies();
  const sessionId = cookieStore.get("session-id");
  const res = await fetch(`https://.../?sessionId=${sessionId}`);
  const user = await res.json();
  return user;
});

ただし、実際にはNext.jsのRequest Memoizationのように、フレームワーク側で同一リクエストの排除が実装されていることが多いと考えられます[3]。そのため、上記のようにfetch()するのみならReact.cache()は不要になるでしょう。

まとめ

Metaでは、自律分散型の設計にバッチングと短命のキャッシュを組み合わせることによって、パフォーマンスと保守性を両立させています。これは大規模開発のみで有用なアプローチではなく、小規模な開発から適用可能で有効な手段です。

実際に、筆者は小規模なアプリケーションでも好んでDataLoaderやReact Cacheを利用した自律分散的な設計を採用していますが、メリットを感じる場面が多く、筆者にとってこれらは必要不可欠な存在です。

本稿を通じて、これらの有効性が伝われば幸いです。

余談: Next.jsにおけるDataLoaderとReact Cache

Next.jsにおけるDataLoaderの使い方や短命のキャッシュについてより詳細に知りたい方は、筆者が以前執筆した「Next.jsの考え方」の以下の章をご参照ください。

https://zenn.dev/akfm/books/nextjs-basic-principle/viewer/part_1_data_loader

https://zenn.dev/akfm/books/nextjs-basic-principle/viewer/part_1_request_memoization

脚注
  1. コードをできるだけ関連性のある場所に配置することを指します。 ↩︎

  2. DataLoaderの実装にあるコメントが参考になります。 ↩︎

  3. 元々はReact側の機能として実装されていたはずですが、現状RSCの必須要件なのかは不明です。 ↩︎

Discussion