Await TcpClient shutdown in STOMP relay

Author: rstoyanchev

spring-messaging/src/main/java/org/springframework/messaging/simp/broker/AbstractBrokerMessageHandler.java

@@ -128,10 +128,13 @@ public boolean isBrokerAvailable() {
 	public final void start() {
 		synchronized (this.lifecycleMonitor) {
 			if (logger.isDebugEnabled()) {
-				logger.debug("Starting " + getClass().getSimpleName());
+				logger.debug("Starting");
 			}
 			startInternal();
 			this.running = true;
+			if (logger.isDebugEnabled()) {
+				logger.debug("Started");
+			}
 		}
 	}
 
@@ -142,10 +145,13 @@ protected void startInternal() {
 	public final void stop() {
 		synchronized (this.lifecycleMonitor) {
 			if (logger.isDebugEnabled()) {
-				logger.debug("Stopping " + getClass().getSimpleName());
+				logger.debug("Stopping");
 			}
 			stopInternal();
 			this.running = false;
+			if (logger.isDebugEnabled()) {
+				logger.debug("Stopped");
+			}
 		}
 	}
 

spring-messaging/src/main/java/org/springframework/messaging/simp/stomp/StompBrokerRelayMessageHandler.java

@@ -20,6 +20,7 @@
 import java.util.Map;
 import java.util.concurrent.Callable;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.TimeUnit;
 
 import org.springframework.messaging.Message;
 import org.springframework.messaging.MessageChannel;
@@ -373,7 +374,7 @@ protected void stopInternal() {
 		this.brokerChannel.unsubscribe(this);
 
 		try {
-			this.tcpClient.shutdown();
+			this.tcpClient.shutdown().get(5000, TimeUnit.MILLISECONDS);
 		}
 		catch (Throwable t) {
 			logger.error("Error while shutting down TCP client", t);