com.microsoft.kiota.http.middleware.ObservabilityHelper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of microsoft-kiota-http-okHttp Show documentation

Microsoft Kiota-Http with okHttp

There is a newer version: 1.8.0

Show newest version

package com.microsoft.kiota.http.middleware;

import com.microsoft.kiota.http.ObservabilityOptions;

import io.opentelemetry.api.GlobalOpenTelemetry;
import io.opentelemetry.api.trace.Span;
import io.opentelemetry.api.trace.SpanBuilder;
import io.opentelemetry.context.Context;

import jakarta.annotation.Nonnull;
import jakarta.annotation.Nullable;

import okhttp3.Request;

import java.util.Objects;

class ObservabilityHelper {
    static Span getSpanForRequest(@Nonnull final Request request, @Nonnull final String spanName) {
        return getSpanForRequest(request, spanName, null);
    }

    static Span getSpanForRequest(
            @Nonnull final Request request,
            @Nonnull final String spanName,
            @Nullable final Span parentSpan) {
        Objects.requireNonNull(request, "parameter request cannot be null");
        Objects.requireNonNull(spanName, "parameter spanName cannot be null");
        final ObservabilityOptions obsOptions = request.tag(ObservabilityOptions.class);
        Span span = null;
        if (obsOptions != null) {
            Span parentToUse = parentSpan;
            if (parentToUse == null) {
                parentToUse = request.tag(Span.class);
            }
            final SpanBuilder builder =
                    GlobalOpenTelemetry.getTracer(obsOptions.getTracerInstrumentationName())
                            .spanBuilder(spanName);
            if (parentToUse != null) {
                builder.setParent(Context.current().with(parentToUse));
            }
            span = builder.startSpan();
        }
        return span;
    }
}