scope.go

package cortex

import "context"

type contextKey int

const (
	tenantKey contextKey = iota
	appKey
)

// TenantFromContext extracts the tenant identifier from the context.
// Returns an empty string if no tenant is set.
func TenantFromContext(ctx context.Context) string {
	v, ok := ctx.Value(tenantKey).(string)
	if !ok {
		return ""
	}
	return v
}

// WithTenant returns a copy of ctx with the tenant identifier attached.
func WithTenant(ctx context.Context, tenant string) context.Context {
	return context.WithValue(ctx, tenantKey, tenant)
}

// AppFromContext extracts the app identifier from the context.
// Returns an empty string if no app is set.
func AppFromContext(ctx context.Context) string {
	v, ok := ctx.Value(appKey).(string)
	if !ok {
		return ""
	}
	return v
}

// WithApp returns a copy of ctx with the app identifier attached.
func WithApp(ctx context.Context, app string) context.Context {
	return context.WithValue(ctx, appKey, app)
}