Good Code vs Bad Code in Golang

0xinhua 发布于

最近在学习Go,也顺便在GCTT社区帮忙翻译了一些文章。

原文地址:Good Code vs Bad Code in Golang 作者:Teivah Harsanyi

最近我被问到关于Golang好代码和不好的代码的一些细节问题,我发现这个问题非常有趣,有趣到可以就这个问题写一篇文章。为了阐明我的观点,我举了一些我在Air Traffic ManagementATM)航空交通管制领域的代码例子,这些代码托管在Github上。

背景

首先,简短地描述一下项目的背景。

Eurocontrol(欧洲航管组织)是一个管理欧洲各国航空交通的组织。EurocontrolAir Navigation Service Provider(ANSP)之间共同的数据交换网络称为AFTN。这个网络主要用来交换两种格式的信息: ADEXPICAO(国际民航组织)格式信息。每一种信息都有它们自己的语法格式,但在语义上,或多或少它们是相同的。在这个背景下,性能必须是实现的关键因素。

这个项目需要提供基于GO语言的用于转换ADEXP格式信息的两种实现(ICAO格式的转换并没有体现在这次对比中):

  • 不好的实现方式(package包名:bad

  • 重构之后的实现方式(package包名:good

  • ADEXP格式的信息例子可以在这里找到。

这次对比的结构里,在ADEXP信息中,解析处理部分只是一个子集,至今它仍然跟Go中常见的一些错误有关系。

解析

概括地讲,一个ADEXP信息是一个tokens的集合,可以是下面任意一种:

  • 简单的

-ARCID ACA878

代表ARCID(航班ID) 是ACA878

  • 循环的

-EETFIR EHAA 0853
-EETFIR EBBU 0908

这个例子代表的是一个FIR(飞行情报区)列表,第一个航班是EHAA 0853,第二个EBBU 0908

  • 复杂的

-GEO -GEOID GEO01 -LATTD 490000N -LONGTD 0500000W
-GEO -GEOID GEO02 -LATTD 500000N -LONGTD 0400000W

一个循环的tockens列表,每一行包含一个子token列表(在这个例子中是GEOIDLATTDLONGTD)。

结合项目背景,实现一个并行执行的转化代码变得很重要。所以有以下算法:

  • 实现一个对输入进行清理和重新排列的预处理过程(我们需要清除可能存在的空格,重新排列例如COMMENT等的多行tokens

  • 然后分割每一行至一个协程中。每一个协程将会负责处理一行tokens,并且返回结果。

  • 最后一个步骤同样重要,整合结果并且返回一个Message消息)结构体,这是一个公共的结构,不管是ADEXP还是ICAO类型的信息。

每一个包包含一个adexp.go文件暴露主要的ParseAdexpMessage()方法。

逐步对比

现在我们逐步来看下我认为的不好的代码,并且我是如何重构它的。

String类型 vs []byte类型

限制输入类型为字符串类型并不好。Gobyte类型的处理提供了强大的支持(例如基础的trim, regexp等),并且输入将会很大程度上类似于[]byte(鉴于AFTN信息是通过TCP协议来接收的),实际上没有理由强制要求字符串类型的输入。

错误处理

错误的处理实现有点糟糕。

我们会发现忽视了在第二个参数中返回的一些可能存在的错误:

Go
1preprocessed, _ := preprocess(string)

好的实现方式是捕获每一个可能的错误:

Go
1preprocessed, err := preprocess(bytes) 2if err != nil { 3return Message{}, err 4}

我们可以在下面这种不好的代码中也能找到犯的一些错误:

Go
1if len(in) == 0 { 2return "", fmt.Errorf("Input is empty") 3}

第一个错误是语法上的。根据Go的语法,错误提示的字符串既不是大写也不是以标点符号结尾。

第二个错误是如果一个错误信息是一个简单的字符串常量(不需要格式化),使用轻量的errors.New()性能会更好。

好的实现如下:

Go
1if len(in) == 0 { 2 return nil, errors.New("input is empty") 3}

避免嵌套

mapLine()函数就是一个很好的例子来说明避免嵌套调用。不好的代码如下:

Go
1func mapLine(msg *Message, in string, ch chan string) { 2 if !startWith(in, stringComment) { 3 token, value := parseLine(in) 4 if token != "" { 5 f, contains := factory[string(token)] 6 if !contains { 7 ch <- "ok" 8 } else { 9 data := f(token, value) 10 enrichMessage(msg, data) 11 ch <- "ok" 12 } 13 } else { 14 ch <- "ok" 15 return 16 } 17 } else { 18 ch <- "ok" 19 return 20 } 21}

相反,好的代码表示得很清晰:

Go
1func mapLine(in []byte, ch chan interface{}) { 2 // Filter empty lines and comment lines 3 if len(in) == 0 || startWith(in, bytesComment) { 4 ch <- nil 5 return 6 } 7 8 token, value := parseLine(in) 9 if token == nil { 10 ch <- nil 11 log.Warnf("Token name is empty on line %v", string(in)) 12 return 13 } 14 15 sToken := string(token) 16 if f, contains := factory[sToken]; contains { 17 ch <- f(sToken, value) 18 return 19 } 20 21 log.Warnf("Token %v is not managed by the parser", string(in)) 22 ch <- nil 23}

在我看来,这使得代码易读性更强。此外,这种扁平化的处理方式也应该加到错误捕获代码中,下面的例子:

Go
1a, err := f1() 2if err == nil { 3 b, err := f2() 4 if err == nil { 5 return b, nil 6 } else { 7 return nil, err 8 } 9} else { 10 return nil, err 11}

应该被修改成:

Go
1a, err := f1() 2if err != nil { 3 return nil, err 4} 5b, err := f2() 6if err != nil { 7 return nil, err 8} 9return b, nil

同样,第二段代码的可读性更好。

传值采用value还是reference

预处理方法的实现并不好:

Go
1func preprocess(in container) (container, error) { 2}

结合项目的背景来说(考虑性能),考虑到一个信息的结构体有可能会比较大,更好的方式是在container结构内传入指针,否则,在上面例子的代码中container的值将会在每一次调用的时候被覆盖掉。

好的实现代码将不会有这个问题因为它单个的处理方式(一个简单的24字节的结构,不管什么类型数据)。

go
1func preprocess(in []byte) ([][]byte, error) { 2}

更广泛地说,无论是根据引用还是数值传递参数都不是一个符合语言习惯的用法。通过数值传递数据也能帮助确定一个方法将不会带来任何的副作用(就像在函数的输入中传递数据一样)。这样做有几个好处,例如单元测试、在代码并发上的重构(否则我们需要检查每个子函数来确定传递是否完成)

我确信这种这种写法需要根据实际项目背景小心地使用。

并发

不好的实现方式源于一个最初的好的想法:利用协程并发处理数据(一个协程处理一行)。

这导致了在一个协程里反复调用mapLine()

go
1for i := 0; i < len(lines); i++ { 2 go mapLine(&msg, lines[i], ch) 3}

mapLine()方法三个参数:

  • 返回指向最后一个Message结构的指针。这意味着每个mapLine()将会被同一个变量填充。

  • 当前行

  • 一个channel通道用于处理行完成时发送消息

为了共享Message消息而去传递一个指针违背了Go基本原则:

不要通过共享内存来通信,而应该通过通信来共享内存

传递共享的变量有两个主要的缺点:

  • 缺点 #1:分割一致的修饰

因为结构中包含一些切片可以被同时修改(同时被两个或者更多的协程)我们得处理互斥的问题。

例如,Message消息结构包含一个Estdata []estdata,通过加上另一个estdata修改这部分必须像下面这样处理:

go
1mutexEstdata.Lock() 2for _, v := range value { 3 fl := extractFlightLevel(v[subtokenFl]) 4 msg.Estdata = append(msg.Estdata, estdata{v[subtokenPtid], v[subtokenEto], fl}) 5} 6mutexEstdata.Unlock()

事实上,排除特定用法,在协程中使用mutex(互斥锁)并不是一个好的选择。

  • 缺点 #2:伪共享

通过线程或者协程分享内存并不是一个好的方式因为可能存在伪共享(一个在CPU缓存中的进程可以被另一个CPU缓存)。这意味着我们需要尽可能地避免通过线程和协程来共享那些需要修改的变量。

在这个例子中,虽然,我不认为伪共享在输入的文件教少的情况下有一个很大的影响(在Message消息结构体中增加文件的性能测试结果或多或少是一样的),但在我看来这也是一个很重要的需要牢记的点。

现在让我们来看下好的并发处理:

Go
1for _, line := range in { 2 go mapLine(line, ch) 3}

现在,mapLine()只接受两个参数:

  • 当前行

  • channel通道,当前行处理完后,这里的通道不再简单地用来发生消息,也用来传送实际的结果。这意味着不应该使用协程去修改最后的消息结构。

结果在父级的协程中整合。(产生的mapLine()方法在各自的协程中被调用)

Go
1msg := Message{} 2 3for range in { 4 data := <-ch 5 6 switch data.(type) { 7 // Modify msg variable 8 } 9}

这个代码更加一致,在我看来,根据Go的原则:通过通信来共享内存。通过单一的协程来修改消息变量防止了可能由于并发导致的修改和伪共享问题。

这部分代码潜在的问题是造成了每一行一个协程,这种实现将能够运行,因为ADEXP信息行数将不会太大,在这个简单实现中,一个请求将产生一个协程,在生成能力上将无法考量。一个更好的选择是创建一个协程池来复用协程。

行处理通知

在上面不好的代码中,一旦mapLine()处理完一行,我们需要在父级的协程中进行标识。这部分将通过使用chan string通道和方法的调用:

Go
1ch <- "ok"

因为父级协程并不会检查通道传过来的结果,较好的处理是通过chan struct{}使用ch <- struct{}{},或者更好的选择(GC会更差)是通过chan interface{}使用ch <- nil处理。

另一个类似的方法(在我看来甚至会更简洁)是使用sync.WaitGroup,因为当每一个mapLine()执行完了,父级的协程还需继续运行。

If条件判断

Go的条件判断语句中,允许在条件前进行赋值。

一个改进版的代码:

Go
1f, contains := factory[string(token)] 2if contains { 3 // Do something 4}

如下:

Go
1if f, contains := factory[sToken]; contains { 2 // Do something 3}

这样代码的可读性更高。

Switch选择

代码中犯得另一个错误是没有设置switch中的default选项:

Go
1switch simpleToken.token { 2case tokenTitle: 3 msg.Title = value 4case tokenAdep: 5 msg.Adep = value 6case tokenAltnz: 7 msg.Alternate = value 8// Other cases 9}

如果开发者能够考虑到所有的情况,switchdefault项是可选的,但是像下面这样捕获特殊的情况肯定会更好。

Go
1switch simpleToken.token { 2case tokenTitle: 3 msg.Title = value 4case tokenAdep: 5 msg.Adep = value 6case tokenAltnz: 7 msg.Alternate = value 8// Other cases 9default: 10 log.Errorf("unexpected token type %v", simpleToken.token) 11 return Message{}, fmt.Errorf("unexpected token type %v", simpleToken.token) 12}

处理default选项会帮助开发者捕获开发过程中可能造成的bugs

递归

parseComplexLines()是一个解析复杂token的方法,在不好的代码中是使用递归来处理:

Go
1func parseComplexLines(in string, currentMap map[string]string, 2 out []map[string]string) []map[string]string { 3 4 match := regexpSubfield.Find([]byte(in)) 5 6 if match == nil { 7 out = append(out, currentMap) 8 return out 9 } 10 11 sub := string(match) 12 13 h, l := parseLine(sub) 14 15 _, contains := currentMap[string(h)] 16 17 if contains { 18 out = append(out, currentMap) 19 currentMap = make(map[string]string) 20 } 21 22 currentMap[string(h)] = string(strings.Trim(l, stringEmpty)) 23 24 return parseComplexLines(in[len(sub):], currentMap, out) 25}

但是Go不支持尾调用优化递归,好的代码使用迭代算法能得到同样的结果:

Go
1func parseComplexToken(token string, value []byte) interface{} { 2 if value == nil { 3 log.Warnf("Empty value") 4 return complexToken{token, nil} 5 } 6 7 var v []map[string]string 8 currentMap := make(map[string]string) 9 10 matches := regexpSubfield.FindAll(value, -1) 11 12 for _, sub := range matches { 13 h, l := parseLine(sub) 14 15 if _, contains := currentMap[string(h)]; contains { 16 v = append(v, currentMap) 17 currentMap = make(map[string]string) 18 } 19 20 currentMap[string(h)] = string(bytes.Trim(l, stringEmpty)) 21 } 22 v = append(v, currentMap) 23 24 return complexToken{token, v} 25}

第二种写法在性能上会优于第一种。

常量管理

我们需要管理一个常量来区分ADEXPICAO类型的消息。不好的写法如下:

Go
1const ( 2 AdexpType = 0 // TODO constant 3 IcaoType = 1 4) 5

反之,利用Goiota(常量计数器)能写出更优雅的代码:

Go
1const ( 2 AdexpType = iota 3 IcaoType 4)

输出的结果是一致的,但是规避了可能存在的错误。

接收方法

每个转换提供一个函数来判断每个Message是非涉及到上一个级别(至少在350级以上的路线点)

不好的代码如下:

Go
1func IsUpperLevel(m Message) bool { 2 for _, r := range m.RoutePoints { 3 if r.FlightLevel > upperLevel { 4 return true 5 } 6 } 7 8 return false 9} 10

意味着我们必须把Message当函数的参数。反之,好的代码是提供一个简单的函数作为Message的接收者。

Go
1func (m *Message) IsUpperLevel() bool { 2 for _, r := range m.RoutePoints { 3 if r.FlightLevel > upperLevel { 4 return true 5 } 6 } 7 8 return false 9}

第二种方法是更可取的,我们简单地让消息实现了一个特定的行为。这也是使用Go实现接口的第一步,如果某天我们需要创建另一个同一行为(IsUpperLevel())的结构,这部分代码不需要被重构(因为Message已经继承该行为)。

注释

注释这部分的问题很明显,但是不好的注释很少被提及。

另一方面,在真实项目中我尝试着去做很好的注释,尽管我并不是那种每行都会注释的人,我依然相信至少在每个函数和在一个复杂的函数的核心部分写上注释是非常重要的事情。

举个例子:

Go
1// Split each line in a goroutine 2for _, line := range in { 3 go mapLine(line, ch) 4} 5 6msg := Message{} 7 8// Gather the goroutine results 9for range in { 10 // ... 11} 12

在注释中提供一个另外的例子也会提供很大的帮助:

Go
1// Parse a line by returning the header (token name) and the value. 2// Example: -COMMENT TEST must returns COMMENT and TEST (in byte slices) 3func parseLine(in []byte) ([]byte, []byte) { 4 // ... 5}

这样的一个额外的例子对另一个开发者更好地了解当前这个项目很有帮助。

最后同样重要的,根据Go的最佳实践,包本身也需要注释。

Go
1/* 2Package good is a library for parsing the ADEXP messages. 3An intermediate format Message is built by the parser. 4*/ 5 6package good 7

日志处理

另一个很明显的问题是在缺少日志处理。因为我并不喜欢Go提供的标准日志package包,我在这个项目中使用一个叫logrus第三方的日志包。

go的fmt包

Go提供一个强力的工具集go fmt。遗憾的是我们忘记去利用它。

DDD

DDD带来了通用语言的概念,以强调所有项目之间共享语言的重要性(business experts, dev, testers 等)。

这点在这个项目上可能无法衡量,但是从整个项目的可维护性上来考虑,保持一个简单的兼容上下文结构的Message也是重要的一点。

性能结果

i7-7700 4x 3.60Ghz的硬件环境下,我通过基准测试来对比两种代码:

  • 好的代码: 60430 ns/op

  • 不好的代码: 45996 ns/op

不好的代码比好的代码慢了超过30%。

结论

在我看来,要对好的代码和不好的代码给出一个普遍的定义非常难。在一个环境中,一段代码可以被认为是好的代码,但在另一个环境中可能被认为是不好的代码。

一个很明显的特征就是好的代码在给定的环境中能很好地解决问题,如果代码高效,但是不满足需求,那也是徒劳。

同时,考虑代码的简洁、可维护性、高效对开发者来说非常重要。

性能的提升伴随着代码复杂性的增加。

一个好的开发者能够在给定的环境中在上面这些特征里找到一个平衡。就像在DDD里,context就是解决方案🙂。