net.winterly.rxjersey.server.SingleMethodDispatcher Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rxjava-server Show documentation

RxJava extension for Jersey and Dropwizard

There is a newer version: 0.11.0

package net.winterly.rxjersey.server;

import org.glassfish.hk2.api.IterableProvider;
import org.glassfish.hk2.api.ServiceLocator;
import org.glassfish.jersey.server.ContainerRequest;
import org.glassfish.jersey.server.internal.process.AsyncContext;
import org.glassfish.jersey.server.spi.internal.ResourceMethodDispatcher;
import rx.Completable;
import rx.Observable;
import rx.Single;

import javax.inject.Inject;
import javax.ws.rs.ProcessingException;
import javax.ws.rs.container.ContainerRequestContext;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.Response;

public class SingleMethodDispatcher extends RxMethodDispatcher {

    private final Response noContent = Response.noContent().build();

    @Context
    private javax.inject.Provider containerRequestContext;

    @Context
    private IterableProvider> requestInterceptors;

    public SingleMethodDispatcher(ResourceMethodDispatcher originalDispatcher) {
        super(originalDispatcher);
    }

    @Override
    public void dispatch(AsyncContext asyncContext, ResourceMethodDispatcher dispatcher, Object resource, ContainerRequest request) throws ProcessingException {
        final ContainerRequestContext requestContext = containerRequestContext.get();

        Completable intercept = Observable.from(requestInterceptors)
                .concatMap(interceptor -> interceptor.intercept(requestContext))
                .lastOrDefault(null)
                .toCompletable();

        Single dispatch = Single.defer(() -> (Single) dispatcher.dispatch(resource, request).getEntity());

        intercept.andThen(dispatch)
                .map(response -> response == null ? noContent : response)
                .subscribe(asyncContext::resume, asyncContext::resume);
    }

    public static class Provider extends RxMethodDispatcherProvider {

        @Inject
        public Provider(ServiceLocator serviceLocator) {
            super(serviceLocator, Observable.class, Single.class, Completable.class);
        }

        @Override
        public RxMethodDispatcher create(ResourceMethodDispatcher dispatcher) {
            return new SingleMethodDispatcher(dispatcher);
        }
    }
}