graphvizify
diff --git a/graph/graph.go b/graph/graph.go
index 8681b8f..911924f 100644
--- a/graph/graph.go
+++ b/graph/graph.go
@@ -13,11 +13,13 @@
)
type MachinePort struct {
+ Machine *Machine
OtherEnd *SwitchPort
Name string
}
type SwitchPort struct {
+ Switch *Switch
OtherEnd *MachinePort
Name string
}
@@ -76,7 +78,8 @@
}
for _, portpb := range machinepb.ManagedPort {
machine.Ports[portpb.Name] = &MachinePort{
- Name: portpb.Name,
+ Name: portpb.Name,
+ Machine: machine,
}
}
g.Machines[machinepb.Name] = machine
@@ -103,7 +106,8 @@
}
for _, portpb := range switchpb.ManagedPort {
sw.Ports[portpb.Name] = &SwitchPort{
- Name: portpb.Name,
+ Name: portpb.Name,
+ Switch: sw,
}
}
g.Switches[switchpb.Name] = sw