feat(gql-utils): add completion handler to create-data-loader

apps/gql/loaders/pano.ts

@@ -67,13 +67,14 @@ const createPanoUpvoteLoaders = ({ prisma }: Clients) => {
         results.map((result) => (result.status === "fulfilled" ? result.value : null))
       );
 
-      results.forEach((upvote) => {
+      return results;
+    },
+    (results) => {
+      results?.forEach((upvote) => {
         if (upvote) {
           byPostID.prime(upvote.id, upvote);
         }
       });
-
-      return results;
     }
   );
 
@@ -87,13 +88,14 @@ const createPanoUpvoteLoaders = ({ prisma }: Clients) => {
         results.map((result) => (result.status === "fulfilled" ? result.value : null))
       );
 
-      results.forEach((upvote) => {
+      return results;
+    },
+    (results) => {
+      results?.forEach((upvote) => {
         if (upvote) {
           byCommentID.prime(upvote.id, upvote);
         }
       });
-
-      return results;
     }
   );
 

packages/gql-utils/data-loader/create-data-loader.ts

@@ -1,5 +1,16 @@
 import DataLoader, { type BatchLoadFn } from "dataloader";
 
-export function createDataLoader<T, U>(batchFn: BatchLoadFn<T, U>) {
-  return new DataLoader(batchFn);
+export function createDataLoader<TKey, TValue>(
+  batchFn: BatchLoadFn<TKey, TValue>,
+  onComplete?: (values: TValue[]) => void
+) {
+  return new DataLoader(async (keys: readonly TKey[]) => {
+    const result = (await batchFn(keys)) ?? [];
+
+    console.log({ result });
+
+    onComplete?.(result as unknown as TValue[]);
+
+    return result;
+  });
 }