package org.cacert.gigi.util;
import java.io.File;
+import java.io.FileReader;
import java.io.IOException;
import java.nio.file.Files;
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 {
- public static String readFile(File f) throws IOException {
- return new String(Files.readAllBytes(f.toPath()));
- }
- public static void main(String[] args) throws SQLException,
- ClassNotFoundException, IOException {
- if (args.length < 4) {
- System.err
- .println("Usage: com.mysql.jdbc.Driver jdbc:mysql://localhost/cacert user password");
- return;
- }
- run(args);
- }
- public static void run(String[] args) 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"));
- File localData = new File("doc/sampleData.sql");
- if (localData.exists()) {
- addFile(stmt, localData);
- }
- stmt.executeBatch();
- stmt.close();
- }
- private static void addFile(Statement stmt, File f) throws IOException,
- SQLException {
- String sql = readFile(f);
- String[] stmts = sql.split(";");
- for (String string : stmts) {
- if (!string.trim().equals("")) {
- stmt.addBatch(string);
- }
- }
- }
+
+ public static String readFile(File f) throws IOException {
+ return new String(Files.readAllBytes(f.toPath()));
+ }
+
+ 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, false);
+ }
+
+ public static void run(String[] args, boolean 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);
+ 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", ""));
+ }
+ }
+ }
}