tracing.go

package flux

import (
	"context"
)

type contextKey string

const queryTracingContextKey contextKey = "query-tracing-enabled"

// WithQueryTracingEnabled will return a child context
// that will turn on experimental query tracing.
func WithQueryTracingEnabled(parentCtx context.Context) context.Context {
	return context.WithValue(parentCtx, queryTracingContextKey, true)
}

// IsQueryTracingEnabled will return true if the context
// contains a key indicating that experimental tracing is enabled.
func IsQueryTracingEnabled(ctx context.Context) bool {
	v := ctx.Value(queryTracingContextKey)
	if v == nil {
		return false
	}
	b, ok := v.(bool)
	if !ok {
		return false
	}
	return b
}