解决go-micro与其它gRPC框架之间的通信问题

go-micro+gRPC

在之前的文章中分别介绍了使用gRPC官方插件和go-micro插件开发gRPC应用程序的方式,都能正常走通。不过当两者混合使用的时候,互相访问就成了问题。比如使用go-micro插件生成的gRPC客户端访问基于gRPC官方插件创建的服务端时就会出现如下错误:

{"id":"go.micro.client","code":501,"status":"Not Implemented"}

经过一番探索,发现是因为go-micro的插件生成代码时丢弃了proto定义中的package,客户端API和服务端API都没有使用这个package,所以它自己也能逻辑自洽,但是和其它框架或者语言的gRPC服务通信时就出现问题了。

这里以 hello.proto 为例:

syntax = "proto3";

option go_package="/proto";

package Business;

service Hello {
  rpc Say (SayRequest) returns (SayResponse);
}

message SayResponse {
  string Message = 1;
}

message SayRequest {
  string Name = 1;
}

对于客户端代理,protoc-gen-go-grpc生成的是:

err := c.cc.Invoke(ctx, "/Business.Hello/Say", in, out, opts...)

protoc-gen-micro生成的是:

req := c.c.NewRequest(c.name, "Hello.Say", in)

可以明显看到,go-micro生成的gRPC method中缺少package。当然这个method的风格也有些差异,不过这个不是问题,因为go-micro还会它进行一些格式化处理,格式化代码在grpc插件中。

plugins/client/grpc/request.go :

func methodToGRPC(service, method string) string {
    // no method or already grpc method
    if len(method) == 0 || method[0] == '/' {
        return method
    }

    // assume method is Foo.Bar
    mParts := strings.Split(method, ".")
    if len(mParts) != 2 {
        return method
    }

    if len(service) == 0 {
        return fmt.Sprintf("/%s/%s", mParts[0], mParts[1])
    }

    // return /pkg.Foo/Bar
    return fmt.Sprintf("/%s.%s/%s", service, mParts[0], mParts[1])
}

可以看到go-micro直接把服务名称作为了package名称,这两者不能等同,不相同时就会出现问题。

网上也没有人提过这个问题,可能混合使用的人不多吧。于是我研究了一下 go-micro 的源码,因为是生成的代码中缺少信息,所以要解决这个问题还是得从protoc-gen-micro入手。

注意这里使用的是go-micro v4版本,其它版本未跟进。

客户端改造

针对客户端问题,我做了如下一些修改:

在生成客户端method时加上package,并直接生成gRPC风格method(go-micro内部其实支持这种风格),修改文件:cmd/protoc-gen-micro/plugin/micro/micro.go

func (g *micro) generateClientMethod(pkg, reqServ, servName, serviceDescVar string, method *pb.MethodDescriptorProto, descExpr string) {
    reqMethod := fmt.Sprintf("%s.%s", servName, method.GetName())
    useGrpc := g.gen.Param["use_grpc"]
    if useGrpc != "" {
        reqMethod = fmt.Sprintf("/%s.%s/%s", pkg, servName, method.GetName())
    }
...

因为还要向前兼容,不能影响现有用户,所以给这个逻辑加了一个开关,使用参数 use_grpc 才会应用新的生成方式。generateClientMethod 方法的 pkg 参数原来并没有,是新加的,从上下文中也比较容易获取到。具体改动可以看这里:https://github.com/asim/go-micro/pull/2474/commits/0d435a690ea21a3f64b0534d1fa244f512601493

现在如果明确只使用gRPC进行通信,或者需要和其它框架或者语言的gRPC应用程序通信,生成代码时可以这样做:

protoc --go_out=. --micro_out=. --micro_opt=use_grpc=1 xxx.proto

关键就是 --micro_opt=use_grpc=1use_grpc这个参数会传递给 protoc-gen-micro,然后就可以在上边修改过的代码中获取到,不管这个参数的值是什么,只要使用了它,就会生成gRPC风格的带package的method。现在生成的代码是这样的:

req := c.c.NewRequest(c.name, "/Business.Hello/Say", in)

用这个客户端代理访问其它框架或者语言开发的gRPC服务就没有问题了,当然访问go-micro的gRPC服务也没有问题。

怎么获取到这个最新版的 protoc-gen-micro 呢?这个修改提了PR之后,目前已经合并到官方的Github仓库中,但是还没有打tag,可以这样安装:

go install go-micro.dev/v4/cmd/protoc-gen-micro@1919048c8f20

这可能不是一个好的修改,因为还需要知道有 use_grpc 这么个参数。肯定还有别的修改方案,但是因为对go-micro了解的不多,所有只选择了这个不会影响现有通信方式的方案。

服务端改造

服务端没有问题,别的框架或者开发语言的gRPC客户端可以调用基于go-micro的gRPC服务。

一开始我测试的时候也遇到了问题,先入为主的以为protoc-gen-micro生成的服务端也有package的问题,因此还提交了个PR,然后被啪啪打脸。然后我又读了读源码,发现go-micro服务端特别巧妙的把客户端请求中的package信息擦除了,所以客户端是否传递package都没有影响,反正服务端不需要。

服务端的注册逻辑在 plugins/server/grpc/server.go 中的 register 方法:

s := new(service)
s.typ = reflect.TypeOf(rcvr)
s.rcvr = reflect.ValueOf(rcvr)
sname := reflect.Indirect(s.rcvr).Type().Name()
...
server.serviceMap[s.name] = s

可以看到这里直接用反射获取的类型名称作为服务名称,没有package什么事。

然后接收到客户端的gRPC请求时,go-micro又把请求中的package擦除了。这段逻辑在 plugins/server/grpc/grpc.go 中的 handler 方法中:

serviceName, methodName, err := mgrpc.ServiceMethod(fullMethod)
service := g.rpc.serviceMap[serviceName]

通过 mgrpc.ServiceMethod 获取服务名称时去掉了package名称,所以客户端带不带package都没有问题。

运行效果

现在把程序跑起来,试试用 protoc-gen-micro 生成的客户端访问 基于 protoc-gen-go-grpc 的服务端。

go-micro+gRPC

以上就是本文的主要内容,示例代码已经上传到Github,欢迎访问:https://github.com/bosima/go-demo/tree/main/go-micro-grpc-hello-compatible

收获更多架构知识,请关注公众号 萤火架构。原创内容,转载请注明出处。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,602评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,442评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,878评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,306评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,330评论 5 373
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,071评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,382评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,006评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,512评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,965评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,094评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,732评论 4 323
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,283评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,286评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,512评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,536评论 2 354
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,828评论 2 345

推荐阅读更多精彩内容