feat: get inserted count

yjl9903 · Apr 9, 2023 · 498af25 · 498af25
1 parent 7ca4f25
commit 498af25
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/packages/worker/src/index.ts b/packages/worker/src/index.ts
@@ -76,8 +76,7 @@ router.get('/resources', async (request, env: Env) => {
 
 router.put('/resources', async (request, env: Env) => {
   try {
-    await handleScheduled(env);
-    return makeResponse({});
+    return makeResponse(await handleScheduled(env));
   } catch (error) {
     console.log(error);
     return makeErrorResponse({}, { status: 400 });

diff --git a/packages/worker/src/scheduled.ts b/packages/worker/src/scheduled.ts
@@ -6,6 +6,7 @@ import { makePrisma } from './prisma';
 export async function handleScheduled(env: Env) {
   const prisma = makePrisma(env);
 
+  let sum = 0;
   const teams = new Set<number>();
   const users = new Set<number>();
 
@@ -71,6 +72,9 @@ export async function handleScheduled(env: Env) {
     });
 
     if (count === 0) break;
+    sum += count;
     console.log(`There are ${count} resources inserted`);
   }
+
+  return { count: sum };
 }