本文首發於我的博客，如果覺得有用，歡迎點贊收藏，讓更多的朋友看到。

最近，我開發了一個非常簡單的小工具，總的代碼量 200 行不到。今天，簡單介紹下它。這是個什麼工具呢？它是一個用於可視化展示 Go Module 依賴關係的工具。

爲何開發

爲什麼會想到開發這個工具？主要有兩點原因：

一是最近經常看到大家在社區討論 Go Module。於是，我也花了一些時間研究了下。期間，遇到了一個需求，如何清晰地識別模塊中依賴項之間的關係。一番瞭解後，發現了 go mod graph。

效果如下：

$ go mod graph
github.com/poloxue/testmod golang.org/x/[email protected]
github.com/poloxue/testmod rsc.io/quote/[email protected]
github.com/poloxue/testmod rsc.io/[email protected]
golang.org/x/[email protected] golang.org/x/[email protected]
rsc.io/quote/[email protected] rsc.io/[email protected]
rsc.io/[email protected] golang.org/x/[email protected]
rsc.io/[email protected] golang.org/x/[email protected]

每一行的格式是 模塊依賴模塊，基本能滿足要求，但總覺得還是不那麼直觀。

二是我之前手裏有一個項目，包管理一直用的是 dep。於是，我也瞭解了下它，把官方文檔仔細讀了一遍。其中的某個章節介紹了依賴項可視化展示的方法。

文檔中給出的包關係圖：

看到這張圖的時候，眼睛瞬間就亮了，圖形化就是優秀，不同依賴之間的關係一目瞭然。這不就是我想要的效果嗎？666，點個贊。

但 ...，隨之而來的問題是，go mod 沒這個能力啊。怎麼辦？

如何實現

先看看是不是已經有人做了這件事了。網上搜了下，沒找到。那是不是能自己實現？應該可以借鑑下 dep 的思路吧？

如下是 dep 依賴實現可視化的方式：

# linux
$ sudo apt-get install graphviz
$ dep status -dot | dot -T png | display

# macOS
$ brew install graphviz
$ dep status -dot | dot -T png | open -f -a /Applications/Preview.app

# Windows
> choco install graphviz.portable
> dep status -dot | dot -T png -o status.png; start status.png

這裏展示了三大系統下的使用方式，它們都安裝了一個軟件包，graphviz。從名字上看，這應該是一個用來實現可視化的軟件，即用來畫圖的。事實也是這樣，可以看看它的官網。

再看下它的使用，發現都是通過管道命令組合的方式，而且前面的部分基本相同，都是 dep status -dot | dot -T png。後面的部分在不同的系統就不同了，Linux 是 display，MacOS 是 open -f -a /Applications/Preview.app，Window 是 start status.png。

稍微分析下就會明白，前面是生成圖片，後面是顯示圖片。因爲不同系統的圖片展示命令不同，所以後面的部分也就不同了。

現在關心的重點在前面，即 dep status -dot | dot -T png 幹了啥，它究竟是如何實現繪圖的？大致猜測，dot -T png 是由 dep status -dot 提供的數據生成圖片。繼續看下 dep status -dot 的執行效果吧。

$ dep status -dot
digraph {
    node [shape=box];
    2609291568 [label="github.com/poloxue/hellodep"];
    953278068 [label="rsc.io/quote\nv3.1.0"];
    3852693168 [label="rsc.io/sampler\nv1.0.0"];
    2609291568 -> 953278068;
    953278068 -> 3852693168;
}

咋一看，輸出的是一段看起來不知道是啥的代碼，這應該是 graphviz 用於繪製圖表的語言。那是不是還有學習下？當然不用啊，這裏用的很簡單，直接套用就行了。

試着分析一下吧，前面兩行可以不用關心，這應該是 graphviz 特定的寫法，表示要畫的是什麼圖。我們主要關心如何將數據以正確形式提供出來。

2609291568 [label="github.com/poloxue/hellodep"];
953278068 [label="rsc.io/quote\nv3.1.0"];
3852693168 [label="rsc.io/sampler\nv1.0.0"];
2609291568 -> 953278068;
953278068 -> 3852693168;

一看就知道，這裏有兩種結構，分別是爲依賴項關聯 ID ，和通過 ID 和 -> 表示依賴間的關係。

