Golang处理gRPC请求/响应元数据的示例代码
元数据
gRPC的元数据（metadata）是基于HTTP/2头部实现的键值对数据，它通常用来实现gRPC的鉴权、链路跟踪以及自定义头部数据等功能。
gRPC的元数据分为两种类型，分别是Header及Trailer。Header可以由客户端或服务端发送，它在客户端请求数据或服务器响应数据前发送。Trailer是一种特殊的头部信息，它仅可由服务端发送，且位于发送的数据之后。
客户端处理
在gRPC客户端中，无论是一元调用还是流调用，可以比较简单地通过google.golang.org/grpc/metadata包提供的AppendToOutgoingContext或NewOutgoingContext方法向请求中加入头部元数据，例如以下几种方式：
// 通过metadata创建新的context
md := metadata.Pairs("k1", "v1", "k2", "v2")
ctx := metadata.NewOutgoingContext(ctx, md)
// 或是向context中添加元数据
ctx = metadata.AppendToOutgoingContext(ctx, "k3", "v3")
// ... 通过ctx进行RPC调用
对于服务端返回的响应中的元数据，一元调用与流调用的处理方式就较为不同。对于一元调用，需要提前定义好用于存储元数据的变量，然后在调用时通过grpc.Header或grpc.Trailer增加调用的选项：
var header, trailer metadata.MD resp, err := cli.UnaryCall(ctx, req, grpc.Header(&header), grpc.Trailer(&trailer)) // 处理header或trailer
而对于任意方式的流调用，都可以简单地通过流调用返回流的Header或Trailer方法获得元数据：
stream, err := cli.StreamCall(ctx) header, err := stream.Header() trailer, err := stream.Trailer()
服务端处理
对于服务端，请求的元数据需要通过metadata.FromIncomingContext从context中获取：
// 一元调用 md, ok := metadata.FromIncomingContext(ctx) // 流调用 ctx := stream.Context() // 需要先从流中得到context md, ok := metadata.FromIncomingContext(ctx)
同样，在服务端发送元数据需要根据一元调用与流调用使用不同的方式。对于一元调用，可以通过grpc.SendHeader、grpc.SetHeader以及grpc.SetTrailer方法设置发送的元数据，例如：
header := metadata.Pairs("header-key", "header-val")
grpc.SendHeader(ctx, header)
trailer := metadata.Pairs("trailer-key", "trailer-val")
grpc.SetTrailer(ctx, trailer)
对于上述的SendHeader及SetHeader方法，其区别为SendHeader方法只能调用一次，而SetHeader方法将会对所有调用的元数据进行合并发送。
对于流调用，服务端发送元数据则是通过流对象中的上述方法：
header := metadata.Pairs("header-key", "header-val")
stream.SendHeader(,header)
trailer := metadata.Pairs("trailer-key", "trailer-val")
stream.SetTrailer(trailer)
服务器拦截器处理
对于gRPC服务端一元调用及流调用拦截器，请求元数据的读取与响应元数据的发送与上一节中的实现相同，便不再赘述。下面我们将讨论一下在拦截器中更新请求元数据，以及读取响应的元数据。
一元调用拦截器更新请求元数据
在服务端拦截器中更新请求的元数据，其实现的方式与客户端发送元数据类似，即需要通过更新后的元数据创建新的context。对于一元调用拦截器，其简单实现如下所示：
md, ok := metadata.FromIncomingContext(ctx)
md.Append("new-key", "new-value")
ctx = metadata.NewIncomingContext(ctx, md)
resp, err := handler(ctx, req) // 传递context至handler中
一元调用拦截器读取响应元数据
对于一元调用响应的元数据，gRPC未提供直接访问的方法响应的元数据。为了在拦截器中能读取到响应的元数据，我们可以通过覆盖原始grpc.ServerTransportStream并对设置的元数据进行备份的方式进行实现。
type WrappedServerTransportStream struct {
grpc.ServerTransportStream
header metadata.MD
trailer metadata.MD
}
func (s *WrappedServerTransportStream) SendHeader(md metadata.MD) error {
if err := s.ServerTransportStream.SendHeader(md); err != nil {
return err
}
s.header = md
return nil
}
// 在需要的情况下继续实现下面的几个方法：
// func (s *WrappedServerTransportStream) SetHeader(metadata.MD) error
// func (s *WrappedServerTransportStream) SetTrailer(metadata.MD) error
在定义带有元数据副本的ServerTransportStream实现后，我们需要通过grpc.ServerTransportStreamFromContext获取到一元调用的原始流，在对其进行封装后，调用grpc.NewContextWithServerTransportStream创建新的context。
stream := grpc.ServerTransportStreamFromContext(ctx)
wrappedStream := &WrappedServerTransportStream{
ServerTransportStream: stream,
}
ctx = grpc.NewContextWithServerTransportStream(ctx, wrappedStream)
resp, err := handler(ctx, req)
// 通过wrappedStream.header、wrappedStream.trailer读取响应的元数据
需要注意，grpc.ServerTransportStream接口是一个实验性的接口，在后续版本中可能会被移除，所以本节中描述的方法在后续版本中可能不再可用。
流调用拦截器更新请求元数据
而对于流调用，gRPC没有提供修改其context的方法，为了实现修改流调用请求元数据，就需要实现grpc.ServerStream接口并加入带有修改后元数据的context。以下是一个简单的实现：
type WrappedStream struct {
grpc.ServerStream
ctx context.Context
}
func (s *WrappedStream) Context() context.Context {
return s.ctx
}
func ExampleStreamInterceptor(srv any, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {
md, ok := metadata.FromIncomingContext(ss.Context())
md.append("new-key", "new-value")
ctx := metadata.NewIncomingContext(ss.Context(), md)
return handler(srv, &WrappedStream{ss, ctx})
}
流调用拦截器读取响应元数据
与在一元调用拦截器中相同，若需要在流调用拦截器中读取响应的元数据，我们可以实现grpc.ServerStream接口，并在其中保存元数据的副本。例如我们可以在上节的WrappedStream的基础上，对其进行一定修改：
type WrappedStream struct {
grpc.ServerStream
header metadata.MD
trailer metadata.MD
}
func (s *WrappedStream) SendHeader(md metadata.MD) error {
if err := s.ServerStream.SendHeader(md); err != nil {
return err
}
s.header = md
return nil
}
// 继续实现SetHeader、SetTrailer等方法
func ExampleStreamInterceptor(srv any, ss grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error {
stream := &WrappedStream{ServerStream: ss}
err := handler(srv, stream)
// 通过stream.header、stream.trailer读取响应元数据
return err
}
以上就是Golang处理gRPC请求/响应元数据的示例代码的详细内容，更多关于Golang处理gRPC请求/响应的资料请关注脚本之家其它相关文章！
相关文章
本文主要介绍了Golang Gin框架实现文件下载功能的示例代码，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-12-12
这篇文章主要介绍了详情介绍golang类型转换问题,分别由介绍类型断言和类型转换，这两者都是不同的概念，下面文章围绕类型断言和类型转换的相关资料展开文章的详细内容,需要的朋友可以参考以下2021-12-12
这篇文章主要介绍了Golang搭建开发环境，本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-11-11
这篇文章主要为大家详细介绍了go语言中如何实现字符切片和字符串互转,文中的示例代码讲解详细,具有一定的学习价值,感兴趣的小伙伴可以了解一下2023-11-11
本文介绍了使用Go语言实现的MapReduce框架,特别是在AWSS3 SDK的MultiPartUpload功能中的应用,包括并发上传和错误处理策略,详细解释了如何通过并发goroutines提高上传效率,并通过MapReduce模型优化代码结构和处理流程,感兴趣的朋友跟随小编一起看看吧2024-09-09
Golang使用pprof和trace进行诊断和修复性能问题
在 Go 中,开发人员可以使用强大的内置工具来帮助诊断和修复性能问题,其中两个工具是 pprof 和 trace 包,下面就跟随小编一起来了解下如何使用pprof和trace进行诊断和修复性能问题吧2024-01-01
这篇文章主要为大家介绍了Hugo 游乐场内容初始化示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-02-02
这篇文章主要为大家介绍了Go语言error的设计理念及背景演化详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-12-12
这篇文章主要介绍了go benchmark 基准测试详解,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-03-03
这篇文章主要介绍了Go语言学习函数+结构体+方法+接口，文章围绕主题的相关资料展开详细的文章说明，具有一定的参考价值，需要的小伙伴可以参考一下2022-05-05
最新评论