Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- package pr04.AddMinion;
- import java.sql.*;
- import java.util.Scanner;
- public class AddMInion {
- private static final String URL = "jdbc:mysql://localhost:3306/minions_db?useSSL=false";
- private static final String USER = "root";
- private static final String PASSWORD = "admin";
- public static void main(String[] args) {
- Scanner scanner = new Scanner(System.in);
- String[] line = scanner.nextLine().split("\\W+");
- String minionName = line[1];
- int minionAge = Integer.parseInt(line[2]);
- String minionTown = line[3];
- String[] secondLine = scanner.nextLine().split("\\W+");
- String villainName = secondLine[1];
- ResultSet resultSet = null;
- try (Connection connection = DriverManager.getConnection(URL, USER,
- PASSWORD); Statement statement = connection.createStatement();) {
- String query = "SELECT t.name FROM towns AS t " + "WHERE t.name = '"
- + minionTown + "';";
- resultSet = statement.executeQuery(query);
- if (!resultSet.isBeforeFirst()) {
- query = "INSERT INTO towns(name) VALUES('" + minionTown + "');";
- statement.executeUpdate(query);
- System.out.println("Town " + minionTown
- + " was added to the database");
- }
- query = "SELECT v.name FROM villains AS v WHERE v.name = '"
- + villainName + "';";
- resultSet = statement.executeQuery(query);
- if (!resultSet.isBeforeFirst()) {
- query = "INSERT INTO villains(name,evilness_factor) VALUES('"
- + villainName + "','evil');";
- statement.executeUpdate(query);
- System.out.println("Villain " + villainName
- + " was added to the database.");
- }
- query = "SELECT id FROM towns WHERE name = '" + minionTown + "';";
- resultSet = statement.executeQuery(query);
- if (resultSet.next()) {
- int townId = resultSet.getInt("id");
- query = "INSERT INTO minions(name,age,town_id) VALUES('"
- + minionName + "'," + minionAge + "," + townId + ");";
- statement.executeUpdate(query);
- }
- query = "SELECT id FROM minions WHERE name = '" + minionName + "';";
- resultSet = statement.executeQuery(query);
- int minionId = 0;
- int villainId = 0;
- if (resultSet.next()) {
- minionId = resultSet.getInt("id");
- }
- query = "SELECT id FROM villains WHERE name = '" + villainName + "';";
- resultSet = statement.executeQuery(query);
- if (resultSet.next()) {
- villainId = resultSet.getInt("id");
- }
- query = "INSERT INTO minions_villains(minion_id,villain_id) VALUES("
- + minionId + "," + villainId + ");";
- System.out.println("Successfully added " + minionName + " to be minion of " + villainName);
- } catch (SQLException e) {
- e.printStackTrace();
- }finally {
- try { if (resultSet != null) resultSet.close();
- } catch (Exception e) {
- e.printStackTrace();
- }
- }
- }
- }
Add Comment
Please, Sign In to add comment