/** * 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.component.seda; import java.util.concurrent.BlockingQueue; import java.util.concurrent.CountDownLatch; import java.util.concurrent.ExecutorService; import java.util.concurrent.TimeUnit; import java.util.concurrent.atomic.AtomicInteger; import org.apache.camel.AsyncCallback; import org.apache.camel.AsyncProcessor; import org.apache.camel.Consumer; import org.apache.camel.Endpoint; import org.apache.camel.Exchange; import org.apache.camel.Processor; import org.apache.camel.ShutdownRunningTask; import org.apache.camel.SuspendableService; import org.apache.camel.impl.LoggingExceptionHandler; import org.apache.camel.impl.ServiceSupport; import org.apache.camel.impl.converter.AsyncProcessorTypeConverter; import org.apache.camel.processor.MulticastProcessor; import org.apache.camel.spi.ExceptionHandler; import org.apache.camel.spi.ShutdownAware; import org.apache.camel.util.AsyncProcessorHelper; import org.slf4j.Logger; import org.slf4j.LoggerFactory; /** * A Consumer for the SEDA component. *
* In this implementation there is a little slack period when you suspend/stop the consumer, by which * the consumer may pickup a newly arrived messages and process it. That period is up till 1 second. * * @version */ public class SedaConsumer extends ServiceSupport implements Consumer, Runnable, ShutdownAware, SuspendableService { private static final transient Logger LOG = LoggerFactory.getLogger(SedaConsumer.class); private final AtomicInteger taskCount = new AtomicInteger(); private CountDownLatch latch; private volatile boolean shutdownPending; private SedaEndpoint endpoint; private AsyncProcessor processor; private ExecutorService executor; private ExceptionHandler exceptionHandler; public SedaConsumer(SedaEndpoint endpoint, Processor processor) { this.endpoint = endpoint; this.processor = AsyncProcessorTypeConverter.convert(processor); } @Override public String toString() { return "SedaConsumer[" + endpoint.getEndpointUri() + "]"; } public Endpoint getEndpoint() { return endpoint; } public ExceptionHandler getExceptionHandler() { if (exceptionHandler == null) { exceptionHandler = new LoggingExceptionHandler(getClass()); } return exceptionHandler; } public void setExceptionHandler(ExceptionHandler exceptionHandler) { this.exceptionHandler = exceptionHandler; } public Processor getProcessor() { return processor; } public boolean deferShutdown(ShutdownRunningTask shutdownRunningTask) { // deny stopping on shutdown as we want seda consumers to run in case some other queues // depend on this consumer to run, so it can complete its exchanges return true; } public int getPendingExchangesSize() { // number of pending messages on the queue return endpoint.getQueue().size(); } public void prepareShutdown() { // signal we want to shutdown shutdownPending = true; LOG.debug("Preparing to shutdown, waiting for {} consumer threads to complete.", latch.getCount()); // wait for all threads to end try { latch.await(); } catch (InterruptedException e) { // ignore } } @Override public boolean isRunAllowed() { if (isSuspending() || isSuspended()) { // allow to run even if we are suspended as we want to // keep the thread task running return true; } return super.isRunAllowed(); } public void run() { taskCount.incrementAndGet(); try { doRun(); } finally { taskCount.decrementAndGet(); } } protected void doRun() { BlockingQueue