001/** 002 * Licensed to the Apache Software Foundation (ASF) under one or more 003 * contributor license agreements. See the NOTICE file distributed with 004 * this work for additional information regarding copyright ownership. 005 * The ASF licenses this file to You under the Apache License, Version 2.0 006 * (the "License"); you may not use this file except in compliance with 007 * the License. You may obtain a copy of the License at 008 * 009 * http://www.apache.org/licenses/LICENSE-2.0 010 * 011 * Unless required by applicable law or agreed to in writing, software 012 * distributed under the License is distributed on an "AS IS" BASIS, 013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 014 * See the License for the specific language governing permissions and 015 * limitations under the License. 016 */ 017package org.apache.camel.component.stub; 018 019import org.apache.camel.AsyncCallback; 020import org.apache.camel.Exchange; 021import org.apache.camel.ExchangePattern; 022import org.apache.camel.WaitForTaskToComplete; 023import org.apache.camel.component.seda.QueueReference; 024import org.apache.camel.component.seda.SedaEndpoint; 025import org.apache.camel.component.seda.SedaProducer; 026 027public class StubProducer extends SedaProducer { 028 029 public StubProducer(SedaEndpoint endpoint, WaitForTaskToComplete waitForTaskToComplete, long timeout, boolean blockWhenFull, long offerTimeout) { 030 super(endpoint, waitForTaskToComplete, timeout, blockWhenFull, offerTimeout); 031 } 032 033 @Override 034 public StubEndpoint getEndpoint() { 035 return (StubEndpoint) super.getEndpoint(); 036 } 037 038 @Override 039 public boolean process(Exchange exchange, AsyncCallback callback) { 040 AsyncCallback cb = callback; 041 042 QueueReference queueReference = getEndpoint().getQueueReference(); 043 boolean empty = queueReference == null || !queueReference.hasConsumers(); 044 045 // if no consumers then use InOnly mode 046 final ExchangePattern pattern = exchange.getPattern(); 047 if (empty && pattern != ExchangePattern.InOnly) { 048 exchange.setPattern(ExchangePattern.InOnly); 049 cb = doneSync -> { 050 // and restore the old pattern after processing 051 exchange.setPattern(pattern); 052 callback.done(doneSync); 053 }; 054 } 055 056 return super.process(exchange, cb); 057 } 058 059}