type WrappedServerStream … // Context returns the wrapper's WrappedContext, overwriting the nested grpc.ServerStream.Context() func (w *WrappedServerStream) Context() context.Context { … } // WrapServerStream returns a ServerStream that has the ability to overwrite context. func WrapServerStream(stream grpc.ServerStream) *WrappedServerStream { … }