当前位置: 代码迷 >> 综合 >> 优雅实现golang默认参数
  详细解决方案

优雅实现golang默认参数

热度:63   发布时间:2024-02-12 11:30:09.0

package main

import "fmt"

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

// 定义修改默认参数的钩子函数
type ModifyOptFunc func(*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))
}