org.apache.cxf.jaxrs.rx3.client.FlowableRxInvokerImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cxf-rt-rs-extension-rx3 Show documentation
Apache CXF JAX-RS Extensions: RxJava3
There is a newer version: 4.0.5
Show newest version
/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.cxf.jaxrs.rx3.client;

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

import javax.ws.rs.HttpMethod;
import javax.ws.rs.client.Entity;
import javax.ws.rs.client.SyncInvoker;
import javax.ws.rs.core.GenericType;
import javax.ws.rs.core.Response;

import io.reactivex.rxjava3.core.BackpressureStrategy;
import io.reactivex.rxjava3.core.Flowable;
import io.reactivex.rxjava3.core.FlowableEmitter;
import io.reactivex.rxjava3.core.FlowableOnSubscribe;
import io.reactivex.rxjava3.core.Scheduler;
import io.reactivex.rxjava3.schedulers.Schedulers;

public class FlowableRxInvokerImpl implements FlowableRxInvoker {
    private final Scheduler sc;
    private final SyncInvoker syncInvoker;
    
    public FlowableRxInvokerImpl(SyncInvoker syncInvoker, ExecutorService ex) {
        this.syncInvoker = syncInvoker;
        this.sc = ex == null ? null : Schedulers.from(ex);
    }

    @Override
    public Flowable get() {
        return get(Response.class);
    }

    @Override
    public  Flowable get(Class responseType) {
        return method(HttpMethod.GET, responseType);
    }

    @Override
    public  Flowable get(GenericType responseType) {
        return method(HttpMethod.GET, responseType);
    }

    @Override
    public Flowable put(Entity entity) {
        return put(entity, Response.class);
    }

    @Override
    public  Flowable put(Entity entity, Class responseType) {
        return method(HttpMethod.PUT, entity, responseType);
    }

    @Override
    public  Flowable put(Entity entity, GenericType responseType) {
        return method(HttpMethod.PUT, entity, responseType);
    }

    @Override
    public Flowable post(Entity entity) {
        return post(entity, Response.class);
    }

    @Override
    public  Flowable post(Entity entity, Class responseType) {
        return method(HttpMethod.POST, entity, responseType);
    }

    @Override
    public  Flowable post(Entity entity, GenericType responseType) {
        return method(HttpMethod.POST, entity, responseType);
    }

    @Override
    public Flowable delete() {
        return delete(Response.class);
    }

    @Override
    public  Flowable delete(Class responseType) {
        return method(HttpMethod.DELETE, responseType);
    }

    @Override
    public  Flowable delete(GenericType responseType) {
        return method(HttpMethod.DELETE, responseType);
    }

    @Override
    public Flowable head() {
        return method(HttpMethod.HEAD);
    }

    @Override
    public Flowable options() {
        return options(Response.class);
    }

    @Override
    public  Flowable options(Class responseType) {
        return method(HttpMethod.OPTIONS, responseType);
    }

    @Override
    public  Flowable options(GenericType responseType) {
        return method(HttpMethod.OPTIONS, responseType);
    }

    @Override
    public Flowable trace() {
        return trace(Response.class);
    }

    @Override
    public  Flowable trace(Class responseType) {
        return method("TRACE", responseType);
    }

    @Override
    public  Flowable trace(GenericType responseType) {
        return method("TRACE", responseType);
    }

    @Override
    public Flowable method(String name) {
        return method(name, Response.class);
    }

    @Override
    public Flowable method(String name, Entity entity) {
        return method(name, entity, Response.class);
    }

    @Override
    public  Flowable method(String name, Entity entity, Class responseType) {
        return create(() -> syncInvoker.method(name, entity, responseType));
    }
    
    @Override
    public  Flowable method(String name, Entity entity, GenericType responseType) {
        return create(() -> syncInvoker.method(name, entity, responseType));
    }

    @Override
    public  Flowable method(String name, Class responseType) {
        return create(() -> syncInvoker.method(name, responseType));
    }

    @Override
    public  Flowable method(String name, GenericType responseType) {
        return create(() -> syncInvoker.method(name, responseType));
    }
    
    private  Flowable create(Supplier supplier) {
        Flowable flowable = Flowable.create(new FlowableOnSubscribe() {
            @Override
            public void subscribe(FlowableEmitter emitter) throws Exception {
                try {
                    T response = supplier.get();
                    if (!emitter.isCancelled()) {
                        emitter.onNext(response);
                    }
                    
                    if (!emitter.isCancelled()) {
                        emitter.onComplete();
                    }
                } catch (Throwable e) {
                    if (!emitter.isCancelled()) {
                        emitter.onError(e);
                    }
                }
            }
        }, BackpressureStrategy.DROP);
        
        if (sc == null) {
            return flowable.subscribeOn(Schedulers.io());
        }
        
        return flowable.subscribeOn(sc).observeOn(sc);
    }

}