天天看点

golang使用graphviz - 蝈蝈俊

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

安装 graphviz

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

golang使用graphviz - 蝈蝈俊

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

golang使用graphviz - 蝈蝈俊

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

golang使用graphviz - 蝈蝈俊

产生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())

}

执行效果如下面截图:

golang使用graphviz - 蝈蝈俊

使用 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()) //输出执行结果

}

生成的效果图:

golang使用graphviz - 蝈蝈俊