优雅实现golang默认参数

​ 原生的golang中,函数定义不支持默认参数。但是在实际开发过程中,经常会有些参数用户可以不关心或者可以根据实际情况去定制实现,这个时候需要使用到默认参数,在C++中,函数的定义和实现本来就支持默认参数,如果需要在golang中实现默认参数,可以参考一下做法:

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))
}
发布了130 篇原创文章 · 获赞 40 · 访问量 8万+

猜你喜欢

转载自blog.csdn.net/u014630623/article/details/103935018