Merge pull request #178 from topcoder-platform/PM-3080

PM-3080 delete aiWorkflowRun data before deleting a submission

Author: himaniraghav3

.circleci/config.yml

@@ -74,11 +74,6 @@ workflows:
                       branches:
                           only:
                               - develop
-                              - feat/ai-workflows
-                              - pm-1955_2
-                              - re-try-failed-jobs
-                              - pm-2539
-                              - pm-2177_fixes
 
 
             - 'build-prod':

prisma/migrations/20251203163619_cascade_delete_workflowrun_deps/migration.sql

@@ -0,0 +1,11 @@
+-- DropForeignKey
+ALTER TABLE "aiWorkflowRunItem" DROP CONSTRAINT "aiWorkflowRunItem_workflowRunId_fkey";
+
+-- DropForeignKey
+ALTER TABLE "aiWorkflowRunItemComment" DROP CONSTRAINT "aiWorkflowRunItemComment_workflowRunItemId_fkey";
+
+-- AddForeignKey
+ALTER TABLE "aiWorkflowRunItem" ADD CONSTRAINT "aiWorkflowRunItem_workflowRunId_fkey" FOREIGN KEY ("workflowRunId") REFERENCES "aiWorkflowRun"("id") ON DELETE CASCADE ON UPDATE CASCADE;
+
+-- AddForeignKey
+ALTER TABLE "aiWorkflowRunItemComment" ADD CONSTRAINT "aiWorkflowRunItemComment_workflowRunItemId_fkey" FOREIGN KEY ("workflowRunItemId") REFERENCES "aiWorkflowRunItem"("id") ON DELETE CASCADE ON UPDATE CASCADE;

src/api/submission/submission.service.ts

@@ -2287,6 +2287,34 @@ export class SubmissionService {
           });
         }
       }
+      const TERMINAL_STATUSES = [
+        'COMPLETED',
+        'FAILURE',
+        'CANCELLED',
+        'SUCCESS',
+      ];
+
+      const runs = await this.prisma.aiWorkflowRun.findMany({
+        where: { submissionId: id },
+        select: { id: true, status: true },
+      });
+
+      if (runs.length > 0) {
+        const nonTerminal = runs.filter(
+          (r) => !TERMINAL_STATUSES.includes(r.status),
+        );
+
+        if (nonTerminal.length > 0) {
+          throw new Error(
+            `Cannot delete submission: ${nonTerminal.length} workflow run(s) still active.`,
+          );
+        }
+
+        await this.prisma.aiWorkflowRun.deleteMany({
+          where: { submissionId: id },
+        });
+      }
+
       await this.prisma.submission.delete({
         where: { id },
       });