Fix saving note

2022-12-27 10:24:47 +01:00
parent 7e3be4c45d
commit 5f1f5b7dcd
1 changed files with 1 additions and 1 deletions
--- a/src/main/java/database/note/NoteDao.java
+++ b/src/main/java/database/note/NoteDao.java
@@ -0,0 +1,88 @@
+package database;
+
+import model.Note;
+import org.jdbi.v3.core.Handle;
+import org.jdbi.v3.core.JdbiException;
+import tools.DatabaseConnection;
+
+import java.util.List;
+import java.util.Optional;
+
+public class NoteDao {
+
+    public static void save(Note note) {
+        try (Handle handle = DatabaseConnection.getHandle()) {
+            handle.createUpdate("""
+                            INSERT INTO notes (`message`, `from`, `to`, `timestamp`, `fame`, `deleted`)
+                            VALUES (?, ?, ?, ?, ?, ?)""")
+                    .bind(0, note.message())
+                    .bind(1, note.from())
+                    .bind(2, note.to())
+                    .bind(3, note.timestamp())
+                    .bind(4, note.fame())
+                    .bind(5, 0)
+                    .execute();
+        } catch (JdbiException e) {
+            throw new DaoException("Failed to save note: %s".formatted(note.toString()), e);
+        }
+    }
+
+    public static List<Note> findAllByTo(String to) {
+        try (Handle handle = DatabaseConnection.getHandle()) {
+            return handle.createQuery("""
+                            SELECT * 
+                            FROM notes
+                            WHERE `deleted` = 0
+                            AND `to` = ?""")
+                    .bind(0, to)
+                    .mapTo(Note.class)
+                    .list();
+        } catch (JdbiException e) {
+            throw new DaoException("Failed to find notes with \"to\": %s".formatted(to), e);
+        }
+    }
+
+    public static Optional<Note> delete(int id) {
+        try (Handle handle = DatabaseConnection.getHandle()) {
+            Optional<Note> note = findById(handle, id);
+            if (note.isEmpty()) {
+                return Optional.empty();
+            }
+            deleteById(handle, id);
+
+            return note;
+        } catch (JdbiException e) {
+            throw new DaoException("Failed to delete note with id: %d".formatted(id), e);
+        }
+    }
+
+    private static Optional<Note> findById(Handle handle, int id) {
+        final Optional<Note> note;
+        try {
+            note = handle.createQuery("""
+                            SELECT *
+                            FROM notes
+                            WHERE `deleted` = 0
+                            AND `id` = ?""")
+                    .bind(0, id)
+                    .mapTo(Note.class)
+                    .findOne();
+        } catch (JdbiException e) {
+            throw new DaoException("Failed find note with id %s".formatted(id), e);
+        }
+        return note;
+    }
+
+    private static void deleteById(Handle handle, int id) {
+        try {
+            handle.createUpdate("""
+                        UPDATE notes
+                        SET `deleted` = 1
+                        WHERE `id` = ?""")
+                    .bind(0, id)
+                    .execute();
+        } catch (JdbiException e) {
+            throw new DaoException("Failed to delete note with id %d".formatted(id), e);
+        }
+    }
+}