golang编码最佳实践(持续更新中)

最近在学习go语言,以此记录日常编码中的最佳实践,欢迎大家一起讨论

注释模板

使用Goanno插件:https://github.com/loveinsky100/goanno

设置模板

以goland为例

  1. 选择"工具-Goanno设置"
  1. 编辑模板

    // ${function_name} ${todo}
    // @receiver ${receiver}
    // @param ${params}
    // @return ${return_types}

案例

Go 复制代码
// GetNode 获取指定结构,从缓存加载
//
//	@receiver c
//	@param ctx
//	@param countryCode	国家码
//	@return *Node
func (c *Cache) GetNode(ctx context.Context, countryCode string) *Node, error {
	...
	return cacheNode, nil
}

日志打印

建议使用zap的零内存分配api,性能比go标准库的api好:GitHub - uber-go/zap: Blazing fast, structured, leveled logging in Go.

格式处理

  • %v: 根据值的类型自动选择合适的格式输出。
  • %+v: 类似 %v,但会输出更多的信息,如字段名称。
  • %#v: 输出值的完整 Go 语法表示,包括类型信息。
  • 不要直接打印byte[]类型等无实际意义的日志,应该转成string打印。

错误处理

go最受争议的部分之一就是错误处理,这里不去讨论其好坏,仍然推荐大家使用官方的处理方式:返回error接口,不建议使用panic + recover,容易导致程序崩溃

Go 复制代码
func f() error {
    if ... {
        return errors.New("xxx")
    }
    return nil
}

解决error无堆栈

部分goer建议使用panic + recover就是因为其有堆栈而error没有,但我们可以实现新的error接口时增加堆栈记录

Go 复制代码
type BizError struct {
	code    string
	message string
	cause   error

	stacktrace struct {
		onCreate *stacktrace
		onPanic  *stacktrace
	}
}

// NewBizError create a new BizError
//   - cause can be nil is no underlying error
//   - omitStacks indicates how many frames should be dropped, if <=0 no stacks will be filled
func NewBizError(code api.ResultCode, message string, cause error, omitStacks int, throwInPlace bool) *BizError {
	err := &BizError{
		code:    code.Code(),
		message: message,
		cause:   cause,
	}
	if omitStacks >= 0 {
		err.stacktrace.onCreate = dumpStacktrace(omitStacks + 1)
		if throwInPlace {
			err.stacktrace.onPanic = err.stacktrace.onCreate
		}
	}
	return err
}

func (this *BizError) Error() string {
	if len(this.message) > 0 {
		return this.code + ": " + this.message
	} else {
		return this.code + ": [NO_MESSAGE]"
	}
}

type stacktrace struct {
	header []byte // goroutine header (e.g., "goroutine 3 [running]:")
	frames []byte // the stacktrace details
}

func dumpStacktrace(skip int) *stacktrace {
	skip += 2 // skip debug.Stack and this frame
	skip *= 2 // 2-line each frame
	stack := debug.Stack()
	var header []byte
	for i, b := range stack { // assumes no unicode in stack, iterate on bytes
		if b == '\n' {
			if header == nil {
				// consume first line as goroutine header
				header = stack[:i]
			} else {
				skip--
				if skip == 0 {
					stack = stack[i:]
					break
				}
			}
		}
	}
	if skip > 0 {
		panic("skip overflow")
	}
	return &stacktrace{header, stack}
}

这样使用时即可记录堆栈

类型转换

因为 Golang 语言是强类型,所以经常会使用到类型转换,所以在这里推荐类型转换三方库:GitHub - spf13/cast: safe and easy casting from one type to another in Go

Go 复制代码
cast.ToString("mayonegg")         // "mayonegg"
cast.ToString(8)                  // "8"
cast.ToString(8.31)               // "8.31"
cast.ToString([]byte("one time")) // "one time"
cast.ToString(nil)                // ""

var foo interface{} = "one more time"
cast.ToString(foo)                // "one more time"

cast.ToInt(8)                  // 8
cast.ToInt(8.31)               // 8
cast.ToInt("8")                // 8
cast.ToInt(true)               // 1
cast.ToInt(false)              // 0

