io.quarkus.vertx.http.runtime.security.VertxBlockingSecurityExecutor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quarkus-vertx-http Show documentation

Vert.x HTTP

The newest version!

package io.quarkus.vertx.http.runtime.security;

import static io.quarkus.vertx.core.runtime.context.VertxContextSafetyToggle.setContextSafe;
import static io.smallrye.common.vertx.VertxContext.getOrCreateDuplicatedContext;

import java.util.concurrent.Callable;
import java.util.function.Supplier;

import jakarta.inject.Inject;

import io.quarkus.runtime.BlockingOperationControl;
import io.quarkus.security.spi.runtime.BlockingSecurityExecutor;
import io.smallrye.mutiny.Uni;
import io.vertx.core.Context;
import io.vertx.core.Vertx;

public class VertxBlockingSecurityExecutor implements BlockingSecurityExecutor {

    @Inject
    Vertx vertx;

    @Override
    public  Uni executeBlocking(Supplier supplier) {
        return Uni.createFrom().deferred(new Supplier>() {
            @Override
            public Uni get() {
                if (BlockingOperationControl.isBlockingAllowed()) {
                    try {
                        return Uni.createFrom().item(supplier.get());
                    } catch (Throwable t) {
                        return Uni.createFrom().failure(t);
                    }
                } else {
                    Context local = getOrCreateDuplicatedContext(vertx);
                    setContextSafe(local, true);
                    return Uni
                            .createFrom()
                            .completionStage(
                                    local
                                            .executeBlocking(new Callable() {
                                                @Override
                                                public T call() {
                                                    return supplier.get();
                                                }
                                            }, false)
                                            .toCompletionStage());
                }
            }
        });
    }
}