From ab3814845650281c64057e26f50e6de6be4379a4 Mon Sep 17 00:00:00 2001
From: Sydonian <794346190@qq.com>
Date: Thu, 24 Apr 2025 11:05:32 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=89=93=E5=8D=B0=E6=97=A5?=
 =?UTF-8?q?=E5=BF=97=E7=9A=84=E5=87=BD=E6=95=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkgs/ioswitch/dag/graph.go | 99 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 99 insertions(+)

diff --git a/pkgs/ioswitch/dag/graph.go b/pkgs/ioswitch/dag/graph.go
index 1a4e0f1..99ed675 100644
--- a/pkgs/ioswitch/dag/graph.go
+++ b/pkgs/ioswitch/dag/graph.go
@@ -1,7 +1,12 @@
 package dag
 
 import (
+	"fmt"
+	"reflect"
+	"strings"
+
 	"gitlink.org.cn/cloudream/common/utils/lo2"
+	"gitlink.org.cn/cloudream/common/utils/reflect2"
 )
 
 type Graph struct {
@@ -55,6 +60,100 @@ func (g *Graph) NewValueVar() *ValueVar {
 	return &ValueVar{}
 }
 
+func (g *Graph) Dump() string {
+	nodeIDs := make(map[Node]int)
+	for i, node := range g.Nodes {
+		nodeIDs[node] = i
+	}
+
+	var sb strings.Builder
+	for _, node := range g.Nodes {
+		id, ok := nodeIDs[node]
+		if !ok {
+			id = len(nodeIDs)
+			nodeIDs[node] = id
+		}
+		sb.WriteString(fmt.Sprintf("[%v]%v\n", id, nodeTypeName(node)))
+		if node.InputStreams().Len() > 0 {
+			sb.WriteString("SIn: ")
+			for i, in := range node.InputStreams().Slots {
+				if i > 0 {
+					sb.WriteString(", ")
+				}
+
+				if in == nil {
+					sb.WriteString("?")
+				} else {
+					sb.WriteString(fmt.Sprintf("%v", nodeIDs[in.Src]))
+				}
+			}
+			sb.WriteString("\n")
+		}
+		if node.OutputStreams().Len() > 0 {
+			sb.WriteString("SOut: ")
+			for i, out := range node.OutputStreams().Slots {
+				if i > 0 {
+					sb.WriteString(", ")
+				}
+
+				sb.WriteString("(")
+				for i2, dst := range out.Dst {
+					if i2 > 0 {
+						sb.WriteString(", ")
+					}
+					sb.WriteString(fmt.Sprintf("%v", nodeIDs[dst]))
+				}
+				sb.WriteString(")")
+			}
+			sb.WriteString("\n")
+		}
+
+		if node.InputValues().Len() > 0 {
+			sb.WriteString("VIn: ")
+			for i, in := range node.InputValues().Slots {
+				if i > 0 {
+					sb.WriteString(", ")
+				}
+
+				if in == nil {
+					sb.WriteString("?")
+				} else {
+					sb.WriteString(fmt.Sprintf("%v", nodeIDs[in.Src]))
+				}
+			}
+			sb.WriteString("\n")
+		}
+		if node.OutputValues().Len() > 0 {
+			sb.WriteString("VOut: ")
+			for i, out := range node.OutputValues().Slots {
+				if i > 0 {
+					sb.WriteString(", ")
+				}
+
+				sb.WriteString("(")
+				for i2, dst := range out.Dst {
+					if i2 > 0 {
+						sb.WriteString(", ")
+					}
+					sb.WriteString(fmt.Sprintf("%v", nodeIDs[dst]))
+				}
+				sb.WriteString(")")
+			}
+			sb.WriteString("\n")
+		}
+
+	}
+	return sb.String()
+}
+
+func nodeTypeName(node Node) string {
+	typ := reflect2.TypeOfValue(node)
+	for typ.Kind() == reflect.Ptr {
+		typ = typ.Elem()
+	}
+	return typ.Name()
+}
+
 func AddNode[N Node](graph *Graph, typ N) N {
 	graph.AddNode(typ)
 	return typ