Go语言设计模式之函数式选项模式

Posted dr-wei

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Go语言设计模式之函数式选项模式相关的知识,希望对你有一定的参考价值。

Go语言设计模式之函数式选项模式

本文主要介绍了Go语言中函数式选项模式及该设计模式在实际编程中的应用。

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

最近看go-micro/options.go源码的时候,发现了一段关于服务注册的代码如下:

type Options struct {
    Broker    broker.Broker
    Cmd       cmd.Cmd
    Client    client.Client
    Server    server.Server
    Registry  registry.Registry
    Transport transport.Transport

    // Before and After funcs
    BeforeStart []func() error
    BeforeStop  []func() error
    AfterStart  []func() error
    AfterStop   []func() error

    // Other options for implementations of the interface
    // can be stored in a context
    Context context.Context
}

func newOptions(opts ...Option) Options {
    opt := Options{
        Broker:    broker.DefaultBroker,
        Cmd:       cmd.DefaultCmd,
        Client:    client.DefaultClient,
        Server:    server.DefaultServer,
        Registry:  registry.DefaultRegistry,
        Transport: transport.DefaultTransport,
        Context:   context.Background(),
    }

    for _, o := range opts {
        o(&opt)
    }

    return opt
}

当时呢,也不是很明白newOptions这个构造函数为什么要这么写,但是后面在微信群里看到有人也再发类似的代码问为什么要这么写,后来在群里讨论的时候才知道了这是一种设计模式–函数式选项模式

可能大家看到现在也不是很明白我说的问题到底是什么,我把它简单提炼一下。

我们现在有一个结构体,定义如下:

type Option struct {
    A string
    B string
    C int
}

现在我们需要为其编写一个构造函数,我们可能会写成下面这种方式:

func newOption(a, b string, c int) *Option {
    return &Option{
        A: a,
        B: b,
        C: c,
    }
}

上面的代码很好理解,也是我们一直在写的。有什么问题吗?

我们现在来思考以下两个问题:

  1. 我们可能需要为Option的字段指定默认值
  2. Option的字段成员可能会发生变更

选项模式

我们先定义一个OptionFunc的函数类型

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 = &Option{
        A: "A",
        B: "B",
        C: 100,
    }
)

最后编写我们新版的构造函数如下:

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

测试一下:

func main() {
    x := newOption("nazha", "小王子", 10)
    fmt.Println(x)
    x = newOption2()
    fmt.Println(x)
    x = newOption2(
        WithA("沙河娜扎"),
        WithC(250),
    )
    fmt.Println(x)
}

输出:

&{nazha 小王子 10}
&{A B 100}
&{沙河娜扎 B 250}

这样一个使用函数式选项设计模式的构造函数就实现了。这样默认值也有了,以后再要为Option添加新的字段也不会影响之前的代码。

推荐阅读:

Go 函数式选项模式

以上是关于Go语言设计模式之函数式选项模式的主要内容,如果未能解决你的问题,请参考以下文章

Go - options模式(函数式选项模式)

Go编程模式 - 5.函数式选项

go 的选项模式

Go语言实践模式 - 函数选项模式(Functional Options Pattern)

学习Go语言之策略模式

go语言设计模式-创建模式之抽象工厂模式(Abstract Factory)