Question

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

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

这是protoc生成的示例界面：

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

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

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

Answer 1

请参阅Sending and receiving metadata - server side

请求Context具有与请求和连接匹配的信息。您可以使用grpc.SetHeader或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定义响应头

1 个答案: