174 lines
3.5 KiB
Go
174 lines
3.5 KiB
Go
|
package container
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"fmt"
|
||
|
"path/filepath"
|
||
|
"reflect"
|
||
|
|
||
|
"github.com/pkg/errors"
|
||
|
|
||
|
"github.com/goccy/go-graphviz"
|
||
|
"github.com/goccy/go-graphviz/cgraph"
|
||
|
)
|
||
|
|
||
|
type config struct {
|
||
|
// logging
|
||
|
loggers []func(string)
|
||
|
indentStr string
|
||
|
|
||
|
// graphing
|
||
|
graphviz *graphviz.Graphviz
|
||
|
graph *cgraph.Graph
|
||
|
visualizers []func(string)
|
||
|
logVisualizer bool
|
||
|
}
|
||
|
|
||
|
func newConfig() (*config, error) {
|
||
|
g := graphviz.New()
|
||
|
graph, err := g.Graph()
|
||
|
if err != nil {
|
||
|
return nil, errors.Wrap(err, "error initializing graph")
|
||
|
}
|
||
|
|
||
|
return &config{
|
||
|
graphviz: g,
|
||
|
graph: graph,
|
||
|
}, nil
|
||
|
}
|
||
|
|
||
|
func (c *config) indentLogger() {
|
||
|
c.indentStr = c.indentStr + " "
|
||
|
}
|
||
|
|
||
|
func (c *config) dedentLogger() {
|
||
|
if len(c.indentStr) > 0 {
|
||
|
c.indentStr = c.indentStr[1:]
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (c config) logf(format string, args ...interface{}) {
|
||
|
s := fmt.Sprintf(c.indentStr+format, args...)
|
||
|
for _, logger := range c.loggers {
|
||
|
logger(s)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (c *config) generateGraph() {
|
||
|
buf := &bytes.Buffer{}
|
||
|
err := c.graphviz.Render(c.graph, graphviz.XDOT, buf)
|
||
|
if err != nil {
|
||
|
c.logf("Error rendering DOT graph: %+v", err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
dot := buf.String()
|
||
|
if c.logVisualizer {
|
||
|
c.logf("DOT Graph: %s", dot)
|
||
|
}
|
||
|
|
||
|
for _, v := range c.visualizers {
|
||
|
v(dot)
|
||
|
}
|
||
|
|
||
|
err = c.graph.Close()
|
||
|
if err != nil {
|
||
|
c.logf("Error closing graph: %+v", err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
err = c.graphviz.Close()
|
||
|
if err != nil {
|
||
|
c.logf("Error closing graphviz: %+v", err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (c *config) addFuncVisualizer(f func(string)) {
|
||
|
c.visualizers = append(c.visualizers, func(dot string) {
|
||
|
f(dot)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *config) enableLogVisualizer() {
|
||
|
c.logVisualizer = true
|
||
|
}
|
||
|
|
||
|
func (c *config) addFileVisualizer(filename string, format string) {
|
||
|
c.visualizers = append(c.visualizers, func(_ string) {
|
||
|
err := c.graphviz.RenderFilename(c.graph, graphviz.Format(format), filename)
|
||
|
if err != nil {
|
||
|
c.logf("Error saving graphviz file %s with format %s: %+v", filename, format, err)
|
||
|
} else {
|
||
|
path, err := filepath.Abs(filename)
|
||
|
if err == nil {
|
||
|
c.logf("Saved graph of container to %s", path)
|
||
|
}
|
||
|
}
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (c *config) locationGraphNode(location Location, scope Scope) (*cgraph.Node, error) {
|
||
|
graph := c.scopeSubGraph(scope)
|
||
|
node, found, err := c.findOrCreateGraphNode(graph, location.Name())
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
if found {
|
||
|
return node, nil
|
||
|
}
|
||
|
|
||
|
node.SetShape(cgraph.BoxShape)
|
||
|
node.SetColor("lightgrey")
|
||
|
return node, nil
|
||
|
}
|
||
|
|
||
|
func (c *config) typeGraphNode(typ reflect.Type) (*cgraph.Node, error) {
|
||
|
node, found, err := c.findOrCreateGraphNode(c.graph, typ.String())
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
if found {
|
||
|
return node, nil
|
||
|
}
|
||
|
|
||
|
node.SetColor("lightgrey")
|
||
|
return node, err
|
||
|
}
|
||
|
|
||
|
func (c *config) findOrCreateGraphNode(subGraph *cgraph.Graph, name string) (node *cgraph.Node, found bool, err error) {
|
||
|
node, err = c.graph.Node(name)
|
||
|
if err != nil {
|
||
|
return nil, false, errors.Wrapf(err, "error finding graph node %s", name)
|
||
|
}
|
||
|
|
||
|
if node != nil {
|
||
|
return node, true, nil
|
||
|
}
|
||
|
|
||
|
node, err = subGraph.CreateNode(name)
|
||
|
if err != nil {
|
||
|
return nil, false, errors.Wrapf(err, "error creating graph node %s", name)
|
||
|
}
|
||
|
|
||
|
return node, false, nil
|
||
|
}
|
||
|
|
||
|
func (c *config) scopeSubGraph(scope Scope) *cgraph.Graph {
|
||
|
graph := c.graph
|
||
|
if scope != nil {
|
||
|
gname := fmt.Sprintf("cluster_%s", scope.Name())
|
||
|
graph = c.graph.SubGraph(gname, 1)
|
||
|
graph.SetLabel(fmt.Sprintf("Scope: %s", scope.Name()))
|
||
|
}
|
||
|
return graph
|
||
|
}
|
||
|
|
||
|
func (c *config) addGraphEdge(from *cgraph.Node, to *cgraph.Node) {
|
||
|
_, err := c.graph.CreateEdge("", from, to)
|
||
|
if err != nil {
|
||
|
c.logf("error creating graph edge")
|
||
|
}
|
||
|
}
|