var eight interface{} = 8
cast.ToInt(eight)              // 8
cast.ToInt(nil)                // 0

json工具

golang原生对json已经做了很好的支持,简单易用,但其性能一直为人垢病,因此建议使用字节开源的工具sonic。除了能平替原生的json使用姿势外,在性能上面也是从底层方面做了很多文章进行优化,性能方面遥遥领先:sonic:基于 JIT 技术的开源全场景高性能 JSON 库_原生云_火山引擎开发者社区_InfoQ写作社区

web/rpc服务

单独为某个下游接口设置超时时间

要知道go的超时时间不像java那样每个调用都固定超时时间,而是以总体时间来计算,但有时部分下游就是需要超出原超时时间进行(当然必须是异步调用,否则就自相矛盾了),具体代码如下:

Go 复制代码
// 注意:必须先cancel再设置,否则只能设置比原来时间更短的时间
func TestTimeout(t *testing.T) {
	ctx := context.Background()
	ctx, _ = context.WithTimeout(ctx, time.Second*3)
	ctx, _ = context.WithTimeout(ctx, time.Second*6)

	deadline, _ := ctx.Deadline()
	fmt.Println(deadline.Sub(time.Now())) // 2.99s, 直接覆盖设置不生效

	ctx, _ = context.WithTimeout(ctx, time.Second*1)
	deadline, _ = ctx.Deadline()
	fmt.Println(deadline.Sub(time.Now())) // 0.99s 设置更短时间, 生效

	ctx = context.WithoutCancel(ctx)                 // 取消
	ctx, _ = context.WithTimeout(ctx, time.Second*6) // 重新设置
	deadline, _ = ctx.Deadline()
	fmt.Println(deadline.Sub(time.Now())) // 5.99s, 取消后再设置, 才生效
}

proto参数校验

建议使用PGV:GitHub - bufbuild/protoc-gen-validate: Protocol Buffer Validation - Being replaced by github.com/bufbuild/protovalidate

这样会再生成一个proto的validate文件

建议consumer在远程调用前先调用validate方法,当参数不合法时提前感知

Go 复制代码
func remote(ctx context.Context) {
    request := ...
    if err := request.ValidateAll(); err != nil {
		return nil, errors.New(err.Error())
	}
    remoteClient.GetXXX(ctx, request)
}

provider在实现时也必须调用validate,防止参数不合法

Go 复制代码
func (s *server) Remote(ctx context.Context, request XXX) Response, error {
    if err := request.ValidateAll(); err != nil {
		return nil, errors.New(err.Error())
	}
    // 处理逻辑
}

但每个方法都要加validate重复代码,有没有办法统一处理参数校验呢,当然是有的

Go 复制代码
type validator interface {
	ValidateAll() error
}

// ValidateAllInterceptor
//
//	@Description: grpc服务注册的参数校验拦截器
//	@return grpc.UnaryServerInterceptor
func ValidateAllInterceptor() grpc.UnaryServerInterceptor {
	return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (reply interface{}, err error) {
		if v, ok := req.(validator); ok {
			if err := v.ValidateAll(); err != nil {
				return nil, err
			}
		}
		return handler(ctx, req)
	}
}
相关推荐
q567315236 分钟前
在 Bash 中获取 Python 模块变量列
开发语言·python·bash
许野平31 分钟前
Rust: 利用 chrono 库实现日期和字符串互相转换
开发语言·后端·rust·字符串·转换·日期·chrono
也无晴也无风雨35 分钟前
在JS中, 0 == [0] 吗
开发语言·javascript
狂奔solar43 分钟前
yelp数据集上识别潜在的热门商家
开发语言·python
blammmp1 小时前
Java:数据结构-枚举
java·开发语言·数据结构
何曾参静谧2 小时前
「C/C++」C/C++ 指针篇 之 指针运算
c语言·开发语言·c++
暗黑起源喵2 小时前
设计模式-工厂设计模式
java·开发语言·设计模式
WaaTong2 小时前
Java反射
java·开发语言·反射
Troc_wangpeng2 小时前
R language 关于二维平面直角坐标系的制作
开发语言·机器学习
努力的家伙是不讨厌的2 小时前
解析json导出csv或者直接入库
开发语言·python·json