tracestorage

type Consumer struct {
	// contains filtered or unexported fields
}

type Event struct {
	Timestamp otelstorage.Timestamp `json:"timestamp"`
	Name      string                `json:"name"`
	Attrs     otelstorage.Attrs     `json:"attrs"`
}

type Inserter interface {
	// SpanWriter creates a new batch.
	SpanWriter(ctx context.Context) (SpanWriter, error)
}

type Link struct {
	TraceID    otelstorage.TraceID `json:"trace_id"`
	SpanID     otelstorage.SpanID  `json:"span_id"`
	TraceState string              `json:"trace_state"`
	Attrs      otelstorage.Attrs   `json:"attrs"`
}

type Querier interface {
	// SearchTags performs search by given tags.
	SearchTags(ctx context.Context, tags map[string]string, opts SearchTagsOptions) (iterators.Iterator[Span], error)

	// TagNames returns all available tag names.
	TagNames(ctx context.Context, opts TagNamesOptions) ([]TagName, error)
	// TagValues returns all available tag values for given tag.
	TagValues(ctx context.Context, attr traceql.Attribute, opts TagValuesOptions) (iterators.Iterator[Tag], error)

	// TraceByID returns spans of given trace.
	TraceByID(ctx context.Context, id otelstorage.TraceID, opts TraceByIDOptions) (iterators.Iterator[Span], error)
}

type SearchTagsOptions struct {
	MinDuration time.Duration
	MaxDuration time.Duration

	// Start defines time range for search.
	//
	// Querier ignores parameter, if it is zero.
	Start time.Time
	// End defines time range for search.
	//
	// Querier ignores parameter, if it is zero.
	End time.Time
}

type Span struct {
	TraceID       otelstorage.TraceID   `json:"trace_id"`
	SpanID        otelstorage.SpanID    `json:"span_id"`
	TraceState    string                `json:"trace_state"`
	ParentSpanID  otelstorage.SpanID    `json:"parent_span_id"`
	Name          string                `json:"name"`
	Kind          int32                 `json:"kind"`
	Start         otelstorage.Timestamp `json:"start"`
	End           otelstorage.Timestamp `json:"end"`
	Attrs         otelstorage.Attrs     `json:"attrs"`
	StatusCode    int32                 `json:"status_code"`
	StatusMessage string                `json:"status_message"`

	BatchID       uuid.UUID         `json:"batch_id"`
	ResourceAttrs otelstorage.Attrs `json:"resource_attrs"`

	ScopeName    string            `json:"scope_name"`
	ScopeVersion string            `json:"scope_version"`
	ScopeAttrs   otelstorage.Attrs `json:"scope_attrs"`

	Events []Event `json:"events"`
	Links  []Link  `json:"links"`
}

type SpanWriter interface {
	// Add adds record to the batch.
	Add(span Span) error
	// Submit sends batch.
	Submit(ctx context.Context) error
	// Close frees resources.
	//
	// Callers should call [SpanWriter.Close] regardless if they called [SpanWriter.Submit] or not.
	Close() error
}

type Tag struct {
	Name  string                 `json:"name"`
	Value string                 `json:"value"`
	Type  traceql.StaticType     `json:"type"`
	Scope traceql.AttributeScope `json:"scope"`
}

type TagName struct {
	Scope traceql.AttributeScope
	Name  string
}

type TagNamesOptions struct {
	// Scope defines attribute scope to lookup.
	//
	// Querier should return attributes from all scopes, if it is zero.
	Scope traceql.AttributeScope
	// Start defines time range for search.
	//
	// Querier ignores parameter, if it is zero.
	Start time.Time
	// End defines time range for search.
	//
	// Querier ignores parameter, if it is zero.
	End time.Time
}

type TagValuesOptions struct {
	// AutocompleteQuery is a set of spanset matchers to only return tags seen
	// on matching spansets.
	AutocompleteQuery traceql.Autocomplete
	// Start defines time range for search.
	//
	// Querier ignores parameter, if it is zero.
	Start time.Time
	// End defines time range for search.
	//
	// Querier ignores parameter, if it is zero.
	End time.Time
}

type TraceByIDOptions struct {
	// Start defines time range for search.
	//
	// Querier ignores parameter, if it is zero.
	Start time.Time
	// End defines time range for search.
	//
	// Querier ignores parameter, if it is zero.
	End time.Time
}