Go语言设计模式：使用Option模式简化类的初始化

Go Golang

发布日期: 2024-07-09

文章字数: 775

阅读时长: 3 分钟

阅读次数:

在面向对象编程中，当我们需要创建一个构造参数众多的类时，不仅使得代码难以阅读，而且在参数较多时，调用者需要记住每个参数的顺序和类型，这无疑增加了使用的复杂性，代码往往变得难以管理。

Go 语言虽然不支持传统意义上的类，但我们也可以使用结构体和函数来模拟面向对象的构造函数。

今天，我们将讨论一种优雅的解决方案——Option 模式。

传统的构造函数方法

先来看一个常见的例子，在 Go 语言中定义了一个 Foo 类，它有四个字段：name、id、age 和 db：

package newdemo

import "fmt"

type Foo struct {
   name string
   id int
   age int
   db interface{}
}

func NewFoo(name string, id int, age int, db interface{}) *Foo {
   return &Foo{
      name: name,
      id:   id,
      age:  age,
      db:   db,
   }
}

func main(){
    foo := NewFoo("jianfengye", 1, 0, nil) // 需要记住每个参数的顺序和类型
    fmt.Println(foo)
}

这种方法在参数较少时工作得很好，但随着参数数量的增加，其局限性也越来越明显。

引入 Option 模式

Option 模式通过使用函数选项来构建对象，为我们提供了一种更为灵活和可扩展的方式来配置类的实例。这种模式允许我们在不改变构造函数签名的情况下，灵活地添加更多的配置选项。

改造后的 Foo 类如下所示：

package newdemo

import "fmt"

type Foo struct {
 name string
 id int
 age int
 db interface{}
}

// FooOption 代表可选参数
type FooOption func(foo *Foo)

// WithName 为 name 字段提供一个设置器
func WithName(name string) FooOption {
   return func(foo *Foo) {
      foo.name = name
   }
}

// WithAge 为 age 字段提供一个设置器
func WithAge(age int) FooOption {
   return func(foo *Foo) {
      foo.age = age
   }
}

// WithDB 为 db 字段提供一个设置器
func WithDB(db interface{}) FooOption {
   return func(foo *Foo) {
      foo.db = db
   }
}

// NewFoo 创建 Foo 实例的构造函数，id为必传参数，其他为可选
func NewFoo(id int, options ...FooOption) *Foo {
   foo := &Foo{
      name: "default",
      id:   id,
      age:  10,
      db:   nil,
   }

   // 遍历每个选项并应用它们
   for _, option := range options {
      option(foo)
   }

   return foo
}

func main(){
    // 使用 Option 模式，仅传递需要设置的字段
    foo := NewFoo(1, WithAge(15), WithName("foo"))
    fmt.Println(foo)
}