Author: davsclaus
Date: Tue Nov 15 13:14:52 2011
New Revision: 1202167

URL: http://svn.apache.org/viewvc?rev=1202167&view=rev
Log:
CAMEL-4682: Improve restarting CamelContext to ensure lifecycle strategies is 
not lost.

Added:
    
camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextRestartTest.java
    
camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextWithLifecycleStrategyRestartTest.java
Modified:
    
camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultCamelContext.java

Modified: 
camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultCamelContext.java
URL: 
http://svn.apache.org/viewvc/camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultCamelContext.java?rev=1202167&r1=1202166&r2=1202167&view=diff
==============================================================================
--- 
camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultCamelContext.java
 (original)
+++ 
camel/trunk/camel-core/src/main/java/org/apache/camel/impl/DefaultCamelContext.java
 Tue Nov 15 13:14:52 2011
@@ -1560,7 +1560,7 @@ public class DefaultCamelContext extends
         // shutdown management as the last one
         shutdownServices(managementStrategy);
         shutdownServices(lifecycleStrategies);
-        lifecycleStrategies.clear();
+        // do not clear lifecycleStrategies as we can start Camel again and 
get the route back as before
 
         // stop the lazy created so they can be re-created on restart
         forceStopLazyInitialization();

Added: 
camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextRestartTest.java
URL: 
http://svn.apache.org/viewvc/camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextRestartTest.java?rev=1202167&view=auto
==============================================================================
--- 
camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextRestartTest.java
 (added)
+++ 
camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextRestartTest.java
 Tue Nov 15 13:14:52 2011
@@ -0,0 +1,67 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.camel.impl;
+
+import org.apache.camel.ContextTestSupport;
+import org.apache.camel.builder.RouteBuilder;
+
+/**
+ *
+ */
+public class DefaultCamelContextRestartTest extends ContextTestSupport {
+
+    public void testRestart() throws Exception {
+        assertTrue(context.getStatus().isStarted());
+        assertFalse(context.getStatus().isStopped());
+        assertEquals(1, context.getRoutes().size());
+
+        getMockEndpoint("mock:result").expectedMessageCount(1);
+        template.sendBody("direct:start", "Hello World");
+        assertMockEndpointsSatisfied();
+
+        // now stop
+        context.stop();
+        assertFalse(context.getStatus().isStarted());
+        assertTrue(context.getStatus().isStopped());
+        assertEquals(0, context.getRoutes().size());
+
+        // now start
+        context.start();
+        assertTrue(context.getStatus().isStarted());
+        assertFalse(context.getStatus().isStopped());
+        assertEquals(1, context.getRoutes().size());
+
+        // must obtain a new template
+        template = context.createProducerTemplate();
+
+        // should still work
+        getMockEndpoint("mock:result").expectedMessageCount(1);
+        template.sendBody("direct:start", "Bye World");
+        assertMockEndpointsSatisfied();
+    }
+
+    @Override
+    protected RouteBuilder createRouteBuilder() throws Exception {
+        return new RouteBuilder() {
+            @Override
+            public void configure() throws Exception {
+                from("direct:start").routeId("foo")
+                    .to("mock:result");
+            }
+        };
+    }
+}

Added: 
camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextWithLifecycleStrategyRestartTest.java
URL: 
http://svn.apache.org/viewvc/camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextWithLifecycleStrategyRestartTest.java?rev=1202167&view=auto
==============================================================================
--- 
camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextWithLifecycleStrategyRestartTest.java
 (added)
+++ 
camel/trunk/camel-core/src/test/java/org/apache/camel/impl/DefaultCamelContextWithLifecycleStrategyRestartTest.java
 Tue Nov 15 13:14:52 2011
