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: 46926 $
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        @XmlAttribute(required = false)
046        private String strategyRef;
047        @XmlAttribute(required = false)
048        private String threadPoolRef;
049        @XmlTransient
050        private AggregationStrategy aggregationStrategy;
051        @XmlTransient
052        private ThreadPoolExecutor threadPoolExecutor;
053    
054        @Override
055        public String toString() {
056            return "Multicast[" + getOutputs() + "]";
057        }
058    
059        @Override
060        public String getShortName() {
061            return "multicast";
062        }
063    
064        @Override
065        public Processor createProcessor(RouteContext routeContext) throws Exception {
066            return createOutputsProcessor(routeContext);
067        }
068    
069        protected Processor createCompositeProcessor(RouteContext routeContext, List<Processor> list) {
070            if (aggregationStrategy == null && strategyRef != null) {
071                aggregationStrategy = routeContext.lookup(strategyRef, AggregationStrategy.class);
072            }
073            if (aggregationStrategy == null) {
074                aggregationStrategy = new UseLatestAggregationStrategy();
075            }
076            if (threadPoolRef != null) {
077                threadPoolExecutor = routeContext.lookup(threadPoolRef, ThreadPoolExecutor.class);
078            }
079            return new MulticastProcessor(list, aggregationStrategy, isParallelProcessing(), threadPoolExecutor);
080        }
081    
082        public AggregationStrategy getAggregationStrategy() {
083            return aggregationStrategy;
084        }
085    
086        public MulticastType setAggregationStrategy(AggregationStrategy aggregationStrategy) {
087            this.aggregationStrategy = aggregationStrategy;
088            return this;
089        }
090    
091        public boolean isParallelProcessing() {
092            return parallelProcessing != null ? parallelProcessing : false;
093        }
094    
095        public MulticastType setParallelProcessing(boolean parallelProcessing) {
096            this.parallelProcessing = parallelProcessing;
097            return this;
098        }
099    
100        public ThreadPoolExecutor getThreadPoolExecutor() {
101            return threadPoolExecutor;
102        }
103    
104        public MulticastType setThreadPoolExecutor(ThreadPoolExecutor executor) {
105            this.threadPoolExecutor = executor;
106            return this;
107        }
108    
109        @Override
110        protected Processor wrapProcessorInInterceptors(RouteContext routeContext, Processor target) throws Exception {
111            // No need to wrap me in interceptors as they are all applied directly to my children
112            return new StreamCachingInterceptor(target);
113        }
114    }