介紹

圖靈完備性(Turing completeness)是通用計算機的一個屬性，它表示一個程序可以寫另一個程序。比如 go test 命令：它會掃描被測試的包，寫一個包含測試內容的程序，然后編譯運行。

可能你聽得比較多的是元編程(meta-program)。這里不細究他們的區(qū)別。我們重點是，用程序生成另一個程序的場景，越來越廣泛了。比如：

• protobufs: 根據(jù) pb 文件（.proto）生成 go 代碼文件(.pb.go)
• yacc: 根據(jù) yacc 文件（.y）生成 go 代碼文件
• bindata: 二進制文件（如 JPEG）轉成 go 的 bytes 數(shù)組
• mockery: 根據(jù) go 的 interface 生成 Mock 對象（依賴 stretchr/testify/mock 包）。

自動生成代碼的命令，該如何集成進項目里面？一般來說，我們可以借助外部的工具，如 Make。使用 Go 1.4 新增的 go generate 命令，就可以避免外部工具了。

Mockery

我們以 mockery 為例，關于 mockery 生成的 Mock 對象，我之前的文章有介紹過：Go 庫：單元測試利器 testify。

比如我們定義了 Greeter 接口，作為 Hello 的參數(shù)（代碼本身無意義，僅做示例）：

// greet.go 文件
package greet
type Greeter interface {
	Greet() string
}
// 加上"hello"前綴
func Hello(g Greeter) string {
	return "hello " + g.Greet()
}

我們要給 Hello 函數(shù)寫單元測試的話，就需要為 Greeter 寫一個 Mock 對象。我們可以使用 mockery 來生成。

安裝 mockery：

go install github.com/vektra/mockery/v2@latest

為當前包下的所有 interface 生成 mock 對象，輸出到 mocks 目錄。

mockery --output mocks/ --dir . --all

我們查看目錄文件：

? tree
.
├── greet.go
└── mocks             // mockery 生成的 mock 對象
    └── Greeter.go
1 directory, 2 files

關于 mockery 的說明，我們可以查看倉庫：github.com/vektra/mock…。現(xiàn)在，我們可以為 Hello 寫單元測試了：

// greet_test.go 文件
package greet
import (
	"example/greet/mocks"
	"testing"
	"github.com/stretchr/testify/assert"
)
func TestHello(t *testing.T) {
  // 實例化 mock 對象
	greeter := new(mocks.Greeter)
  // 設置預期，當請求 greeter.Greet() 時，返回 "world"
	greeter.On("Greet").Return("world")
	want := "hello world"
	got := Hello(greeter)
  // 斷言相等
	assert.Equal(t, want, got)
}

Go Generate

我們接下來的問題是，Mockery 命令需要集成到項目里，我們可以寫 shell 腳本、或者 Make 文件，但這些都不是 go tool 的工具。

用法

go generate 非常方便使用，只要當前目錄的 .go 文件（如 greet.go 文件），加上備注：

//go:generate mockery --output mocks/ --dir . --all

我們先把剛剛生成的 mocks 目錄刪除，當前目錄結構是：

?   tree
.
├── greet.go
└── greet_test.go
0 directories, 2 files

然后進入 greet 包，執(zhí)行 go generate:

?  go generate  
01 Dec 22 20:06 CST INF Starting mockery dry-run=false version=v2.10.0
01 Dec 22 20:06 CST INF Walking dry-run=false version=v2.10.0
01 Dec 22 20:06 CST INF Generating mock dry-run=false interface=Greeter qualified-name=example/greet version=v2.10.0

打印目錄結構，會發(fā)現(xiàn) mocks 對象又生成了：

? tree
.
├── greet.go
├── greet_test.go
└── mocks
    └── Greeter.go
1 directory, 3 files

總結

Rob Pike 大神在 go generate 的提案有說過，希望 go generate 能替換 go 倉庫中的 Makefile。本人沒有明顯的偏向，但是 go generate 確實不失為一個優(yōu)秀的工具。

引用

• go.dev/blog/genera…
• go.googlesource.com/proposal/+/…

以上就是Go Generate 代替 Makefile使用方法詳解的詳細內容，更多關于Go Generate代替Makefile的資料請關注腳本之家其它相關文章！

最新評論