micro/tracer/options.go

package tracer

import (
	"context"

	"go.unistack.org/micro/v3/logger"
)

// SpanOptions contains span option
type SpanOptions struct {
	Labels []interface{}
}

// SpanOption func signature
type SpanOption func(o *SpanOptions)

// EventOptions contains event options
type EventOptions struct{}

// EventOption func signature
type EventOption func(o *EventOptions)

func SpanLabels(labels ...interface{}) SpanOption {
	return func(o *SpanOptions) {
		o.Labels = labels
	}
}

// Options struct
type Options struct {
	// Logger used for logging
	Logger logger.Logger
	// Name of the tracer
	Name string
	// Context used to store custome tracer options
	Context context.Context
}

// Option func signature
type Option func(o *Options)

// Logger sets the logger
func Logger(l logger.Logger) Option {
	return func(o *Options) {
		o.Logger = l
	}
}

// NewOptions returns default options
func NewOptions(opts ...Option) Options {
	options := Options{
		Logger: logger.DefaultLogger,
	}
	for _, o := range opts {
		o(&options)
	}
	return options
}

// Name sets the name
func Name(n string) Option {
	return func(o *Options) {
		o.Name = n
	}
}