增加JVM退出钩子处理.

baomidou · Feb 10, 2025 · 62dd02e · 62dd02e
1 parent c4091c9
commit 62dd02e
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 0 deletions.
diff --git a/...lparser-4.9/src/main/java/com/baomidou/mybatisplus/extension/parser/JsqlParserGlobal.java b/...lparser-4.9/src/main/java/com/baomidou/mybatisplus/extension/parser/JsqlParserGlobal.java
@@ -21,6 +21,8 @@
 import net.sf.jsqlparser.parser.CCJSqlParserUtil;
 import net.sf.jsqlparser.statement.Statement;
 import net.sf.jsqlparser.statement.Statements;
+import org.apache.ibatis.logging.Log;
+import org.apache.ibatis.logging.LogFactory;
 
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.LinkedBlockingQueue;
@@ -33,6 +35,8 @@
  */
 public class JsqlParserGlobal {
 
+    private static final Log LOG = LogFactory.getLog(JsqlParserGlobal.class);
+
     /**
      * 默认线程数大小
      *
@@ -54,6 +58,17 @@ public class JsqlParserGlobal {
         return thread;
     });
 
+    static {
+        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
+            if (!executorService.isShutdown()) {
+                if (LOG.isDebugEnabled()) {
+                    LOG.debug("close jsqlParser thread pool ...");
+                }
+                executorService.shutdown();
+            }
+        }, "mybatis-plus-jsqlParser-shutdown-hook"));
+    }
+
     @Setter
     private static JsqlParserFunction<String, Statement> parserSingleFunc = sql -> CCJSqlParserUtil.parse(sql, executorService, null);
 

diff --git a/...lparser-5.0/src/main/java/com/baomidou/mybatisplus/extension/parser/JsqlParserGlobal.java b/...lparser-5.0/src/main/java/com/baomidou/mybatisplus/extension/parser/JsqlParserGlobal.java
@@ -21,6 +21,8 @@
 import net.sf.jsqlparser.parser.CCJSqlParserUtil;
 import net.sf.jsqlparser.statement.Statement;
 import net.sf.jsqlparser.statement.Statements;
+import org.apache.ibatis.logging.Log;
+import org.apache.ibatis.logging.LogFactory;
 
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.LinkedBlockingQueue;
@@ -33,6 +35,8 @@
  */
 public class JsqlParserGlobal {
 
+    private static final Log LOG = LogFactory.getLog(JsqlParserGlobal.class);
+
     /**
      * 默认线程数大小
      *
@@ -54,6 +58,17 @@ public class JsqlParserGlobal {
         return thread;
     });
 
+    static {
+        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
+            if (!executorService.isShutdown()) {
+                if (LOG.isDebugEnabled()) {
+                    LOG.debug("close jsqlParser thread pool ...");
+                }
+                executorService.shutdown();
+            }
+        }, "mybatis-plus-jsqlParser-shutdown-hook"));
+    }
+
     @Setter
     private static JsqlParserFunction<String, Statement> parserSingleFunc = sql -> CCJSqlParserUtil.parse(sql, executorService, null);
 

diff --git a/...-jsqlparser/src/main/java/com/baomidou/mybatisplus/extension/parser/JsqlParserGlobal.java b/...-jsqlparser/src/main/java/com/baomidou/mybatisplus/extension/parser/JsqlParserGlobal.java
@@ -21,6 +21,8 @@
 import net.sf.jsqlparser.parser.CCJSqlParserUtil;
 import net.sf.jsqlparser.statement.Statement;
 import net.sf.jsqlparser.statement.Statements;
+import org.apache.ibatis.logging.Log;
+import org.apache.ibatis.logging.LogFactory;
 
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.LinkedBlockingQueue;
@@ -33,6 +35,8 @@
  */
 public class JsqlParserGlobal {
 
+    private static final Log LOG = LogFactory.getLog(JsqlParserGlobal.class);
+
     /**
      * 默认线程数大小
      *
@@ -54,6 +58,17 @@ public class JsqlParserGlobal {
         return thread;
     });
 
+    static {
+        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
+            if (!executorService.isShutdown()) {
+                if (LOG.isDebugEnabled()) {
+                    LOG.debug("close jsqlParser thread pool ...");
+                }
+                executorService.shutdown();
+            }
+        }, "mybatis-plus-jsqlParser-shutdown-hook"));
+    }
+
     @Setter
     private static JsqlParserFunction<String, Statement> parserSingleFunc = sql -> CCJSqlParserUtil.parse(sql, executorService, null);