Function WrapServerStream

internal/api/grpc.go:35–40

WrapServerStream returns a ServerStream that has the ability to overwrite context.

(stream grpc.ServerStream)

// WrapServerStream returns a ServerStream that has the ability to overwrite context.
func WrapServerStream(stream grpc.ServerStream) *WrappedServerStream {
	if existing, ok := stream.(*WrappedServerStream); ok {
		return existing
	}
	return &WrappedServerStream{ServerStream: stream, WrappedContext: stream.Context()}
}

type GrpcServer struct {
	*grpc.Server