Add JDBC and count method for DAO

hardingadonis · Jan 22, 2024 · fca51be · fca51be
1 parent 593869f
commit fca51be
Show file tree

Hide file tree

Showing 8 changed files with 159 additions and 0 deletions.
diff --git a/src/main/java/io/hardingadonis/saledock/dao/IDAO.java b/src/main/java/io/hardingadonis/saledock/dao/IDAO.java
@@ -9,4 +9,8 @@ public interface IDAO<T> {
     public Optional<T> getByID(Integer ID);
 
     public List<T> getAll();
+
+    default Long count() {
+        return 0L;
+    }
 }
diff --git a/src/main/java/io/hardingadonis/saledock/dao/impl/CustomerDAOImpl.java b/src/main/java/io/hardingadonis/saledock/dao/impl/CustomerDAOImpl.java
@@ -3,6 +3,7 @@
 import io.hardingadonis.saledock.dao.*;
 import io.hardingadonis.saledock.model.*;
 import io.hardingadonis.saledock.utils.*;
+import java.sql.*;
 import java.util.*;
 import org.hibernate.*;
 
@@ -38,4 +39,27 @@ public List<Customer> getAll() {
             return session.createQuery("FROM Customer", Customer.class).getResultList();
         }
     }
+
+    @Override
+    public Long count() {
+        Long count = 0L;
+
+        try {
+            Connection conn = Singleton.dbContext.getConnection();
+
+            PreparedStatement smt = conn.prepareStatement("SELECT COUNT(*) FROM `customer`");
+
+            ResultSet rs = smt.executeQuery();
+
+            if (rs.next()) {
+                count = rs.getLong(1);
+            }
+
+            Singleton.dbContext.closeConnection(conn);
+        } catch (SQLException ex) {
+            System.err.println(ex.getMessage());
+        }
+
+        return count;
+    }
 }
diff --git a/src/main/java/io/hardingadonis/saledock/dao/impl/OrderDAOImpl.java b/src/main/java/io/hardingadonis/saledock/dao/impl/OrderDAOImpl.java
@@ -3,6 +3,7 @@
 import io.hardingadonis.saledock.dao.*;
 import io.hardingadonis.saledock.model.*;
 import io.hardingadonis.saledock.utils.*;
+import java.sql.*;
 import java.util.*;
 import org.hibernate.*;
 
@@ -38,4 +39,27 @@ public List<Order> getAll() {
             return session.createQuery("FROM Order", Order.class).getResultList();
         }
     }
+
+    @Override
+    public Long count() {
+        Long count = 0L;
+
+        try {
+            Connection conn = Singleton.dbContext.getConnection();
+
+            PreparedStatement smt = conn.prepareStatement("SELECT COUNT(*) FROM `order`");
+
+            ResultSet rs = smt.executeQuery();
+
+            if (rs.next()) {
+                count = rs.getLong(1);
+            }
+
+            Singleton.dbContext.closeConnection(conn);
+        } catch (SQLException ex) {
+            System.err.println(ex.getMessage());
+        }
+
+        return count;
+    }
 }
diff --git a/src/main/java/io/hardingadonis/saledock/dao/impl/ProductDAOImpl.java b/src/main/java/io/hardingadonis/saledock/dao/impl/ProductDAOImpl.java
@@ -3,6 +3,7 @@
 import io.hardingadonis.saledock.dao.*;
 import io.hardingadonis.saledock.model.*;
 import io.hardingadonis.saledock.utils.*;
+import java.sql.*;
 import java.util.*;
 import org.hibernate.*;
 
@@ -38,4 +39,27 @@ public List<Product> getAll() {
             return session.createQuery("FROM Product", Product.class).getResultList();
         }
     }
+
+    @Override
+    public Long count() {
+        Long count = 0L;
+
+        try {
+            Connection conn = Singleton.dbContext.getConnection();
+
+            PreparedStatement smt = conn.prepareStatement("SELECT COUNT(*) FROM `product`");
+
+            ResultSet rs = smt.executeQuery();
+
+            if (rs.next()) {
+                count = rs.getLong(1);
+            }
+
+            Singleton.dbContext.closeConnection(conn);
+        } catch (SQLException ex) {
+            System.err.println(ex.getMessage());
+        }
+
+        return count;
+    }
 }
