1 files changed, 76 insertions, 0 deletions
diff --git a/db/seeds/seedIssueStage.ts b/db/seeds/seedIssueStage.ts
new file mode 100644
index 00000000..117e2a99
--- /dev/null
+++ b/db/seeds/seedIssueStage.ts
@@ -0,0 +1,76 @@
+/**
+ * seedIssueStages.ts
+ * - issueStages 테이블에 테스트 데이터 삽입
+ */
+
+import db from "@/db/db"
+import { faker } from "@faker-js/faker"
+import { documents, issueStages } from "../schema/vendorDocu"
+
+// 하나의 문서당 스테이지 개수 범위
+const STAGE_MIN = 1
+const STAGE_MAX = 3
+
+async function seedIssueStages() {
+  try {
+    console.log("Seeding issue stages started...")
+    
+    // 먼저 documents 테이블에서 모든 문서 ID 가져오기
+    const allDocs = await db.select({ id: documents.id }).from(documents)
+    console.log(`Found ${allDocs.length} documents to create stages for.`)
+    
+    if (allDocs.length === 0) {
+      console.warn("No documents found. Run seedDocuments.ts first.")
+      return
+    }
+    
+    // 각 문서에 대해 스테이지 생성 및 삽입
+    let totalStagesCreated = 0
+    
+    for (const doc of allDocs) {
+      const stageCount = faker.number.int({ min: STAGE_MIN, max: STAGE_MAX })
+      const stagesData = []
+      
+      for (let j = 0; j < stageCount; j++) {
+        stagesData.push({
+          documentId: doc.id,
+          stageName: faker.helpers.arrayElement([
+            "Issued for Review",
+            "IFC",
+            "AFC",
+            "As-Built",
+          ]),
+          planDate: faker.helpers.maybe(() => faker.date.future({ years: 0.5 }), {
+            probability: 0.5,
+          }),
+          actualDate: faker.helpers.maybe(() => faker.date.future({ years: 0.5 }), {
+            probability: 0.5,
+          })
+        })
+      }
+      
+      // 한 번에 많은 데이터를 삽입하지 않고 문서별로 삽입
+      try {
+        await db.insert(issueStages).values(stagesData)
+        totalStagesCreated += stagesData.length
+        
+        // 진행 상황 로깅 (50개마다)
+        if (totalStagesCreated % 50 === 0) {
+          console.log(`Created ${totalStagesCreated} stages so far...`)
+        }
+      } catch (error) {
+        console.error(`Error inserting stages for document ${doc.id}:`, error)
+      }
+    }
+
+    console.log("Seeding issue stages completed successfully.")
+  } catch (err) {
+    console.error("Seeding issue stages error:", err)
+    process.exit(1)
+  }
+}
+
+// 스크립트 직접 실행 시
+seedIssueStages()
+  .then(() => process.exit(0))
+  .catch(() => process.exit(1))
+\ No newline at end of file