Go中的Option设计模式

关于什么是设计模式，笔者在这里就不做过多的解释。大家自行百度谷歌就好。下面我们聊聊Option设计模式，这个单词翻译过来就是选项的意思。

它一般都是用在初始化数据的时候使用。

type User struct {
    // 必须初始化的值
    Username string
    Age int
    
    // 非必要初始化的值
    Password string
}

结构体中，有些字段是必须要初始化的，而有些则是不用。那些不用初始化的字段就被称为可选项。用户根据自己的实际需求对其进行初始化。

一般来说，对于这种结构体，不论是必须初始化的字段还是非必须的，都是私有的字段，不会暴露给外界。

下面是一个标准的Option的设计模式

// 定义一个Option函数签名
type UserOption func(user *User)

// 提供一个可选函数
func WithUserPassword(password string) UserOption {
    return func(user *User) {
        user.Password = password
    }
}

type User struct {
    // 必须初始化的值
    Username string
    Age int
    
    // 非必要初始化的值
    Password string
}

// 定义一个初始化User的方法
func NewUser(username string, age int, opts...UserOption) *User {
    user := &User{
        Username: username,
        Age: age,
    }
    for _, opt := range opts {
        opt(user)
    }
    return user
}

步骤

1. 定义一个结构体，内部字段全部都是私有的，并且存在必须初始化字段和非必要初始化字段
2. 定义一个Option函数签名，参数是结构体指针，必须是结构体指针，因为只有指针才能设置上值
3. 定义一个初始化结构体的方法，参数是结构体必须要初始化的字段值，外加一个Option函数切片，返回值可以是结构体，也可以是指针结构体
4. 初始化结构体方法内部先初始化好一个最基本的结构体，然后遍历Option函数切片
5. 定义的Option函数实现最好用With开头，规范，不遵守也可以

Option模式变种

// 定义一个Option函数签名
type UserOptionErr func(user *User) error

// 提供一个可选函数
func WithUserPassword(password string) UserOption {
    return func(user *User) error {
        if password == "" {
            return errors.New("password 不能为空")
        }
        user.Password = password
        return nil
    }
}

type User struct {
    // 必须初始化的值
    Username string
    Age int
    
    // 非必要初始化的值
    Password string
}

// 定义一个初始化User的方法
func NewUser(username string, age int, opts...UserOption) (*User, error) {
    user := &User{
        Username: username,
        Age: age,
    }
    for _, opt := range opts {
        if err := opt(user); err != nil {
            return nil, err
        }
    }
    return user, nil
}

大家应该发现了，就是修改Option函数签名的返回值，这种模式可以用作对Option函数的参数进行校验。由于这里发生了变更，导致初始化结构体的方法也需要做出相应的变化。

对于Option模式，在很多场景中都能用到，并且也是非常好用的。大家记住它是在初始化数据的时候用就行，代码模式相对固定。