【Golang 实现函数选择模式的变参传参】

方法一

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("76：%+v",cc.dopts)
}


func main(){
	//NewClient("127.0.0.1",withPasswordd("654321"),withService("habox"))
	NewClient("127.0.0.1",withService("habox"))
	//NewClient("127.0.0.1")
}

方法二

package main

import "fmt"

// 可以由用户自定义的参数集合
type Option struct {
	retryNum int
}

// 定义修改默认参数的钩子函数
type ModifyOptFunc func(opt *Option)

// 实际修改默认参数的函数
func WithRetryNum(num int) ModifyOptFunc {
	return func(opt *Option) {
		opt.retryNum = num
	}
}

func OperateRedis(modOptions ...ModifyOptFunc) {
	// 定义参数并添加默认值
	opt := Option{
		retryNum : 10,
	}

	// 调用钩子函数，并对默认值进行修改
	for _,fun := range modOptions {
		fun(&opt)
	}

	// 使用参数做相关实际逻辑
	fmt.Println("current retry number:", opt.retryNum)
}

func main() {
	OperateRedis()
	OperateRedis(WithRetryNum(1))
	OperateRedis(WithRetryNum(200))
}