From e5fc1b0dad362512179053dbfd69084ec5b03217 Mon Sep 17 00:00:00 2001
From: Noel Kwan <noelkwan1998@gmail.com>
Date: Mon, 16 Sep 2024 17:01:11 +0800
Subject: [PATCH] build relation dep edges

---
 dashboard/lib/api/streaming.ts |  5 +++--
 dashboard/pages/ddl_graph.tsx  | 24 ++++++++++++++++++++++--
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/dashboard/lib/api/streaming.ts b/dashboard/lib/api/streaming.ts
index 95ad89fdf5c58..b0e6c8492ffdf 100644
--- a/dashboard/lib/api/streaming.ts
+++ b/dashboard/lib/api/streaming.ts
@@ -126,7 +126,7 @@ export async function getRelations() {
   return relations
 }
 
-export async function getRelationDependencies() {
+export async function getRelationDependencies(): Map<number, number[]> {
   return await getObjectDependencies()
 }
 
@@ -200,7 +200,8 @@ export async function getSchemas() {
   return schemas
 }
 
-export async function getObjectDependencies() {
+// Returns a map of object id to a list of object ids that it depends on
+export async function getObjectDependencies(): Map<number, number[]> {
   let objDependencies: ObjectDependencies[] = (
     await api.get("/object_dependencies")
   ).map(ObjectDependencies.fromJSON)
diff --git a/dashboard/pages/ddl_graph.tsx b/dashboard/pages/ddl_graph.tsx
index eb109d8501536..3638a95395a1c 100644
--- a/dashboard/pages/ddl_graph.tsx
+++ b/dashboard/pages/ddl_graph.tsx
@@ -47,7 +47,7 @@ import {
 import {
   getFragmentsByJobId, getRelationDependencies,
   getRelationIdInfos,
-  getStreamingJobs,
+  getStreamingJobs, Relation,
 } from "../lib/api/streaming"
 import {DdlBox, FragmentBox} from "../lib/layout"
 import { TableFragments, TableFragments_Fragment } from "../proto/gen/meta"
@@ -225,6 +225,26 @@ function buildFragmentDependencyAsEdges(
   return nodes
 }
 
+function buildDdlDependencyAsEdges(
+    relations: Relation[],
+    dependencies: Map<number, number[]>,
+): DdlBox[] {
+  const nodes: DdlBox[] = []
+  for (const relation of relations) {
+    let parentIds = dependencies.get(relation.id) || []
+    nodes.push({
+      id: relation.id.toString(),
+      order: relation.id,
+      width: 0,
+      height: 0,
+      parentIds: parentIds.map((x) => x.toString()),
+      ddl_name: relation.name,
+      schema_name: relation.schemaName
+    })
+  }
+  return nodes
+}
+
 const SIDEBAR_WIDTH = 200
 
 type BackPressureDataSource = "Embedded" | "Prometheus"
@@ -246,7 +266,7 @@ export default function Streaming() {
   const { response: relationList } = useFetch(getStreamingJobs)
   const { response: relationIdInfos } = useFetch(getRelationIdInfos)
   // 1. Get the relation dependendencies.
-  // const { response: relationDeps } = useFetch(getRelationDependencies)
+  const { response: relationDeps } = useFetch(getRelationDependencies)
   // 2. Get the relation -> input fragment_id mapping.
   // 3. Get the relation -> output fragment_id mapping.
   // 4. Construct the BP graph for relation ids using 1-3.