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

微信扫一扫,分享到朋友圈

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 += 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
爱好:爬山

微信扫一扫,分享到朋友圈

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

百度网盘空间“大”调整 原2T免费空间将调整为100G

上一篇

自动化运维工具Ansible

下一篇

你也可能喜欢

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

长按储存图像,分享给朋友