设计模式 函数式选项模式

需要函数式选项模式

为什么需要函数式选项模式

范例源码

https://github.com/micro/go-micro/blob/master/options.go

函数选项模式

我们先定义一个 结构体Option 和对应的 OptionFunc 的函数类型

type Option struct {
    A string
    B string
    C int
}
type OptionFunc func(*Option)

然后利用闭包为每个字段编写一个设置值的 With 函数

func WithA(a string) OptionFunc {
    return func(o *Option) {
        o.A = a
    }
}

func WithB(b string) OptionFunc {
    return func(o *Option) {
        o.B = b
    }
}

func WithC(c int) OptionFunc {
    return func(o *Option) {
        o.C = c
    }
}

然后，我们定义一个 默认的Option 函数 如下

var (
    defaultOption = setDefaultOption()
)

func setDefaultOption() *Option {
    return &Option{
        A: "A",
        B: "B",
        C: 100,
    }
}

最后编写,我们new 构造函数如下

func NewOption(opts ...OptionFunc) (opt *Option) {
    opt = defaultOption
    for _, o := range opts {
        o(opt)
    }
    defaultOption = setDefaultOption()
    return
}

这里需要注意，defaultOption = setDefaultOption() 一定别忘记。不然会相互影响

测试选项模式

func TestNewOption(t *testing.T) {
    absOptions := newOptionABS("A", "B", 100)
    optionFast := NewOption()
    changeOption := NewOption(
        WithA("B"),
        WithB("A"),
    )
    // do
    t.Logf("absOptions %v", absOptions)
    t.Logf("optionFast %v", optionFast)
    t.Logf("changeOption %v", changeOption)
}

详细代码见