23 lines · 674 B · Go
RawDownload
package executor import "context" type downstreamWebsocketContextKey struct{} // WithDownstreamWebsocket marks the current request as coming from a downstream websocket connection.func WithDownstreamWebsocket(ctx context.Context) context.Context {	if ctx == nil {		ctx = context.Background()	}	return context.WithValue(ctx, downstreamWebsocketContextKey{}, true)} // DownstreamWebsocket reports whether the current request originates from a downstream websocket connection.func DownstreamWebsocket(ctx context.Context) bool {	if ctx == nil {		return false	}	raw := ctx.Value(downstreamWebsocketContextKey{})	enabled, ok := raw.(bool)	return ok && enabled}