log the scheduled function in work:start

Author: ianmacartney

src/component/loop.ts

@@ -560,19 +560,29 @@ async function beginWork(
   if (!work) {
     throw new Error("work not found");
   }
-  recordStarted(console, work, lagMs);
   const { attempts: attempt, fnHandle, fnArgs, payloadId } = work;
   const args = { workId, fnHandle, fnArgs, payloadId, logLevel, attempt };
+  let scheduleId;
   if (work.fnType === "action") {
-    return ctx.scheduler.runAfter(0, internal.worker.runActionWrapper, args);
+    scheduleId = await ctx.scheduler.runAfter(
+      0,
+      internal.worker.runActionWrapper,
+      args,
+    );
   } else if (work.fnType === "mutation" || work.fnType === "query") {
-    return ctx.scheduler.runAfter(0, internal.worker.runMutationWrapper, {
-      ...args,
-      fnType: work.fnType,
-    });
+    scheduleId = await ctx.scheduler.runAfter(
+      0,
+      internal.worker.runMutationWrapper,
+      {
+        ...args,
+        fnType: work.fnType,
+      },
+    );
   } else {
     throw new Error(`Unexpected fnType ${work.fnType}`);
   }
+  recordStarted(console, work, lagMs, scheduleId);
+  return scheduleId;
 }
 
 /**

src/component/stats.ts

@@ -38,11 +38,13 @@ export function recordStarted(
   console: Logger,
   work: Doc<"work">,
   lagMs: number,
+  scheduledFunctionId: Id<"_scheduled_functions">,
 ) {
   console.event("started", {
     workId: work._id,
     fnName: work.fnName,
     enqueuedAt: work._creationTime,
+    scheduledFunctionId,
     startedAt: Date.now(),
     startLag: lagMs,
   });