package/mq
1
0
Fork 0
mirror of https://github.com/oarkflow/mq.git synced 2025-10-29 00:11:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update: dependencies

Browse Source
This commit is contained in:
Oarkflow
2024-11-22 19:47:47 +05:45
parent ff8c6fc3f1
commit cbe93dd9ca
1 changed files with 45 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
dag/v2/ui.go
View File

@@ -143,39 +143,60 @@ func (tm *DAG) ExportDOT() string {
var sb strings.Builder var sb strings.Builder
sb.WriteString(fmt.Sprintf(`digraph "%s" {`, tm.name)) sb.WriteString(fmt.Sprintf(`digraph "%s" {`, tm.name))
sb.WriteString("\n") sb.WriteString("\n")
sb.WriteString(fmt.Sprintf(` label="%s";`, tm.name)) sb.WriteString(` label="Enhanced DAG Representation";`)
sb.WriteString("\n") sb.WriteString("\n")
sb.WriteString(` labelloc="t";`) sb.WriteString(` labelloc="t"; fontsize=22; fontname="Helvetica";`)
sb.WriteString("\n") sb.WriteString("\n")
sb.WriteString(` fontsize=20;`) sb.WriteString(` node [shape=box, fontname="Helvetica", fillcolor="#B3CDE0", fontcolor="#2C3E50", fontsize=10, margin="0.25,0.15", style="rounded,filled"];`)
sb.WriteString("\n") sb.WriteString("\n")
sb.WriteString(` node [shape=box, style="rounded,filled", fillcolor="lightgray", fontname="Arial", margin="0.2,0.1"];`) sb.WriteString(` edge [fontname="Helvetica", fontsize=12, arrowsize=0.8];`)
sb.WriteString("\n") sb.WriteString("\n")
sb.WriteString(` edge [fontname="Arial", fontsize=12, arrowsize=0.8];`) sb.WriteString(` rankdir=TB;`)
sb.WriteString("\n")
sb.WriteString(` size="10,10";`)
sb.WriteString("\n")
sb.WriteString(` ratio="fill";`)
sb.WriteString("\n") sb.WriteString("\n")
sortedNodes := tm.TopologicalSort() sortedNodes := tm.TopologicalSort()
for _, nodeKey := range sortedNodes { for _, nodeKey := range sortedNodes {
node, _ := tm.nodes.Get(nodeKey) node, _ := tm.nodes.Get(nodeKey)
nodeColor := "lightblue" nodeColor := "lightgray"
sb.WriteString(fmt.Sprintf(` "%s" [label=" %s", fillcolor="%s", id="node_%s"];`, node.ID, node.Label, nodeColor, node.ID)) nodeShape := "box"
labelSuffix := ""
// Apply styles based on NodeType
switch node.NodeType {
case Function:
nodeColor = "#D4EDDA"
labelSuffix = " [Function]"
case Page:
nodeColor = "#F08080"
labelSuffix = " [Page]"
}
sb.WriteString(fmt.Sprintf(
` "%s" [label="%s%s", fontcolor="#2C3E50", fillcolor="%s", shape="%s", style="rounded,filled", id="node_%s"];`,
node.ID, node.Label, labelSuffix, nodeColor, nodeShape, node.ID))
sb.WriteString("\n") sb.WriteString("\n")
} }
// Define edges with unique styling by EdgeType
for _, nodeKey := range sortedNodes { for _, nodeKey := range sortedNodes {
node, _ := tm.nodes.Get(nodeKey) node, _ := tm.nodes.Get(nodeKey)
for _, edge := range node.Edges { for _, edge := range node.Edges {
var edgeStyle string edgeStyle := "solid"
edgeColor := "black"
labelSuffix := ""
// Apply styles based on EdgeType
switch edge.Type { switch edge.Type {
case Iterator: case Iterator:
edgeStyle = "dashed" edgeStyle = "dashed"
default: edgeColor = "blue"
labelSuffix = " [Iter]"
case Simple:
edgeStyle = "solid" edgeStyle = "solid"
edgeColor = "black"
labelSuffix = ""
} }
edgeColor := "black" sb.WriteString(fmt.Sprintf(
sb.WriteString(fmt.Sprintf(` "%s" -> "%s" [label=" %s", color="%s", style=%s, fontsize=10, arrowsize=0.6];`, node.ID, edge.To.ID, edge.Label, edgeColor, edgeStyle)) ` "%s" -> "%s" [label="%s%s", color="%s", style="%s"];`,
node.ID, edge.To.ID, edge.Label, labelSuffix, edgeColor, edgeStyle))
sb.WriteString("\n") sb.WriteString("\n")
} }
} }
@@ -187,6 +208,8 @@ func (tm *DAG) ExportDOT() string {
} }
} }
} }
// Optional: Group related nodes into subgraphs (e.g., loops)
for _, nodeKey := range sortedNodes { for _, nodeKey := range sortedNodes {
node, _ := tm.nodes.Get(nodeKey) node, _ := tm.nodes.Get(nodeKey)
if node.processor != nil { if node.processor != nil {
@@ -194,35 +217,28 @@ func (tm *DAG) ExportDOT() string {
if subDAG != nil { if subDAG != nil {
sb.WriteString(fmt.Sprintf(` subgraph "cluster_%s" {`, subDAG.name)) sb.WriteString(fmt.Sprintf(` subgraph "cluster_%s" {`, subDAG.name))
sb.WriteString("\n") sb.WriteString("\n")
sb.WriteString(fmt.Sprintf(` label=" %s";`, subDAG.name)) sb.WriteString(fmt.Sprintf(` label="Subgraph: %s";`, subDAG.name))
sb.WriteString("\n") sb.WriteString("\n")
sb.WriteString(` style=dashed;`) sb.WriteString(` style=filled; color=gray90;`)
sb.WriteString("\n")
sb.WriteString(` bgcolor="lightgray";`)
sb.WriteString("\n")
sb.WriteString(` node [shape=rectangle, style="filled", fillcolor="lightblue", fontname="Arial", margin="0.2,0.1"];`)
sb.WriteString("\n") sb.WriteString("\n")
subDAG.nodes.ForEach(func(subNodeKey string, subNode *Node) bool { subDAG.nodes.ForEach(func(subNodeKey string, subNode *Node) bool {
sb.WriteString(fmt.Sprintf(` "%s" [label=" %s"];`, subNodeKey, subNode.Label)) sb.WriteString(fmt.Sprintf(` "%s" [label="%s"];`, subNode.ID, subNode.Label))
sb.WriteString("\n") sb.WriteString("\n")
return true return true
}) })
subDAG.nodes.ForEach(func(subNodeKey string, subNode *Node) bool { subDAG.nodes.ForEach(func(subNodeKey string, subNode *Node) bool {
for _, edge := range subNode.Edges { for _, edge := range subNode.Edges {
sb.WriteString(fmt.Sprintf(` "%s" -> "%s" [label=" %s", color="black", style=solid, arrowsize=0.6];`, subNodeKey, edge.To.ID, edge.Label)) sb.WriteString(fmt.Sprintf(` "%s" -> "%s" [label="%s"];`, subNodeKey, edge.To.ID, edge.Label))
sb.WriteString("\n") sb.WriteString("\n")
} }
return true return true
}) })
sb.WriteString(` }`) sb.WriteString(" }\n")
sb.WriteString("\n")
sb.WriteString(fmt.Sprintf(` "%s" -> "%s" [label=" %s", color="black", style=solid, arrowsize=0.6];`, node.ID, subDAG.startNode, subDAG.name))
sb.WriteString("\n")
} }
} }
} }
sb.WriteString(`}`)
sb.WriteString("\n") sb.WriteString("}\n")
return sb.String() return sb.String()
} }