Java Code Examples for org.h2.tools.RunScript#execute()

private TestServer() {
    DataSource ds = (DataSource) SingletonServiceFactory.getBean(DataSource.class);
    try (Connection connection = ds.getConnection()) {
        String schemaResourceName = "/create_h2.sql";
        InputStream in = TestServer.class.getResourceAsStream(schemaResourceName);

        if (in == null) {
            throw new RuntimeException("Failed to load resource: " + schemaResourceName);
        }
        InputStreamReader reader = new InputStreamReader(in, UTF_8);
        RunScript.execute(connection, reader);

    } catch (SQLException e) {
        e.printStackTrace();
    }

}
 

@BeforeClass
public static void runOnceBeforeClass() {
    System.out.println("@BeforeClass - runOnceBeforeClass");
    DataSource ds = (DataSource) SingletonServiceFactory.getBean(DataSource.class);
    try (Connection connection = ds.getConnection()) {
        String schemaResourceName = "/create_h2.sql";
        InputStream in = AuditHandlerTest.class.getResourceAsStream(schemaResourceName);

        if (in == null) {
            throw new RuntimeException("Failed to load resource: " + schemaResourceName);
        }
        InputStreamReader reader = new InputStreamReader(in, UTF_8);
        RunScript.execute(connection, reader);

    } catch (SQLException e) {
        e.printStackTrace();
    }

}
 

private TestServer() {
    DataSource ds = (DataSource) SingletonServiceFactory.getBean(DataSource.class);
    try (Connection connection = ds.getConnection()) {
        String schemaResourceName = "/create_h2.sql";
        InputStream in = TestServer.class.getResourceAsStream(schemaResourceName);

        if (in == null) {
            throw new RuntimeException("Failed to load resource: " + schemaResourceName);
        }
        InputStreamReader reader = new InputStreamReader(in, UTF_8);
        RunScript.execute(connection, reader);

    } catch (SQLException e) {
        e.printStackTrace();
    }

}
 

private TestServer() {
    DataSource ds = (DataSource) SingletonServiceFactory.getBean(DataSource.class);
    try (Connection connection = ds.getConnection()) {
        String schemaResourceName = "/create_h2.sql";
        InputStream in = TestServer.class.getResourceAsStream(schemaResourceName);

        if (in == null) {
            throw new RuntimeException("Failed to load resource: " + schemaResourceName);
        }
        InputStreamReader reader = new InputStreamReader(in, UTF_8);
        RunScript.execute(connection, reader);

    } catch (SQLException e) {
        e.printStackTrace();
    }

}
 

private TestServer() {
    DataSource ds = (DataSource) SingletonServiceFactory.getBean(DataSource.class);
    try (Connection connection = ds.getConnection()) {
        String schemaResourceName = "/create_h2.sql";
        InputStream in = TestServer.class.getResourceAsStream(schemaResourceName);

        if (in == null) {
            throw new RuntimeException("Failed to load resource: " + schemaResourceName);
        }
        InputStreamReader reader = new InputStreamReader(in, UTF_8);
        RunScript.execute(connection, reader);

    } catch (SQLException e) {
        e.printStackTrace();
    }

}
 

@BeforeClass
public static void loadDriverTest() throws Exception {
    Class.forName("org.h2.Driver");
    activeConn = getConnection();

    Connection conn = getConnection();
    RunScript.execute(conn, new FileReader("src/test/resources/testdb.ddl"));
    conn.close();
}
 

@Before
public void createTableAndImportDataSet() throws Exception {
	for (DataSourceEntry entry : getDataSourceEntryArray()) {
		RunScript.execute(entry.getJdbcUrl(), USER, PASSWORD, getSchema(), "UTF8", false);
		cleanlyInsert(JDBC_DRIVER, entry.getJdbcUrl(), USER, PASSWORD, readDataSet(entry.getDataSets()));
		if (entry.isWrite()) {
			writeDs = createRealDataSource(entry.getJdbcUrl(), USER, PASSWORD, JDBC_DRIVER);
		} else {
			readDses.add(createRealDataSource(entry.getJdbcUrl(), USER, PASSWORD, JDBC_DRIVER));
		}
	}
}
 

@Override
public void shutdown() {
  try (Connection conn = getDataSource().getConnection()) {
    RunScript.execute(conn, new StringReader("SHUTDOWN"));
  } catch (SQLException x) {
    throw new RuntimeException(x);
  }
}
 

private static void dropTable(final SchemaEnvironment databaseEnvironmentSchema, final DatabaseType databaseType) {
    for (String each : databaseEnvironmentSchema.getDatabases()) {
        DataSource dataSource = DataSourceUtil.createDataSource(databaseType, each);
        try (Connection connection = dataSource.getConnection();
             StringReader stringReader = new StringReader(Joiner.on(";\n").join(databaseEnvironmentSchema.getTableDropSQLs()))) {
            RunScript.execute(connection, stringReader);
        } catch (final SQLException ex) {
            // TODO table maybe not exist
        }
    }
}
 

