package org.cacert.gigi.util;
import java.io.File;
-import java.io.FileReader;
+import java.io.FileInputStream;
import java.io.IOException;
-import java.nio.file.Files;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.io.Reader;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Properties;
-import java.util.regex.Matcher;
-import java.util.regex.Pattern;
-public class DatabaseManager {
+import org.cacert.gigi.database.DatabaseConnection;
+import org.cacert.gigi.database.SQLFileManager;
+import org.cacert.gigi.database.SQLFileManager.ImportType;
- public static String readFile(File f) throws IOException {
- return new String(Files.readAllBytes(f.toPath()));
- }
+public class DatabaseManager {
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();
- 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")
};
System.err.println("Usage: com.mysql.jdbc.Driver jdbc:mysql://localhost/cacert user password");
return;
}
- run(args, false);
+ run(args, test ? ImportType.TEST : ImportType.PRODUCTION);
}
- public static void run(String[] args, boolean truncate) throws ClassNotFoundException, SQLException, IOException {
+ 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]);
- Statement stmt = conn.createStatement();
- addFile(stmt, new File("doc/tableStructure.sql"), truncate);
- File localData = new File("doc/sampleData.sql");
- if (localData.exists()) {
- addFile(stmt, localData, false);
- }
- stmt.executeBatch();
- stmt.close();
- }
-
- private static void addFile(Statement stmt, File f, boolean truncate) throws IOException, SQLException {
- String sql = readFile(f);
- sql = sql.replaceAll("--[^\n]+\n", "\n");
- String[] stmts = sql.split(";");
- Pattern p = Pattern.compile("\\s*DROP TABLE IF EXISTS `([^`]+)`");
- for (String string : stmts) {
- Matcher m = p.matcher(string);
- if (m.matches()) {
- String sql2 = "TRUNCATE `" + m.group(1) + "`";
- stmt.addBatch(sql2);
- }
- if ( !string.trim().equals("") && ( !truncate || string.contains("INSERT"))) {
- stmt.addBatch(string.replace("ENGINE=Memory", ""));
+ 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();
}
}
}