Function AddDedicatedAttributes

pkg/util/test/req.go:309–375

AddDedicatedAttributes adds resource and span attributes to a trace that are stored in dedicated columns when a backend.BlockMeta is created with the column assignments from MakeDedicatedColumns.

(trace *tempopb.Trace)

// AddDedicatedAttributes adds resource and span attributes to a trace that are stored in dedicated
// columns when a backend.BlockMeta is created with the column assignments from MakeDedicatedColumns.
func AddDedicatedAttributes(trace *tempopb.Trace) *tempopb.Trace {
	makeVal := func(c backend.DedicatedColumn, i int) *v1_common.AnyValue {
		if c.Type == backend.DedicatedColumnTypeInt {
			return &v1_common.AnyValue{
				Value: &v1_common.AnyValue_IntValue{
					IntValue: int64(i + 1),
				},
			}
		}
		if slices.Contains(c.Options, backend.DedicatedColumnOptionBlob) {
			return &v1_common.AnyValue{
				Value: &v1_common.AnyValue_StringValue{
					StringValue: DedicatedBlobTestString(),
				},
			}
		}
		return &v1_common.AnyValue{
			Value: &v1_common.AnyValue_StringValue{
				StringValue: fmt.Sprintf("dedicated-%s-attr-value-%d", c.Scope, i+1),
			},
		}
	}

	spanAttrs := make([]*v1_common.KeyValue, 0, len(dedicatedColumnsSpan))
	for i, c := range dedicatedColumnsSpan {
		spanAttrs = append(spanAttrs, &v1_common.KeyValue{
			Key:   c.Name,
			Value: makeVal(c, i),
		})
	}
	resourceAttrs := make([]*v1_common.KeyValue, 0, len(dedicatedColumnsResource))
	for i, c := range dedicatedColumnsResource {
		resourceAttrs = append(resourceAttrs, &v1_common.KeyValue{
			Key:   c.Name,
			Value: makeVal(c, i),
		})
	}
	eventAttrs := make([]*v1_common.KeyValue, 0, len(dedicatedColumnsEvent))
	for i, c := range dedicatedColumnsEvent {
		eventAttrs = append(eventAttrs, &v1_common.KeyValue{
			Key:   c.Name,
			Value: makeVal(c, i),
		})
	}

	for _, batch := range trace.ResourceSpans {
		attr := make([]*v1_common.KeyValue, 0, len(resourceAttrs)+len(batch.Resource.Attributes))
		attr = append(attr, resourceAttrs...)
		batch.Resource.Attributes = append(attr, batch.Resource.Attributes...)

		for _, ss := range batch.ScopeSpans {
			for _, span := range ss.Spans {
				attr = make([]*v1_common.KeyValue, 0, len(spanAttrs)+len(span.Attributes))
				attr = append(attr, spanAttrs...)
				span.Attributes = append(attr, span.Attributes...)

				for _, e := range span.Events {
					attr = make([]*v1_common.KeyValue, 0, len(eventAttrs)+len(e.Attributes))