From 59b4ee891bb9355278d8f76379fa8bba1f9a9a47 Mon Sep 17 00:00:00 2001
From: Jon Sundin <sundinjon@gmail.com>
Date: Wed, 31 Jan 2024 01:49:09 +0000
Subject: [PATCH] added updateMessageById

---
 src/main/java/DAO/MessageDAO.java | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

diff --git a/src/main/java/DAO/MessageDAO.java b/src/main/java/DAO/MessageDAO.java
index 7412cd5..ad0ceea 100644
--- a/src/main/java/DAO/MessageDAO.java
+++ b/src/main/java/DAO/MessageDAO.java
@@ -18,8 +18,8 @@ public Message insertMessage(int posted_by, String message_text, long time_poste
         try {
             Connection conn = ConnectionUtil.getConnection();
             String sql = "INSERT INTO message (posted_by, message_text, time_posted_epoch) "
-                        + "VALUES (?, ?, ?)";
-            
+                    + "VALUES (?, ?, ?)";
+
             PreparedStatement ps = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
             ps.setInt(1, posted_by);
             ps.setString(2, message_text);
@@ -96,6 +96,25 @@ public Message getMessageById(int message_id) {
         return null;
     }
 
+    public Message updateMessageById(int message_id, String updated_message_text) {
+        try {
+            Connection conn = ConnectionUtil.getConnection();
+            String sql = "UPDATE message SET message_text = ? WHERE message_id = ?;";
+            PreparedStatement ps = conn.prepareStatement(sql);
+
+            ps.setString(1, updated_message_text);
+            ps.setInt(2, message_id);
+            int updatedRows = ps.executeUpdate();
+
+            while (updatedRows > 0) {
+                return getMessageById(message_id);
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        return null;
+    }
+
     public Message deleteMessageById(int message_id) {
         Message message = getMessageById(message_id);
         if (message != null) {
@@ -103,11 +122,11 @@ public Message deleteMessageById(int message_id) {
                 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;
                 }
@@ -117,5 +136,5 @@ public Message deleteMessageById(int message_id) {
         }
         return null;
     }
-    
+
 }