Fixed the connection leak in Database Manager

DeathGOD7 · Dec 26, 2024 · 25fdf1c · 25fdf1c
1 parent 626e617
commit 25fdf1c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/build.gradle.kts b/build.gradle.kts
@@ -12,7 +12,7 @@ plugins {
 }
 
 group = "io.github.deathgod7.SE7ENLib"
-version = "1.1.1"
+version = "1.1.2-rc1"
 description = "A lib to aid in development for my java stuff."
 
 repositories {

diff --git a/src/main/java/io/github/deathgod7/SE7ENLib/database/DatabaseManager.java b/src/main/java/io/github/deathgod7/SE7ENLib/database/DatabaseManager.java
@@ -61,7 +61,20 @@ public DatabaseInfo getDbInfo() {
 	 * @return {@link Boolean}
 	 */
 	public boolean isConnected() {
-		return (this.getConnection() != null);
+		if (_dbInfo.getDbType() == DatabaseType.SQLite || _dbInfo.getDbType() == DatabaseType.MySQL) {
+			try (Connection con = (Connection) this.getConnection()) {
+				return con != null;
+			} catch (SQLException ex) {
+				Logger.log("[CONNECTION CHECK ERROR] " + ex.getMessage());
+				return false;
+			}
+		}
+		else if (_dbInfo.getDbType() == DatabaseType.MongoDB) {
+			return this.getMongoDB().isConnected();
+		}
+		else {
+			return false;
+		}
 	}
 
 	private boolean debugMode = false;