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.spring.handler;
018    
019    import java.util.HashMap;
020    import java.util.HashSet;
021    import java.util.Map;
022    import java.util.Set;
023    
024    import javax.xml.bind.Binder;
025    import javax.xml.bind.JAXBContext;
026    import javax.xml.bind.JAXBException;
027    import org.w3c.dom.Element;
028    import org.w3c.dom.Node;
029    import org.w3c.dom.NodeList;
030    
031    import org.apache.camel.builder.xml.Namespaces;
032    import org.apache.camel.model.FromDefinition;
033    import org.apache.camel.model.SendDefinition;
034    import org.apache.camel.spi.NamespaceAware;
035    import org.apache.camel.spring.CamelBeanPostProcessor;
036    import org.apache.camel.spring.CamelConsumerTemplateFactoryBean;
037    import org.apache.camel.spring.CamelContextFactoryBean;
038    import org.apache.camel.spring.CamelEndpointFactoryBean;
039    import org.apache.camel.spring.CamelJMXAgentDefinition;
040    import org.apache.camel.spring.CamelProducerTemplateFactoryBean;
041    import org.apache.camel.spring.remoting.CamelProxyFactoryBean;
042    import org.apache.camel.spring.remoting.CamelServiceExporter;
043    import org.apache.camel.util.ObjectHelper;
044    import org.apache.camel.view.ModelFileGenerator;
045    import org.apache.commons.logging.Log;
046    import org.apache.commons.logging.LogFactory;
047    import org.springframework.beans.factory.BeanDefinitionStoreException;
048    import org.springframework.beans.factory.config.BeanDefinition;
049    import org.springframework.beans.factory.config.RuntimeBeanReference;
050    import org.springframework.beans.factory.parsing.BeanComponentDefinition;
051    import org.springframework.beans.factory.support.BeanDefinitionBuilder;
052    import org.springframework.beans.factory.xml.NamespaceHandlerSupport;
053    import org.springframework.beans.factory.xml.ParserContext;
054    
055    /**
056     * Camel namespace for the spring XML configuration file.
057     */
058    public class CamelNamespaceHandler extends NamespaceHandlerSupport {
059    
060        private static final Log LOG = LogFactory.getLog(CamelNamespaceHandler.class);
061        protected BeanDefinitionParser endpointParser = new BeanDefinitionParser(CamelEndpointFactoryBean.class);
062        protected BeanDefinitionParser beanPostProcessorParser = new BeanDefinitionParser(CamelBeanPostProcessor.class);
063        protected Set<String> parserElementNames = new HashSet<String>();
064        protected Binder<Node> binder;
065        private JAXBContext jaxbContext;
066        private Map<String, BeanDefinitionParser> parserMap = new HashMap<String, BeanDefinitionParser>();
067    
068        public ModelFileGenerator createModelFileGenerator() throws JAXBException {
069            return new ModelFileGenerator(getJaxbContext());
070        }
071    
072        public void init() {
073            // remoting
074            addBeanDefinitionParser("proxy", CamelProxyFactoryBean.class);
075            addBeanDefinitionParser("template", CamelProducerTemplateFactoryBean.class);
076            addBeanDefinitionParser("consumerTemplate", CamelConsumerTemplateFactoryBean.class);
077            addBeanDefinitionParser("export", CamelServiceExporter.class);
078           
079            // jmx agent
080            addBeanDefinitionParser("jmxAgent", CamelJMXAgentDefinition.class);
081    
082            // endpoint
083            addBeanDefinitionParser("endpoint", CamelEndpointFactoryBean.class);
084    
085            // camel context
086            boolean osgi = false;
087            Class cl = CamelContextFactoryBean.class;
088            try {
089                cl = Class.forName("org.apache.camel.osgi.CamelContextFactoryBean");
090                osgi = true;
091            } catch (Throwable t) {
092                // not running with camel-osgi so we fallback to the regular factory bean
093                LOG.trace("Cannot find class so assuming not running in OSGI container: " + t.getMessage());
094            }
095    
096            if (osgi) {
097                LOG.info("camel-osgi.jar detected in classpath");
098            } else {
099                LOG.info("camel-osgi.jar not detected in classpath");
100            }
101    
102            if (LOG.isDebugEnabled()) {
103                LOG.debug("Using " + cl.getCanonicalName() + " as CamelContextBeanDefinitionParser");
104            }
105            registerParser("camelContext", new CamelContextBeanDefinitionParser(cl));
106        }
107    
108        private void addBeanDefinitionParser(String elementName, Class<?> type) {
109            BeanDefinitionParser parser = new BeanDefinitionParser(type);
110            registerParser(elementName, parser);
111            parserMap.put(elementName, parser);
112        }
113    
114        protected void createBeanPostProcessor(ParserContext parserContext, String contextId, Element childElement, BeanDefinitionBuilder parentBuilder) {
115            String beanPostProcessorId = contextId + ":beanPostProcessor";
116            childElement.setAttribute("id", beanPostProcessorId);
117            BeanDefinition definition = beanPostProcessorParser.parse(childElement, parserContext);
118            // only register to camel context id as a String. Then we can look it up later
119            // otherwise we get a circular reference in spring and it will not allow custom bean post processing
120            // see more at CAMEL-1663
121            definition.getPropertyValues().addPropertyValue("camelId", contextId);
122            parentBuilder.addPropertyReference("beanPostProcessor", beanPostProcessorId);
123        }
124    
125        protected void registerScriptParser(String elementName, String engineName) {
126            registerParser(elementName, new ScriptDefinitionParser(engineName));
127        }
128    
129        protected void registerParser(String name, org.springframework.beans.factory.xml.BeanDefinitionParser parser) {
130            parserElementNames.add(name);
131            registerBeanDefinitionParser(name, parser);
132        }
133    
134        public Set<String> getParserElementNames() {
135            return parserElementNames;
136        }
137    
138        protected Object parseUsingJaxb(Element element, ParserContext parserContext) {
139            try {
140                binder = getJaxbContext().createBinder();
141                return binder.unmarshal(element);
142            } catch (JAXBException e) {
143                throw new BeanDefinitionStoreException("Failed to parse JAXB element: " + e, e);
144            }
145        }
146    
147        public JAXBContext getJaxbContext() throws JAXBException {
148            if (jaxbContext == null) {
149                jaxbContext = createJaxbContext();
150            }
151            return jaxbContext;
152        }
153    
154        protected JAXBContext createJaxbContext() throws JAXBException {
155            StringBuilder packages = new StringBuilder();
156            for (Class cl : getJaxbPackages()) {
157                if (packages.length() > 0) {
158                    packages.append(":");
159                }
160                packages.append(cl.getName().substring(0, cl.getName().lastIndexOf('.')));
161            }
162            return JAXBContext.newInstance(packages.toString(), getClass().getClassLoader());
163        }
164    
165        protected Set<Class> getJaxbPackages() {
166            Set<Class> classes = new HashSet<Class>();
167            classes.add(org.apache.camel.spring.CamelContextFactoryBean.class);
168            classes.add(org.apache.camel.ExchangePattern.class);
169            classes.add(org.apache.camel.model.RouteDefinition.class);
170            classes.add(org.apache.camel.model.config.StreamResequencerConfig.class);
171            classes.add(org.apache.camel.model.dataformat.DataFormatsDefinition.class);
172            classes.add(org.apache.camel.model.language.ExpressionDefinition.class);
173            classes.add(org.apache.camel.model.loadbalancer.RoundRobinLoadBalancerDefinition.class);
174            return classes;
175        }
176    
177        protected class CamelContextBeanDefinitionParser extends BeanDefinitionParser {
178            public CamelContextBeanDefinitionParser(Class type) {
179                super(type);
180            }
181    
182            @Override
183            protected void doParse(Element element, ParserContext parserContext, BeanDefinitionBuilder builder) {
184                super.doParse(element, parserContext, builder);
185    
186                String contextId = element.getAttribute("id");
187    
188                // lets avoid folks having to explicitly give an ID to a camel context
189                if (ObjectHelper.isEmpty(contextId)) {
190                    contextId = "camelContext";
191                    element.setAttribute("id", contextId);
192                }
193    
194                // now lets parse the routes with JAXB
195                Object value = parseUsingJaxb(element, parserContext);
196                
197                if (value instanceof CamelContextFactoryBean) {
198                    // set the property value with the JAXB parsed value
199                    CamelContextFactoryBean factoryBean = (CamelContextFactoryBean)value;
200                    builder.addPropertyValue("id", contextId);
201                    builder.addPropertyValue("routes", factoryBean.getRoutes());
202                    builder.addPropertyValue("intercepts", factoryBean.getIntercepts());
203                    builder.addPropertyValue("interceptFroms", factoryBean.getInterceptFroms());
204                    builder.addPropertyValue("interceptSendToEndpoints", factoryBean.getInterceptSendToEndpoints());
205                    builder.addPropertyValue("dataFormats", factoryBean.getDataFormats());
206                    builder.addPropertyValue("onCompletions", factoryBean.getOnCompletions());
207                    builder.addPropertyValue("onExceptions", factoryBean.getOnExceptions());
208                    builder.addPropertyValue("builderRefs", factoryBean.getBuilderRefs());
209                    builder.addPropertyValue("properties", factoryBean.getProperties());
210                    builder.addPropertyValue("packageScan", factoryBean.getPackageScan());
211                    if (factoryBean.getPackages().length > 0) {
212                        builder.addPropertyValue("packages", factoryBean.getPackages());
213                    }
214                }
215    
216                boolean createdBeanPostProcessor = false;
217                NodeList list = element.getChildNodes();
218                int size = list.getLength();
219                for (int i = 0; i < size; i++) {
220                    Node child = list.item(i);
221                    if (child instanceof Element) {
222                        Element childElement = (Element)child;
223                        String localName = child.getLocalName();
224                        if (localName.equals("beanPostProcessor")) {
225                            createBeanPostProcessor(parserContext, contextId, childElement, builder);
226                            createdBeanPostProcessor = true;
227                        } else if (localName.equals("endpoint")) {
228                            registerEndpoint(childElement, parserContext, contextId);
229                        } else {
230                            BeanDefinitionParser parser = parserMap.get(localName);
231                            if (parser != null) {
232                                BeanDefinition definition = parser.parse(childElement, parserContext);
233                                String id = childElement.getAttribute("id");
234                                if (ObjectHelper.isNotEmpty(id)) {
235                                    parserContext.registerComponent(new BeanComponentDefinition(definition, id));
236                                    if (localName.equals("jmxAgent")) {
237                                        builder.addPropertyReference("camelJMXAgent", id);
238                                    }
239                                }
240                            }
241    
242                        }
243                    }
244                }
245    
246                // register as endpoint defined indirectly in the routes by from/to types having id explict set
247                registerEndpointsWithIdsDefinedInFromOrToTypes(element, parserContext, contextId);
248    
249                // lets inject the namespaces into any namespace aware POJOs
250                injectNamespaces(element);
251                if (!createdBeanPostProcessor) {
252                    // no bean processor element so lets create it by ourself
253                    Element childElement = element.getOwnerDocument().createElement("beanPostProcessor");
254                    element.appendChild(childElement);
255                    createBeanPostProcessor(parserContext, contextId, childElement, builder);
256                }
257            }
258        }
259    
260        protected void injectNamespaces(Element element) {
261            NodeList list = element.getChildNodes();
262            Namespaces namespaces = null;
263            int size = list.getLength();
264            for (int i = 0; i < size; i++) {
265                Node child = list.item(i);
266                if (child instanceof Element) {
267                    Element childElement = (Element)child;
268                    Object object = binder.getJAXBNode(child);
269                    if (object instanceof NamespaceAware) {
270                        NamespaceAware namespaceAware = (NamespaceAware)object;
271                        if (namespaces == null) {
272                            namespaces = new Namespaces(element);
273                        }
274                        namespaces.configure(namespaceAware);
275                    }
276                    injectNamespaces(childElement);
277                }
278            }
279        }
280    
281        /**
282         * Uses for auto registering endpoints from the <tt>from</tt> or <tt>to</tt> DSL if they have an id attribute set
283         */
284        protected void registerEndpointsWithIdsDefinedInFromOrToTypes(Element element, ParserContext parserContext, String contextId) {
285            NodeList list = element.getChildNodes();
286            int size = list.getLength();
287            for (int i = 0; i < size; i++) {
288                Node child = list.item(i);
289                if (child instanceof Element) {
290                    Element childElement = (Element)child;
291                    Object object = binder.getJAXBNode(child);
292                    // we only want from/to types to be registered as endpoints
293                    if (object instanceof FromDefinition || object instanceof SendDefinition) {
294                        registerEndpoint(childElement, parserContext, contextId);
295                    }
296                    // recursive
297                    registerEndpointsWithIdsDefinedInFromOrToTypes(childElement, parserContext, contextId);
298                }
299            }
300        }
301    
302        private void registerEndpoint(Element childElement, ParserContext parserContext, String contextId) {
303            String id = childElement.getAttribute("id");
304            // must have an id to be registered
305            if (ObjectHelper.isNotEmpty(id)) {
306                BeanDefinition definition = endpointParser.parse(childElement, parserContext);
307                definition.getPropertyValues().addPropertyValue("camelContext", new RuntimeBeanReference(contextId));
308                parserContext.registerComponent(new BeanComponentDefinition(definition, id));
309            }
310        }
311        
312    }