内容简介:作为golang开发人员,您将遇到的许多问题之一是尝试将函数的参数设置为可选。有时候使用默认设置,但有时候需要提供自定义设置。在许多语言中,这很容易;在c系列语言中,您可以使用不同数量的参数提供相同函数的多个版本,在php这样的语言中,您可以为参数提供默认值,并在调用方法时忽略它们。但在golang您不能做到这两点。那么在go中该如何实现呢?我们来看一个例子吧,假设我们有一个名为StuffClient的服务,它可以执行一些操作并具有两个配置选项(超时和重试):
作为golang开发人员,您将遇到的许多问题之一是尝试将函数的参数设置为可选。有时候使用默认设置,但有时候需要提供自定义设置。
在许多语言中,这很容易;在c系列语言中,您可以使用不同数量的参数提供相同函数的多个版本,在 php 这样的语言中,您可以为参数提供默认值,并在调用方法时忽略它们。但在golang您不能做到这两点。那么在 go 中该如何实现呢?
我们来看一个例子吧,假设我们有一个名为StuffClient的服务,它可以执行一些操作并具有两个配置选项(超时和重试):
type StuffClient interface { DoStuff() error } type stuffClient struct { conn Connection timeout int retries int }
该结构是私有的,所以我们应该为它提供某种构造函数
func NewStuffClient(conn Connection, timeout, retries int) StuffClient { return &stuffClient { conn: conn, timeout: timeout, retries: retries, } }
但现在我们总是要在每次调用NewStuffClient时提供超时和重试。大多数时候我们只想使用默认值。我们无法使用不同数量的参数定义多个版本的NewStuffClient, 否则我们将得到一个编译错误。
一种选择时创建另一个具有不同名称的构造函数,例如
func NewStuffClient (conn Connection) StuffClient { return &stuffClient { conn: conn, timeout: DefaultTimeout, retries: DefaultRetries, } } func NewStuffClienWithOptions(conn Connection, timeout, retries int) StuffClient { return &stuffClient { conn: conn, timeout: timeout, retries: retries, } }
我们还可以做的更好,将所有选项放到配置对象中
type StuffClientOptions struct { Retries int Timeout int } func NewStuffClient(conn Connection, options StuffClientOptions) StuffClient { return &stuffClient { conn: conn, timeout: options.Timeout, retries: options.Retries, } }
但那也不是很好,现在我们必须这个结构并传入它,即使我们不想指定任何选项,我们也没有自动填写的默认值,除非我们在代码中添加了一堆检查或者我们可以传入一个DefaultSuffClientOptions变量(但这可能会导致在一个地方被修改,影响别的地方)
那么解决方案是什么?解决这个难题的最好方法就是使用函数选项模式,利用go闭包的方便支持,让我们保留上面定义的 StuffClientOptions,但我们会添加一些东西:
type StuffClientOption func(*StuffClientOptions) type StuffClientOptions struct { Retries int Timeout int } func WithRetries(r int) StuffClientOption { return func(o *StuffClientOptions) { o.retries = r } } func WithTimeout(t int) StuffClientOption { return func(o *StuffClientOptions) { o.timeout = t } } var defaultStuffClientOptions = StuffClientOptions { Retries: 3, Timeout: 2, } func NewStuffClient(conn Connection, opts ...StuffClientOption) StuffClient { options := defaultStuffClientOptions for _, o := range opts { o(&options) } return &stuffClient{ conn: conn, timeout: options.Timeout, retries: options.Retries, } }
现在看起来已经非常好用了。关于它的好处是我们可以随时添加新选项,只需要对代码进行少量的更改。
var defaultStuffClientOptions = StuffClientOptions{ Retries: 3, Timeout: 2, } type StuffClientOption func(*StuffClientOptions) type StuffClientOptions struct { Retries int //number of times to retry the request before giving up Timeout int //connection timeout in seconds } func WithRetries(r int) StuffClientOption { return func(o *StuffClientOptions) { o.Retries = r } } func WithTimeout(t int) StuffClientOption { return func(o *StuffClientOptions) { o.Timeout = t } } type StuffClient interface { DoStuff() error } type stuffClient struct { conn Connection timeout int retries int } type Connection struct {} func NewStuffClient(conn Connection, opts ...StuffClientOption) StuffClient { options := defaultStuffClientOptions for _, o := range opts { o(&options) } return &stuffClient{ conn: conn, timeout: options.Timeout, retries: options.Retries, } } func (c stuffClient) DoStuff() error { return nil }
我们也可以通过删除 StuffClientOptions 结构并将选项直接应用于我们的StuffClient, 可以进一步简化这一过程
var defaultStuffClient = stuffClient{ retries: 3, timeout: 2, } type StuffClientOption func(*stuffClient) func WithRetries(r int) StuffClientOption { return func(o *stuffClient) { o.retries = r } } func WithTimeout(t int) StuffClientOption { return func(o *stuffClient) { o.timeout = t } } type StuffClient interface { DoStuff() error } type stuffClient struct { conn Connection timeout int retries int } type Connection struct{} func NewStuffClient(conn Connection, opts ...StuffClientOption) StuffClient { client := defaultStuffClient for _, o := range opts { o(&client) } client.conn = conn return client } func (c stuffClient) DoStuff() error { return nil }
在我们的示例中,只是将配置直接应用于结构,在中间有一个额外的配置结构是没有意义的,但请注意,在许多情况下,您可能仍希望使用上一个示例中的config结构,例如:如果你的构造函数使用配置选项来执行某些操作但并没有将它们存储到结构中,或者他们被传递到其他地方。config结构变量是更通用的实现。
使用步骤
-
定义选项config结构体
type options struct{ timeout time.Duration }
-
定义默认config结构体变量
var defaultOptions = options{}
-
定义配置选项函数
type option func(*options) func WithTimeout(t time.Duration) options { return func(o *options) { o.timeout = t } }
-
应用函数选项配置
func Do(opts ...option) { d := defaultOptions for _, o := range opts { o(&d) } }
原文链接:
https://halls-of-valhalla.org/beta/articles/functional-options-pattern-in-go,54/
相关参考连接
Dave Cheney: https://dave.cheney.net/2014/10/17/functional-options-for-friendly-apis
Rob Pike: https://commandcenter.blogspot.com/2014/01/self-referential-functions-and-design.html
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持 码农网
猜你喜欢:- Vue render函数实战--实现tabs选项卡组件
- go-micro 框架源码剖析 之 函数选项模式
- 横向的选项卡效果
- .NET Core微服务开发选项
- UNP 学习笔记——套接字选项
- 实用的可选项(Optional)扩展
本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们。
热搜:搜索排名营销大揭秘
【美】肖恩·布拉德利 / 中国人民大学出版社有限公司 / 2018-7-30 / CNY 55.00
首部大数据在我国政府管理场景中的应用实践案例读本,全面展示我国电子政务与数字化建设的成果,深度理解实施国家大数据战略的重要意义。 本书作者作为国内最早从事大数据应用研究的实践者之一,亲历了中国大数据的发展历程、主要事件、应用案例以及行业变化。 在本书中,作者将其所亲历的大数据发展历程进行了阐述,从大数据的基本概念、特点到实践解读,通俗易懂,给我们的实际工作提供了重要参考。作者将帮助读者......一起来看看 《热搜:搜索排名营销大揭秘》 这本书的介绍吧!