diff --git a/src/main/java/io/hardingadonis/saledock/utils/IDBContext.java b/src/main/java/io/hardingadonis/saledock/utils/IDBContext.java
@@ -0,0 +1,10 @@
+package io.hardingadonis.saledock.utils;
+
+import java.sql.*;
+
+public interface IDBContext {
+
+    public Connection getConnection();
+
+    public void closeConnection(Connection connection);
+}
diff --git a/src/main/java/io/hardingadonis/saledock/utils/Singleton.java b/src/main/java/io/hardingadonis/saledock/utils/Singleton.java
@@ -2,6 +2,7 @@
 
 import io.hardingadonis.saledock.dao.*;
 import io.hardingadonis.saledock.dao.impl.*;
+import io.hardingadonis.saledock.utils.impl.*;
 
 public class Singleton {
 
@@ -14,6 +15,8 @@ public class Singleton {
     public static IOrderDAO orderDAO;
 
     public static IProductDAO productDAO;
+
+    public static IDBContext dbContext;
 
     static {
         categoryDAO = new CategoryDAOImpl();
@@ -25,5 +28,7 @@ public class Singleton {
         orderDAO = new OrderDAOImpl();
 
         productDAO = new ProductDAOImpl();
+
+        dbContext = new DBContextMySQLImpl();
     }
 }
diff --git a/src/main/java/io/hardingadonis/saledock/utils/impl/DBContextMySQLImpl.java b/src/main/java/io/hardingadonis/saledock/utils/impl/DBContextMySQLImpl.java
@@ -0,0 +1,64 @@
+package io.hardingadonis.saledock.utils.impl;
+
+import java.io.*;
+import java.sql.*;
+import java.util.*;
+
+import io.hardingadonis.saledock.utils.*;
+
+public class DBContextMySQLImpl implements IDBContext {
+
+    private String dbURL;
+    private String user;
+    private String password;
+
+    public DBContextMySQLImpl() {
+        try (InputStream input = DBContextMySQLImpl.class.getClassLoader().getResourceAsStream("config.mysql.properties")) {
+            Properties prop = new Properties();
+
+            if (input == null) {
+                this.dbURL = "jdbc:mysql://localhost:3306/saledock";
+                this.user = "root";
+                this.password = "";
+            } else {
+                prop.load(input);
+
+                this.dbURL = "jdbc:mysql://" + prop.getProperty("config.jdbc.url") + ":3306/" + prop.getProperty("config.jdbc.database_name");
+                this.user = prop.getProperty("config.jdbc.user");
+                this.password = prop.getProperty("config.jdbc.password");
+            }
+        } catch (IOException ex) {
+            System.err.println(ex.getMessage());
+        }
+    }
+
+    @Override
+    public Connection getConnection() {
+        Connection connection = null;
+
+        try {
+            DriverManager.registerDriver(new com.mysql.cj.jdbc.Driver());
+
+            connection = DriverManager.getConnection(this.dbURL, this.user, this.password);
+
+            System.out.println("Connect MySQL successfully!");
+        } catch (SQLException ex) {
+            System.err.println(ex.getMessage());
+        }
+
+        return connection;
+    }
+
+    @Override
+    public void closeConnection(Connection connection) {
+        try {
+            if (connection != null) {
+                connection.close();
+
+                System.out.println("Close MySQL successfully!");
+            }
+        } catch (SQLException ex) {
+            System.err.println(ex.getMessage());
+        }
+    }
+}
diff --git a/src/main/resources/config.mysql.properties b/src/main/resources/config.mysql.properties
@@ -0,0 +1,4 @@
+config.jdbc.url=localhost
+config.jdbc.database_name=saledock
+config.jdbc.user=root
+config.jdbc.password=