[DOP-25788] Make output.type a list

Author: dolfinus

src/components/lineage/converters/getGraphEdges.tsx

@@ -44,6 +44,22 @@ const getMinimalEdge = (relation: BaseRelationLineageResponseV1): Edge => {
     };
 };
 
+const getOutputEdgeColor = (
+    relation: OutputRelationLineageResponseV1,
+): string => {
+    for (const type of relation.types) {
+        switch (type) {
+            case "DROP":
+            case "TRUNCATE":
+                return "red";
+            case "ALTER":
+            case "RENAME":
+                return "yellow";
+        }
+    }
+    return "green";
+};
+
 const getOutputEdge = (
     relation: OutputRelationLineageResponseV1,
     raw_response: LineageResponseV1,
@@ -79,19 +95,7 @@ const getOutputEdge = (
         }
     }
 
-    let color = "green";
-    switch (relation.type) {
-        case "DROP":
-        case "TRUNCATE":
-            color = "red";
-            break;
-        case "ALTER":
-        case "RENAME":
-            color = "yellow";
-            break;
-        default:
-            color = "green";
-    }
+    const color = getOutputEdgeColor(relation);
 
     // Too many animated edges is heavy load for browser
     const totalIOEdges =

src/components/lineage/edges/IOEdge.tsx

@@ -39,6 +39,10 @@ const IOEdge = ({
         return <ReactFlowBaseEdge id={id} path={edgePath} {...props} />;
     }
 
+    const types = ((data.types ?? []) as string[]).map((type) =>
+        translate(`edges.ioTypes.${type}`, { _: type }),
+    );
+
     return (
         <>
             <ReactFlowBaseEdge id={id} path={edgePath} {...props} />
@@ -53,13 +57,11 @@ const IOEdge = ({
                     }}
                     className={`nodrag nopan ${selected ? "selected" : ""}`}
                 >
-                    {data.type ? (
+                    {types.length ? (
                         <Chip
                             size="small"
                             color="secondary"
-                            label={translate(`edges.ioTypes.${data.type}`, {
-                                _: data.type,
-                            })}
+                            label={types.join(", ")}
                             style={props.labelStyle}
                         />
                     ) : null}

src/dataProvider/types.ts

@@ -190,7 +190,7 @@ type OutputRelationTypeLineageResponseV1 =
 interface OutputRelationLineageResponseV1
     extends BaseRelationLineageResponseV1 {
     last_interaction_at: string;
-    type: OutputRelationTypeLineageResponseV1 | null;
+    types: OutputRelationTypeLineageResponseV1[];
     num_rows: number | null;
     num_bytes: number | null;
     num_files: number | null;