From 5a06ee53b6986e92dfceb967c5e23f4c0c66d7ef Mon Sep 17 00:00:00 2001
From: ChomeNS <95471003+ChomeNS@users.noreply.github.com>
Date: Sat, 14 Dec 2024 16:54:28 +0700
Subject: [PATCH] fix: actually handle stopping the database when the bot is
 stopping

---
 build-number.txt                                       |  2 +-
 src/main/java/me/chayapak1/chomens_bot/Main.java       |  2 ++
 .../chayapak1/chomens_bot/plugins/DatabasePlugin.java  | 10 ++++++++++
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/build-number.txt b/build-number.txt
index 832d4caf..1e36b913 100644
--- a/build-number.txt
+++ b/build-number.txt
@@ -1 +1 @@
-1256
\ No newline at end of file
+1258
\ No newline at end of file
diff --git a/src/main/java/me/chayapak1/chomens_bot/Main.java b/src/main/java/me/chayapak1/chomens_bot/Main.java
index 3e3e385a..0a838b1c 100644
--- a/src/main/java/me/chayapak1/chomens_bot/Main.java
+++ b/src/main/java/me/chayapak1/chomens_bot/Main.java
@@ -148,6 +148,8 @@ public class Main {
 
         executorService.shutdown();
 
+        database.stop();
+
         try {
             final boolean ignoredExecutorDone = executor.awaitTermination(5, TimeUnit.SECONDS);
             final boolean ignoredExecutorServiceDone = executorService.awaitTermination(5, TimeUnit.SECONDS);
diff --git a/src/main/java/me/chayapak1/chomens_bot/plugins/DatabasePlugin.java b/src/main/java/me/chayapak1/chomens_bot/plugins/DatabasePlugin.java
index 8e308879..fd67518a 100644
--- a/src/main/java/me/chayapak1/chomens_bot/plugins/DatabasePlugin.java
+++ b/src/main/java/me/chayapak1/chomens_bot/plugins/DatabasePlugin.java
@@ -50,4 +50,14 @@ public class DatabasePlugin {
 
         return statement.executeUpdate(query);
     }
+
+    public void stop () {
+        executorService.shutdown();
+
+        try {
+            connection.close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+    }
 }