Function SpanTimer

x/x.go:990–1033

SpanTimer returns a function used to record the duration of the given span. It supports both OpenCensus and OpenTelemetry spans

(spanInterface interface{}, name string)

// SpanTimer returns a function used to record the duration of the given span.
// It supports both OpenCensus and OpenTelemetry spans
func SpanTimer(spanInterface interface{}, name string) func() {
	// OpenCensus span case
	if span, ok := spanInterface.(*trace.Span); ok && span != nil {
		uniq := int64(rand.Int31()) //nolint:gosec // unique id for tracing does not require cryptographic precision
		attrs := trace.WithAttributes(
			attribute.Int64("funcId", uniq),
			attribute.String("funcName", name),
		)
		(*span).AddEvent("Start", attrs)
		start := time.Now()

		return func() {
			(*span).AddEvent("End", trace.WithAttributes(
				attribute.Int64("funcId", uniq),
				attribute.String("funcName", name),
				attribute.String("duration", time.Since(start).String()),
			))
			// TODO: We can look into doing a latency record here.
		}
	}

	// OpenTelemetry span case
	if span, ok := spanInterface.(trace.Span); ok {
		uniq := int64(rand.Int31()) //nolint:gosec // unique id for tracing does not require cryptographic precision
		attrs := trace.WithAttributes(
			attribute.Int64("funcId", uniq),
			attribute.String("funcName", name),
		)
		span.AddEvent("Start", attrs)
		start := time.Now()

		return func() {
			span.AddEvent("End", trace.WithAttributes(
				attribute.Int64("funcId", uniq),
				attribute.String("funcName", name),
				attribute.String("duration", time.Since(start).String()),
			))
			// TODO: We can look into doing a latency record here.
		}
	}

	// Fall back for nil or unknown span type
	return func() {}
}

// CloseFunc needs to be called to close all the client connections.
type CloseFunc func()