org.apache.camel.rx.ReactiveCamel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of camel-rx Show documentation
Camel Reactive Extensions support
There is a newer version: 2.25.4
/**
 * 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.camel.rx;

import java.util.concurrent.ExecutorService;

import org.apache.camel.CamelContext;
import org.apache.camel.Endpoint;
import org.apache.camel.Exchange;
import org.apache.camel.Message;
import org.apache.camel.rx.support.EndpointObservable;
import org.apache.camel.rx.support.EndpointSubscribeFunc;
import org.apache.camel.rx.support.ExchangeToBodyFunc1;
import org.apache.camel.rx.support.ObserverSender;
import org.apache.camel.util.CamelContextHelper;
import rx.Observable;
import rx.functions.Func1;

/**
 * Provides the Reactive Extensions support for
 * Camel via the RxJava library
 */
public class ReactiveCamel {
    private final CamelContext camelContext;
    // a worker pool for running tasks such as stopping consumers which should not use the event loop
    // thread from rx-java but use our own thread to process such tasks
    private final ExecutorService workerPool;

    /**
     * Wrap the CamelContext as reactive.
     * 
     * Uses a default value of 10 as maximum number of threads in the worker pool used for reactive background tasks.
     *
     * @param camelContext  the CamelContext
     */
    public ReactiveCamel(CamelContext camelContext) {
        this(camelContext, 10);
    }

    /**
     * Wrap the CamelContext as reactive.
     *
     * @param camelContext  the CamelContext
     * @param maxWorkerPoolSize  maximum number of threads in the worker pool used for reactive background tasks
     */
    public ReactiveCamel(CamelContext camelContext, int maxWorkerPoolSize) {
        this.camelContext = camelContext;
        this.workerPool = camelContext.getExecutorServiceManager().newThreadPool(this, "ReactiveCamelWorker", 0, maxWorkerPoolSize);
    }

    /**
     * Returns an {@link rx.Observable < org.apache.camel.Message >} to allow the messages sent on the endpoint
     * to be processed using  Reactive Extensions
     */
    public Observable toObservable(String uri) {
        return toObservable(endpoint(uri));
    }

    /**
     * Returns an {@link rx.Observable } for the messages with their payload converted to the given type
     * to allow the messages sent on the endpoint
     * to be processed using  Reactive Extensions
     */
    public  Observable toObservable(String uri, final Class bodyType) {
        return toObservable(endpoint(uri), bodyType);
    }

    /**
     * Returns an {@link rx.Observable < org.apache.camel.Message >} to allow the messages sent on the endpoint
     * to be processed using  Reactive Extensions
     */
    public Observable toObservable(Endpoint endpoint) {
        return toObservable(endpoint, Message.class);
    }

    /**
     * Returns an {@link rx.Observable } for the messages with their payload converted to the given type
     * to allow the messages sent on the endpoint
     * to be processed using  Reactive Extensions
     */
    public  Observable toObservable(Endpoint endpoint, final Class bodyType) {
        return createEndpointObservable(endpoint, new ExchangeToBodyFunc1(bodyType));
    }

    /**
     * Sends events on the given {@link Observable} to the given camel endpoint
     */
    public  void sendTo(Observable observable, String endpointUri) {
        sendTo(observable, endpoint(endpointUri));
    }
    /**
     * Sends events on the given {@link Observable} to the given camel endpoint
     */
    public  void sendTo(Observable observable, Endpoint endpoint) {
        try {
            ObserverSender observer = new ObserverSender(endpoint);
            observable.subscribe(observer);
        } catch (Exception e) {
            throw new RuntimeCamelRxException(e);
        }
    }

    /**
     * Convenience method for beginning the route
     */
    public Observable from(Endpoint endpoint) {
        return createEndpointObservable(endpoint);
    }

    /**
     * Convenience method for beginning the route
     */
    public Observable from(String uri) {
        return from(endpoint(uri));
    }

    /**
     * Convenience method for creating CamelOperator instances
     */
    public CamelOperator to(String uri) throws Exception {
        return new CamelOperator(camelContext, uri);
    }

    /**
     * Convenience method for creating CamelOperator instances
     */
    public CamelOperator to(Endpoint endpoint) throws Exception {
        return new CamelOperator(endpoint);
    }

    public CamelContext getCamelContext() {
        return camelContext;
    }

    public Endpoint endpoint(String endpointUri) {
        return CamelContextHelper.getMandatoryEndpoint(camelContext, endpointUri);
    }

    /**
     * Returns a newly created {@link Observable} given a function which converts
     * the {@link Exchange} from the Camel consumer to the required type
     */
    protected  Observable createEndpointObservable(final Endpoint endpoint,
                                                         final Func1 converter) {
        Observable.OnSubscribe func = new EndpointSubscribeFunc(workerPool, endpoint, converter);
        return new EndpointObservable(endpoint, func);
    }

    /**
     * Return a newly created {@link Observable} without conversion
     */
    protected Observable createEndpointObservable(final Endpoint endpoint) {
        return new EndpointObservable(endpoint, new EndpointSubscribeFunc(workerPool, endpoint, new Func1() {
            @Override
            public Exchange call(Exchange exchange) {
                return exchange;
            }
        }));
    }
}