博客地址:note.mogutou.xyz/articles/20…html
风格是指规范代码的共同约定。风格一词实际上是有点用词不当的,由于共同约定的范畴远远不止 gofmt 所作的源代码格式化这些。git
本指南旨在经过详尽描述 Uber 在编写 Go 代码中的注意事项(规定)来解释其中复杂之处。制定这些注意事项(规定)是为了提升代码可维护性同时也让工程师们高效的使用 Go 的特性。github
这份指南最初由 Prashant Varanasi 和 Simon Newton 编写,目的是让一些同事快速上手 Go 。多年来,已经根据其余人的反馈不断修改。golang
这份文档记录了咱们在 Uber 遵照的 Go 惯用准则。其中不少准则是 Go 的通用准则,其余方面依赖于外部资源:shell
全部的代码都应该经过 golint
和 go vet
检查。咱们建议您设置编辑器:c#
goimports
golint
和 go vet
来检查错误您能够在这找到关于编辑器设定 Go tools 的相关信息:api
你基本永远不须要一个指向接口的指针。你应该直接将接口做为值传递,由于接口的底层数据就是指针。bash
一个接口包含两个字段:app
若是你须要接口方法来修改这些底层数据,那你必须使用指针。
具备值类型接收器的方法能够被值类型和指针类型调用。
例如,
type S struct {
data string
}
func (s S) Read() string {
return s.data
}
func (s *S) Write(str string) {
s.data = str
}
sVals := map[int]S{1: {"A"}}
// 值类型变量只能调用 Read 方法
sVals[1].Read()
// 没法编译经过:
// sVals[0].Write("test")
sPtrs := map[int]*S{1: {"A"}}
// 指针类型变量能够调用 Read 和 Write 方法:
sPtrs[1].Read()
sPtrs[1].Write("test")
复制代码
同理,即便方法是值类型接收器,接口也能够经过指针来知足调用需求。
type F interface {
f()
}
type S1 struct{}
func (s S1) f() {}
type S2 struct{}
func (s *S2) f() {}
s1Val := S1{}
s1Ptr := &S1{}
s2Val := S2{}
s2Ptr := &S2{}
var i F
i = s1Val
i = s1Ptr
i = s2Ptr
// 没法编译经过, 由于 s2Val 是一个值类型变量, 而且 f 方法不具备值类型接收器。
// i = s2Val
复制代码
Effective Go 中关于 Pointers vs. Values 写的很棒。
零值的 sync.Mutex
和 sync.RWMutex
是有效的,因此基本是不须要一个指向 Mutex
的指针的。
Bad | Good |
---|---|
mu := new(sync.Mutex)
mu.Lock()
复制代码 |
var mu sync.Mutex
mu.Lock()
复制代码 |
若是你但愿经过指针操做结构体,mutex 能够做为其非指针结构体字段,或者最好直接嵌入结构体中。
type smap struct {
sync.Mutex
data map[string]string
}
func newSMap() *smap {
return &smap{
data: make(map[string]string),
}
}
func (m *smap) Get(k string) string {
m.Lock()
defer m.Unlock()
return m.data[k]
}
复制代码 |
type SMap struct {
mu sync.Mutex
data map[string]string
}
func NewSMap() *SMap {
return &SMap{
data: make(map[string]string),
}
}
func (m *SMap) Get(k string) string {
m.mu.Lock()
defer m.mu.Unlock()
return m.data[k]
}
复制代码 |
嵌入到非导出类型或者须要实现 Mutex 接口的类型。 | 对于导出类型,将 mutex 做为私有成员变量。 |
切片和 map 包含一个指针来指向底层数据,因此当须要复制他们时须要特别注意。
请记住,若是存储了对 slice 或 map 的引用,那么用户是能够对其进行修改。
Bad | Good |
---|---|
func (d *Driver) SetTrips(trips []Trip) {
d.trips = trips
}
trips := ...
d1.SetTrips(trips)
// 是想修改 d1.trips 吗?
trips[0] = ...
复制代码 |
func (d *Driver) SetTrips(trips []Trip) {
d.trips = make([]Trip, len(trips))
copy(d.trips, trips)
}
trips := ...
d1.SetTrips(trips)
// 修改 trips[0] 而且不影响 d1.trips 。
trips[0] = ...
复制代码 |
同理,谨慎提防用户修改暴露内部状态的 slices 和 maps 。
Bad | Good |
---|---|
type Stats struct {
sync.Mutex
counters map[string]int
}
// Snapshot 返回当前状态
func (s *Stats) Snapshot() map[string]int {
s.Lock()
defer s.Unlock()
return s.counters
}
// snapshot 再也不受锁保护了!
snapshot := stats.Snapshot()
复制代码 |
type Stats struct {
sync.Mutex
counters map[string]int
}
func (s *Stats) Snapshot() map[string]int {
s.Lock()
defer s.Unlock()
result := make(map[string]int, len(s.counters))
for k, v := range s.counters {
result[k] = v
}
return result
}
// snapshot 是一分拷贝的内容了
snapshot := stats.Snapshot()
复制代码 |
使用 defer 来作资源的清理工做,例如文件的关闭和锁的释放。
Bad | Good |
---|---|
p.Lock()
if p.count < 10 {
p.Unlock()
return p.count
}
p.count++
newCount := p.count
p.Unlock()
return newCount
// 当有多处 return 时容易忘记释放锁
复制代码 |
p.Lock()
defer p.Unlock()
if p.count < 10 {
return p.count
}
p.count++
return p.count
// 可读性更高
复制代码 |
defer 只有很是小的性能开销,只有当你能证实你的函数执行时间在纳秒级别时才能够不使用它。使用 defer 对代码可读性的提升是很是值得的,由于使用 defer 的成本真的很是小。特别是在一些主要是作内存操做的长函数中,函数中的其余计算操做远比 defer
重要。
通道的大小一般应该设为 1 或者设为无缓冲类型。默认状况下,通道是无缓冲类型的,大小为 0 。将通道大小设为其余任何数值都应该通过深思熟虑。认真考虑如何肯定其大小,是什么阻止了工做中的通道被填满并阻塞了写入操做,以及何种状况会发生这样的现象。
Bad | Good |
---|---|
// 足以知足任何人!
c := make(chan int, 64)
复制代码 |
// 大小 为 1
c := make(chan int, 1) // or
// 无缓冲 channel, 大小为 0
c := make(chan int)
复制代码 |
在 Go 中使用枚举的标准方法是声明一个自定义类型并经过 iota 关键字来声明一个 const 组。可是因为 Go 中变量的默认值都为该类型的零值,因此枚举变量的值应该从非零值开始。
Bad | Good |
---|---|
type Operation int
const (
Add Operation = iota
Subtract
Multiply
)
// Add=0, Subtract=1, Multiply=2
复制代码 |
type Operation int
const (
Add Operation = iota + 1
Subtract
Multiply
)
// Add=1, Subtract=2, Multiply=3
复制代码 |
在某些状况下,从零值开始也是能够的。例如,当零值是咱们指望的默认行为时。
type LogOutput int
const (
LogToStdout LogOutput = iota
LogToFile
LogToRemote
)
// LogToStdout=0, LogToFile=1, LogToRemote=2
复制代码
有不少种方法来声明 errors:
errors.New
声明简单的静态字符串错误信息fmt.Errorf
声明格式化的字符串错误信息Error()
方法"pkg/errors".Wrap
包装错误类型返回错误时,请考虑如下因素来做出最佳选择:
error.New
。Error()
方法。fmt.Errorf
。若是客户须要检测错误,而且是经过 errors.New
建立的一个简单的错误,请使用var 声明这个错误类型。
Bad | Good |
---|---|
// package foo
func Open() error {
return errors.New("could not open")
}
// package bar
func use() {
if err := foo.Open(); err != nil {
if err.Error() == "could not open" {
// handle
} else {
panic("unknown error")
}
}
}
复制代码 |
// package foo
var ErrCouldNotOpen = errors.New("could not open")
func Open() error {
return ErrCouldNotOpen
}
// package bar
if err := foo.Open(); err != nil {
if err == foo.ErrCouldNotOpen {
// handle
} else {
panic("unknown error")
}
}
复制代码 |
若是你有一个错误须要客户端来检测,而且你想向其添加更多信息(例如,它不是一个简单的静态字符串),那么应该声明一个自定义类型。
Bad | Good |
---|---|
func open(file string) error {
return fmt.Errorf("file %q not found", file)
}
func use() {
if err := open(); err != nil {
if strings.Contains(err.Error(), "not found") {
// handle
} else {
panic("unknown error")
}
}
}
复制代码 |
type errNotFound struct {
file string
}
func (e errNotFound) Error() string {
return fmt.Sprintf("file %q not found", e.file)
}
func open(file string) error {
return errNotFound{file: file}
}
func use() {
if err := open(); err != nil {
if _, ok := err.(errNotFound); ok {
// handle
} else {
panic("unknown error")
}
}
}
复制代码 |
直接将自定义的错误类型设为导出须要特别当心,由于这意味着他们已经成为包的公开 API 的一部分了。更好的方式是暴露一个匹配函数来检测错误。
// package foo
type errNotFound struct {
file string
}
func (e errNotFound) Error() string {
return fmt.Sprintf("file %q not found", e.file)
}
func IsNotFoundError(err error) bool {
_, ok := err.(errNotFound)
return ok
}
func Open(file string) error {
return errNotFound{file: file}
}
// package bar
if err := foo.Open("foo"); err != nil {
if foo.IsNotFoundError(err) {
// handle
} else {
panic("unknown error")
}
}
复制代码
下面提供三种主要的方法来传递函数调用失败返回的错误:
"pkg/errors".Wrap
来增长上下文信息,这样返回的错误信息中就会包含更多的上下文信息,而且经过 "pkg/errors".Cause
能够提取出原始错误信息。fmt.Errorf
。状况容许的话建议增长更多的上下文信息来代替诸如 "connection refused"
之类模糊的错误信息。返回 "failed to call service foo: connection refused"
用户能够知道更多有用的错误信息。
在将上下文信息添加到返回的错误时,请避免使用 "failed to" 之类的短语以保持信息简洁,这些短语描述的状态是显而易见的,而且会随着错误在堆栈中的传递而逐渐堆积:
Bad | Good |
---|---|
s, err := store.New()
if err != nil {
return fmt.Errorf(
"failed to create new store: %s", err)
}
复制代码 |
s, err := store.New()
if err != nil {
return fmt.Errorf(
"new store: %s", err)
}
复制代码 |
failed to x: failed to y: failed to create new store: the error
复制代码 |
x: y: new store: the error
复制代码 |
可是,若是这个错误信息是会被发送到另外一个系统时,必须清楚的代表这是一个错误(例如,日志中 err
标签或者 Failed
前缀)。
另见 Don't just check errors, handle them gracefully。
类型断言的单返回值形式在遇到类型错误时会直接 panic 。所以,请始终使用 "comma ok" 惯用方法。
Bad | Good |
---|---|
t := i.(string)
复制代码 |
t, ok := i.(string)
if !ok {
// handle the error gracefully
}
复制代码 |
生产级的代码必须避免 panics 。panics 是级联故障的主要源头。若是错误发生,函数应该返回错误而且容许调用者决定若是处理它。
Bad | Good |
---|---|
func foo(bar string) {
if len(bar) == 0 {
panic("bar must not be empty")
}
// ...
}
func main() {
if len(os.Args) != 2 {
fmt.Println("USAGE: foo <bar>")
os.Exit(1)
}
foo(os.Args[1])
}
复制代码 |
func foo(bar string) error {
if len(bar) == 0
return errors.New("bar must not be empty")
}
// ...
return nil
}
func main() {
if len(os.Args) != 2 {
fmt.Println("USAGE: foo <bar>")
os.Exit(1)
}
if err := foo(os.Args[1]); err != nil {
panic(err)
}
}
复制代码 |
Panic/recover 并非错误处理策略。程序只有在遇到没法处理的状况下才能够 panic ,例如,nil 引用。程序初始化时是一个例外状况:程序启动时遇到须要终止执行的错误可能会 painc 。
var _statusTemplate = template.Must(template.New("name").Parse("_statusHTML"))
复制代码
即便是在测试中,也应优先选择 t.Fatal
或 t.FailNow
而非 panic,以确保测试标记为失败。
Bad | Good |
---|---|
// func TestFoo(t *testing.T)
f, err := ioutil.TempFile("", "test")
if err != nil {
panic("failed to set up test")
}
复制代码 |
// func TestFoo(t *testing.T)
f, err := ioutil.TempFile("", "test")
if err != nil {
t.Fatal("failed to set up test")
}
复制代码 |
Go 的 sync/atomic
包仅仅提供针对原始类型(int32, int64, ...)的原子操做。所以,很容易忘记使用原子操做来读写变量。
go.uber.org/atomic 经过隐藏基础类型,使这些操做类型安全。而且,它还提供一个方便的 atomic.Bool
类型。
Bad | Good |
---|---|
type foo struct {
running int32 // atomic
}
func (f* foo) start() {
if atomic.SwapInt32(&f.running, 1) == 1 {
// already running…
return
}
// start the Foo
}
func (f *foo) isRunning() bool {
return f.running == 1 // race!
}
复制代码 |
type foo struct {
running atomic.Bool
}
func (f *foo) start() {
if f.running.Swap(true) {
// already running…
return
}
// start the Foo
}
func (f *foo) isRunning() bool {
return f.running.Load()
}
复制代码 |
性能方面的特定准则,仅适用于热路径。
将原语转换为字符串或从字符串转换时,strconv
速度比 fmt
更快。
Bad | Good |
---|---|
for i := 0; i < b.N; i++ {
s := fmt.Sprint(rand.Int())
}
复制代码 |
for i := 0; i < b.N; i++ {
s := strconv.Itoa(rand.Int())
}
复制代码 |
BenchmarkFmtSprint-4 143 ns/op 2 allocs/op
复制代码 |
BenchmarkStrconv-4 64.2 ns/op 1 allocs/op
复制代码 |
不要反复地从字符串字面量建立 byte 切片。相反,执行一次转换后存储结果供后续使用。
Bad | Good |
---|---|
for i := 0; i < b.N; i++ {
w.Write([]byte("Hello world"))
}
复制代码 |
data := []byte("Hello world")
for i := 0; i < b.N; i++ {
w.Write(data)
}
复制代码 |
BenchmarkBad-4 50000000 22.2 ns/op
复制代码 |
BenchmarkGood-4 500000000 3.25 ns/op
复制代码 |
Go 支持将类似的声明分组:
Bad | Good |
---|---|
import "a"
import "b"
复制代码 |
import (
"a"
"b"
)
复制代码 |
分组一样适用于常量、变量和类型的声明:
Bad | Good |
---|---|
const a = 1
const b = 2
var a = 1
var b = 2
type Area float64
type Volume float64
复制代码 |
const (
a = 1
b = 2
)
var (
a = 1
b = 2
)
type (
Area float64
Volume float64
)
复制代码 |
仅将类似的声明放在同一组。不相关的声明不要放在同一个组内。
Bad | Good |
---|---|
type Operation int
const (
Add Operation = iota + 1
Subtract
Multiply
ENV_VAR = "MY_ENV"
)
复制代码 |
type Operation int
const (
Add Operation = iota + 1
Subtract
Multiply
)
const ENV_VAR = "MY_ENV"
复制代码 |
声明分组能够在任意位置使用。例如,能够在函数内部使用。
Bad | Good |
---|---|
func f() string {
var red = color.New(0xff0000)
var green = color.New(0x00ff00)
var blue = color.New(0x0000ff)
...
}
复制代码 |
func f() string {
var (
red = color.New(0xff0000)
green = color.New(0x00ff00)
blue = color.New(0x0000ff)
)
...
}
复制代码 |
import 有两类导入组:
goimports 默认的分组以下:
Bad | Good |
---|---|
import (
"fmt"
"os"
"go.uber.org/atomic"
"golang.org/x/sync/errgroup"
)
复制代码 |
import (
"fmt"
"os"
"go.uber.org/atomic"
"golang.org/x/sync/errgroup"
)
复制代码 |
当为包命名时,请注意以下事项:
net/url
, 而不是 net/urls
另见 Package Names 和 Style guideline for Go packages
咱们遵循 Go 社区关于使用的 MixedCaps for function names。有一种状况例外,对相关的测试用例进行分组时,函数名可能包含下划线,如: TestMyFunction_WhatIsBeingTested
。
若是包的名称与导入路径的最后一个元素不匹配,那必须使用导入别名。
import (
"net/http"
client "example.com/client-go"
trace "example.com/trace/v2"
)
复制代码
在其余状况下,除非导入的包名之间有直接冲突,不然应避免使用导入别名。
Bad | Good |
---|---|
import (
"fmt"
"os"
nettrace "golang.net/x/trace"
)
复制代码 |
import (
"fmt"
"os"
"runtime/trace"
nettrace "golang.net/x/trace"
)
复制代码 |
因此,公开的函数应排布在文件首,并在 struct、const 和 var 定义以后。
newXYZ()/ NewXYZ() 之类的函数应该排布在声明类型以后,具备接收器的其他方法以前。
由于函数是按接收器类别分组的,因此普通工具函数应排布在文件末尾。
Bad | Good |
---|---|
func (s *something) Cost() {
return calcCost(s.weights)
}
type something struct{ ... }
func calcCost(n int[]) int {...}
func (s *something) Stop() {...}
func newSomething() *something {
return &something{}
}
复制代码 |
type something struct{ ... }
func newSomething() *something {
return &something{}
}
func (s *something) Cost() {
return calcCost(s.weights)
}
func (s *something) Stop() {...}
func calcCost(n int[]) int {...}
复制代码 |
代码应该经过尽量地先处理错误状况/特殊状况,而且及早返回或继续下一循环来减小嵌套。尽可能减小嵌套于多个级别的代码数量。
Bad | Good |
---|---|
for _, v := range data {
if v.F1 == 1 {
v = process(v)
if err := v.Call(); err == nil {
v.Send()
} else {
return err
}
} else {
log.Printf("Invalid v: %v", v)
}
}
复制代码 |
for _, v := range data {
if v.F1 != 1 {
log.Printf("Invalid v: %v", v)
continue
}
v = process(v)
if err := v.Call(); err != nil {
return err
}
v.Send()
}
复制代码 |
若是一个变量在 if 的两个分支中都设置了,那应该使用单个 if 。
Bad | Good |
---|---|
var a int
if b {
a = 100
} else {
a = 10
}
复制代码 |
a := 10
if b {
a = 100
}
复制代码 |
在顶层使用标准 var 关键字声明变量时,不要显式指定类型,除非它与表达式的返回类型不一样。
Bad | Good |
---|---|
var _s string = F()
func F() string { return "A" }
复制代码 |
var _s = F()
// F 已经明确声明返回一个字符串类型,咱们没有必要显式指定 _s 的类型
func F() string { return "A" }
复制代码 |
若是表达式的返回类型与所需的类型不彻底匹配,请显示指定类型。
type myError struct{}
func (myError) Error() string { return "error" }
func F() myError { return myError{} }
var _e error = F()
// F 返回一个 myError 类型的实例,可是咱们要 error 类型
复制代码
非导出的全局变量和常量前面加上前缀 _
,以明确表示它们是全局符号。
例外:未导出的错误类型变量,应以 err
开头。
解释:顶级(全局)变量和常量具备包范围做用域。使用通用名称命名,可能在其余文件中不经意间地使用一个错误值。
Bad | Good |
---|---|
// foo.go
const (
defaultPort = 8080
defaultUser = "user"
)
// bar.go
func Bar() {
defaultPort := 9090
...
fmt.Println("Default port", defaultPort)
// We will not see a compile error if the first line of
// Bar() is deleted.
}
复制代码 |
// foo.go
const (
_defaultPort = 8080
_defaultUser = "user"
)
复制代码 |
嵌入式类型(例如 mutex )应该放置在结构体字段列表的顶部,而且必须以空行与常规字段隔开。
Bad | Good |
---|---|
type Client struct {
version int
http.Client
}
复制代码 |
type Client struct {
http.Client
version int
}
复制代码 |
初始化结构体时,必须指定字段名称。go vet
强制执行。
Bad | Good |
---|---|
k := User{"John", "Doe", true}
复制代码 |
k := User{
FirstName: "John",
LastName: "Doe",
Admin: true,
}
复制代码 |
例外:在测试文件中,若是结构体只有3个或更少的字段,则能够省略字段名称。
tests := []struct{
}{
op Operation
want string
}{
{Add, "add"},
{Subtract, "subtract"},
}
复制代码
若是声明局部变量时须要明确设值,应使用短变量声明形式(:=)。
Bad | Good |
---|---|
var s = "foo"
复制代码 |
s := "foo"
复制代码 |
可是,在某些状况下,使用 var 关键字声明变量,默认的初始化值会更清晰。例如,声明空切片。
Bad | Good |
---|---|
func f(list []int) {
filtered := []int{}
for _, v := range list {
if v > 10 {
filtered = append(filtered, v)
}
}
}
复制代码 |
func f(list []int) {
var filtered []int
for _, v := range list {
if v > 10 {
filtered = append(filtered, v)
}
}
}
复制代码 |
nil 是一个有效的长度为 0 的 slice,这意味着:
不该明确返回长度为零的切片,而应该直接返回 nil 。
Bad | Good |
---|---|
if x == "" {
return []int{}
}
复制代码 |
if x == "" {
return nil
}
复制代码 |
若要检查切片是否为空,始终使用 len(s) == 0
,不要与 nil 比较来检查。
Bad | Good |
---|---|
func isEmpty(s []string) bool {
return s == nil
}
复制代码 |
func isEmpty(s []string) bool {
return len(s) == 0
}
复制代码 |
零值切片(经过 var 声明的切片)可直接使用,无需调用 make 建立。
Bad | Good |
---|---|
nums := []int{}
// or, nums := make([]int)
if add1 {
nums = append(nums, 1)
}
if add2 {
nums = append(nums, 2)
}
复制代码 |
var nums []int
if add1 {
nums = append(nums, 1)
}
if add2 {
nums = append(nums, 2)
}
复制代码 |
若是有可能,尽可能缩小变量做用范围,除非这样与减小嵌套的规则冲突。
Bad | Good |
---|---|
err := ioutil.WriteFile(name, data, 0644)
if err != nil {
return err
}
复制代码 |
if err := ioutil.WriteFile(name, data, 0644); err != nil {
return err
}
复制代码 |
若是须要在 if 以外使用函数调用的结果,则不该尝试缩小范围。
Bad | Good |
---|---|
if data, err := ioutil.ReadFile(name); err == nil {
err = cfg.Decode(data)
if err != nil {
return err
}
fmt.Println(cfg)
return nil
} else {
return err
}
复制代码 |
data, err := ioutil.ReadFile(name)
if err != nil {
return err
}
if err := cfg.Decode(data); err != nil {
return err
}
fmt.Println(cfg)
return nil
复制代码 |
函数调用中的裸参数可能会下降代码可读性。因此当参数名称的含义不明显时,请为参数添加 C 样式的注释(/* … */)。
Bad | Good |
---|---|
// func printInfo(name string, isLocal, done bool)
printInfo("foo", true, true)
复制代码 |
// func printInfo(name string, isLocal, done bool)
printInfo("foo", true /* isLocal */, true /* done */)
复制代码 |
上面更好的做法是将 bool 类型替换为自定义类型,从而使代码更易读且类型安全。未来须要拓展时,该参数也能够不止两个状态(true/false)。
type Region int
const (
UnknownRegion Region = iota
Local
)
type Status int
const (
StatusReady = iota + 1
StatusDone
// 也许未来咱们会有 StatusInProgress。
)
func printInfo(name string, region Region, status Status) 复制代码
Go 支持原始字符串字面值,能够多行并包含引号。使用它能够避免使用肉眼阅读较为困难的手工转义的字符串。
Bad | Good |
---|---|
wantError := "unknown name:\"test\""
复制代码 |
wantError := `unknown error:"test"`
复制代码 |
在初始化结构引用时,使用 &T{}
而非 new(T)
,以使其与结构体初始化方式保持一致。
Bad | Good |
---|---|
sval := T{Name: "foo"}
// 定义方式不一致
sptr := new(T)
sptr.Name = "bar"
复制代码 |
sval := T{Name: "foo"}
sptr := &T{Name: "bar"}
复制代码 |
若是为 Printf-style 函数声明格式化字符串,将格式化字符串放在函数外面 ,并将其设置为 const 常量。
这有助于 go vet
对格式字符串进行静态分析。
Bad | Good |
---|---|
msg := "unexpected values %v, %v\n"
fmt.Printf(msg, 1, 2)
复制代码 |
const msg = "unexpected values %v, %v\n"
fmt.Printf(msg, 1, 2)
复制代码 |
声明 Printf-style 函数时,请确保 go vet
能够检查它的格式化字符串。
这意味着应尽量使用预约义的 Printf-style 函数名称。go vet
默认会检查它们。更多相关信息,请参见 Printf系列。
若是不能使用预约义的名称,请以 f 结尾:Wrapf,而非 Wrap。由于 go vet
能够指定检查特定的 Printf 样式名称,但名称必须以 f 结尾。
$ go vet -printfuncs=wrapf,statusf
...
复制代码
另见 go vet: Printf family check
在核心测试逻辑重复时,将表驱动测试与子测试一块儿使用,以免重复代码。
Bad | Good |
---|---|
// func TestSplitHostPort(t *testing.T)
host, port, err := net.SplitHostPort("192.0.2.0:8000")
require.NoError(t, err)
assert.Equal(t, "192.0.2.0", host)
assert.Equal(t, "8000", port)
host, port, err = net.SplitHostPort("192.0.2.0:http")
require.NoError(t, err)
assert.Equal(t, "192.0.2.0", host)
assert.Equal(t, "http", port)
host, port, err = net.SplitHostPort(":8000")
require.NoError(t, err)
assert.Equal(t, "", host)
assert.Equal(t, "8000", port)
host, port, err = net.SplitHostPort("1:8")
require.NoError(t, err)
assert.Equal(t, "1", host)
assert.Equal(t, "8", port)
复制代码 |
// func TestSplitHostPort(t *testing.T)
tests := []struct{
give string
wantHost string
wantPort string
}{
{
give: "192.0.2.0:8000",
wantHost: "192.0.2.0",
wantPort: "8000",
},
{
give: "192.0.2.0:http",
wantHost: "192.0.2.0",
wantPort: "http",
},
{
give: ":8000",
wantHost: "",
wantPort: "8000",
},
{
give: "1:8",
wantHost: "1",
wantPort: "8",
},
}
for _, tt := range tests {
t.Run(tt.give, func(t *testing.T) {
host, port, err := net.SplitHostPort(tt.give)
require.NoError(t, err)
assert.Equal(t, tt.wantHost, host)
assert.Equal(t, tt.wantPort, port)
})
}
复制代码 |
测试表使得向错误消息注入上下文信息,减小重复的逻辑,添加新的测试用例变得更加容易。
咱们遵循这样的约定:将结构体切片称为 tests。 每一个测试用例称为 tt 。此外,咱们鼓励使用 give 和 want 前缀说明每一个测试用例的输入和输出值。
tests := []struct{
give string
wantHost string
wantPort string
}{
// ...
}
for _, tt := range tests {
// ...
}
复制代码
功能选项是一种模式,声明一个不透明 Option 类型,该类型记录某些内部结构体的信息。您的函数接受这些不定数量的选项参数,并将选项参数上的信息做用于内部结构上。
此模式可用于扩展构造函数和实现其余公共 API 中的可选参数,特别是这些参数已经有三个或者超过三个的状况下。
Bad | Good |
---|---|
// package db
func Connect( addr string, timeout time.Duration, caching bool, ) (*Connection, error) {
// ...
}
// Timeout and caching must always be provided,
// even if the user wants to use the default.
db.Connect(addr, db.DefaultTimeout, db.DefaultCaching)
db.Connect(addr, newTimeout, db.DefaultCaching)
db.Connect(addr, db.DefaultTimeout, false /* caching */)
db.Connect(addr, newTimeout, false /* caching */)
复制代码 |
type options struct {
timeout time.Duration
caching bool
}
// Option overrides behavior of Connect.
type Option interface {
apply(*options)
}
type optionFunc func(*options) func (f optionFunc) apply(o *options) {
f(o)
}
func WithTimeout(t time.Duration) Option {
return optionFunc(func(o *options) {
o.timeout = t
})
}
func WithCaching(cache bool) Option {
return optionFunc(func(o *options) {
o.caching = cache
})
}
// Connect creates a connection.
func Connect( addr string, opts ...Option, ) (*Connection, error) {
options := options{
timeout: defaultTimeout,
caching: defaultCaching,
}
for _, o := range opts {
o.apply(&options)
}
// ...
}
// Options must be provided only if needed.
db.Connect(addr)
db.Connect(addr, db.WithTimeout(newTimeout))
db.Connect(addr, db.WithCaching(false))
db.Connect(
addr,
db.WithCaching(false),
db.WithTimeout(newTimeout),
)
复制代码 |
另见,