Batch fetching associated PR information (#160)

Batch fetching associated PR information

Author: lucieleblanc

Diff for: .github/workflows/check-dist.yml

@@ -46,7 +46,7 @@ jobs:
         id: diff
 
       # If index.js was different than expected, upload the expected version as an artifact
-      - uses: actions/upload-artifact@v2
+      - uses: actions/upload-artifact@v4
         if: ${{ failure() && steps.diff.conclusion == 'failure' }}
         with:
           name: dist

Diff for: dist/index.js

@@ -125,43 +125,66 @@ function branchFromVersion(version: string, channel: string): string {
   )}`
 }
 
+function chunkArray(originalArray: string[], chunkSize: number): string[][] {
+  const arrayChunks: string[][] = []
+  let currentChunk: string[] = []
+  for (const item of originalArray) {
+    currentChunk.push(item)
+    if (currentChunk.length >= chunkSize) {
+      arrayChunks.push(currentChunk)
+      currentChunk = []
+    }
+  }
+  if (currentChunk.length > 0) {
+    arrayChunks.push(currentChunk)
+  }
+  return arrayChunks
+}
+
 // Fetches PR body text from a series of commits.
 async function fetchPullRequestBodyFromCommits(
   commits: string[],
   graphqlWithAuth: Function
 ): Promise<string[]> {
   let commitsSubQuery = ''
-  for (const oid of commits) {
-    commitsSubQuery += `
-      commit_${oid}: object(oid: "${oid}") {
-        ... on Commit {
-          oid
-          author {
-            name
-          }
-          associatedPullRequests(first: 1) {
-            nodes {
-                body
+
+  // Split the list of commits into batches of 25,
+  // then request one batch at a time.
+  const commitBatches = chunkArray(commits, 25)
+  const commitsInfo: string[] = []
+
+  for (const commitBatch of commitBatches) {
+    for (const oid of commitBatch) {
+      commitsSubQuery += `
+        commit_${oid}: object(oid: "${oid}") {
+          ... on Commit {
+            oid
+            author {
+              name
+            }
+            associatedPullRequests(first: 1) {
+              nodes {
+                  body
+              }
             }
           }
         }
-      }
-  `
-  }
+    `
+    }
 
-  const response = await graphqlWithAuth(`
-  {
-    repository(owner: "warpdotdev", name: "warp-internal") {
-      ${commitsSubQuery}
+    const response = await graphqlWithAuth(`
+    {
+      repository(owner: "warpdotdev", name: "warp-internal") {
+        ${commitsSubQuery}
+      }
     }
-  }
-`)
+  `)
 
-  const commitsInfo: string[] = []
-  for (const oid of commits) {
-    const commitResponse = response.repository[`commit_${oid}`]
-    if (commitResponse.associatedPullRequests.nodes.length > 0) {
-      commitsInfo.push(commitResponse.associatedPullRequests.nodes[0].body)
+    for (const oid of commitBatch) {
+      const commitResponse = response.repository[`commit_${oid}`]
+      if (commitResponse.associatedPullRequests.nodes.length > 0) {
+        commitsInfo.push(commitResponse.associatedPullRequests.nodes[0].body)
+      }
     }
   }
   return commitsInfo