只有Host和Port是必選項(xiàng),Timeout和RetryNum會給默認(rèn)值,實(shí)例化這個(gè)結(jié)構(gòu)體的時(shí)候這兩個(gè)參數(shù)是可選的。

場景帶入
日常開發(fā)中,封裝相關(guān)的庫時(shí),例如封裝微信支付sdk、MySQL ORM、自定義的Server端等,都會給出很多相關(guān)的配置,例如:
type Client struct {
host string
port string
timeout time.Duration
retryNum int
}
其中只有host和port是必選項(xiàng),timeout和retryNum會給默認(rèn)值,實(shí)例化這個(gè)結(jié)構(gòu)體的時(shí)候這兩個(gè)參數(shù)是可選的。
封裝的庫一般需要做到以下幾點(diǎn):
- 需要一個(gè)實(shí)例化結(jié)構(gòu)體的方法,需要必選參數(shù),可選項(xiàng)需要處理為可選參數(shù)。
- 可選配置可以按需靈活傳入,任意順序傳遞參數(shù)。
- 新增一個(gè)可選配置項(xiàng)不需要大面積改代碼,容易擴(kuò)展和維護(hù)。
- 結(jié)構(gòu)體里面的屬性需要設(shè)置為不可導(dǎo)出,即在包外部不能修改。
使用函數(shù)式編程方式的實(shí)現(xiàn)方法
先通過一個(gè)代碼片段來看一下使用函數(shù)式編程方式是怎么實(shí)現(xiàn)的:
package main
import (
"fmt"
"time"
)
type Client struct {
host string
port string
timeout time.Duration
retryNum int
}
type Option func(*Client)
func NewClient(host, port string, opt ...Option) *Client {
c := Client{
host: host,
port: port,
timeout: time.Second,
retryNum: 1,
}
for _, option := range opt {
option(&c)
}
return &c
}
func Timeout(timeout time.Duration) Option {
return func(c *Client) {
c.timeout = timeout
}
}
func RetryNum(num int) Option {
return func(c *Client) {
c.retryNum = num
}
}
使用方法是這樣的,只傳入必傳參數(shù)。
func main() {
client := NewClient("localhost", "8080")
}
傳入可選參數(shù)timeout:
func main() {
client := NewClient("localhost", "8080", Timeout(2*time.Second))
}
傳入可選參數(shù)retryNum。
func main() {
client := NewClient("localhost", "8080", RetryNum(3))
}
傳入timeout和retryNum。
func main() {
client = NewClient("localhost", "8080", Timeout(2*time.Second), RetryNum(3))
// 或者
// client = NewClient("localhost", "8080", RetryNum(3), Timeout(2*time.Second))
}
以上面的Timeout函數(shù)為例,這個(gè)函數(shù)返回的一個(gè)函數(shù)來設(shè)置傳入的配置項(xiàng),Timeout就是一個(gè)高階函數(shù)。
小結(jié)
上面的實(shí)現(xiàn)方式有個(gè)專業(yè)術(shù)語叫做 Functional Options,使用這種編程模式有很多好處:
- 易讀性強(qiáng),容易理解。
- 方便擴(kuò)展,容易維護(hù)。
- 參數(shù)傳遞不需要先后順序。