一个Go方法，五种变换
Go语言中文网 原文地址：https://mp.weixin.qq.com/s/GR-8Nv65N66xbOPL_Q1mxg
我们先来模拟一段开车去商场购物的代码。主要经历三个阶段：开车去商店、到商店购物、然后购物完成开车回家。每个过程都需要去check其中返回的err错误并返回shopper对象。
// 开车去商店
shopper, err := shopper.Drive(FuelNeededToGetToStore)
ifnil!= err {
log.Fatalf("could not complete shopping: %s", err)
}
// 买鸡蛋
shopper, err = shopper.BuyEggs(EggsRequired)
ifnil!= err {
log.Fatalf("could not complete shopping: %s", err)
}
// 买完鸡蛋开车回家
shopper, err = shopper.Drive(FuelNeededToGetHome)
ifnil!= err {
log.Fatalf("could not complete shopping: %s", err)
}
变换1:err集中判断
针对这段代码，错误处理代码较多，可以将三个重复代码块抽离成一个代码块。代码如下：
func main(){
shopper, err := shopper.Drive(FuelNeededToGetToStore)
FatalIfErrNotNil(err)
shopper, err = shopper.BuyEggs(EggsRequired)
FatalIfErrNotNil(err)
shopper, err = shopper.Drive(FuelNeededToGetHome)
FatalIfErrNotNil(err)
}
func FatalIfErrNotNil(err error){
if nil!= err {
log.Fatalf("could not complete shopping: %s", err)
}
}
不过在生产项目中，正常业务我们一般不会使用fatal去对非nil的err处理，fatal会以非0状态退出程序。而是使用https://github.com/pkg/errors对err进行wrap和cause 或者使用定义的业务相关的err并返回上一层做处理。这里也可以尝试使用bool判断来对err处理（不过之前土拨鼠之前群里也提出这样的想法，大家都说多此一举，尴尬）。
func main(){
shopper, err := shopper.Drive(FuelNeededToGetToStore)
ifErrorHandled(err) {
return...
}
}
func ErrorHandled(err)bool{
if nil!= err {
return true
}
// 也可以对error进行其他判断操作
...
return false
}
变换2:方法中包括err处理
是不是觉得上面的err处理还是不够优雅，所以咱们这里考虑把err的处理移出到主线之外。每个方法里都包含了对err的非nil判断和返回。每个操作也只有在err为nil的情况下正常执行 。只有在三个操作都结束后才真正对err进行处理。
func main(){
shopper, err := shopper.Drive(FuelNeededToGetToStore,nil)
shopper, err = shopper.BuyEggs(EggsRequired, err)
shopper, err = shopper.Drive(FuelNeededToGetHome, err)
if nil!= err {
log.Fatalf("could not complete shopping: %s", err)
}
}
func (s Shopper) Drive(fuelRequired int, err error)(Shopper, error){
if nil!= err {
return s, err
}
变换3:使用函数对err分解
下面这个变换，抛弃了之前Shopper作为receiver的做法，而是将Shopper作为函数的参数。其实跟变换2很相似。这儿是单独抽离出一个公共的函数ErrCheckFunc进行err处理和函数执行。
func main(){
drive := ErrCheckFunc(Drive)
buy := ErrCheckFunc(BuyEggs)
err, shopper := drive(nil, shopper, FuelNeededToGetToStore)
err, shopper = buy(err, shopper, EggsRequired)
err, shopper = drive(err, shopper, FuelNeededToGetHome)
if nil!= err {
log.Fatalf("could not complete shopping: %s", err)
}
}
funcDrive(s Shopper, fuelRequired int)(Shopper, error){
if nil!= err {
returns, err
}
...
}
funcErrCheckFunc(f func(Shopper, int)(Shopper, error))func(error, Shopper, int)(error, Shopper){
return func(err error, s Shopper, arg int)(error, Shopper){
if nil!= err {
return err, s
}
s, err = f(s, arg)
return err, s
}
}
变换4:单行操作
这次改进是采用了装饰器模式的思想把err的check逻辑和函数的执行操作都封装在了ErrCheckFunc一个方法中。相比变换3，是将ErrCheckFunc中的返回值中的函数入参arg转移到了ErrCheckFunc入参中的arg参数。
func main(){
driveToStore := ErrCheckFunc(Drive, FuelNeededToGetToStore)
buyEggs := ErrCheckFunc(BuyEggs, EggsRequired)
driveHome := ErrCheckFunc(Drive, FuelNeededToGetHome)
err, shopper := driveHome(buyEggs(driveToStore(nil, shopper)))
if nil!= err {
log.Fatalf("could not complete shopping: %s", err)
}
}
funcErrCheckFunc(f func(Shopper, int)(Shopper, error),arg int)func(error, Shopper)(error, Shopper){
return func(err error, s Shopper)(error, Shopper){
if nil!= err {
return err, s
}
s, err = f(s, arg)
returnerr, s
}
}
变换5:迭代变换
终极变换来了，这次变换采用了迭代的思想。ProcessSteps这里使用的是一个可变的函数参数。
func main(){
driveToStore := Flavorize(Drive, FuelNeededToGetToStore)
buyEggs := Flavorize(BuyEggs, EggsRequired)
driveHome := Flavorize(Drive, FuelNeededToGetHome)
shopper, err := ProcessSteps(shopper,
driveToStore,
buyEggs,
driveHome,
)
if nil!= err {
log.Fatalf("could not complete shopping: %s", err)
}
}
funcProcessSteps(s Shopper, steps ...func(Shopper)(Shopper, error))(Shopper, error){
for_, step :=range steps {
var err error
s, err = step(s)
if nil!= err {
returns, err
}
}
return s, nil
}
func Flavorize(f func(Shopper, int)(Shopper, error),arg int)func(Shopper)(Shopper, error){
return func(s Shopper)(Shopper, error){
return f(s, arg)
}
}
土拨鼠今天拿这个例子来展示给大家，主要是因为在项目中也会经常遇到这样的代码逻辑场景，就想着看看别人是怎么去重构改进的，学习怎么去优化重构，使代码变得更好维护、易于扩展、复用性高、可读性高。希望大家也可以借鉴学习，如果你有更好的变换和想法欢迎大家留言讨论。
网友评论