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

在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的方式來修改的。
唯一不好的地方可能是代碼量相應的增加了。但是為了更優雅，這種做法還是值得的。