ca patine

Author: BarbDev

src/main/java/bdd/BaseDeDonnee.java

@@ -2,16 +2,15 @@
 
 import org.h2.tools.Csv;
 
-import java.io.*;
-import java.math.BigDecimal;
 import java.sql.*;
-import java.util.ArrayList;
+import java.util.HashMap;
 
 /**
  * TODO : mettre a jour les expirations de location de film (à chaque lancement ?)
  */
 public class BaseDeDonnee {
 	private Connection bdd;
+	private HashMap<String, Table> tables = new HashMap<>();
 
 	public BaseDeDonnee() {
 		try {
@@ -22,298 +21,28 @@ public BaseDeDonnee() {
 			 * TODO : voir le chiffrement de la BDD (simple option normalement)
 			 */
 			bdd = DriverManager.getConnection("jdbc:h2:./test", "sa", "");
-			//createTables(null, null);
-			
+			tables.put("films", new TableFilms(this));
+			tables.put("utilisateurs", new TableUtilisateurs(this));
+
 		} catch (SQLException e) {
 			e.printStackTrace();
 		} catch (ClassNotFoundException e1) {
 			e1.printStackTrace();
 		}
 	}
 
-
-	public void createTables(String csvFilm, String csvUser) {
-		/* On vérifie que les tables ne sont pas déjà créés. */
-		/* On crée les tables. */
-		String films = "", users = "";
-		/*if (csvFilm != null && csvUser != null) {
-			films = "as select * from CSVREAD('"+csvFilm+"', null, 'fieldSeparator=|')";
-			users = "as select * from CSVREAD('"+csvUser+"', null, 'fieldSeparator=|')";
-		} */
-
-		try {
-			Statement statement = bdd.createStatement();
-			deleteAll();
-			statement.execute("CREATE TABLE IF NOT EXISTS films (" +
-					"id INT AUTO_INCREMENT (0, 1) PRIMARY KEY," +
-					"titre VARCHAR(255) NOT NULL UNIQUE," +
-					"duree SMALLINT NOT NULL," +
-					"note DECIMAL NOT NULL," +
-					"prix DECIMAL NOT NULL," +
-					"date_sortie DATE NOT NULL," +
-					"genres VARCHAR(15) NOT NULL," +
-					"synopsis VARCHAR(1500) NOT NULL," +
-					"acteurs VARCHAR(300) NOT NULL" +
-					") "+ films +";");
-			statement.execute("CREATE TABLE IF NOT EXISTS utilisateurs (" +
-					"id INT AUTO_INCREMENT (0, 1) PRIMARY KEY," +
-					"nom VARCHAR(35) NOT NULL," +
-					"prenom VARCHAR(35) NOT NULL," +
-					"pseudo VARCHAR(35) NOT NULL UNIQUE," +
-					"email VARCHAR(35) NOT NULL UNIQUE," +
-					"films_loue VARCHAR(1000)," +
-					"solde DECIMAL NOT NULL," +
-					"role VARCHAR(6) NOT NULL," +
-					"pwd VARCHAR(256) NOT NULL" +
-					") " + users +";");
-
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-	}
-
-	/* TODO :
-	 * Doit y avoir moyen de simplifier tout ca.
-	 * Créer une classe/interface Table, avoir deux spécifications, TableUtilisateur et TableFilm
-	 * Avoir qu'une seule méthode pr ajouter/modifier/supprimer quelque soit le type ?!
-	 */
 	/**
-	 * Ajoute une entrée à la BDD
-	 * TODO : définir ce qui fait qu'un utilisateur est unique (je dirais le PSEUDO)
-	 * @param user l'utilisateur à ajouter (contient déjà toutes les données nécessaire).
-	 *
-	 * Doit provoquer une erreur/exception lorsque le speudo est déjà utilisé.
+	 * Retourne la connection à la BDD
+	 * TODO : devrait être enlevé et prévoir des méthodes qui permettent une certaines abstraction (cf. execute())
+	 * @return
 	 */
-	public void ajouter(Utilisateur user, String pwd) {
-		try {
-			PreparedStatement pst = bdd.prepareStatement("INSERT INTO utilisateurs (" +
-					"nom," +
-					"prenom," +
-					"pseudo," +
-					"email," +
-					"solde," +
-					"role," +
-					"pwd" +
-					") values (?,?,?,?,?,?,?);");
-			pst.setString(1, user.getNom());
-			pst.setString(2, user.getPrenom());
-			pst.setString(3, user.getPseudo());
-			pst.setString(4, user.getEmail());
-			if (user.getRole().equals("user"))
-				pst.setBigDecimal(5, BigDecimal.valueOf(((Client) user).getSolde()));
-			else
-				pst.setBigDecimal(5, BigDecimal.valueOf(0));
-			pst.setString(6, user.getRole());
-			pst.setString(7, pwd);
-			pst.executeUpdate();
-			pst.close();
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-	}
-
-	public void ajouter(Film film) {
-		try {
-			PreparedStatement pst = bdd.prepareStatement("INSERT INTO films (" +
-					"titre," +
-					"duree," +
-					"note," +
-					"prix," +
-					"date_sortie," +
-					"genres," +
-					"synopsis," +
-					"acteurs" +
-					") values (?,?,?,?,?,?,?,?);");
-			pst.setString(1, film.getTitre());
-			pst.setInt(2, film.getDuree());
-			pst.setBigDecimal(3, BigDecimal.valueOf(film.getNote()));
-			pst.setBigDecimal(4, BigDecimal.valueOf(film.getPrix()));
-			pst.setDate(5, film.getDateSortie());
-			pst.setString(6, film.getGenre());
-			pst.setString(7, film.getSynopsis());
-
-			StringBuilder st = new StringBuilder();
-			for (String s : film.getActeurs()) {
-				st.append(s);
-				st.append(";");
-			}
-			pst.setString(8, st.toString());
-			pst.executeUpdate();
-			pst.close();
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
+	public Connection getConnection() {
+		return bdd;
 	}
 
-	public void modifier(Utilisateur user, String pwd) {
-		try {
-			PreparedStatement pst = bdd.prepareStatement("UPDATE utilisateurs" +
-					"SET nom=?," +
-					"prenom=?," +
-					"pseudo=?," +
-					"email=?," +
-					"solde=?," +
-					"role=?," +
-					"films_loue=?," +
-					"pwd=?" +
-					"WHERE id = ?;");
-			pst.setString(1, user.getNom());
-			pst.setString(2, user.getPrenom());
-			pst.setString(3, user.getPseudo());
-			pst.setString(4, user.getEmail());
-			if (user.getRole().equals("user")) {
-				pst.setBigDecimal(5, BigDecimal.valueOf(((Client) user).getSolde()));
-				pst.setString(7, Client.formatFilmsForBDD((Client)user));
-			}
-			else {
-				pst.setBigDecimal(5, BigDecimal.valueOf(0));
-				pst.setString(7, "");
-			}
-			pst.setString(6, user.getRole());
-			pst.setString(8, pwd);
-			pst.setInt(9, user.getId());
-			pst.executeUpdate();
-			pst.close();
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-	}
 
-	public void modifier(Film film) {
-		try {
-			PreparedStatement pst = bdd.prepareStatement("UPDATE films" +
-					"SET titre=?," +
-					"duree=?," +
-					"note=?," +
-					"prix=?," +
-					"date_sortie=?," +
-					"genres=?," +
-					"synopsis=?," +
-					"acteurs=?" +
-					"WHERE id = ?;");
-			pst.setString(1, film.getTitre());
-			pst.setInt(2, film.getDuree());
-			pst.setBigDecimal(3, BigDecimal.valueOf(film.getNote()));
-			pst.setBigDecimal(4, BigDecimal.valueOf(film.getPrix()));
-			pst.setDate(5, film.getDateSortie());
-			pst.setString(6, film.getGenre());
-			pst.setString(7, film.getSynopsis());
-
-			StringBuilder st = new StringBuilder();
-			for (String s : film.getActeurs()) {
-				st.append(s);
-				st.append(";");
-			}
-			pst.setString(8, st.toString());
-			pst.executeUpdate();
-			pst.close();
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-	}
-
-	public void supprimer(Utilisateur user) {
-		try {
-			PreparedStatement pst = bdd.prepareStatement("DELETE FROM utilisateurs" +
-					"WHERE id=?;");
-			pst.setInt(1, user.getId());
-			pst.executeUpdate();
-			pst.close();
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-	}
-
-	public void supprimer(Film film) {
-		try {
-			PreparedStatement pst = bdd.prepareStatement("DELETE FROM films" +
-					"WHERE id=?;");
-			pst.setInt(1, film.getId());
-			pst.executeUpdate();
-			pst.close();
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-	}
-
-	public ArrayList<Film> getFilms() {
-		ArrayList<Film> films = new ArrayList<>();
-
-		try {
-			ResultSet set = bdd.prepareStatement("SELECT * FROM films").executeQuery();
-			while (set.next()) {
-				Film film = new Film(set.getInt("id"),
-						set.getString("titre"),
-						set.getInt("duree"),
-						set.getInt("note"),
-						set.getBigDecimal("prix").floatValue(),
-						set.getDate("date_sortie"),
-						set.getString("genres"),
-						set.getString("synopsis"),
-						set.getString("acteurs").split(";"));
-				films.add(film);
-			}
-			set.close();
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-		return films;
-	}
-
-
-	public Film getFilm(String titre) {
-		Film film = null;
-		try {
-			PreparedStatement pst = bdd.prepareStatement("SELECT * FROM films WHERE titre = ?");
-			pst.setString(1, titre);
-			ResultSet set = pst.executeQuery();
-			while (set.next()) {
-				film = new Film(set.getInt("id"),
-					set.getString("titre"),
-					set.getInt("duree"),
-					set.getInt("note"),
-					set.getBigDecimal("prix").floatValue(),
-					set.getDate("date_sortie"),
-					set.getString("genres"),
-					set.getString("synopsis"),
-					set.getString("acteurs").split(";"));
-			}
-			set.close();
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-		return film;
-	}
-
-	/**
-	 *
-	 * @param pseudo
-	 * @param pwd
-	 * @return null si pas le bon login
-	 */
-	public Utilisateur getUser(String pseudo, String pwd) {
-		Utilisateur user = null;
-		try {
-			PreparedStatement pst = bdd.prepareStatement("SELECT * FROM utilisateurs WHERE pseudo = ?;");
-			pst.setString(1, pseudo);
-			ResultSet set = pst.executeQuery();
-			if (set.next()) {
-				if (set.getString("role").equals("user")) {
-					user = new Client(set.getInt("id"),
-							pseudo, set.getString("nom"), set.getString("prenom"),
-							set.getString("email"), set.getString("role"),
-							set.getBigDecimal("solde").floatValue(), Client.formatFilmsForClient(set.getString("films_loue"), this));
-				} else {
-					user = new Utilisateur(set.getInt("id"),
-							pseudo, set.getString("nom"), set.getString("prenom"),
-							set.getString("email"), set.getString("role"));
-				}
-			}
-			set.close();
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-		return user;
+	public void createTable(Table t, String csv) {
+		t.create(csv);
 	}
 
 	public void addFromCSV(String path, String table) {
@@ -337,16 +66,10 @@ public void addFromCSV(String path, String table) {
 		}
 	}
 
-	public void deleteAll() {
-		Statement statement = null;
-		try {
-			statement = bdd.createStatement();
-			statement.execute("DROP TABLE IF EXISTS films, utilisateurs;");
-		} catch (SQLException e) {
-			e.printStackTrace();
-		}
-	}
-
+	/**
+	 * Permet d'executer une action simple sur la BDD (sans paramètre ni retour)
+	 * @param action l'action au format SQL à effectuer
+	 */
 	public void execute(String action) {
 		try {
 			Statement statement = bdd.createStatement();
@@ -361,9 +84,9 @@ static public void main(String[] args) {
 		BaseDeDonnee baseDeDonnee = new BaseDeDonnee();
 		//baseDeDonnee.ajouter(new Film(1, "Mr Bean", 160, 7.8f, 20, new Date(Date.valueOf("1994-6-10").getTime()),
 		//		"drole", "Mr bean par en vacs :D", new String[]{"Le magnifique", "lul"}));
-		baseDeDonnee.createTables(null, null);
+		baseDeDonnee.createTable(null, null);
 
-		for (Film f : baseDeDonnee.getFilms())
-			System.out.println(f);
+		//for (Film f : baseDeDonnee.getFilms())
+		//	System.out.println(f);
 	}
 }