Herança e uso de listas nos DAOs

Author: danielledsena

src/main/java/com/example/AppBd.java

@@ -17,7 +17,10 @@ public static void main(String[] args){
     public AppBd() {
         try (var conn = ConnectionManager.getConnection()) {
             var estadoDAO = new EstadoDAO(conn);
-            estadoDAO.listar();
+            var listaEstados = estadoDAO.listar();
+            for (var estado : listaEstados) {
+                System.out.println(estado);
+            }
             estadoDAO.localizar("CE");
 
             var marca = new Marca();

src/main/java/com/example/dao/DAO.java

@@ -4,7 +4,7 @@
 import java.sql.SQLException;
 
 public class DAO {
-    private Connection conn;
+    protected Connection conn;
 
     public DAO(Connection conn) {
         this.conn = conn;

src/main/java/com/example/dao/EstadoDAO.java

@@ -2,24 +2,29 @@
 
 import java.sql.Connection;
 import java.sql.SQLException;
+import java.util.LinkedList;
+import java.util.List;
 
-public class EstadoDAO {
-    private Connection conn;
+import com.example.model.Estado;
 
+public class EstadoDAO extends DAO {
     public EstadoDAO(Connection conn) {
-        this.conn = conn;
+        super(conn);
     }
 
-    public void listar() {
-        try {
-            var statement = conn.createStatement();
-            var result = statement.executeQuery("select * from estado");
-            while(result.next()) {
-                System.out.printf("Id: %d Nome: %s UF: %s\n", result.getInt("id"), result.getString("nome"), result.getString("uf"));
-            }
-        } catch(SQLException e) {
-            System.err.println("Não foi possível executar a consulta ao banco" + e.getMessage());
-        } 
+    public List<Estado> listar() throws SQLException {
+        var lista = new LinkedList<Estado>();
+        var statement = conn.createStatement();
+        var result = statement.executeQuery("select * from estado");
+        while(result.next()) {
+            var estado = new Estado();
+            estado.setId(result.getLong("id"));
+            estado.setNome(result.getString("nome"));
+            estado.setUf(result.getString("uf"));
+            lista.add(estado);
+        }
+
+        return lista;
     }
 
     public void localizar(String uf) {

src/main/java/com/example/dao/ProdutoDAO.java

@@ -2,13 +2,12 @@
 
 import java.sql.Connection;
 import java.sql.SQLException;
-import com.example.model.Produto;
 
-public class ProdutoDAO {
-    private Connection conn;
+import com.example.model.Produto;
 
+public class ProdutoDAO extends DAO {
     public ProdutoDAO(Connection conn) {
-        this.conn = conn;
+        super(conn);
     }
 
     public void excluir(long id) {

src/main/java/com/example/model/Estado.java

@@ -44,4 +44,8 @@ public int getPopulacao() {
     public void setPopulacao(int populacao) {
         this.populacao = populacao;
     }
+
+    public String toString() {
+        return "Estado: " + nome + " Id: " + id + " UF: " + uf;
+    }
 }