org.jboss.resteasy.reactive.client.handlers.ClientSwitchToRequestContextRestHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of resteasy-reactive-client Show documentation

There is a newer version: 3.17.5

Show newest version

package org.jboss.resteasy.reactive.client.handlers;

import java.util.concurrent.Executor;

import org.jboss.resteasy.reactive.client.impl.ClientRequestContextImpl;
import org.jboss.resteasy.reactive.client.impl.RestClientRequestContext;
import org.jboss.resteasy.reactive.client.spi.ClientRestHandler;

import io.smallrye.mutiny.infrastructure.Infrastructure;
import io.vertx.core.Context;
import io.vertx.core.Handler;
import io.vertx.core.Vertx;

/**
 * This handler ensures that the context to use is the same as the client request context, which is important to keep the
 * request context in sync when updating the response.
 */
public class ClientSwitchToRequestContextRestHandler implements ClientRestHandler {
    @Override
    public void handle(RestClientRequestContext requestContext) throws Exception {
        Context current = Vertx.currentContext();
        ClientRequestContextImpl clientRequestContext = requestContext.getClientRequestContext();
        if (clientRequestContext == null) {
            return;
        }

        Context captured = clientRequestContext.getContext();
        if (captured != null && current != captured) {
            requestContext.suspend();
            requestContext.resume(new Executor() {
                @Override
                public void execute(Runnable command) {
                    // This is necessary to propagate the Smallrye context which is required for OpenTelemetry
                    Runnable decorated = Infrastructure.decorate(command);
                    captured.runOnContext(new Handler() {
                        @Override
                        public void handle(Void unused) {
                            decorated.run();
                        }
                    });
                }
            });
        }
    }
}