Fix mysql connection timeout issue

junghyun397 · Jan 31, 2021 · 2e1e697 · 2e1e697
1 parent 0f4f077
commit 2e1e697
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/main/java/do1phin/mine2021/data/db/MysqlDatabaseHelper.java b/src/main/java/do1phin/mine2021/data/db/MysqlDatabaseHelper.java
@@ -43,6 +43,9 @@ public boolean connect() {
     @SuppressWarnings("SqlResolve")
     @Override
     public void initDatabase() {
+        Executors.newSingleThreadScheduledExecutor()
+                .scheduleAtFixedRate(this::validateConnection, 1, 1, TimeUnit.MINUTES);
+
         try {
             final Statement stmt = this.connection.createStatement();
 
@@ -70,17 +73,14 @@ public void initDatabase() {
         } catch (SQLException e) {
             e.printStackTrace();
         }
-
-        Executors.newSingleThreadScheduledExecutor()
-                .scheduleAtFixedRate(this::validateConnection, 1, 1, TimeUnit.MINUTES);
     }
 
     @SuppressWarnings("SqlResolve")
     @Override
     public int getNextAutoIncrement() {
         try {
             final PreparedStatement pstmt = this.connection.prepareStatement("SHOW TABLE STATUS LIKE 'user_info'");
-            ResultSet resultSet = pstmt.executeQuery();
+            final ResultSet resultSet = pstmt.executeQuery();
 
             if (resultSet.next()) return resultSet.getInt("Auto_increment");
         } catch (SQLException e) {
@@ -94,7 +94,8 @@ private void validateConnection() {
         try {
             final PreparedStatement pstmt = this.connection.prepareStatement("SELECT 1");
             if (pstmt.executeQuery().next()) return;
-        } catch (SQLException ignored) {}
+        } catch (Exception ignored) {}
+        this.disconnect();
         this.connect();
     }