fix: restore discovery task pipeline

2026-04-22 14:20:54 +08:00
parent 804fd756e7
commit 79744667c6
10 changed files with 1426 additions and 20 deletions
@@ -0,0 +1,51 @@
+DO $$
+BEGIN
+  IF NOT EXISTS (
+    SELECT 1
+    FROM pg_type
+    WHERE typname = 'TaskStatus'
+  ) THEN
+    CREATE TYPE "TaskStatus" AS ENUM ('PENDING', 'IN_PROGRESS', 'COMPLETED', 'FAILED');
+  END IF;
+END
+$$;
+
+CREATE TABLE IF NOT EXISTS "project_discovery_tasks" (
+  "id" TEXT NOT NULL,
+  "status" "TaskStatus" NOT NULL DEFAULT 'PENDING',
+  "sourceUrl" TEXT NOT NULL,
+  "sourceType" TEXT NOT NULL DEFAULT 'manual',
+  "explorationData" JSONB,
+  "explorationSummary" TEXT,
+  "errorMessage" TEXT,
+  "retryCount" INTEGER NOT NULL DEFAULT 0,
+  "lastRetryAt" TIMESTAMP(3),
+  "projectId" TEXT,
+  "createdAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  "startedAt" TIMESTAMP(3),
+  "completedAt" TIMESTAMP(3),
+  "updatedAt" TIMESTAMP(3) NOT NULL DEFAULT CURRENT_TIMESTAMP,
+
+  CONSTRAINT "project_discovery_tasks_pkey" PRIMARY KEY ("id")
+);
+
+CREATE INDEX IF NOT EXISTS "idx_task_project_id" ON "project_discovery_tasks"("projectId");
+CREATE INDEX IF NOT EXISTS "idx_task_source_url" ON "project_discovery_tasks"("sourceUrl");
+CREATE INDEX IF NOT EXISTS "idx_task_status_created" ON "project_discovery_tasks"("status", "createdAt");
+
+DO $$
+BEGIN
+  IF NOT EXISTS (
+    SELECT 1
+    FROM pg_constraint
+    WHERE conname = 'project_discovery_tasks_projectId_fkey'
+  ) THEN
+    ALTER TABLE "project_discovery_tasks"
+      ADD CONSTRAINT "project_discovery_tasks_projectId_fkey"
+      FOREIGN KEY ("projectId")
+      REFERENCES "projects"("id")
+      ON DELETE SET NULL
+      ON UPDATE CASCADE;
+  END IF;
+END
+$$;