@Before
public void initTable() {
    try {
        ShardingSphereConnection conn = shardingSphereDataSource.getConnection();
        RunScript.execute(conn, new InputStreamReader(AbstractSQLTest.class.getClassLoader().getResourceAsStream("jdbc_data.sql")));
        conn.close();
    } catch (final SQLException ex) {
        ex.printStackTrace();
    }
}
 

private void runScript(String fileName) throws SQLException, IOException {
    Connection connection = null;
    try {
        connection = connectionBuilder.getConnection();
        RunScript.execute(connection, load(fileName));
    } finally {
        connection.close();
    }
}
 

/**
 * Prepares test table containing a single data row.
 */
private static void setupDb() throws SQLException, URISyntaxException {
    File script = new File(CamelSqlTest.class.getResource("/init-db.sql").toURI());
    RunScript.execute("jdbc:h2:mem:jbpm-db;MVCC=true",
                      "sa",
                      "",
                      script.getAbsolutePath(),
                      StandardCharsets.UTF_8,
                      false);
}
 

@Before
public void initTable() {
    try (ShardingSphereConnection connection = encryptDataSource.getConnection()) {
        RunScript.execute(connection, new InputStreamReader(AbstractSQLTest.class.getClassLoader().getResourceAsStream("encrypt_data.sql")));
    } catch (final SQLException ex) {
        ex.printStackTrace();
    }
}
 

/**
 * Populate sample database.
 *
 * @throws SQLException if
 */
public static void populateDatabase() throws SQLException {
    // Try to connect to database TCP server.
    JdbcConnectionPool dataSrc = JdbcConnectionPool.create("jdbc:h2:tcp://localhost/mem:ExampleDb", "sa", "");

    // Create Person table in database.
    RunScript.execute(dataSrc.getConnection(), new StringReader(CREATE_PERSON_TABLE));

    // Populates Person table with sample data in database.
    RunScript.execute(dataSrc.getConnection(), new StringReader(POPULATE_PERSON_TABLE));
}
 

private void changeInternalDataServerPassword(String oldPassword, String newPassword) throws Exception {
	org.h2.Driver.load();
	for (Entry<String, DataServer> en : Config.nodes().dataServers().entrySet()) {
		DataServer o = en.getValue();
		if (BooleanUtils.isTrue(o.getEnable())) {
			try (Connection conn = DriverManager.getConnection(
					"jdbc:h2:tcp://" + en.getKey() + ":" + o.getTcpPort() + "/X", "sa", oldPassword)) {
				RunScript.execute(conn, new StringReader("ALTER USER SA SET PASSWORD '" + newPassword + "'"));
			} catch (Exception e) {
				throw new Exception("Verify that the dataServer:" + en.getKey()
						+ " is started and that the dataServer password is updated synchronously.", e);
			}
		}
	}
}
 

/**
 * Start H2 database TCP server.
 *
 * @param args Command line arguments, none required.
 * @throws IgniteException If start H2 database TCP server failed.
 */
public static void main(String[] args) throws IgniteException {
    try {
        // Start H2 database TCP server in order to access sample in-memory database from other processes.
        Server.createTcpServer("-tcpDaemon").start();

        populateDatabase();

        // Try to connect to database TCP server.
        JdbcConnectionPool dataSrc = JdbcConnectionPool.create("jdbc:h2:tcp://localhost/mem:ExampleDb", "sa", "");

        // Create Person table in database.
        RunScript.execute(dataSrc.getConnection(), new StringReader(CREATE_PERSON_TABLE));

        // Populates Person table with sample data in database.
        RunScript.execute(dataSrc.getConnection(), new StringReader(POPULATE_PERSON_TABLE));
    }
    catch (SQLException e) {
        throw new IgniteException("Failed to start database TCP server", e);
    }

    try {
        do {
            System.out.println("Type 'q' and press 'Enter' to stop H2 TCP server...");
        }
        while ('q' != System.in.read());
    }
    catch (IOException ignored) {
        // No-op.
    }
}
 

@Override
public void clean() {
  super.clean();
  if (server != null) {
    server.shutdown();
  } else {
    try (Connection conn = dataSource.getConnection()) {
      RunScript.execute(conn, new StringReader("SHUTDOWN"));
    } catch (SQLException x) {
      throw new RuntimeException(x.getMessage(), x);
    }
  }
}
 

@BeforeClass
public static void createSchema() throws Exception {
    RunScript.execute(CONNECTION_URL, USER_NAME, PASSWORD, DB_SETUP_SCRIPT, StandardCharsets.UTF_8, false);
}
 

@Before
public void createTableAndImportDataSet() throws Exception {
	RunScript.execute(JDBC_URL, USER, PASSWORD, getSchema(), "UTF8", false);
	cleanlyInsert(JDBC_DRIVER, JDBC_URL, USER, PASSWORD, readDataSet());
}
 

private ResultSet execQuery(String query) throws SQLException {
  try (Connection conn = dataSource.getConnection()) {
    return RunScript.execute(conn, new StringReader(query));
  }
}