mellowdeep

add minion

Oct 23rd, 2016
84
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Java 2.77 KB | None | 0 0
  1. package pr04.AddMinion;
  2.  
  3. import java.sql.*;
  4. import java.util.Scanner;
  5.  
  6. public class AddMInion {
  7.  
  8.     private static final String URL = "jdbc:mysql://localhost:3306/minions_db?useSSL=false";
  9.  
  10.     private static final String USER = "root";
  11.  
  12.     private static final String PASSWORD = "admin";
  13.  
  14.     public static void main(String[] args) {
  15.  
  16.         Scanner scanner = new Scanner(System.in);
  17.  
  18.         String[] line = scanner.nextLine().split("\\W+");
  19.         String minionName = line[1];
  20.         int minionAge = Integer.parseInt(line[2]);
  21.         String minionTown = line[3];
  22.  
  23.         String[] secondLine = scanner.nextLine().split("\\W+");
  24.         String villainName = secondLine[1];
  25.  
  26.         ResultSet resultSet = null;
  27.         try (Connection connection = DriverManager.getConnection(URL, USER,
  28.                 PASSWORD); Statement statement = connection.createStatement();) {
  29.             String query = "SELECT t.name FROM towns AS t " + "WHERE t.name = '"
  30.                     + minionTown + "';";
  31.             resultSet = statement.executeQuery(query);
  32.             if (!resultSet.isBeforeFirst()) {
  33.                 query = "INSERT INTO towns(name) VALUES('" + minionTown + "');";
  34.                 statement.executeUpdate(query);
  35.                 System.out.println("Town " + minionTown
  36.                         + " was added to the database");
  37.             }
  38.             query = "SELECT v.name FROM villains AS v WHERE v.name = '"
  39.                     + villainName + "';";
  40.             resultSet = statement.executeQuery(query);
  41.             if (!resultSet.isBeforeFirst()) {
  42.                 query = "INSERT INTO villains(name,evilness_factor) VALUES('"
  43.                         + villainName + "','evil');";
  44.                 statement.executeUpdate(query);
  45.                 System.out.println("Villain " + villainName
  46.                         + " was added to the database.");
  47.             }
  48.             query = "SELECT id FROM towns WHERE name = '" + minionTown + "';";
  49.             resultSet = statement.executeQuery(query);
  50.             if (resultSet.next()) {
  51.                 int townId = resultSet.getInt("id");
  52.                 query = "INSERT INTO minions(name,age,town_id) VALUES('"
  53.                         + minionName + "'," + minionAge + "," + townId + ");";
  54.                 statement.executeUpdate(query);
  55.             }
  56.             query = "SELECT id FROM minions WHERE name = '" + minionName + "';";
  57.             resultSet = statement.executeQuery(query);
  58.             int minionId = 0;
  59.             int villainId = 0;
  60.             if (resultSet.next()) {
  61.                 minionId = resultSet.getInt("id");
  62.  
  63.             }
  64.             query = "SELECT id FROM villains WHERE name = '" + villainName + "';";
  65.             resultSet = statement.executeQuery(query);
  66.             if (resultSet.next()) {
  67.                 villainId = resultSet.getInt("id");
  68.  
  69.             }
  70.             query = "INSERT INTO minions_villains(minion_id,villain_id) VALUES("
  71.                     + minionId + "," + villainId + ");";
  72.             System.out.println("Successfully added " + minionName + " to be minion of " + villainName);
  73.         } catch (SQLException e) {
  74.             e.printStackTrace();
  75.         }finally {
  76.             try { if (resultSet != null) resultSet.close();
  77.             } catch (Exception e) {
  78.                 e.printStackTrace();
  79.             }
  80.         }
  81.     }
  82. }
Add Comment
Please, Sign In to add comment