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.servicemix.eip.packaging;
018    
019    import java.util.ArrayList;
020    import java.util.List;
021    
022    import org.apache.servicemix.common.Endpoint;
023    import org.apache.servicemix.common.packaging.Consumes;
024    import org.apache.servicemix.common.xbean.AbstractXBeanServiceUnitAnalyzer;
025    import org.apache.servicemix.eip.EIPEndpoint;
026    import org.apache.servicemix.eip.patterns.ContentBasedRouter;
027    import org.apache.servicemix.eip.patterns.MessageFilter;
028    import org.apache.servicemix.eip.patterns.Pipeline;
029    import org.apache.servicemix.eip.patterns.SplitAggregator;
030    import org.apache.servicemix.eip.patterns.StaticRecipientList;
031    import org.apache.servicemix.eip.patterns.StaticRoutingSlip;
032    import org.apache.servicemix.eip.patterns.WireTap;
033    import org.apache.servicemix.eip.patterns.XPathSplitter;
034    import org.apache.servicemix.eip.support.ExchangeTarget;
035    import org.apache.servicemix.eip.support.RoutingRule;
036    
037    public class EipServiceUnitAnalyzer extends AbstractXBeanServiceUnitAnalyzer {
038    
039        protected List<Consumes> getConsumes(Endpoint endpoint) {
040            List<Consumes> consumes;
041            if (endpoint instanceof ContentBasedRouter) {
042                consumes = resolveContentBasedRouter((ContentBasedRouter) endpoint);
043            } else if (endpoint instanceof MessageFilter) {
044                consumes = resolveMessageFilter((MessageFilter) endpoint);
045            } else if (endpoint instanceof Pipeline) {
046                consumes = resolvePipeline((Pipeline) endpoint);
047            } else if (endpoint instanceof SplitAggregator) {
048                consumes = resolveSplitAggregator((SplitAggregator) endpoint);
049            } else if (endpoint instanceof StaticRecipientList) {
050                consumes = resolveStaticRecipientList((StaticRecipientList) endpoint);
051            } else if (endpoint instanceof StaticRoutingSlip) {
052                consumes = resolveStaticRoutingSlip((StaticRoutingSlip) endpoint);
053            } else if (endpoint instanceof WireTap) {
054                consumes = resolveWireTap((WireTap) endpoint);
055            } else if (endpoint instanceof XPathSplitter) {
056                consumes = resolveXPathSplitter((XPathSplitter) endpoint);
057            } else {
058                consumes = new ArrayList<Consumes>();
059            }
060            return consumes;
061        }
062    
063        private List<Consumes> resolveXPathSplitter(XPathSplitter splitter) {
064            return generateConsumesFromTarget(splitter.getTarget(), new ArrayList<Consumes>());
065        }
066    
067        private List<Consumes> resolveWireTap(WireTap tap) {
068            List<Consumes> consumes = new ArrayList<Consumes>();
069            consumes = generateConsumesFromTarget(tap.getTarget(), consumes);
070            consumes = generateConsumesFromTarget(tap.getInListener(), consumes);
071            consumes = generateConsumesFromTarget(tap.getOutListener(), consumes);
072            consumes = generateConsumesFromTarget(tap.getFaultListener(), consumes);
073            return consumes;
074        }
075    
076        private List<Consumes> resolveStaticRoutingSlip(StaticRoutingSlip slip) {
077            List<Consumes> consumes = new ArrayList<Consumes>();
078            for (int i = 0; i < slip.getTargets().length; i++) {
079                consumes = generateConsumesFromTarget(slip.getTargets()[i], consumes);
080            }
081            return consumes;
082        }
083    
084        private List<Consumes> resolveStaticRecipientList(StaticRecipientList list) {
085            List<Consumes> consumes = new ArrayList<Consumes>();
086            for (int i = 0; i < list.getRecipients().length; i++) {
087                consumes = generateConsumesFromTarget(list.getRecipients()[i], consumes);
088            }
089            return consumes;
090        }
091    
092        private List<Consumes> resolveSplitAggregator(SplitAggregator aggregator) {
093            return generateConsumesFromTarget(aggregator.getTarget(), new ArrayList<Consumes>());
094        }
095    
096        private List<Consumes> resolvePipeline(Pipeline pipeline) {
097            List<Consumes> consumes = generateConsumesFromTarget(pipeline.getTarget(), new ArrayList<Consumes>());
098            consumes = generateConsumesFromTarget(pipeline.getTransformer(), consumes);
099            return consumes;
100        }
101    
102        private List<Consumes> resolveMessageFilter(MessageFilter filter) {
103            return generateConsumesFromTarget(filter.getTarget(), new ArrayList<Consumes>());
104        }
105    
106        private List<Consumes> resolveContentBasedRouter(ContentBasedRouter router) {
107            List<Consumes> consumes = new ArrayList<Consumes>();
108            for (int i = 0; i < router.getRules().length; i++) {
109                RoutingRule rule = router.getRules()[i];
110                consumes = generateConsumesFromTarget(rule.getTarget(), consumes);
111            }
112            return consumes;
113        }
114    
115        private List<Consumes> generateConsumesFromTarget(ExchangeTarget target, List<Consumes> consumes) {
116            if (target != null) {
117                Consumes consume = new Consumes();
118                consume.setEndpointName(target.getEndpoint());
119                consume.setServiceName(target.getService());
120                consume.setInterfaceName(target.getInterface());
121                consumes.add(consume);
122            }
123            return consumes;
124        }
125    
126        protected String getXBeanFile() {
127            return "xbean.xml";
128        }
129    
130        protected boolean isValidEndpoint(Object bean) {
131            return bean instanceof EIPEndpoint;
132        }
133    
134    }