wehub-resource-sync
24 lines
674 B
Go
24 lines
674 B
Go
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
|
|
}
|