From fdd849fa0d9f2e536bc4e489d7023c6643b1d87a Mon Sep 17 00:00:00 2001
From: hyeonsu <hyeonsu@wace.me>
Date: Wed, 10 Sep 2025 15:40:49 +0900
Subject: [PATCH] =?UTF-8?q?=EB=8D=B0=EC=9D=B4=ED=84=B0=20=ED=9D=90?=
 =?UTF-8?q?=EB=A6=84=20=EA=B4=80=EA=B3=84=EB=8F=84=20=ED=85=8C=EC=9D=B4?=
 =?UTF-8?q?=EB=B8=94=20=ED=99=94=EB=A9=B4=EC=97=90=20=ED=9A=8C=EC=82=AC?=
 =?UTF-8?q?=EC=BD=94=EB=93=9C=20=EC=BB=AC=EB=9F=BC=20=EC=B6=94=EA=B0=80?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 frontend/components/dataflow/DataFlowList.tsx | 43 ++-----------------
 frontend/lib/api/dataflow.ts                  |  1 +
 2 files changed, 4 insertions(+), 40 deletions(-)

diff --git a/frontend/components/dataflow/DataFlowList.tsx b/frontend/components/dataflow/DataFlowList.tsx
index ec962824..2da1200a 100644
--- a/frontend/components/dataflow/DataFlowList.tsx
+++ b/frontend/components/dataflow/DataFlowList.tsx
@@ -64,6 +64,7 @@ export default function DataFlowList({ onDiagramSelect, selectedDiagram, onDesig
         relationshipCount: diagram.relationships?.relationships?.length || 0,
         tableCount: diagram.relationships?.tables?.length || 0,
         tables: diagram.relationships?.tables || [],
+        companyCode: diagram.company_code, // 회사 코드 추가
         createdAt: new Date(diagram.created_at || new Date()),
         createdBy: diagram.created_by || "SYSTEM",
         updatedAt: new Date(diagram.updated_at || diagram.created_at || new Date()),
@@ -180,44 +181,6 @@ export default function DataFlowList({ onDiagramSelect, selectedDiagram, onDesig
     }
   };
 
-  // 관계 타입에 따른 배지 색상
-  const getRelationshipTypeBadge = (relationshipType: string) => {
-    switch (relationshipType) {
-      case "one-to-one":
-        return (
-          <Badge variant="secondary" className="bg-gray-100 text-gray-700">
-            1:1
-          </Badge>
-        );
-      case "one-to-many":
-        return (
-          <Badge variant="secondary" className="bg-orange-100 text-orange-700">
-            1:N
-          </Badge>
-        );
-      case "many-to-one":
-        return (
-          <Badge variant="secondary" className="bg-yellow-100 text-yellow-700">
-            N:1
-          </Badge>
-        );
-      case "many-to-many":
-        return (
-          <Badge variant="secondary" className="bg-red-100 text-red-700">
-            N:N
-          </Badge>
-        );
-      case "multi-relationship":
-        return (
-          <Badge variant="secondary" className="bg-purple-100 text-purple-700">
-            다중 관계
-          </Badge>
-        );
-      default:
-        return <Badge variant="secondary">{relationshipType}</Badge>;
-    }
-  };
-
   if (loading) {
     return (
       <div className="flex items-center justify-center py-8">
@@ -262,7 +225,7 @@ export default function DataFlowList({ onDiagramSelect, selectedDiagram, onDesig
               <TableRow>
                 <TableHead>관계도명</TableHead>
                 <TableHead>연결 타입</TableHead>
-                <TableHead>관계 타입</TableHead>
+                <TableHead>회사 코드</TableHead>
                 <TableHead>테이블 수</TableHead>
                 <TableHead>관계 수</TableHead>
                 <TableHead>최근 수정</TableHead>
@@ -291,7 +254,7 @@ export default function DataFlowList({ onDiagramSelect, selectedDiagram, onDesig
                     </div>
                   </TableCell>
                   <TableCell>{getConnectionTypeBadge(diagram.connectionType)}</TableCell>
-                  <TableCell>{getRelationshipTypeBadge(diagram.relationshipType)}</TableCell>
+                  <TableCell>{diagram.companyCode || "*"}</TableCell>
                   <TableCell>
                     <div className="flex items-center">
                       <Database className="mr-1 h-3 w-3 text-gray-400" />
diff --git a/frontend/lib/api/dataflow.ts b/frontend/lib/api/dataflow.ts
index 9fd91e0c..55c46414 100644
--- a/frontend/lib/api/dataflow.ts
+++ b/frontend/lib/api/dataflow.ts
@@ -97,6 +97,7 @@ export interface DataFlowDiagram {
   tableCount: number;
   relationshipCount: number;
   tables: string[];
+  companyCode: string; // 회사 코드 추가
   createdAt: Date;
   createdBy: string;
   updatedAt: Date;