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.concurrent.LinkedBlockingQueue;
020    import java.util.concurrent.ThreadPoolExecutor;
021    import java.util.concurrent.TimeUnit;
022    
023    import javax.xml.bind.annotation.XmlAccessType;
024    import javax.xml.bind.annotation.XmlAccessorType;
025    import javax.xml.bind.annotation.XmlAttribute;
026    import javax.xml.bind.annotation.XmlRootElement;
027    import javax.xml.bind.annotation.XmlTransient;
028    
029    import org.apache.camel.Expression;
030    import org.apache.camel.Processor;
031    import org.apache.camel.model.language.ExpressionType;
032    import org.apache.camel.processor.Splitter;
033    import org.apache.camel.processor.aggregate.AggregationStrategy;
034    import org.apache.camel.processor.aggregate.UseLatestAggregationStrategy;
035    import org.apache.camel.spi.RouteContext;
036    
037    /**
038     * Represents an XML <splitter/> element
039     *
040     * @version $Revision: 61170 $
041     */
042    @XmlRootElement(name = "splitter")
043    @XmlAccessorType(XmlAccessType.FIELD)
044    public class SplitterType extends ExpressionNode {
045        @XmlTransient
046        private AggregationStrategy aggregationStrategy;
047        @XmlAttribute(required = false)
048        private Boolean parallelProcessing;
049        @XmlTransient
050        private ThreadPoolExecutor threadPoolExecutor;
051        @XmlAttribute(required = false)
052        private String threadPoolExecutorRef;
053        @XmlAttribute(required = false)
054        private Boolean streaming = false;
055        
056        public SplitterType() {
057        }
058    
059        public SplitterType(Expression expression) {
060            super(expression);
061        }
062    
063        public SplitterType(ExpressionType expression) {
064            super(expression);
065        }
066    
067        @Override
068        public String toString() {
069            return "Splitter[" + getExpression() + " -> " + getOutputs() + "]";
070        }
071    
072        @Override
073        public String getShortName() {
074            return "splitter";
075        }
076    
077        @Override
078        public Processor createProcessor(RouteContext routeContext) throws Exception {
079            Processor childProcessor = routeContext.createProcessor(this);
080            if (aggregationStrategy == null) {
081                aggregationStrategy = new UseLatestAggregationStrategy();
082            }
083            threadPoolExecutor = createThreadPoolExecutor(routeContext);
084            return new Splitter(getExpression().createExpression(routeContext), childProcessor, aggregationStrategy,
085                    isParallelProcessing(), threadPoolExecutor, streaming);
086        }
087        
088        public AggregationStrategy getAggregationStrategy() {
089            return aggregationStrategy;
090        }
091    
092        public void setAggregationStrategy(AggregationStrategy aggregationStrategy) {
093            this.aggregationStrategy = aggregationStrategy;
094        }
095        
096        public boolean isParallelProcessing() {
097            return parallelProcessing != null ? parallelProcessing : false;
098        }
099    
100        public void setParallelProcessing(boolean parallelProcessing) {
101            this.parallelProcessing = parallelProcessing;
102        }
103        
104        /**
105         * The splitter should use streaming -- exchanges are being sent as the data for them becomes available.
106         * This improves throughput and memory usage, but it has a drawback: 
107         * - the sent exchanges will no longer contain the {@link Splitter#SPLIT_SIZE} header property 
108         * 
109         * @return whether or not streaming should be used
110         */
111        public boolean isStreaming() {
112            return streaming != null ? streaming : false;
113        }
114    
115        public void setStreaming(boolean streaming) {
116            this.streaming = streaming;
117        }
118        
119        /**
120         * Enables streaming. 
121         * Cfr. {@link SplitterType#setStreaming(boolean)} for more information
122         */
123        public SplitterType streaming() {
124            setStreaming(true);
125            return this;
126        }
127    
128        private ThreadPoolExecutor createThreadPoolExecutor(RouteContext routeContext) {
129            ThreadPoolExecutor threadPoolExecutor = getThreadPoolExecutor();
130            if (threadPoolExecutor == null && threadPoolExecutorRef != null) {
131                threadPoolExecutor = routeContext.lookup(threadPoolExecutorRef, ThreadPoolExecutor.class);
132            }
133            if (threadPoolExecutor == null) {
134                // fall back and use default
135                threadPoolExecutor = new ThreadPoolExecutor(4, 16, 0L, TimeUnit.MILLISECONDS, new LinkedBlockingQueue());
136            }
137            return threadPoolExecutor;
138        }    
139       
140        public ThreadPoolExecutor getThreadPoolExecutor() {
141            return threadPoolExecutor;
142        }
143    
144        public void setThreadPoolExecutor(ThreadPoolExecutor threadPoolExecutor) {
145            this.threadPoolExecutor = threadPoolExecutor;
146        }
147    }