@Test
public void testCreateFromURI() throws Exception {
URI uri = testFile.toURI();
sm.register("test", uri);
assertTrue(sm.getSource("test").isFileSource());
uri =
URI.create(
"postgresql://www.host.com:1234/db_name?table=toto&schema=tata&"
+ "user=me&password=changeme");
sm.register("test2", uri);
assertTrue(sm.getSource("test2").isDBSource());
DBSource s = sm.getSource("test2").getDBSource();
assertEquals("www.host.com", s.getHost());
assertEquals("db_name", s.getDbName());
assertEquals(1234, s.getPort());
assertEquals("jdbc:postgresql", s.getPrefix());
assertEquals("toto", s.getTableName());
assertEquals("tata", s.getSchemaName());
assertEquals("me", s.getUser());
assertEquals("changeme", s.getPassword());
}
public boolean isConnected() throws Exception {
File f = new File(sqlScriptFile);
if (!f.exists()) {
return false;
}
Class.forName(jdbcDriver);
String connectionString = dbSource.getPrefix() + ":";
if (dbSource.getHost() != null) {
connectionString += "//" + dbSource.getHost();
if (dbSource.getPort() != -1) {
connectionString += (":" + dbSource.getPort());
}
connectionString += "/";
}
connectionString += (dbSource.getDbName());
try {
Connection c =
DriverManager.getConnection(connectionString, dbSource.getUser(), dbSource.getPassword());
} catch (SQLException ex) {
return false;
}
return true;
}
public void create(DataSourceFactory dsf) throws Exception {
FileInputStream fis = new FileInputStream(sqlScriptFile);
DataInputStream dis = new DataInputStream(fis);
byte[] buffer = new byte[(int) fis.getChannel().size()];
dis.readFully(buffer);
String script = new String(buffer);
Class.forName(jdbcDriver);
String connectionString = dbSource.getPrefix() + ":";
if (dbSource.getHost() != null) {
connectionString += "//" + dbSource.getHost();
if (dbSource.getPort() != -1) {
connectionString += (":" + dbSource.getPort());
}
connectionString += "/";
}
connectionString += (dbSource.getDbName());
DriverManager.setLoginTimeout(2);
Connection c =
DriverManager.getConnection(connectionString, dbSource.getUser(), dbSource.getPassword());
Statement st = c.createStatement();
String[] statements = script.split("\\Q;\\E");
for (String statement : statements) {
try {
st.execute(statement);
} catch (SQLException e) {
System.err.println(statement);
e.printStackTrace();
}
}
st.close();
c.close();
DBTableSourceDefinition def = new DBTableSourceDefinition(dbSource);
dsf.getSourceManager().remove(name);
dsf.getSourceManager().register(name, def);
}