按上面的猜想，我們可以試着畫出一個簡單的圖, 用於表示 a 模塊依賴 b 模塊。執行命令如下，將繪圖代碼通過 each 管道的方式發送給 dot 命令。

$ echo 'digraph {
node [shape=box];
1 [label="a"];
2 [label="b"];
1 -> 2;
}' | dot -T png | open -f -a /Applications/Preview.app

效果如下：

繪製一個依賴關係圖竟然這麼簡單。

看到這裏，是不是發現問題已經變得非常簡單了。我們只要將 go mod graph 的輸出轉化爲類似的結構就能實現可視化了。

開發流程介紹

接下來，開發這個小程序吧，我將這個小程序命名爲 modv，即 module visible 的意思。項目源碼位於 poloxue/modv。

接收管道的輸入

先要檢查數據輸入管道是否正常。

我們的目標是使用類似 dep 中作圖的方式，go mod graph 通過管道將數據傳遞給 modv。因此，要先檢查 os.Stdin，即檢查標準輸入狀態是否正常，以及是否是管道傳輸。

下面是 main 函數的代碼，位於 main.go 中。

func main() {
    info, err := os.Stdin.Stat()
    if err != nil {
        fmt.Println("os.Stdin.Stat:", err)
        PrintUsage()
        os.Exit(1)
    }

    // 是否是管道傳輸
    if info.Mode()&os.ModeNamedPipe == 0 {
        fmt.Println("command err: command is intended to work with pipes.")
        PrintUsage()
        os.Exit(1)
    }

一旦確認輸入設備一切正常，我們就可以進入到數據讀取、解析與渲染的流程了。

    mg := NewModuleGraph(os.Stdin)
    mg.Parse()
    mg.Render(os.Stdout)
}

接下來，開始具體看看如何實現數據的處理流程。

抽象實現結構

先定義一個結構體，並大致定義整個流程。

type ModGraph struct {
    Reader io.Reader  // 讀取數據流
}

func NewModGraph(r io.Reader) *ModGraph {
    return &ModGraph{Reader: r}
}

// 執行數據的處理轉化
func (m *ModGraph) Parse() error {}

// 結果渲染與輸出
func (m *ModGraph) Render(w io.Writer) error {}

再看下 go mod graph 的輸出吧，如下：

github.com/poloxue/testmod golang.org/x/[email protected]
github.com/poloxue/testmod rsc.io/quote/[email protected]
...

每一行的結構是 模塊依賴項。現在的目標是要它解析成下面這樣的結構：

digraph {
    node [shape=box];
    1 github.com/poloxue/testmod;
    2 golang.org/x/[email protected];
    3 rsc.io/quote/[email protected];
    1 -> 2;
    1 -> 3;
}

前面說過，這裏包含了兩種不同的結構，分別是模塊與 ID 關聯關係，以及模塊 ID 表示模塊間的依賴關聯。爲 ModGraph 結構體增加兩個成員表示它們。

type ModGraph struct {
    r io.Reader  // 數據流讀取實例，這裏即 os.Stdin
 
    // 每一項名稱與 ID 的映射
    Mods         map[string]int
    // ID 和依賴 ID 關係映射，一個 ID 可能依賴多個項
    Dependencies map[int][]int
}

要注意的是，增加了兩個 map 成員後，記住要在 NewModGraph 中初始化下它們。

mod graph 輸出解析

如何進行解析？

介紹到這裏，目標已經很明白了。就是要將輸入數據解析到 Mods 和 Dependencies 兩個成員中，實現代碼都在 Parse 方法中。

爲了方便進行數據讀取，首先，我們利用 bufio 基於 reader 創建一個新的 bufReader，

