Go - 学习 grpc.Dial(target string, opts …DialOption) 的写法

咱们平时是这样使用 grpc.Dial 方法的，比如：

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

咱们怎么能写出类似这样的调用方式，它是怎么实现的？

这篇文章咱们写一个 Demo，其实很简单，一步步往下看。

一

opts …DialOption，这个是不定参数传递，参数的类型为 DialOption，不定参数是指函数传入的参数个数为不定数量，可以不传，也可以为多个。

写一个不定参数传递的方法也很简单，看看下面这个方法 1 + 2 + 3 = 6。

func Add(a int, args ...int) (result int) {result += afor _, arg := range args {result += arg}return
}fmt.Println(Add(1, 2, 3))// 输出 6

其实平时我们用的 fmt.Println()、fmt.Sprintf() 都属于不定参数的传递。

二

WithInsecure()、WithBlock() 类似于这样的 With 方法，其实作用就是修改 dialOptions 结构体的配置，之所以这样写我个人认为是面向对象的思想，当配置项调整的时候调用方无需修改。

场景

咱们模拟一个场景，使用 不定参数 和 WithXXX 这样的写法，写个 Demo，比如我们要做一个从附近找朋友的功能，配置项有：性别、年龄、身高、体重、爱好，我们要找性别为女性，年龄为30岁，身高为160cm，体重为55kg，爱好为爬山的人，希望是这样的调用方式：

friends, err := friend.Find("附近的人",friend.WithSex(1),friend.WithAge(30),friend.WithHeight(160),friend.WithWeight(55),friend.WithHobby("爬山"))

代码实现

// option.gopackage friendimport ("sync"
)var (cache = &sync.Pool{New: func() interface{} {return &option{sex: 0}},}
)type Option func(*option)type option struct {sex    intage    intheight intweight inthobby  string
}func (o *option) reset() {o.sex = 0o.age = 0o.height = 0o.weight = 0o.hobby = ""
}func getOption() *option {return cache.Get().(*option)
}func releaseOption(opt *option) {opt.reset()cache.Put(opt)
}// WithSex setup sex, 1=female 2=male
func WithSex(sex int) Option {return func(opt *option) {opt.sex = sex}
}// WithAge setup age
func WithAge(age int) Option {return func(opt *option) {opt.age = age}
}// WithHeight set up height
func WithHeight(height int) Option {return func(opt *option) {opt.height = height}
}// WithWeight set up weight
func WithWeight(weight int) Option {return func(opt *option) {opt.weight = weight}
}// WithHobby set up Hobby
func WithHobby(hobby string) Option {return func(opt *option) {opt.hobby = hobby}
}

// friend.gopackage friendimport ("fmt"
)func Find(where string, options ...Option) (string, error) {friend := fmt.Sprintf("从 %s 找朋友\n", where)opt := getOption()defer func() {releaseOption(opt)}()for _, f := range options {f(opt)}if opt.sex == 1 {sex := "性别：女性"friend += fmt.Sprintf("%s\n", sex)}if opt.sex == 2 {sex := "性别：男性"friend += fmt.Sprintf("%s\n", sex)}if opt.age != 0 {age := fmt.Sprintf("年龄：%d岁", opt.age)friend += fmt.Sprintf("%s\n", age)}if opt.height != 0 {height := fmt.Sprintf("身高：%dcm", opt.height)friend += fmt.Sprintf("%s\n", height)}if opt.weight != 0 {weight := fmt.Sprintf("体重：%dkg", opt.weight)friend += fmt.Sprintf("%s\n", weight)}if opt.hobby != "" {hobby := fmt.Sprintf("爱好：%s", opt.hobby)friend += fmt.Sprintf("%s\n", hobby)}return friend, nil
}

// main.gopackage mainimport ("demo/friend""fmt"
)func main() {friends, err := friend.Find("附近的人",friend.WithSex(1),friend.WithAge(30),friend.WithHeight(160),friend.WithWeight(55),friend.WithHobby("爬山"))if err != nil {fmt.Println(err)}fmt.Println(friends)
}

输出

从 附近的人 找朋友
性别：女性
年龄：30岁
身高：160cm
体重：55kg
爱好：爬山