com.microsoft.kiota.http.middleware.HeadersInspectionHandler 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.middleware.options.HeadersInspectionOption;

import io.opentelemetry.api.trace.Span;
import io.opentelemetry.context.Scope;

import jakarta.annotation.Nonnull;

import kotlin.Pair;

import okhttp3.Interceptor;
import okhttp3.Request;
import okhttp3.Response;

import java.io.IOException;
import java.util.HashSet;
import java.util.Objects;

/**
 * The middleware responsible for inspecting the request and response headers
 */
public class HeadersInspectionHandler implements Interceptor {
    /**
     * Create a new instance of the HeadersInspectionHandler class with the default options
     */
    public HeadersInspectionHandler() {
        this(new HeadersInspectionOption());
    }

    /**
     * Create a new instance of the HeadersInspectionHandler class with the provided options
     * @param options The options to use for the handler
     */
    public HeadersInspectionHandler(@Nonnull final HeadersInspectionOption options) {
        this.options = Objects.requireNonNull(options);
    }

    private final HeadersInspectionOption options;

    /** {@inheritDoc} */
    @Override
    @SuppressWarnings("UnknownNullness")
    @Nonnull public Response intercept(final Chain chain) throws IOException {
        Objects.requireNonNull(chain, "parameter chain cannot be null");
        Request request = chain.request();
        HeadersInspectionOption inspectionOption = request.tag(HeadersInspectionOption.class);
        if (inspectionOption == null) {
            inspectionOption = options;
        }
        final Span span =
                ObservabilityHelper.getSpanForRequest(
                        request, "HeadersInspectionHandler_Intercept");
        Scope scope = null;
        if (span != null) {
            scope = span.makeCurrent();
            span.setAttribute("com.microsoft.kiota.handler.headersInspection.enable", true);
        }
        try {
            if (span != null) {
                request = request.newBuilder().tag(Span.class, span).build();
            }
            if (inspectionOption.getInspectRequestHeaders()) {
                for (final Pair header : request.headers()) {
                    HashSet value = new HashSet<>();
                    value.add(header.getSecond());
                    inspectionOption.getRequestHeaders().put(header.getFirst(), value);
                }
            }
            final Response response = chain.proceed(request);
            if (inspectionOption.getInspectResponseHeaders()) {
                for (final Pair header : response.headers()) {
                    HashSet value = new HashSet<>();
                    value.add(header.getSecond());
                    inspectionOption.getResponseHeaders().put(header.getFirst(), value);
                }
            }
            return response;
        } finally {
            if (scope != null) {
                scope.close();
            }
            if (span != null) {
                span.end();
            }
        }
    }
}