fix: adding more time log for the run status

web/src/app/(app)/api/update-run/route.ts

@@ -16,96 +16,114 @@ import { z } from "zod";
 const Request = z.object({
   run_id: z.string(),
   status: WorkflowRunStatusSchema.optional(),
-  time: z.date().optional(),
+  time: z.coerce.date().optional(),
   output_data: z.any().optional(),
 });
 
 export async function POST(request: Request) {
-  const [data, error] = await parseDataSafe(Request, request);
-  if (!data || error) return error;
+  try {
+    const [data, error] = await parseDataSafe(Request, request);
 
-  const { run_id, status, time, output_data } = data;
+    if (!data || error) return error;
 
-  if (status == "started" && time != undefined) {
-    // It successfully started, update the started_at time
-    await db
-      .update(workflowRunsTable)
-      .set({
-        started_at: time,
-      })
-      .where(eq(workflowRunsTable.id, run_id));
-  }
+    const { run_id, status, time, output_data } = data;
 
-  if (status == "queued" && time != undefined) {
-    // It successfully started, update the started_at time
-    await db
-      .update(workflowRunsTable)
-      .set({
-        queued_at: time,
-      })
-      .where(eq(workflowRunsTable.id, run_id));
-  }
+    if (status == "started" && time != undefined) {
+      // It successfully started, update the started_at time
+      await db
+        .update(workflowRunsTable)
+        .set({
+          started_at: time,
+        })
+        .where(eq(workflowRunsTable.id, run_id));
+    }
 
-  if (output_data) {
-    const workflow_run_output = await db.insert(workflowRunOutputs).values({
-      run_id: run_id,
-      data: output_data,
-    });
-  } else if (status) {
-    const [workflow_run] = await db
-      .update(workflowRunsTable)
-      .set({
-        status: status,
-        ended_at:
-          status === "success" || status === "failed" ? new Date() : null,
-      })
-      .where(eq(workflowRunsTable.id, run_id))
-      .returning();
+    if (status == "queued" && time != undefined) {
+      // It successfully started, update the started_at time
+      await db
+        .update(workflowRunsTable)
+        .set({
+          queued_at: time,
+        })
+        .where(eq(workflowRunsTable.id, run_id));
+    }
 
-    // Need to filter out only comfy deploy serverless
-    // Also multiply with the gpu selection
-    if (workflow_run.machine_type == "comfy-deploy-serverless") {
-      if (
-        (status === "success" || status === "failed") &&
-        workflow_run.user_id
-      ) {
-        const sub = await getCurrentPlan({
-          user_id: workflow_run.user_id,
-          org_id: workflow_run.org_id,
-        });
+    if (output_data) {
+      const workflow_run_output = await db.insert(workflowRunOutputs).values({
+        run_id: run_id,
+        data: output_data,
+      });
+    } else if (status) {
+      const [workflow_run] = await db
+        .update(workflowRunsTable)
+        .set({
+          status: status,
+          ended_at:
+            status === "success" || status === "failed" ? new Date() : null,
+        })
+        .where(eq(workflowRunsTable.id, run_id))
+        .returning();
 
-        if (sub && sub.subscription_item_api_id && workflow_run.ended_at) {
-          let durationInSec = Math.abs(
-            (workflow_run.ended_at.getTime() -
-              workflow_run.created_at.getTime()) /
-              1000,
-          );
-          durationInSec = Math.ceil(durationInSec);
-          switch (workflow_run.gpu) {
-            case "A100":
-              durationInSec *= 7;
-              break;
-            case "A10G":
-              durationInSec *= 4;
-              break;
+      // Need to filter out only comfy deploy serverless
+      // Also multiply with the gpu selection
+      if (workflow_run.machine_type == "comfy-deploy-serverless") {
+        if (
+          (status === "success" || status === "failed") &&
+          workflow_run.user_id
+        ) {
+          const sub = await getCurrentPlan({
+            user_id: workflow_run.user_id,
+            org_id: workflow_run.org_id,
+          });
+
+          if (sub && sub.subscription_item_api_id && workflow_run.ended_at) {
+            let durationInSec = Math.abs(
+              (workflow_run.ended_at.getTime() -
+                workflow_run.created_at.getTime()) /
+                1000,
+            );
+            durationInSec = Math.ceil(durationInSec);
+            switch (workflow_run.gpu) {
+              case "A100":
+                durationInSec *= 7;
+                break;
+              case "A10G":
+                durationInSec *= 4;
+                break;
+            }
+            await stripe.subscriptionItems.createUsageRecord(
+              sub.subscription_item_api_id,
+              {
+                quantity: durationInSec,
+              },
+            );
           }
-          await stripe.subscriptionItems.createUsageRecord(
-            sub.subscription_item_api_id,
-            {
-              quantity: durationInSec,
-            },
-          );
         }
       }
     }
-  }
 
-  return NextResponse.json(
-    {
-      message: "success",
-    },
-    {
-      status: 200,
-    },
-  );
+    return NextResponse.json(
+      {
+        message: "success",
+      },
+      {
+        status: 200,
+      },
+    );
+  } catch (error: unknown) {
+    console.log("An error here");
+
+    const errorMessage =
+      error instanceof Error ? error.message : "Unknown error";
+    console.log(errorMessage);
+
+    return NextResponse.json(
+      {
+        error: errorMessage,
+      },
+      {
+        status: 500,
+      },
+    );
+  }
 }