net.redpipe.engine.rxjava.CompletableProvider Maven / Gradle / Ivy

Go to download
package net.redpipe.engine.rxjava;

import rx.Completable;
import rx.Subscription;
import org.jboss.resteasy.spi.AsyncResponseProvider;

import javax.ws.rs.core.Response;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.CompletionStage;

public class CompletableProvider implements AsyncResponseProvider {

    private static class CompletableAdaptor extends CompletableFuture
    {
        private Subscription subscription;

        public CompletableAdaptor(Completable observable)
        {
            this.subscription = observable.subscribe(this::completeEmpty, this::completeExceptionally);
        }

        @Override
        public boolean cancel(boolean mayInterruptIfRunning)
        {
            subscription.unsubscribe();
            return super.cancel(mayInterruptIfRunning);
        }

        void completeEmpty() {
            super.complete(Response.status(204).build());
        }

    }

    @Override
    public CompletionStage toCompletionStage(Completable asyncResponse) {
        return new CompletableAdaptor(asyncResponse);
    }

}