func (m *ModGraph) Parse() error {
    bufReader := bufio.NewReader(m.Reader)
    ...

爲便於按行解析數據，我們通過 bufReader 的 ReadBytes() 方法循環一行一行地讀取 os.Stdin 中的數據。然後，對每一行數據按空格切分，獲取到依賴關係的兩項。代碼如下：

for {
    relationBytes, err := bufReader.ReadBytes('\n')
    if err != nil {
        if err == io.EOF {
            return nil
        }
        return err
    }

    relation := bytes.Split(relationBytes, []byte(" "))
    // module and dependency
    mod, depMod := strings.TrimSpace(string(relation[0])), strings.TrimSpace(string(relation[1]))

    ...
}

接下來，就是將解析出來的依賴關係組織到 Mods 和 Dependencies 兩個成員中。模塊 ID 是生成規則採用的是最簡單的實現方式，從 1 自增。實現代碼如下：

modId, ok := m.Mods[mod]
if !ok {
    modId = serialID
    m.Mods[mod] = modId
    serialID += 1
}

depModId, ok := m.Mods[depMod]
if !ok {
    depModId = serialID
    m.Mods[depMod] = depModId
    serialID += 1
}

if _, ok := m.Dependencies[modId]; ok {
    m.Dependencies[modId] = append(m.Dependencies[modId], depModId)
} else {
    m.Dependencies[modId] = []int{depModId}
}

解析的工作到這裏就結束了。

渲染解析的結果

這個小工具還剩下最後一步工作要做，即將解析出來的數據渲染出來，以滿足 graphviz 工具的作圖要求。實現代碼是 Render部分：

首先，定義一個模板，以生成滿足要求的輸出格式。

var graphTemplate = `digraph {
node [shape=box];
{{ range $mod, $modId := .mods -}}
{{ $modId }} [label="{{ $mod }}"];
{{ end -}}
{{- range $modId, $depModIds := .dependencies -}}
{{- range $_, $depModId := $depModIds -}}
{{ $modId }} -> {{ $depModId }};
{{  end -}}
{{- end -}}
}
`

這一塊沒啥好介紹的，主要是要熟悉 Go 中的 text/template 模板的語法規範。爲了展示友好，這裏通過 - 實現換行的去除，整體而言不影響閱讀。

接下來，看 Render 方法的實現，把前面解析出來的 Mods 和 Dependencies 放入模板進行渲染。

func (m *ModuleGraph) Render(w io.Writer) error {
    templ, err := template.New("graph").Parse(graphTemplate)
    if err != nil {
        return fmt.Errorf("templ.Parse: %v", err)
    }

    if err := templ.Execute(w, map[string]interface{}{
        "mods":         m.Mods,
        "dependencies": m.Dependencies,
    }); err != nil {
        return fmt.Errorf("templ.Execute: %v", err)
    }

    return nil
}

現在，全部工作都完成了。最後，將這個流程整合到 main 函數。接下來就是使用了。

使用體驗

開始體驗下吧。補充一句，這個工具，我現在只測試了 Mac 下的使用，如有問題，歡迎提出來。

首先，要先安裝一下 graphviz，安裝的方式在本文開頭已經介紹了，選擇你的系統安裝方式。

接着是安裝 modv，命令如下：

$ go get github.com/poloxue/modv

安裝完成！簡單測試下它的使用。

以 MacOS 爲例。先下載測試庫，github.com/poloxue/testmod。進入 testmod 目錄執行命令：

$ go mod graph | modv | dot -T png | open -f -a /Applications/Preview.app

如果執行成功，將看到如下的效果：

完美地展示了各個模塊之間的依賴關係。

一些思考

本文是篇實踐性的文章，從一個簡單想法到成功呈現出一個可以使用的工具。雖然，開發起來並不難，從開發到完成，僅僅花了一兩個小時。但我的感覺，這確實是個有實際價值的工具。

還有一些想法沒有實現和驗證，比如一旦項目較大，是否可以方便的展示某個指定節點的依賴樹，而非整個項目。還有，在其他項目向 Go Module 遷移的時候，這個小工具是否能產生一些價值。

歡迎關注我的微信公衆號。

如何實現 Go Module 依賴關係的可視化

爲何開發

如何實現

開發流程介紹

接收管道的輸入

抽象實現結構

mod graph 輸出解析

渲染解析的結果

使用體驗

一些思考

從詞法分析角度看 Go 代碼的組成

如何實現 Go Module 依賴關係的可視化

Go 中實例演示 json 的 map 和 struct 便捷處理轉化

一文理清 Go 引用的常見疑惑

如何靈活地進行 Go 版本管理

https://yachay.unat.edu.pe/blog/index.php?comment_area=format_blog&comment_component=blog&comment_co

linux以太網驅動總結