1. 程式人生 > >Golang中設置函數默認參數的優雅實現

Golang中設置函數默認參數的優雅實現

我不 今天 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中設置函數默認參數的優雅實現