diff --git a/src/library/Database.java b/src/library/Database.java index 32a0849..9418ad2 100644 --- a/src/library/Database.java +++ b/src/library/Database.java @@ -6,7 +6,6 @@ package library; import java.sql.Connection; import java.sql.DriverManager; -import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.SQLException; import java.sql.Statement; @@ -84,7 +83,7 @@ public class Database { return books; } - public List selectAlbums() { + public List selectAlbums() { List albums = new LinkedList(); try { ResultSet result = stat.executeQuery("SELECT * FROM albums"); @@ -105,6 +104,26 @@ public class Database { return albums; } + public List selectMovies() { + List movies = new LinkedList(); + try { + ResultSet result = stat.executeQuery("SELECT * FROM movies"); + int id, year; + String name, category, director; + while(result.next()) { + id = result.getInt("id"); + year = result.getInt("year"); + name = result.getString("name"); + director = result.getString("director"); + category = result.getString("category"); + + movies.add(new Movie(id, name, director, year, category)); + } + } catch (SQLException e) { + JOptionPane.showMessageDialog(null, "Błąd przy odczycie z bazy.", "Error:", JOptionPane.INFORMATION_MESSAGE); + } + return movies; + } public void closeConnection() { try {