Use WrappedServerStream
@@ -67,7 +67,7 @@ func StreamCanceler(ctx context.Context) grpc.StreamServerInterceptor {
cancelMu.Unlock()
// invoke rpc
- err := handler(srv, &ctxServerStream{cctx, ss})
+ err := handler(srv, NewWrappedServerStream(cctx, ss))
// remove the cancel function
cancelMu.Lock()
@@ -77,12 +77,3 @@ func StreamCanceler(ctx context.Context) grpc.StreamServerInterceptor {
return err
})
}
-
-type ctxServerStream struct {
- ctx context.Context
- grpc.ServerStream
-}
-
-func (c *ctxServerStream) Context() context.Context {
- return c.ctx
-}
