From 4f993edaf8fb4ff5b0f54f5003053990a771c1e2 Mon Sep 17 00:00:00 2001
From: Niilyx _ <nait.meddour.idir100@gmail.com>
Date: Sun, 6 Nov 2022 15:44:12 +0100
Subject: [PATCH] =?UTF-8?q?g=C3=A9rant=20de=20connexion=20=C3=A0=20la=20db?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../efreicraft/ecatup/utils/DBConnection.java | 76 +++++++++++++++++++
 1 file changed, 76 insertions(+)
 create mode 100644 src/main/java/fr/efreicraft/ecatup/utils/DBConnection.java

diff --git a/src/main/java/fr/efreicraft/ecatup/utils/DBConnection.java b/src/main/java/fr/efreicraft/ecatup/utils/DBConnection.java
new file mode 100644
index 0000000..f8e1076
--- /dev/null
+++ b/src/main/java/fr/efreicraft/ecatup/utils/DBConnection.java
@@ -0,0 +1,76 @@
+package fr.efreicraft.ecatup.utils;
+
+import org.bukkit.Bukkit;
+import org.mariadb.jdbc.Configuration;
+import org.mariadb.jdbc.Connection;
+import org.mariadb.jdbc.Driver;
+
+import javax.annotation.Nullable;
+import java.sql.SQLException;
+
+public class DBConnection {
+
+    private Connection connection = null;
+    private final String HOST;
+    private final String DATABASE;
+    private final String USER;
+    private final String PASSWORD;
+    private final int PORT;
+
+    public DBConnection(String host, int port, String database, String user, String password) {
+        this.HOST = host;
+        this.PORT = port;
+        this.DATABASE = database;
+        this.USER = user;
+        this.PASSWORD = password;
+    }
+
+    public boolean open() {
+        try {
+            connection = Driver.connect(new Configuration.Builder()
+                    .addHost(this.HOST, this.PORT)
+                    .database(this.DATABASE)
+                    .user(this.USER)
+                    .password(this.PASSWORD)
+                    .socketTimeout(0)
+                    .maxIdleTime(0)
+                    .build());
+        } catch (SQLException e) {
+            e.printStackTrace();
+            Bukkit.getLogger().severe("Couldn't open connection to database!");
+            return false;
+        }
+        return true;
+    }
+
+    public void close() {
+        try {
+            if (connection != null)
+                connection.close();
+        } catch (SQLException e) {
+            e.printStackTrace();
+            Bukkit.getLogger().severe("Couldn't close connection to database!");
+        }
+    }
+
+    @Nullable
+    public Connection openThenGetConnection() {
+        if (connection == null || connection.isClosed()) {
+            if (!this.open()) {
+                Bukkit.getLogger().severe("Tried to reopen a connection, and failed.");
+                return null;
+            }
+        }
+
+        return connection;
+    }
+
+    /**
+     * Cette fonction retourne directement la connexion, sans vérifier si elle existe. Elle est bien plus dangereuse !
+     * @return connection La connexion à la connexion à la DB
+     */
+    @Nullable
+    public Connection getConnection() {
+        return connection;
+    }
+}