Merge branch '1.3.x'

spring-cloud-context/src/main/java/org/springframework/cloud/context/restart/RestartEndpoint.java

@@ -16,13 +16,18 @@
 
 package org.springframework.cloud.context.restart;
 
+import java.io.Closeable;
+import java.io.IOException;
+
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
+
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.actuate.endpoint.annotation.Endpoint;
 import org.springframework.boot.actuate.endpoint.annotation.WriteOperation;
 import org.springframework.boot.context.event.ApplicationPreparedEvent;
 import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.context.ApplicationContext;
 import org.springframework.context.ApplicationEvent;
 import org.springframework.context.ApplicationListener;
 import org.springframework.context.ConfigurableApplicationContext;
@@ -39,8 +44,7 @@ import org.springframework.util.ClassUtils;
  *
  */
 @Endpoint(id = "restart", enableByDefault = false)
-public class RestartEndpoint
-		implements ApplicationListener<ApplicationPreparedEvent> {
+public class RestartEndpoint implements ApplicationListener<ApplicationPreparedEvent> {
 
 	private static Log logger = LogFactory.getLog(RestartEndpoint.class);
 
@@ -82,7 +86,7 @@ public class RestartEndpoint
 	}
 
 	@WriteOperation
-	//FIXME: map with "message: Restarting" or couldn't restart
+	// FIXME: map with "message: Restarting" or couldn't restart
 	public Boolean restart() {
 		try {
 			doRestart();
@@ -142,7 +146,7 @@ public class RestartEndpoint
 				this.integrationShutdown.stop(this.timeout);
 			}
 			this.application.setEnvironment(this.context.getEnvironment());
-			this.context.close();
+			close();
 			// If running in a webapp then the context classloader is probably going to
 			// die so we need to revert to a safe place before starting again
 			overrideClassLoaderForRestart();
@@ -151,6 +155,19 @@ public class RestartEndpoint
 		return this.context;
 	}
 
+	private void close() {
+		ApplicationContext context = this.context;
+		while (context instanceof Closeable) {
+			try {
+				((Closeable) context).close();
+			}
+			catch (IOException e) {
+				logger.error("Cannot close context: " + context.getId(), e);
+			}
+			context = context.getParent();
+		}
+	}
+
 	// @ManagedAttribute
 	public boolean isRunning() {
 		if (this.context != null) {

spring-cloud-context/src/test/java/org/springframework/cloud/context/restart/RestartIntegrationTests.java

@@ -16,16 +16,19 @@
 
 package org.springframework.cloud.context.restart;
 
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertNotSame;
-import static org.junit.Assert.assertNull;
-
 import org.junit.After;
 import org.junit.Test;
+
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
 import org.springframework.context.ConfigurableApplicationContext;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.context.support.LiveBeansView;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNotSame;
+import static org.junit.Assert.assertNull;
 
 public class RestartIntegrationTests {
 
@@ -41,7 +44,10 @@ public class RestartIntegrationTests {
 	@Test
 	public void testRestartTwice() throws Exception {
 
-		context = SpringApplication.run(TestConfiguration.class, "--management.endpoint.restart.enabled=true", "--server.port=0");
+		context = SpringApplication.run(TestConfiguration.class,
+				"--management.endpoint.restart.enabled=true", "--server.port=0",
+				"--spring.liveBeansView.mbeanDomain=livebeans");
+
 		RestartEndpoint endpoint = context.getBean(RestartEndpoint.class);
 		assertNotNull(context.getParent());
 		assertNull(context.getParent().getParent());
@@ -59,6 +65,10 @@ public class RestartIntegrationTests {
 		assertNotNull(context.getParent());
 		assertNull(context.getParent().getParent());
 
+		LiveBeansView beans = new LiveBeansView();
+		String json = beans.getSnapshotAsJson();
+		assertThat(json).containsOnlyOnce("parent\": \"bootstrap");
+		assertThat(json).containsOnlyOnce("parent\": null");
 	}
 
 	public static void main(String[] args) {