咱们平时是这样使用 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 += a
     for _, 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.go
 ​
 package friend
 ​
 import (
     "sync"
 )
 ​
 var (
     cache = &sync.Pool{
         New: func() interface{} {
             return &option{sex: 0}
         },
     }
 )
 ​
 type Option func(*option)
 ​
 type option struct {
     sex    int
     age    int
     height int
     weight int
     hobby  string
 }
 ​
 func (o *option) reset() {
     o.sex = 0
     o.age = 0
     o.height = 0
     o.weight = 0
     o.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.go
 ​
 package friend
 ​
 import (
     "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.go
 ​
 package main
 ​
 import (
     "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
 爱好：爬山