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; 018 019 import java.util.HashMap; 020 import java.util.LinkedList; 021 import java.util.Map; 022 import java.util.Set; 023 024 import javax.xml.bind.JAXBException; 025 026 import org.apache.camel.CamelContext; 027 import org.apache.camel.ProducerTemplate; 028 import org.apache.camel.spring.handler.CamelNamespaceHandler; 029 import org.apache.camel.util.MainSupport; 030 import org.apache.camel.view.ModelFileGenerator; 031 import org.apache.commons.logging.Log; 032 import org.apache.commons.logging.LogFactory; 033 import org.springframework.context.ApplicationContext; 034 import org.springframework.context.support.AbstractApplicationContext; 035 import org.springframework.context.support.ClassPathXmlApplicationContext; 036 import org.springframework.context.support.FileSystemXmlApplicationContext; 037 038 /** 039 * A command line tool for booting up a CamelContext using an optional Spring 040 * ApplicationContext 041 * 042 * @version $Revision: 66961 $ 043 */ 044 public class Main extends MainSupport { 045 private static Main instance; 046 047 private String applicationContextUri = "META-INF/spring/*.xml"; 048 private String fileApplicationContextUri; 049 private AbstractApplicationContext applicationContext; 050 private AbstractApplicationContext parentApplicationContext; 051 private String parentApplicationContextUri; 052 053 public Main() { 054 055 addOption(new ParameterOption("ac", "applicationContext", 056 "Sets the classpath based spring ApplicationContext", "applicationContext") { 057 protected void doProcess(String arg, String parameter, LinkedList<String> remainingArgs) { 058 setApplicationContextUri(parameter); 059 } 060 }); 061 062 addOption(new ParameterOption("fa", "fileApplicationContext", 063 "Sets the filesystem based spring ApplicationContext", "fileApplicationContext") { 064 protected void doProcess(String arg, String parameter, LinkedList<String> remainingArgs) { 065 setFileApplicationContextUri(parameter); 066 } 067 }); 068 069 } 070 071 /** 072 * A class for intercepting the hang up signal and do a graceful shutdown of 073 * the Camel. 074 */ 075 private class HangupInterceptor extends Thread { 076 Log log = LogFactory.getLog(this.getClass()); 077 Main mainInstance; 078 079 public HangupInterceptor(Main main) { 080 mainInstance = main; 081 } 082 083 @Override 084 public void run() { 085 log.info("Recieved hang up - stopping the main instance."); 086 try { 087 mainInstance.stop(); 088 } catch (Exception ex) { 089 log.warn(ex); 090 } 091 } 092 } 093 094 095 public static void main(String... args) { 096 Main main = new Main(); 097 instance = main; 098 main.enableHangupSupport(); 099 main.run(args); 100 } 101 102 /** 103 * Returns the currently executing main 104 * 105 * @return the current running instance 106 */ 107 public static Main getInstance() { 108 return instance; 109 } 110 111 /** 112 * Enables the hangup support. Gracefully stops by calling stop() on a 113 * Hangup signal. 114 */ 115 public void enableHangupSupport() { 116 HangupInterceptor interceptor = new HangupInterceptor(this); 117 Runtime.getRuntime().addShutdownHook(interceptor); 118 } 119 120 @Override 121 public void enableDebug() { 122 super.enableDebug(); 123 setParentApplicationContextUri("/META-INF/services/org/apache/camel/spring/debug.xml"); 124 } 125 126 @Override 127 public void enableTrace() { 128 // TODO 129 super.enableTrace(); 130 setParentApplicationContextUri("/META-INF/services/org/apache/camel/spring/trace.xml"); 131 } 132 133 // Properties 134 // ------------------------------------------------------------------------- 135 public AbstractApplicationContext getApplicationContext() { 136 return applicationContext; 137 } 138 139 public void setApplicationContext(AbstractApplicationContext applicationContext) { 140 this.applicationContext = applicationContext; 141 } 142 143 public String getApplicationContextUri() { 144 return applicationContextUri; 145 } 146 147 public void setApplicationContextUri(String applicationContextUri) { 148 this.applicationContextUri = applicationContextUri; 149 } 150 151 public String getFileApplicationContextUri() { 152 return fileApplicationContextUri; 153 } 154 155 public void setFileApplicationContextUri(String fileApplicationContextUri) { 156 this.fileApplicationContextUri = fileApplicationContextUri; 157 } 158 159 public AbstractApplicationContext getParentApplicationContext() { 160 if (parentApplicationContext == null) { 161 if (parentApplicationContextUri != null) { 162 parentApplicationContext = new ClassPathXmlApplicationContext(parentApplicationContextUri); 163 parentApplicationContext.start(); 164 } 165 } 166 return parentApplicationContext; 167 } 168 169 public void setParentApplicationContext(AbstractApplicationContext parentApplicationContext) { 170 this.parentApplicationContext = parentApplicationContext; 171 } 172 173 public String getParentApplicationContextUri() { 174 return parentApplicationContextUri; 175 } 176 177 public void setParentApplicationContextUri(String parentApplicationContextUri) { 178 this.parentApplicationContextUri = parentApplicationContextUri; 179 } 180 181 // Implementation methods 182 // ------------------------------------------------------------------------- 183 184 @Override 185 protected void doStart() throws Exception { 186 super.doStart(); 187 if (applicationContext == null) { 188 applicationContext = createDefaultApplicationContext(); 189 } 190 applicationContext.start(); 191 192 postProcessContext(); 193 } 194 195 protected ProducerTemplate findOrCreateCamelTemplate() { 196 String[] names = getApplicationContext().getBeanNamesForType(ProducerTemplate.class); 197 if (names != null && names.length > 0) { 198 return (ProducerTemplate) getApplicationContext().getBean(names[0], ProducerTemplate.class); 199 } 200 for (CamelContext camelContext : getCamelContexts()) { 201 return camelContext.createProducerTemplate(); 202 } 203 throw new IllegalArgumentException("No CamelContexts are available so cannot create a ProducerTemplate!"); 204 } 205 206 protected AbstractApplicationContext createDefaultApplicationContext() { 207 // file based 208 if (getFileApplicationContextUri() != null) { 209 String[] args = getFileApplicationContextUri().split(";"); 210 211 ApplicationContext parentContext = getParentApplicationContext(); 212 if (parentContext != null) { 213 return new FileSystemXmlApplicationContext(args, parentContext); 214 } else { 215 return new FileSystemXmlApplicationContext(args); 216 } 217 } 218 219 // default to classpath based 220 String[] args = getApplicationContextUri().split(";"); 221 ApplicationContext parentContext = getParentApplicationContext(); 222 if (parentContext != null) { 223 return new ClassPathXmlApplicationContext(args, parentContext); 224 } else { 225 return new ClassPathXmlApplicationContext(args); 226 } 227 } 228 229 protected void doStop() throws Exception { 230 LOG.info("Apache Camel terminating"); 231 232 if (applicationContext != null) { 233 applicationContext.close(); 234 } 235 } 236 237 protected Map<String, CamelContext> getCamelContextMap() { 238 Map<String, SpringCamelContext> map = applicationContext.getBeansOfType(SpringCamelContext.class); 239 Set<Map.Entry<String, SpringCamelContext>> entries = map.entrySet(); 240 Map<String, CamelContext> answer = new HashMap<String, CamelContext>(); 241 for (Map.Entry<String, SpringCamelContext> entry : entries) { 242 String name = entry.getKey(); 243 CamelContext camelContext = entry.getValue(); 244 answer.put(name, camelContext); 245 } 246 return answer; 247 } 248 249 protected ModelFileGenerator createModelFileGenerator() throws JAXBException { 250 return new ModelFileGenerator(new CamelNamespaceHandler().getJaxbContext()); 251 } 252 }