函數的不定參數你會用嗎？

如果一個方法中需要傳遞多個參數且某些參數又是非必傳，應該如何處理？

案例

// NewFriend 尋找志同道合朋友
func NewFriend(sex int, age int, hobby string) (string, error) {
	
	// 邏輯處理 ...

	return "", nil
}
           

NewFriend()

，方法中參數

sex

和

age

為非必傳參數，這時方法如何怎麼寫？

傳參使用不定參數！

想一想怎麼去實作它？

看一下這樣寫可以嗎？

// Sex 性别
type Sex int

// Age 年齡
type Age int

// NewFriend 尋找志同道合的朋友
func NewFriend(hobby string, args ...interface{}) (string, error) {
	for _, arg := range args {
		switch arg.(type) {
		case Sex:
			fmt.Println(arg, "is sex")
		case Age:
			fmt.Println(arg, "is age")
		default:
			fmt.Println("未知的類型")
		}
	}
	return "", nil
}
           

有沒有更好的方案呢？

傳遞結構體... 恩，這也是一個辦法。

咱們看看别人的開源代碼怎麼寫的呢，我學習的是

grpc.Dial(target string, opts …DialOption)

方法，它都是通過

WithXX

方法進行傳遞的參數，例如：

conn, err := grpc.Dial("127.0.0.1:8000",
	grpc.WithChainStreamInterceptor(),
	grpc.WithInsecure(),
	grpc.WithBlock(),
	grpc.WithDisableRetry(),
)
           

比着葫蘆畫瓢，我實作的是這樣的，大家可以看看：

// Option custom setup config
type Option func(*option)

// option 參數配置項
type option struct {
	sex int
	age int
}

// NewFriend 尋找志同道合的朋友
func NewFriend(hobby string, opts ...Option) (string, error) {
	opt := new(option)
	for _, f := range opts {
		f(opt)
	}

	fmt.Println(opt.sex, "is sex")
	fmt.Println(opt.age, "is age")

	return "", nil
}

// WithSex sex 1=female 2=male
func WithSex(sex int) Option {
	return func(opt *option) {
		opt.sex = sex
	}
}

// WithAge age
func WithAge(age int) Option {
	return func(opt *option) {
		opt.age = age
	}
}
           

使用的時候這樣調用：

friends, err := friend.NewFriend(
	"看書",
	friend.WithAge(30),
	friend.WithSex(1),
)

if err != nil {
	fmt.Println(friends)
}
           

這樣寫如果新增其他參數，是不是也很好配置呀。

以上。

對以上有疑問，快來我的星球交流吧 ~ https://t.zsxq.com/iIUVVnA

作者：新亮筆記（關注公衆号，可申請添加微信好友）

出處：https://www.cnblogs.com/xinliangcoder

本文版權歸作者和部落格園共有，歡迎轉載，但未經作者同意必須保留此段聲明，且在文章頁面明顯位置給出原文連接配接，否則保留追究法律責任的權利。