Golang的一些优雅设计

最新推荐文章于 2023-12-06 13:39:14 发布

Spring_java_gg

最新推荐文章于 2023-12-06 13:39:14 发布

阅读量249

点赞数

文章标签： golang ios 开发语言后端

原文链接：https://mp.weixin.qq.com/s?__biz=MzI3NTEwOTA4OQ==&mid=2649183071&idx=1&sn=250c4f84c3daaea64961eb1cbd0bb88f&chksm=f31a36d1c46dbfc72e95bc00d98d5725f409fd6c874f49733fdecaf2ddaa77ac4d10e4be70c3&scene=126&&sessionid=0

版权

借助一些设计模式、流式编程、函数编程的方法可以让我们的Golang代码更清晰优雅，本文中描述了在错误处理、可选配置、并发控制等方面的优化手段。

链式错误处理

很多人不喜欢Go的错误处理，需要写大量if err != nil的代码，特别是在一些复杂步骤场景中，每一步都要判断结果是否出错。在这种情况中，可以通过类似链式调用将错误封装在其中。

比如在对象中附带一个error属性，在每一步调用中如果error不为空直接返回

type Handler struct {
    props interface

    err error
}

func (h *Handler) Err() error {
    return h.err
}

func (h *Handler) Step1() *Handler {
    if h.err != nil {
        return h
    }

    // do something for step2
    return h
}

func (h *Handler) Step2() *Handler {
    if h.err != nil {
        return h
    }

    // do something fot step2
    return h
}

// ... StepN()

调用时直接通过链式调用即可，最后再判断错误

h := &Handler{}
if err := h.Step1().Step2().StepN().Err(); err != nil {
    // handle error
}

这种方式在一些数据库包中有大量使用，比如etcd、gorm。

可选配置

在创建对象时，如果可配置的属性很多，通常会引入一个配置文件

type Config struct {
    Port string
    Host string
    Timeout time.Time
    // ...
}

func NewServer(conf *Config) *Server {

}

通常这些配置都有默认值，config也不是必须的，通过建造者模式可以轻松解决此类问题

builder := &Builder{}
server := builder.WithPort("8080").WithHost("0.0.0.0").WithTimeOut(10*time.Second).Complete()

但建造者需要写一个建造类，配置对应的属性设置方法

type Builder struct {
    server Server
}

func (b *Builder) WithPort(port string) *Builder {
    b.server.port = port
    return b
}

func (b *Builder) WithHost(host string) *Builder {
    b.server.host = host
    return b
}

除了建造者模式，还可以通过可选配置，对调用者更友好，将配置项封装成Option，需要的时候注入对应的Option即可

type Option func(*Server)

type WithPort(port int) Option {
    return func(s *Server) {
        s.port = port
    }
}

type WithHost(host int) Option {
    return func(s *Server) {
        s.host = host
    }
}

type NewServer(opts ...Option) *Server {
    s := defaultServer() // 默认配置
    for _, opt := range opts {
        opt(s) // 添加可选配置
    }
    return s
}

调用时，只需在NewServer配置对应的Option即可

// 默认配置
s := NewServer()

// 可选配置
s := NewServer(WithPort("8080"), WithHost("127.0.0.1"))

可选配置相比直接使用配置和建造者模式，更加清晰，也非常容易扩展和维护，在kuberentes、etcd、go-redis库中都有非常多的应用。

并发控制

Golang基础库中已经提供不少并发控制工具，比如Channel、WaitGroup、各种锁等等。

ErrGroup、 WaitGroup可以等待多个Goroutine执行结束，但很多时候并发执行多个任务，如果其中一个任务出错那么整体失败，需要直接返回，这种情况下我们可以使用ErrGroup

ErrGroup借助封装了WaitGroup、Once以及Context，调用Wait时如果一个任务失败取消Context直接返回，核心逻辑如下

type ErrGroup struct {
 ctx    context.Context
 cancel func()

 wg sync.WaitGroup

 errOnce sync.Once
 err     error
}

func (g *ErrGroup) Wait() error {
 g.wg.Wait()

 if g.cancel != nil {
  g.cancel()
 }

 return g.err
}

func (g *ErrGroup) Go(f func(ctx context.Context) error) {
 g.wg.Add(1)

 go func() {
  defer g.wg.Done()
  if err := f(g.ctx); err != nil {
            // 执行失败则运行cancel
   g.errOnce.Do(func() {
    g.err = err
    if g.cancel != nil {
     g.cancel()
    }
   })
  }
 }()
}

控制并发数

借助有缓冲的Channel，可以实现控制Goroutine并发数，逻辑如下：

func NewCtrlGroup(number int) *CtrlGroup {
 return &CtrlGroup{
  ch: make(chan struct{}, number),
 }
}

type CtrlGroup struct {
 ch chan struct{}
 wg sync.WaitGroup
}

func (g *CtrlGroup) Enter() {
 g.ch <- struct{}{}
}

func (g *CtrlGroup) Leave() {
 <-g.ch
}

func (g *CtrlGroup) Go(f func()) {
 g.Enter() // 接收到新任务，发送到Channel，如果Channel满需要等待
 g.wg.Add(1)

 go func() {
  defer g.Leave() // 任务结束，取出一个元素
  defer g.wg.Done()
  f()
 }()
}

func (g *CtrlGroup) Wait() {
 g.wg.Wait()
}