Golang中設置函數默認參數的優雅實現
阿新 • • 發佈:2019-01-26
我不 今天 client pass urn dial type import address
在Golang中,我們經常碰到要設置一個函數的默認值,或者說我定義了參數值,但是又不想傳遞值,這個在python或php一類的語言中很好實現,但Golang中好像這種方法又不行。今天在看Grpc源碼時,發現了一個方法可以很優雅的實現,叫做 Functional Options Patter.通過定義函數的方式來實現
比如我們以如下的構造函數為例說明下,用這個的好處
func NewClient(address string,timeout,trynums int){}
如果我們要實例化這個函數,timeout,trynums這個是必須要傳的,那如果我不想傳呢,一般可能是通過傳對象(struct,map)或定義多個func,感覺都不太方便。
func NewClient(address string){}
func NewClientNoTimeout(address string,trynums int){}
另一種傳一個對象
type Options struct{
timeout int,
trynums int
}
func NewClient(address string,opts Options){}
用對象的形式,還得檢查參數的合法性。比如傳遞了不存在的參數等。
那麽,我們看下用Functional Options Patter的方式,我寫了一個簡單的例子。
package main import "fmt" //如何向func傳遞默認值 type dialOption struct { Username string Password string Service string } type DialOption interface { apply(*dialOption) } type funcOption struct { f func(*dialOption) } func(fdo *funcOption) apply(do *dialOption){ fdo.f(do) } func newFuncOption(f func(*dialOption))*funcOption{ return &funcOption{ f:f, } } func withUserName(s string) DialOption{ return newFuncOption(func(o *dialOption){ o.Username = s }) } func withPasswordd(s string) DialOption{ return newFuncOption(func(o *dialOption){ o.Password = s }) } func withService(s string) DialOption{ return newFuncOption(func(o *dialOption){ o.Service = s }) } //默認參數 func defaultOptions() dialOption{ return dialOption{ Service:"test", } } type clientConn struct { timeout int dopts dialOption } func NewClient(address string, opts ...DialOption){ cc :=&clientConn{ timeout:30, dopts:defaultOptions(), } //循環調用opts for _,opt := range opts { opt.apply(&cc.dopts) } fmt.Printf("%+v",cc.dopts) } func main(){ NewClient("127.0.0.1",withPasswordd("654321"),withService("habox")) NewClient("127.0.0.1",withService("habox")) }
實例化時,通過func的方式來傳遞參數,也可以定義一些默認參數。如果以後要加,只需要更改很少的代碼。
而且,這種方式也不會傳遞不相關的參數,因為參數都在通過func的方式來修改的。
唯一不好的地方可能是代碼量相應的增加了。但是為了更優雅,這種做法還是值得的。
Golang中設置函數默認參數的優雅實現