
場景帶入
日常開發(fā)中,封裝相關的庫時,例如封裝微信支付sdk、MySQL ORM、自定義的Server端等,都會給出很多相關的配置,例如:
type Client struct {
host string
port string
timeout time.Duration
retryNum int
}
其中只有host和port是必選項,timeout和retryNum會給默認值,實例化這個結構體的時候這兩個參數是可選的。
封裝的庫一般需要做到以下幾點:
- 需要一個實例化結構體的方法,需要必選參數,可選項需要處理為可選參數。
- 可選配置可以按需靈活傳入,任意順序傳遞參數。
- 新增一個可選配置項不需要大面積改代碼,容易擴展和維護。
- 結構體里面的屬性需要設置為不可導出,即在包外部不能修改。
使用函數式編程方式的實現方法
先通過一個代碼片段來看一下使用函數式編程方式是怎么實現的:
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
}
}
使用方法是這樣的,只傳入必傳參數。
func main() {
client := NewClient("localhost", "8080")
}
傳入可選參數timeout:
func main() {
client := NewClient("localhost", "8080", Timeout(2*time.Second))
}
傳入可選參數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函數為例,這個函數返回的一個函數來設置傳入的配置項,Timeout就是一個高階函數。
小結
上面的實現方式有個專業(yè)術語叫做 Functional Options,使用這種編程模式有很多好處:
- 易讀性強,容易理解。
- 方便擴展,容易維護。
- 參數傳遞不需要先后順序。