go函数类型的小理解

派序猿

已于 2023-07-21 15:19:22 修改

阅读量348

点赞数 1

分类专栏： go 文章标签： golang

于 2023-07-21 15:10:52 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_44559085/article/details/131852474

版权

go 专栏收录该内容

1 篇文章

订阅专栏

1.type Option func(*Options) 怎么理解

type Option func(*Options) 是一个函数类型，它接受一个指向 Options 结构体的指针作为参数，且不返回任何结果。

在Go语言中，这种函数类型常用于实现一种配置模式，其中函数被用作选项来配置某个对象或函数的行为。函数类型 Option 定义了一种方式，通过传递不同的函数作为选项，可以对 Options 结构体进行特定的配置。

这种模式的一个常见的设计模式是可选参数模式，它允许用户在调用函数或初始化对象时，根据需要选择性地提供一些自定义选项，而不是强制使用预定义的固定参数。

以下是一个示例，演示了如何使用 Option 函数类型和 Options 结构体完成配置模式的操作：

package main

import "fmt"

// Options 结构体用于保存配置选项的值
type Options struct {
	Option1 string
	Option2 int
	Option3 bool
}

// WithOption1 是一个 Option 类型的函数，用于设置 Option1 的值
func WithOption1(val string) Option {
	return func(opts *Options) {
		opts.Option1 = val
	}
}

// WithOption2 是一个 Option 类型的函数，用于设置 Option2 的值
func WithOption2(val int) Option {
	return func(opts *Options) {
		opts.Option2 = val
	}
}

// WithOption3 是一个 Option 类型的函数，用于设置 Option3 的值
func WithOption3(val bool) Option {
	return func(opts *Options) {
		opts.Option3 = val
	}
}

// CreateObject 创建一个对象，根据提供的选项进行配置
func CreateObject(options ...Option) *Options {
	opts := &Options{}
	for _, opt := range options {
		opt(opts)
	}
	return opts
}

func main() {
	// 使用不同的选项创建对象
	obj1 := CreateObject(WithOption1("Value1"), WithOption2(2))
	obj2 := CreateObject(WithOption3(true))

	fmt.Println(obj1)
	fmt.Println(obj2)
}

在上面的示例中，我们定义了一个 Options 结构体，用于保存配置选项的值。然后，我们定义了几个 WithOptionX 函数，每个函数用于设置不同的选项的值。这些函数的返回值都是 Option 函数类型，它们接受一个指向 Options 结构体的指针作为参数，并在函数体内对其进行相应的赋值操作。

最后，我们定义了一个 CreateObject 函数，它接受多个 Option 类型的参数，并使用这些选项对 Options 结构体进行配置。通过调用不同的选项函数，可以创建具有不同配置的对象。

通过这种方式，我们可以根据需要选择性地配置对象的属性，而不是通过固定的参数列表。这样的配置模式更加灵活，可以方便地扩展和定制化对象的行为。

2.type Option func(*Options) 有什么用

type Option func(*Options) 的设计模式被广泛应用于Go语言中的库和框架，它具有以下几个优点和用途：

灵活的配置：通过使用 Option 函数来设置选项，可以根据具体需求灵活地配置对象或函数的行为。用户可以选择性地提供不同的选项，而不是强制使用固定的参数列表。
可扩展性：使用 Option 函数作为配置选项可以轻松地添加新的选项，而不会破坏现有的代码。当需要添加、修改或删除配置选项时，只需编写/更改相应的 Option 函数即可，而不需要修改大量的代码。
可读性和可维护性：使用配置选项模式可以使代码更加清晰和可读。每个选项的含义明确，并且可以通过选项函数的命名来传达其作用。这样，代码的可理解性和可维护性都会得到提高。
提供默认值：选项函数可以设置选项的默认值，以确保对象在没有显式配置选项的情况下也可以正常工作。如果用户没有提供某个选项，将使用默认值，从而提供更好的用户体验。
可测试性：对于单元测试和集成测试，可以为对象提供不同的选项，以验证不同的行为。使用配置选项模式可以方便地创建和管理不同场景下的测试数据。

总之，使用 type Option func(*Options) 这种配置选项模式能够提供更灵活、可扩展和可维护的代码，同时增强代码的可读性和测试性，是一个常见的设计模式。

3.例子

下面是一个简单的例子，演示了如何使用 Option 函数来配置一个文件读取器对象：

