knowledge-base

我的知识库 / Go / Golang 函数可选参数模式

Golang 函数可选参数模式

函数可选参数模式

type Server struct {
    Addr    string
    Timeout time.Duration
}

type Option func(*Server)

func newServer(addr string, options ...Option) (*Server, error) {
    s := &Server{
        Addr: addr,
    }
    for _, opt := range options {
        opt(s)
    }
    // ...
    return s, nil
}

func WithTimeout(timeout time.Duration) Option {
    return func(s *Server) {
        s.Timeout = timeout
    }
}

通用函数可选参数模式

type BasicService struct {
    redisClient string
}

type ServiceOption func(*BasicService)

func WithRedisClient(redisClient string) ServiceOption {
    return func(s *BasicService) {
        s.redisClient = redisClient
    }
}

type UserService struct {
    *BasicService
}

type OrderService struct {
    *BasicService
}

func newUserService(opts ...ServiceOption) *UserService {
    us := &UserService{BasicService: new(BasicService)}
    for _, opt := range opts {
        opt(us.BasicService)
    }
    return us
}

func newOrderService(opts ...ServiceOption) *OrderService {
    os := &OrderService{BasicService: new(BasicService)}
    for _, opt := range opts {
        opt(os.BasicService)
    }
    return os
}

func TestNewService(t *testing.T) {
    us := newUserService(WithRedisClient("redis"))
    t.Log(us.redisClient)

    os := newOrderService(WithRedisClient("redis"))
    t.Log(os.redisClient)
}

---

« Go 开发环境配置

» Golang 密钥对、数字签名和证书管理