- public static void main(String[] args) throws SQLException, ClassNotFoundException, IOException {
- if (args.length == 0) {
- Properties p = new Properties();
- p.load(new FileReader("config/gigi.properties"));
- args = new String[] { p.getProperty("sql.driver"), p.getProperty("sql.url"), p.getProperty("sql.user"),
- p.getProperty("sql.password") };
- }
- if (args.length < 4) {
- System.err.println("Usage: com.mysql.jdbc.Driver jdbc:mysql://localhost/cacert user password");
- return;
- }
- run(args);
- }
+ public static void main(String[] args) throws SQLException, ClassNotFoundException, IOException {
+ boolean test = false;
+ if (args.length >= 1 && args[0].equals("--test")) {
+ test = true;
+ String[] ne = new String[args.length - 1];
+ System.arraycopy(args, 1, ne, 0, ne.length);
+ args = ne;
+ }
+ if (args.length == 0) {
+ Properties p = new 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")
+ };
+ }
+ if (args.length < 4) {
+ System.err.println("Usage: com.mysql.jdbc.Driver jdbc:mysql://localhost/cacert user password");
+ return;
+ }
+ run(args, test ? ImportType.TEST : ImportType.PRODUCTION);
+ }
+
+ 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();
+ 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();
+ stmt.close();