delete message by id dao method

jmsundin · Jan 30, 2024 · 673c27b · 673c27b
1 parent e17a19e
commit 673c27b
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/main/java/DAO/MessageDAO.java b/src/main/java/DAO/MessageDAO.java
@@ -95,5 +95,27 @@ public Message getMessageById(int message_id) {
         }
         return null;
     }
+
+    public Message deleteMessageById(int message_id) {
+        Message message = getMessageById(message_id);
+        if (message != null) {
+            try {
+                Connection conn = ConnectionUtil.getConnection();
+                String sql = "DELETE FROM message WHERE message_id = ?";
+                PreparedStatement ps = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
+
+                ps.setInt(1, message_id);
+
+                int deletedEntity = ps.executeUpdate();
+
+                if (deletedEntity > 0) {
+                    return message;
+                }
+            } catch (SQLException e) {
+                e.printStackTrace();
+            }
+        }
+        return null;
+    }
 
 }