graph.go

package main

import (
	"github.com/awalterschulze/gographviz"
)

type Edge struct {
	host1 string
	host2 string
	port  string
}

type Graph struct {
	graph   *gographviz.Graph
	edgeMap map[Edge]bool
}

func (g *Graph) NewGraph() {
	graphAst, _ := gographviz.Parse([]byte(`graph G{}`))
	g.graph = gographviz.NewGraph()
	gographviz.Analyse(graphAst, g.graph)
	g.edgeMap = make(map[Edge]bool)
}

func (g *Graph) AddNode(hostName string) {
	host := "\"" + hostName + "\""
	g.graph.AddNode("G", host, nil)
}

func (g *Graph) AddConnectionOnce(h Host, conn Connection) {
	var edge Edge
	h1 := "\"" + h.hostName + "\""
	h2 := "\"" + conn.hostName + "\""
	if h.hostName < conn.hostName {
		edge = Edge{h1, h2, conn.port}
	} else {
		edge = Edge{h2, h1, conn.port}
	}
	_, ok := g.edgeMap[edge]
	if !ok {
		g.edgeMap[edge] = true
        port := "\"" + conn.port + "\""
		g.graph.AddEdge(h1, h2, false, map[string]string{"label": port})
	}
}