Something went wrong. Try again.
sloproxy
Something went wrong. Try again.
2.5 kB · 79 lines
Go
at main
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980package executor
import ( "context" "sync/atomic")
type downstreamWebsocketContextKey struct{}type requireUpstreamWebsocketContextKey struct{}type upstreamAttemptTrackerContextKey struct{}
type upstreamAttemptTracker struct { attempted atomic.Bool}
// 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}
// WithRequiredUpstreamWebsocket marks a request whose incremental context is valid only on the current upstream websocket.func WithRequiredUpstreamWebsocket(ctx context.Context) context.Context { if ctx == nil { ctx = context.Background() } return context.WithValue(ctx, requireUpstreamWebsocketContextKey{}, true)}
// RequiredUpstreamWebsocket reports whether falling back to an HTTP upstream would lose request context.func RequiredUpstreamWebsocket(ctx context.Context) bool { if ctx == nil { return false } raw := ctx.Value(requireUpstreamWebsocketContextKey{}) enabled, ok := raw.(bool) return ok && enabled}
// WithUpstreamAttemptTracker installs a fresh tracker for one provider execution attempt.func WithUpstreamAttemptTracker(ctx context.Context) context.Context { if ctx == nil { ctx = context.Background() } return context.WithValue(ctx, upstreamAttemptTrackerContextKey{}, &upstreamAttemptTracker{})}
// MarkUpstreamAttempt records that the provider execution reached an upstream transport boundary.func MarkUpstreamAttempt(ctx context.Context) { if ctx == nil { return } tracker, ok := ctx.Value(upstreamAttemptTrackerContextKey{}).(*upstreamAttemptTracker) if !ok || tracker == nil { return } tracker.attempted.Store(true)}
// UpstreamAttempted reports whether the tracked provider execution reached an upstream transport boundary.func UpstreamAttempted(ctx context.Context) bool { if ctx == nil { return false } tracker, ok := ctx.Value(upstreamAttemptTrackerContextKey{}).(*upstreamAttemptTracker) return ok && tracker != nil && tracker.attempted.Load()}