1 package org.cacert.gigi.util;
4 import java.io.FileInputStream;
5 import java.io.FileReader;
6 import java.io.IOException;
7 import java.io.InputStream;
8 import java.sql.Connection;
9 import java.sql.DriverManager;
10 import java.sql.SQLException;
11 import java.sql.Statement;
12 import java.util.Properties;
14 import org.cacert.gigi.database.DatabaseConnection;
15 import org.cacert.gigi.database.SQLFileManager;
16 import org.cacert.gigi.database.SQLFileManager.ImportType;
18 public class DatabaseManager {
20 public static void main(String[] args) throws SQLException, ClassNotFoundException, IOException {
22 if (args.length >= 1 && args[0].equals("--test")) {
24 String[] ne = new String[args.length - 1];
25 System.arraycopy(args, 1, ne, 0, ne.length);
28 if (args.length == 0) {
29 Properties p = new Properties();
30 try (FileReader reader = new FileReader("config/gigi.properties")) {
34 p.getProperty("sql.driver"), p.getProperty("sql.url"), p.getProperty("sql.user"), p.getProperty("sql.password")
37 if (args.length < 4) {
38 System.err.println("Usage: com.mysql.jdbc.Driver jdbc:mysql://localhost/cacert user password");
41 run(args, test ? ImportType.TEST : ImportType.PRODUCTION);
44 public static void run(String[] args, ImportType truncate) throws ClassNotFoundException, SQLException, IOException {
45 Class.forName(args[0]);
46 Connection conn = DriverManager.getConnection(args[1], args[2], args[3]);
47 conn.setAutoCommit(false);
48 Statement stmt = conn.createStatement();
49 try (InputStream structure = DatabaseConnection.class.getResourceAsStream("tableStructure.sql")) {
50 SQLFileManager.addFile(stmt, structure, truncate);
52 File localData = new File("doc/sampleData.sql");
53 if (localData.exists()) {
54 try (FileInputStream f = new FileInputStream(localData)) {
55 SQLFileManager.addFile(stmt, f, ImportType.PRODUCTION);