Complete pets DAO

src/main/java/es/uvigo/esei/daa/dao/PetsDAO.java

@@ -2,10 +2,9 @@ package es.uvigo.esei.daa.dao;
 
 import es.uvigo.esei.daa.entities.Pet;
 
-import java.sql.Connection;
-import java.sql.PreparedStatement;
-import java.sql.ResultSet;
-import java.sql.SQLException;
+import java.sql.*;
+import java.util.LinkedList;
+import java.util.List;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
@@ -41,4 +40,103 @@ public class PetsDAO extends DAO {
                 row.getString("name")
         );
     }
+
+    public void delete(int id)
+            throws DAOException, IllegalArgumentException {
+        try (final Connection conn = this.getConnection()) {
+            final String query = "DELETE FROM pet WHERE id=?";
+
+            try (final PreparedStatement statement = conn.prepareStatement(query)) {
+                statement.setInt(1, id);
+
+                if (statement.executeUpdate() != 1) {
+                    throw new IllegalArgumentException("Invalid id");
+                }
+            }
+        } catch (SQLException e) {
+            LOG.log(Level.SEVERE, "Error deleting a pet", e);
+            throw new DAOException(e);
+        }
+
+    }
+
+    public List<Pet> list() throws DAOException {
+        try (final Connection conn = this.getConnection()) {
+            final String query = "SELECT * FROM pet";
+
+            try (final PreparedStatement statement = conn.prepareStatement(query)) {
+                try (final ResultSet result = statement.executeQuery()) {
+                    final List<Pet> pets = new LinkedList<>();
+
+                    while (result.next()) {
+                        pets.add(rowToEntity(result));
+                    }
+
+                    return pets;
+                }
+            }
+        } catch (SQLException e) {
+            LOG.log(Level.SEVERE, "Error listing pets", e);
+            throw new DAOException(e);
+        }
+    }
+
+
+    public void modify(Pet pet)
+            throws DAOException, IllegalArgumentException {
+        if (pet == null) {
+            throw new IllegalArgumentException("pet can't be null");
+        }
+
+        try (Connection conn = this.getConnection()) {
+            final String query = "UPDATE pet SET name=?, owner=? WHERE id=?";
+
+            try (PreparedStatement statement = conn.prepareStatement(query)) {
+                statement.setString(1, pet.getName());
+                statement.setInt(2, pet.getOwner());
+                statement.setInt(3, pet.getId());
+
+                if (statement.executeUpdate() != 1) {
+                    throw new IllegalArgumentException("name and owner can't be null");
+                }
+            }
+        } catch (SQLException e) {
+            LOG.log(Level.SEVERE, "Error modifying a pet", e);
+            throw new DAOException();
+        }
+    }
+
+
+    public Pet add(String name, int owner)
+            throws DAOException, IllegalArgumentException {
+        if (name == null) {
+            throw new IllegalArgumentException("name and owner can't be null");
+        }
+
+        try (Connection conn = this.getConnection()) {
+            final String query = "INSERT INTO pet VALUES(null, ?, ?)";
+
+            try (PreparedStatement statement = conn.prepareStatement(query, Statement.RETURN_GENERATED_KEYS)) {
+                statement.setString(1, name);
+                statement.setInt(2, owner);
+
+                if (statement.executeUpdate() == 1) {
+                    try (ResultSet resultKeys = statement.getGeneratedKeys()) {
+                        if (resultKeys.next()) {
+                            return new Pet(resultKeys.getInt(1), owner, name);
+                        } else {
+                            LOG.log(Level.SEVERE, "Error retrieving inserted id");
+                            throw new SQLException("Error retrieving inserted id");
+                        }
+                    }
+                } else {
+                    LOG.log(Level.SEVERE, "Error inserting value");
+                    throw new SQLException("Error inserting value");
+                }
+            }
+        } catch (SQLException e) {
+            LOG.log(Level.SEVERE, "Error adding a pet", e);
+            throw new DAOException(e);
+        }
+    }
 }