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     */
017    package org.apache.camel.model;
018    
019    import java.util.List;
020    import java.util.concurrent.ThreadPoolExecutor;
021    
022    import javax.xml.bind.annotation.XmlAccessType;
023    import javax.xml.bind.annotation.XmlAccessorType;
024    import javax.xml.bind.annotation.XmlAttribute;
025    import javax.xml.bind.annotation.XmlRootElement;
026    import javax.xml.bind.annotation.XmlTransient;
027    
028    import org.apache.camel.Processor;
029    import org.apache.camel.processor.MulticastProcessor;
030    import org.apache.camel.processor.aggregate.AggregationStrategy;
031    import org.apache.camel.processor.aggregate.UseLatestAggregationStrategy;
032    import org.apache.camel.processor.interceptor.StreamCachingInterceptor;
033    import org.apache.camel.spi.RouteContext;
034    
035    /**
036     * Represents an XML <multicast/> element
037     *
038     * @version $Revision: 42222 $
039     */
040    @XmlRootElement(name = "multicast")
041    @XmlAccessorType(XmlAccessType.FIELD)
042    public class MulticastType extends OutputType<ProcessorType> {
043        @XmlAttribute(required = false)
044        private Boolean parallelProcessing;
045        @XmlTransient
046        private AggregationStrategy aggregationStrategy;
047        @XmlTransient
048        private ThreadPoolExecutor threadPoolExecutor;
049    
050        @Override
051        public String toString() {
052            return "Multicast[" + getOutputs() + "]";
053        }
054    
055        @Override
056        public Processor createProcessor(RouteContext routeContext) throws Exception {
057            return createOutputsProcessor(routeContext);
058        }
059    
060        protected Processor createCompositeProcessor(List<Processor> list) {
061            if (aggregationStrategy == null) {
062                aggregationStrategy = new UseLatestAggregationStrategy();
063            }
064            return new MulticastProcessor(list, aggregationStrategy, isParallelProcessing(), threadPoolExecutor);
065        }
066    
067        public AggregationStrategy getAggregationStrategy() {
068            return aggregationStrategy;
069        }
070    
071        public MulticastType setAggregationStrategy(AggregationStrategy aggregationStrategy) {
072            this.aggregationStrategy = aggregationStrategy;
073            return this;
074        }
075    
076        public boolean isParallelProcessing() {
077            return parallelProcessing != null ? parallelProcessing : false;
078        }
079    
080        public MulticastType setParallelProcessing(boolean parallelProcessing) {
081            this.parallelProcessing = parallelProcessing;
082            return this;
083        }
084    
085        public ThreadPoolExecutor getThreadPoolExecutor() {
086            return threadPoolExecutor;
087        }
088    
089        public MulticastType setThreadPoolExecutor(ThreadPoolExecutor executor) {
090            this.threadPoolExecutor = executor;
091            return this;
092        }
093    
094        @Override
095        protected Processor wrapProcessorInInterceptors(RouteContext routeContext, Processor target) throws Exception {
096            // No need to wrap me in interceptors as they are all applied directly to my children
097            return new StreamCachingInterceptor(target);
098        }
099    }