@@ -0,0 +1,155 @@
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.camel.impl;
+
+import java.util.Collection;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import org.apache.camel.CamelContext;
+import org.apache.camel.Component;
+import org.apache.camel.ContextTestSupport;
+import org.apache.camel.Endpoint;
+import org.apache.camel.ErrorHandlerFactory;
+import org.apache.camel.Processor;
+import org.apache.camel.Route;
+import org.apache.camel.Service;
+import org.apache.camel.VetoCamelContextStartException;
+import org.apache.camel.builder.RouteBuilder;
+import org.apache.camel.spi.LifecycleStrategy;
+import org.apache.camel.spi.RouteContext;
+
+/**
+ *
+ */
+public class DefaultCamelContextWithLifecycleStrategyRestartTest extends 
ContextTestSupport {
+
+    private MyStrategy strategy = new MyStrategy();
+
+    public void testRestart() throws Exception {
+        assertTrue(context.getStatus().isStarted());
+        assertFalse(context.getStatus().isStopped());
+        assertEquals(1, context.getRoutes().size());
+        assertEquals(1, strategy.getCounter().get());
+
+        getMockEndpoint("mock:result").expectedMessageCount(1);
+        template.sendBody("direct:start", "Hello World");
+        assertMockEndpointsSatisfied();
+
+        // now stop
+        context.stop();
+        assertFalse(context.getStatus().isStarted());
+        assertTrue(context.getStatus().isStopped());
+        assertEquals(0, context.getRoutes().size());
+
+        // now start
+        context.start();
+        assertTrue(context.getStatus().isStarted());
+        assertFalse(context.getStatus().isStopped());
+        assertEquals(1, context.getRoutes().size());
+        assertEquals(2, strategy.getCounter().get());
+
+        // must obtain a new template
+        template = context.createProducerTemplate();
+
+        // should still work
+        getMockEndpoint("mock:result").expectedMessageCount(1);
+        template.sendBody("direct:start", "Bye World");
+        assertMockEndpointsSatisfied();
+    }
+
+    @Override
+    protected CamelContext createCamelContext() throws Exception {
+        CamelContext context = super.createCamelContext();
+        context.addLifecycleStrategy(strategy);
+        return context;
+    }
+
+    @Override
+    protected RouteBuilder createRouteBuilder() throws Exception {
+        return new RouteBuilder() {
+            @Override
+            public void configure() throws Exception {
+                from("direct:start").routeId("foo")
+                    .to("mock:result");
+            }
+        };
+    }
+
+    private class MyStrategy implements LifecycleStrategy {
+
+        private AtomicInteger counter = new AtomicInteger();
+
+        @Override
+        public void onContextStart(CamelContext context) throws 
VetoCamelContextStartException {
+            counter.incrementAndGet();
+        }
+
+        @Override
+        public void onContextStop(CamelContext context) {
+        }
+
+        @Override
+        public void onComponentAdd(String name, Component component) {
+        }
+
+        @Override
+        public void onComponentRemove(String name, Component component) {
+        }
+
+        @Override
+        public void onEndpointAdd(Endpoint endpoint) {
+        }
+
+        @Override
+        public void onEndpointRemove(Endpoint endpoint) {
+        }
+
+        @Override
+        public void onServiceAdd(CamelContext context, Service service, Route 
route) {
+        }
+
+        @Override
+        public void onServiceRemove(CamelContext context, Service service, 
Route route) {
+        }
+
+        @Override
+        public void onRoutesAdd(Collection<Route> routes) {
+        }
+
+        @Override
+        public void onRoutesRemove(Collection<Route> routes) {
+        }
+
+        @Override
+        public void onRouteContextCreate(RouteContext routeContext) {
+        }
+
+        @Override
+        public void onErrorHandlerAdd(RouteContext routeContext, Processor 
errorHandler, ErrorHandlerFactory errorHandlerBuilder) {
+        }
+
+        @Override
+        public void onThreadPoolAdd(CamelContext camelContext, 
ThreadPoolExecutor threadPool, String id, String sourceId, String routeId, 
String threadPoolProfileId) {
+        }
+
+        public AtomicInteger getCounter() {
+            return counter;
+        }
+    }
+}


Reply via email to