package main

import (
	"fmt"
	"os"
)

// FileReader 是文件读取器对象
type FileReader struct {
	filePath  string
	bufferSize int
}

// Option 是配置选项函数类型
type Option func(*FileReader)

// WithFilePath 是一个 Option 类型的函数，用于设置文件路径
func WithFilePath(path string) Option {
	return func(fr *FileReader) {
		fr.filePath = path
	}
}

// WithBufferSize 是一个 Option 类型的函数，用于设置缓冲区大小
func WithBufferSize(size int) Option {
	return func(fr *FileReader) {
		fr.bufferSize = size
	}
}

// NewFileReader 创建一个文件读取器对象，并根据提供的选项进行配置
func NewFileReader(options ...Option) *FileReader {
	fr := &FileReader{}
	for _, opt := range options {
		opt(fr)
	}
	return fr
}

// Read 读取文件内容
func (fr *FileReader) Read() (string, error) {
	file, err := os.Open(fr.filePath)
	if err != nil {
		return "", err
	}
	defer file.Close()

	buffer := make([]byte, fr.bufferSize)
	_, err = file.Read(buffer)
	if err != nil {
		return "", err
	}

	return string(buffer), nil
}

func main() {
	// 创建文件读取器对象，并配置选项
	fr := NewFileReader(WithFilePath("data.txt"), WithBufferSize(1024))

	// 读取文件内容
	content, err := fr.Read()
	if err != nil {
		fmt.Println("File read error:", err)
		return
	}

	fmt.Println("File contents:", content)
}

在上面的示例中，我们定义了一个 FileReader 结构体，表示一个文件读取器对象。我们使用两个 Option 类型的函数来设置文件路径 (WithFilePath) 和缓冲区大小 (WithBufferSize)。

接下来，我们定义了一个 NewFileReader 函数，它根据提供的选项创建一个文件读取器对象，并将选项应用于该对象。在 NewFileReader 函数的循环中，我们迭代提供的选项函数切片，并在每次迭代中调用选项函数，将选项应用于文件读取器对象。

最后，我们定义了一个 Read 方法，用于读取指定路径的文件内容。在 main 函数中，我们创建一个文件读取器对象并配置选项。然后，我们调用 Read 方法读取文件的内容，并打印输出。

通过传递不同的选项，我们可以轻松地配置不同的文件路径和缓冲区大小。这种可选参数模式使得对象的配置和使用更加灵活和可扩展。

opt(fr) 这个作用是什么？

在示例代码中，opt(fr) 的作用是将选项函数 opt 应用于文件读取器对象 fr。

在 NewFileReader 函数中，我们使用一个循环遍历提供的选项函数切片 options。对于每个选项函数 opt，我们通过调用 opt(fr) 将其应用于文件读取器对象 fr。

选项函数的定义如下：

type Option func(*FileReader)

选项函数的参数是一个指向文件读取器对象 FileReader 的指针，因此 fr 是指向文件读取器对象的指针。

通过调用 opt(fr)，我们实际上是在每次迭代期间调用选项函数，将当前的文件读取器对象 fr 作为参数传递给选项函数。这样，选项函数可以根据具体的实现和需求，从而修改文件读取器对象的内部状态或属性，以实现特定的配置效果。

通过将选项函数应用于文件读取器对象，我们可以根据需要定制文件读取器的行为，而不需要在 FileReader 结构体中预定义所有可能的配置属性。这种动态配置的方式使得我们能够编写更灵活、可扩展和可配置的代码。

假设我 opts 修改了一个Signal 为false，那么通过这个函数后 Signal 是true 还是 false？

func newOptions(opts ...Option) Options {
	opt := Options{
		Auth:      auth.DefaultAuth,
		Broker:    broker.DefaultBroker,
		Cache:     cache.DefaultCache,
		Cmd:       cmd.DefaultCmd,
		Config:    config.DefaultConfig,
		Client:    client.DefaultClient,
		Server:    server.DefaultServer,
		Store:     store.DefaultStore,
		Registry:  registry.DefaultRegistry,
		Runtime:   runtime.DefaultRuntime,
		Transport: transport.DefaultTransport,
		Context:   context.Background(),
		Signal:    true,
		Logger:    logger.DefaultLogger,
	}
    for _, o := range opts {
        o(&opt)
    }

	return opt
}