From 6567addfae7c36247743ba6324b71e77d6aefec7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=94=D0=B0=D0=BD=D0=B8=D0=BB?=
 <119668441+Jiga228@users.noreply.github.com>
Date: Wed, 28 May 2025 21:26:50 +0700
Subject: [PATCH] Fix bugs lose connection with DB

---
 .../Repository/DataBase.java                  | 34 +++++++++++++++++--
 1 file changed, 31 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/Samsung/TaskTraker_Server/Repository/DataBase.java b/src/main/java/com/Samsung/TaskTraker_Server/Repository/DataBase.java
index 0ed3447..290d41c 100644
--- a/src/main/java/com/Samsung/TaskTraker_Server/Repository/DataBase.java
+++ b/src/main/java/com/Samsung/TaskTraker_Server/Repository/DataBase.java
@@ -9,7 +9,8 @@ import java.util.*;
 
 public class DataBase {
     private static DataBase instance;
-    private final Connection connection;
+    private Connection connection;
+    private final String login, password;
 
     public void close() {
         try {
@@ -20,15 +21,21 @@ public class DataBase {
         instance = null;
     }
 
-    private DataBase(String login, String pass) {
+    private void OpenConnection() {
         try {
-            connection = DriverManager.getConnection(API_KEYS.DB_URL, login, pass);
+            connection = DriverManager.getConnection(API_KEYS.DB_URL, login, password);
         } catch (SQLException e) {
             System.out.println("[!] DB: " + e.getMessage());
             throw new RuntimeException(e);
         }
     }
 
+    private DataBase(String login, String password) {
+        this.login = login;
+        this.password = password;
+        OpenConnection();
+    }
+
     public static DataBase getInstance() {
         if(instance == null) {
             instance = new DataBase(API_KEYS.DB_LOGIN, API_KEYS.DB_PASSWORD);
@@ -37,6 +44,9 @@ public class DataBase {
     }
 
     public User SaveUser(String login, String password) throws SQLException {
+        if(!connection.isValid(100))
+            OpenConnection();
+
         PreparedStatement saveUser_stmt = connection.prepareStatement("INSERT INTO users(login, password) VALUE (?, ?)");
         saveUser_stmt.setString(1, login);
         saveUser_stmt.setString(2, password);
@@ -51,6 +61,9 @@ public class DataBase {
     }
 
     public void AddTask(Task task, User user) throws SQLException {
+        if(!connection.isValid(100))
+            OpenConnection();
+
         PreparedStatement addTask = connection.prepareStatement("INSERT INTO tasks(ID," +
                                                              " TaskName," +
                                                              " TaskDescription," +
@@ -69,6 +82,9 @@ public class DataBase {
     }
 
     public void UpdateTask(Task task, User user) throws SQLException {
+        if(!connection.isValid(100))
+            OpenConnection();
+
         PreparedStatement addTask = connection.prepareStatement("UPDATE tasks SET TaskName = ?," +
                                                                      " TaskDescription = ?," +
                                                                      " TaskDate = ?," +
@@ -85,6 +101,9 @@ public class DataBase {
     }
 
     public void RemoveTask(long taskID, User user) throws SQLException {
+        if(!connection.isValid(100))
+            OpenConnection();
+
         PreparedStatement remove = connection.prepareStatement("DELETE FROM tasks WHERE ID = ? AND UserOwner = ?");
         remove.setLong(1, taskID);
         remove.setLong(2, user.getID());
@@ -92,6 +111,9 @@ public class DataBase {
     }
 
     public void UpdatePassword(String password, int userID) throws SQLException {
+        if(!connection.isValid(100))
+            OpenConnection();
+
         PreparedStatement update = connection.prepareStatement("UPDATE users SET password = ? WHERE id = ?");
         update.setString(1, password);
         update.setInt(2, userID);
@@ -99,6 +121,9 @@ public class DataBase {
     }
 
     public List<Task> getTaskList(User user) throws SQLException {
+        if(!connection.isValid(100))
+            OpenConnection();
+
         PreparedStatement findTasks = connection.prepareStatement("SELECT * FROM tasks WHERE UserOwner = ?");
         findTasks.setInt(1, user.getID());
         ResultSet result = findTasks.executeQuery();
@@ -117,6 +142,9 @@ public class DataBase {
     }
 
     public User FindUser(String login) throws SQLException {
+        if(!connection.isValid(100))
+            OpenConnection();
+
         PreparedStatement getUser_stmt = connection.prepareStatement("SELECT * FROM users WHERE login = ?");
         getUser_stmt.setString(1, login);
         ResultSet resultSet = getUser_stmt.executeQuery();