导航

golang使用graphviz

Posted on 2015-11-03 18:29  蝈蝈俊  阅读(4250)  评论(0编辑  收藏  举报

graphviz的介绍请参考: http://www.cnblogs.com/ghj1976/p/4539788.html 

安装 graphviz

需要在 http://www.graphviz.org/Download_macos.php 下载对应操作系统的版本。

image

安装完成后,可以用 dot –V  命令查看安装的版本,确认安装

image

mac 下安装后,会有一个 Graphviz 的应用可以查看 *.gv 的文件, 如下图效果所示:

image

 

产生Graphviz 文件

用 Golang 产生Graphviz 文件的封装方法如下:

https://github.com/awalterschulze/gographviz 

 

我们使用它的范例代码如下:

package main

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

func main() {
    graphAst, _ := gographviz.Parse([]byte(`digraph G{}`))
    graph := gographviz.NewGraph()
    gographviz.Analyse(graphAst, graph)
    graph.AddNode("G", "a", nil)
    graph.AddNode("G", "b", nil)
    graph.AddEdge("a", "b", true, nil)
    fmt.Println(graph.String())
}

执行效果如下面截图:

image

 

 

使用 dot 生成 png 的命令行如下:

dot 11.gv -T png -o 11.png

 

完整的 Go 代码如下:

 

package main

import (
    "bytes"
    "fmt"
    "github.com/awalterschulze/gographviz"
    "io/ioutil"
    "os/exec"
)

func main() {
    graphAst, _ := gographviz.Parse([]byte(`digraph G{}`))
    graph := gographviz.NewGraph()
    gographviz.Analyse(graphAst, graph)
    graph.AddNode("G", "a", nil)
    graph.AddNode("G", "b", nil)
    graph.AddEdge("a", "b", true, nil)
    fmt.Println(graph.String())

    // 输出文件
    ioutil.WriteFile("11.gv", []byte(graph.String()), 0666)

    // 产生图片
    system("dot 11.gv -T png -o 12.png")
}

//调用系统指令的方法,参数s 就是调用的shell命令
func system(s string) {
    cmd := exec.Command(`/bin/sh`, `-c`, s) //调用Command函数
    var out bytes.Buffer                    //缓冲字节

    cmd.Stdout = &out //标准输出
    err := cmd.Run()  //运行指令 ,做判断
    if err != nil {
        fmt.Println(err)
    }
    fmt.Printf("%s", out.String()) //输出执行结果
}

生成的效果图:

image