✨ バッチですべてのメンバーを取得する機能

SystemEngineeringTeam · Sep 11, 2024 · 1a06219 · 1a06219
1 parent 4dea72f
commit 1a06219
Show file tree

Hide file tree

Showing 6 changed files with 57 additions and 14 deletions.
diff --git a/src/hooks/teams.ts b/src/hooks/teams.ts
@@ -1,6 +1,8 @@
+/* eslint-disable no-console */
+
 import { useStore } from "@nanostores/react";
 import { match } from "ts-pattern";
-import { A } from "@/lib/consts";
+import { A, range } from "@/lib/consts";
 import { getEsaClient } from "@/lib/services/esa";
 import { $selectedTeamName } from "@/lib/stores/teams";
 import {
@@ -50,18 +52,54 @@ export function useTeam() {
       .with(A.Success, ({ data }) => data)
       .otherwise(handleError);
 
-  const fetchMembers = async (): Promise<
+  const fetchMembers = async (
+    /**
+     * 1 度に取得するメンバー数
+     */
+    prePage?: number,
+    /**
+     * ページ番号
+     */
+    page: number = 1,
+  ): Promise<
     InferResponseType<"/teams/{team_name}/members", "get">["members"]
   > =>
     await match(
-      await getEsaClient().GET(
-        "/teams/{team_name}/members",
-        paramsWithTeamName,
-      ),
+      await getEsaClient().GET("/teams/{team_name}/members", {
+        ...paramsWithTeamName,
+        params: {
+          ...paramsWithTeamName.params,
+          query: { page, per_page: prePage },
+        },
+      }),
     )
       .with(A.Success, ({ data }) => data.members)
       .otherwise(handleError);
 
+  const fetchMembersAll = async (): Promise<
+    InferResponseType<"/teams/{team_name}/members", "get">["members"]
+  > => {
+    const PER_PAGE_MAX = 100;
+
+    const stats = await fetchStats();
+    console.log(`[fetchMembersAll] Num of members: ${stats.members}`);
+
+    const pageCount = Math.ceil(stats.members / 100);
+    const membersPromises = range(pageCount, 1).map(
+      // eslint-disable-next-line @typescript-eslint/promise-function-async
+      (pageAt, _) => {
+        const start = PER_PAGE_MAX * (pageAt - 1);
+        const end = Math.min(PER_PAGE_MAX * pageAt, stats.members);
+        console.log(
+          `[fetchMembersAll] Fetching members as a batch... ${pageAt}/${pageCount}: ${start}..${end}`,
+        );
+        return fetchMembers(PER_PAGE_MAX, pageAt);
+      },
+    );
+
+    return (await Promise.all(membersPromises)).flat();
+  };
+
   const createNewPost = async (
     postBody: InferRequestBodyType<"/teams/{team_name}/posts", "post">["post"],
   ): Promise<InferResponseType<"/teams/{team_name}/posts", "post", 201>> =>
@@ -162,6 +200,7 @@ export function useTeam() {
     fetchStats,
     fetchAbout,
     fetchMembers,
+    fetchMembersAll,
     fetchPostByPostId,
     fetchPostsByCategory,
     fetchEmojis,

diff --git a/src/lib/consts.ts b/src/lib/consts.ts
@@ -35,3 +35,7 @@ export function getLocalStorageKey(key: string, trailingColon = false): string {
 }
 
 export const DB_VERSION = "2";
+
+export function range(size: number, startAt: number = 0): readonly number[] {
+  return [...Array(size).keys()].map((i) => i + startAt);
+}
diff --git a/src/pages/achievements/[id]/index.tsx b/src/pages/achievements/[id]/index.tsx
@@ -21,12 +21,12 @@ const BoxStyle = styled(Box)`
 
 export default function Page(): ReactElement {
   const { id } = useParams();
-  const { fetchMembers } = useTeam();
+  const { fetchMembersAll } = useTeam();
   const { fetch: fetchAchievements } = useAchievements(useTeam);
   const { fetch: fetchUnlockedAchievements } = useUnlockedAchievements(useTeam);
   const swrAMU = useSWRImmutable("amu", async () => ({
     achievements: await fetchAchievements(),
-    members: await fetchMembers(),
+    members: await fetchMembersAll(),
     unlockedAchievements: await fetchUnlockedAchievements(),
   }));
 

diff --git a/src/pages/members/[id]/index.tsx b/src/pages/members/[id]/index.tsx
@@ -22,12 +22,12 @@ const BoxStyle = styled(Box)`
 
 export default function Page(): ReactElement {
   const { id } = useParams();
-  const { fetchMembers } = useTeam();
+  const { fetchMembersAll } = useTeam();
   const { fetch: fetchAchievements } = useAchievements(useTeam);
   const { fetch: fetchUnlockedAchievements } = useUnlockedAchievements(useTeam);
   const swrAMU = useSWRImmutable("amu", async () => ({
     achievements: await fetchAchievements(),
-    members: await fetchMembers(),
+    members: await fetchMembersAll(),
     unlockedAchievements: await fetchUnlockedAchievements(),
   }));
 

diff --git a/src/pages/members/index.tsx b/src/pages/members/index.tsx
@@ -18,10 +18,10 @@ const BoxStyle = styled(Box)`
 `;
 
 export default function Page(): ReactElement {
-  const { fetchMembers } = useTeam();
+  const { fetchMembersAll } = useTeam();
   const { fetch: fetchUnlockedAchievements } = useUnlockedAchievements(useTeam);
   const swrMU = useSWRImmutable("mu", async () => ({
-    members: await fetchMembers(),
+    members: await fetchMembersAll(),
     unlockedAchievements: await fetchUnlockedAchievements(),
   }));
 

diff --git a/src/pages/ranking/index.tsx b/src/pages/ranking/index.tsx
@@ -48,12 +48,12 @@ const LogRecentUnlockedStyle = styled.div`
 `;
 
 export default function Page(): ReactElement {
-  const { fetchMembers } = useTeam();
+  const { fetchMembersAll } = useTeam();
   const { fetch: fetchAchievements } = useAchievements(useTeam);
   const { fetch: fetchUnlockedAchievements } = useUnlockedAchievements(useTeam);
   const swrAMU = useSWRImmutable("amu", async () => ({
     achievements: await fetchAchievements(),
-    members: await fetchMembers(),
+    members: await fetchMembersAll(),
     unlockedAchievements: await fetchUnlockedAchievements(),
   }));