import java.io.File;
import java.io.FileInputStream;
-import java.io.FileReader;
import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.io.Reader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
}
if (args.length == 0) {
Properties p = new Properties();
- p.load(new FileReader("config/gigi.properties"));
+ try (Reader reader = new InputStreamReader(new FileInputStream("config/gigi.properties"), "UTF-8")) {
+ p.load(reader);
+ }
args = new String[] {
p.getProperty("sql.driver"), p.getProperty("sql.url"), p.getProperty("sql.user"), p.getProperty("sql.password")
};
public static void run(String[] args, ImportType truncate) throws ClassNotFoundException, SQLException, IOException {
Class.forName(args[0]);
- Connection conn = DriverManager.getConnection(args[1], args[2], args[3]);
- conn.setAutoCommit(false);
- Statement stmt = conn.createStatement();
- SQLFileManager.addFile(stmt, DatabaseConnection.class.getResourceAsStream("tableStructure.sql"), truncate);
- File localData = new File("doc/sampleData.sql");
- if (localData.exists()) {
- SQLFileManager.addFile(stmt, new FileInputStream(localData), ImportType.PRODUCTION);
+ final Connection conn = DriverManager.getConnection(args[1], args[2], args[3]);
+ try {
+ conn.setAutoCommit(false);
+ Statement stmt = conn.createStatement();
+ try {
+ try (InputStream structure = DatabaseConnection.class.getResourceAsStream("tableStructure.sql")) {
+ SQLFileManager.addFile(stmt, structure, truncate);
+ }
+ File localData = new File("doc/sampleData.sql");
+ if (localData.exists()) {
+ try (FileInputStream f = new FileInputStream(localData)) {
+ SQLFileManager.addFile(stmt, f, ImportType.PRODUCTION);
+ }
+ }
+ stmt.executeBatch();
+ conn.commit();
+ } finally {
+ stmt.close();
+ }
+ } finally {
+ conn.close();
}
- stmt.executeBatch();
- conn.commit();
- stmt.close();
}
-
}