rest - grpc REST 网关中的 ETags 和 304 Not Modified 响应

Question

我正在使用 grpc-gateway 从 gRPC 规范自动生成 REST API来自 Github 的项目。在此 REST API 中，我想支持 ETag header 和 304 Not Modified 响应。

据我了解，通常您会在 gRPC 服务器中创建一个具有特定状态代码的响应，然后该状态代码将由 grpc-gateway 转换为 HTTP 状态代码。但是，由于标准 gRPC 并不真正支持缓存概念，因此没有映射到 HTTP 304 状态代码的 gRPC 状态代码。

使用 grpc-gateway，似乎可以在 gRPC 状态代码为错误代码时自定义 HTTP 状态代码(覆盖 runtime.HTTPError 函数)。但是，我还没有找到任何方法来在 gRPC 响应代码正常时自定义 HTTP 响应代码。

那么，有什么推荐的方法可以实现这一目标吗？

Answer 1

这是一个使用自定义转发器实现基本 etag 和 304 响应的示例。

可以引用这些directions进行设置，然后按照以下方式实现该方法:

func forwardGetPost(ctx context.Context, mux *runtime.ServeMux, marshaler runtime.Marshaler, w http.ResponseWriter, req *http.Request, resp proto.Message, opts ...func(context.Context, http.ResponseWriter, proto.Message) error) {
    // add cache-control rules for this proxy endpoint
    w.Header().Add("Cache-Control", "max-age=60")

    // create an etag
    // (when the response represents some entity from the db it may have a last edited timestamp)
    p := resp.(*Post)
    etag := fmt.Sprintf("W/%q", p.GetLastEdited())
    w.Header().Add("ETag", etag)

    // check whether the request provides an etag
    inm := req.Header.Get("If-None-Match")
    if inm != "" {
        if inm == etag {
            w.WriteHeader(http.StatusNotModified)
        }
    }
    runtime.ForwardResponseMessage(ctx, mux, marshaler, w, req, resp, opts...)
}

这不会阻止代理向 grpc 服务器发出请求，但会阻止在 etags 匹配时将这些字节发送回客户端。