gRPC 流拦截器未将上下文传递给请求方法

本篇文章给大家分享《gRPC 流拦截器未将上下文传递给请求方法》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。

我有以下运行服务器端的 grpc 拦截器，它包装服务器流并将其传递给下一个处理程序：

// harmonycontext contains a custom context for passing data from middleware to handlers
type harmonycontext struct {
    context.context
    request interface{}
    userid  uint64
    limiter *rate.limiter
}

type iharmonywrappedserverstream interface {
    getwrappedcontext() harmonycontext
}

type harmonywrappedserverstream struct {
    grpc.serverstream
    wrappedcontext harmonycontext
}

func (ss harmonywrappedserverstream) getwrappedcontext() harmonycontext {
    return ss.wrappedcontext
}

func (m middlewares) harmonycontextinterceptorstream(srv interface{}, ss grpc.serverstream, info *grpc.streamserverinfo, handler grpc.streamhandler) error {
    wrapped := wrapserverstream(ss)
    return handler(srv, wrapped)
}

func wrapserverstream(stream grpc.serverstream) harmonywrappedserverstream {
    if existing, ok := stream.(harmonywrappedserverstream); ok {
        return existing
    }
    return harmonywrappedserverstream{serverstream: stream, wrappedcontext: harmonycontext{
        context: stream.context(),
    }}
}

在处理程序本身中，我有以下代码：

func (v1 *V1) StreamGuildEvents(r *corev1.StreamGuildEventsRequest, s corev1.CoreService_StreamGuildEventsServer) error {
    wrappedStream := s.(middleware.IHarmonyWrappedServerStream)
    println(wrappedStream)
    return nil
}

但是，在发送流请求时出现以下运行时错误： 接口转换：*corev1.coreservicestreamguildeventsserver不是中间件。iharmonywrappedserverstream：缺少方法getwrappedcontext 事实上，处理程序中的serverstream与拦截器中的serverstream完全不同。有没有办法让拦截器正确传递自定义的serverstream？

解决方案

我们已经开放了一个存储库来演示这一点（如果它对任何人有帮助的话）。

https://github.com/drud/api-common/blob/main/interceptors/state.go#L207

与@Bluskript impl 非常相似，但我没有遇到上面遇到的相同问题。我相信这可能来自返回类型 grpc.StreamServerInterceptor 以及我使用上下文 getter 的位置。

终于介绍完啦！小伙伴们，这篇关于《gRPC 流拦截器未将上下文传递给请求方法》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！