shutdown hook

Author: qq254963746

lts-admin/src/main/java/com/lts/admin/support/SystemInitListener.java

@@ -47,12 +47,12 @@ public void contextInitialized(ServletContextEvent servletContextEvent) {
             if (StringUtils.isEmpty(ltsMonitorCfgPath)) {
                 ltsMonitorCfgPath = this.getClass().getResource("/").getPath();
             }
-            MonitorAgentStartup.main(new String[]{ltsMonitorCfgPath});
+            MonitorAgentStartup.start(ltsMonitorCfgPath);
         }
     }
 
     @Override
     public void contextDestroyed(ServletContextEvent servletContextEvent) {
-
+        MonitorAgentStartup.stop();
     }
 }

lts-monitor/src/main/java/com/lts/monitor/MonitorAgentStartup.java

@@ -3,24 +3,30 @@
 import com.lts.core.commons.utils.StringUtils;
 
 import java.util.Map;
+import java.util.concurrent.atomic.AtomicBoolean;
 
 /**
  * @author Robert HG ([email protected]) on 3/5/16.
  */
 public class MonitorAgentStartup {
 
+    private final static MonitorAgent agent = new MonitorAgent();
+    private static final AtomicBoolean started = new AtomicBoolean(false);
+
     public static void main(String[] args) {
         String cfgPath = args[0];
         start(cfgPath);
     }
 
-    private static void start(String cfgPath) {
+    public static void start(String cfgPath) {
+
+        if (!started.compareAndSet(false, true)) {
+            return;
+        }
 
         try {
             MonitorCfg cfg = MonitorCfgLoader.load(cfgPath);
 
-            final MonitorAgent agent = new MonitorAgent();
-
             agent.setRegistryAddress(cfg.getRegistryAddress());
             agent.setClusterName(cfg.getClusterName());
             if (StringUtils.isNotEmpty(cfg.getBindIp())) {
@@ -46,4 +52,10 @@ public void run() {
         }
     }
 
+    public static void stop() {
+        if (started.compareAndSet(true, false)) {
+            agent.stop();
+        }
+    }
+
 }