如果一个方法中需要传递多个参数且某些参数又是非必传，应该如何处理？

案例

  1. // NewFriend 寻找志同道合朋友
  2. func NewFriend(sex int, age int, hobby string) (string, error) {
  4.     // 逻辑处理 ...
  6.     return "", nil
  7. }

NewFriend()，方法中参数 sex 和 age 为非必传参数，这时方法如何怎么写？

传参使用不定参数！

想一想怎么去实现它？

看一下这样写可以吗？

  1. // Sex 性别
  2. type Sex int
  4. // Age 年龄
  5. type Age int
  7. // NewFriend 寻找志同道合的朋友
  8. func NewFriend(hobby string, args ...interface{}) (string, error) {
  9.     for _, arg := range args {
  10.         switch arg.(type) {
  11.         case Sex:
  12.             fmt.Println(arg, "is sex")
  13.         case Age:
  14.             fmt.Println(arg, "is age")
  15.         default:
  16.             fmt.Println("未知的类型")
  17.         }
  18.     }
  19.     return "", nil
  20. }

有没有更好的方案呢？

传递结构体… 恩，这也是一个办法。

咱们看看别人的开源代码怎么写的呢，我学习的是 grpc.Dial(target string, opts …DialOption) 方法，它都是通过 WithXX 方法进行传递的参数，例如：

  1. conn, err := grpc.Dial("127.0.0.1:8000",
  2.     grpc.WithChainStreamInterceptor(),
  3.     grpc.WithInsecure(),
  4.     grpc.WithBlock(),
  5.     grpc.WithDisableRetry(),
  6. )

比着葫芦画瓢，我实现的是这样的，大家可以看看：

  1. // Option custom setup config
  2. type Option func(*option)
  4. // option 参数配置项
  5. type option struct {
  6.     sex int
  7.     age int
  8. }
  10. // NewFriend 寻找志同道合的朋友
  11. func NewFriend(hobby string, opts ...Option) (string, error) {
  12.     opt := new(option)
  13.     for _, f := range opts {
  14.         f(opt)
  15.     }
  17.     fmt.Println(opt.sex, "is sex")
  18.     fmt.Println(opt.age, "is age")
  20.     return "", nil
  21. }
  23. // WithSex sex 1=female 2=male
  24. func WithSex(sex int) Option {
  25.     return func(opt *option) {
  26.         opt.sex = sex
  27.     }
  28. }
  30. // WithAge age
  31. func WithAge(age int) Option {
  32.     return func(opt *option) {
  33.         opt.age = age
  34.     }
  35. }

使用的时候这样调用：

  1. friends, err := friend.NewFriend(
  2.     "看书",
  3.     friend.WithAge(30),
  4.     friend.WithSex(1),
  5. )
  7. if err != nil {
  8.     fmt.Println(friends)
  9. }

这样写如果新增其他参数，是不是也很好配置呀。

以上。

对以上有疑问，快来我的星球交流吧 ~ https://t.zsxq.com/iIUVVnA