如何为一元 rpc 定义响应 headers

Question

在 Go 中，我想从我的 rpc 处理程序实现中设置一些响应 headers。具体来说，我想设置一些与缓存控制相关的headers。

但是当rpc定义为一元响应时，protoc生成的方法签名只包含请求上下文和请求消息。这与从流响应生成的代码形成对比，后者包括请求消息和 grpc.ServerStream.

这是协议生成的示例接口：

type CmsServer interface {
  GetPost(context.Context, *PostRequest) (*Post, error)
  GetPosts(*GetPostsOptions, Cms_GetPostsServer) error
}

在示例中，GetPosts 包含 Cms_GetPostsServer 参数，这是一个 grpc.ServerStream 似乎包含设置自定义响应的方法 headers https://godoc.org/google.golang.org/grpc#ServerStream（我还没有尝试过，如果这是错误的请纠正我）。

同时，GetPost 似乎没有任何方法可以通过查看方法签名来设置响应 headers（context.Context 只是请求上下文，而不是响应语境）。那么如何为一元响应设置自定义响应 headers？

Answer 1

参见发送和接收元数据 - 服务器端

请求上下文具有将请求与连接相匹配的信息。您使用 grpc.SetHeader or grpc.SendHeader 从服务器写入额外的元数据

func (s *server) SomeRPC(ctx context.Context, in *pb.someRequest) (*pb.someResponse, error) {
    // create and send header
    header := metadata.Pairs("header-key", "val")
    grpc.SendHeader(ctx, header)
    // create and set trailer
    trailer := metadata.Pairs("trailer-key", "val")
    grpc.SetTrailer(ctx, trailer)
}

如何为一元 rpc 定义响应 headers

How to define response headers for unary rpc

go

protocol-buffers

grpc

protoc