admin 管理员组

文章数量: 1184232

gRPC

代码地址


逻辑都在拦截器里实现了，可以直接看 拦截器 的代码

介绍

像一些 token 等数据 我们可能希望通过 header头传递，gRPC的数据先传递到 context 里，然后再写到 header 头里，然后
传递给另一端。

直接上代码

客户端

header 数据的写入

需要导入 google.golang/grpc/metadata 这个包

headerData := metadata.Pairs("timestamp", strconv.Itoa(int(time.Now().Unix())), "token", "123")
ctxH := metadata.NewOutgoingContext(ctx, headerData)
// 后续也可以往后面添加数据
ctxH = metadata.AppendToOutgoingContext(ctxH, "kay1", "val1", "key2", "val2")

ctxH就是一个上下文，请求服务端时，传的 context.Context 替换成 ctxH 就行

读取 header 信息

header 头信息也可能通过 trailer 传递

一元模式读取 header 信息
var header, trailer metadata.MD
// 调用服务端方法的时候可以在后面传参数
serverRetMsg, err := client.RPCMethod(ctx, &Struct, grpc.Header(&header), grpc.Trailer(&trailer))
流模式读取 header 信息
stream, err := client.RequestServerStreamMethod(ctx)
header, err := stream.Header()
trailer := stream.Trailer()

服务端

header 头信息的写入
一元模式 数据的写入
header := metadata.Pairs("token", "123")
grpc.SendHeader(ctx, header)
trailer := metadata.Pairs("is-end", "end")
grpc.SetTrailer(ctx, trailer)
流模式 数据的写入

跟一元的一样，服务端提供的方法传进来的流参数里面有相关的方法

header := metadata.Pairs("token", "123")
stream.SendHeader(ctx, header)
trailer := metadata.Pairs("is-end", "end")
stream.SetTrailer(ctx, trailer)
header 头信息的读取
一元模式 数据的读取
md, ok := metadata.FromIncomingContext(ctx)
流模式 数据的读取
md, ok := metadata.FromIncomingContext(stream.Context())

本文标签： gRPC

版权声明：本文标题：gRPC 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/p/1700276492a376035.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。