modified AlterBenchmarkTest to support multiple SQL

Author: Dingshilun

script/testing/junit/AlterBenchmarkTest.java

@@ -33,7 +33,13 @@ public class AlterBenchmarkTest extends PLTestBase {
                     "year integer," +
                     "month integer);";
 
+    private static String selectSQL;
+    private static String alterAdd;
+    private static String alterDrop;
+    private static String alterChangeInline;
+    private static String alterChangeNotInline;
 
+    private static enum DBMS {MySQL, Postgres, Peloton};
     @Rule
     public ExpectedException thrown = ExpectedException.none();
 
@@ -50,21 +56,53 @@ public static Connection makePostgresConnection(String host,
                                             int port,
                                             String username,
                                             String pass) throws SQLException {
-        String url = String.format("jdbc:postgresql://%s:%d/postgres",
-                host, port);
+        String url = String.format("jdbc:postgresql://%s:%d/postgres", host, port);
+        Connection conn = DriverManager.getConnection(url, username, pass);
+        return conn;
+    }
+
+    private static Connection makeMySQLConnection(String host,
+                                                  int port,
+                                                  String username,
+                                                  String pass) throws SQLException {
+        String url = String.format("jdbc:mysql://%s:%d/mysql", host, port);
         Connection conn = DriverManager.getConnection(url, username, pass);
         return conn;
     }
 
     /**
      * Setup the connection to peloton or other DBMS
+     * If you want to run other
      * @throws SQLException
      */
     @Before
     public void Setup() throws SQLException {
-        //connection to Postgres
-        //conn = makePostgresConnection("localhost", 5432, "dingshilun", "");
-        conn = makeDefaultConnection();
+        DBMS testingDB = DBMS.Peloton;
+        String userName = "";
+        String passWD = "";
+        switch (testingDB){
+            case Peloton:
+                alterAdd = "alter table tbl add day integer;";
+                alterDrop = "alter table tbl drop month;";
+                alterChangeInline = "alter table tbl alter year type varchar";
+                alterChangeNotInline = "alter table tbl alter year type integer USING year::INTEGER";
+                conn = makeDefaultConnection();
+                break;
+            case Postgres:
+                alterAdd = "alter table tbl add day integer;";
+                alterDrop = "alter table tbl drop month;";
+                alterChangeInline = "alter table tbl alter year type varchar";
+                alterChangeNotInline = "alter table tbl alter year type integer USING year::INTEGER";
+                conn = makePostgresConnection("localhost", 5432, userName, passWD);
+                break;
+            case MySQL:
+                alterAdd = "alter table tbl add day integer;";
+                alterDrop = "alter table tbl drop month;";
+                alterChangeInline = "alter table tbl modify year type varchar";
+                alterChangeNotInline = "alter table tbl modify year type integer";
+                conn = makeMySQLConnection("localhost", 3306, userName, passWD);
+                break;
+        }
         conn.setAutoCommit(true);
         InitDatabase();
     }
@@ -82,6 +120,7 @@ public void Teardown() throws SQLException {
      */
     @Test
     public void test_tuple_number_varies() throws SQLException {
+        //define tuple number
         int[] workload = {};
         for (int i = 0; i< workload.length;i++) {
             // firstly use select * to make sure all tuples are in memory
@@ -96,29 +135,31 @@ public void test_tuple_number_varies() throws SQLException {
             } catch (Exception e) {
                 e.printStackTrace();
             }
-            
-            String alterSql1 = "alter table tbl add day integer;";
+
+            String alterSql1 = alterAdd;
             long startTime1 = System.currentTimeMillis();
             conn.createStatement().execute(alterSql1);
             long endTime1 = System.currentTimeMillis();
 
-            String alterSql2 = "alter table tbl drop month;";
+            String alterSql2 = alterDrop;
             long startTime2 = System.currentTimeMillis();
             conn.createStatement().execute(alterSql2);
             long endTime2 = System.currentTimeMillis();
 
-            String alterSql3 = "alter table tbl alter year type varchar";
+            String alterSql3 = alterChangeInline;
             long startTime3 = System.currentTimeMillis();
             conn.createStatement().execute(alterSql3);
             long endTime3 = System.currentTimeMillis();
 
-            String alterSql4 = "alter table tbl alter year type integer USING year::INTEGER";
+            String alterSql4 = alterChangeNotInline;
             long startTime4 = System.currentTimeMillis();
             conn.createStatement().execute(alterSql4);
             long endTime4 = System.currentTimeMillis();
 
-            System.out.println("Alter add column " + workload[i] + " tuples took: " + (endTime1 - startTime1) + " milliseconds");
-            System.out.println("Alter drop column " + workload[i] + " tuples took: " + (endTime2 - startTime2) + " milliseconds");
+            System.out.println("Alter add column " + workload[i] + " tuples took: "
+                    + (endTime1 - startTime1) + " milliseconds");
+            System.out.println("Alter drop column " + workload[i] + " tuples took: "
+                    + (endTime2 - startTime2) + " milliseconds");
             System.out.println("Alter change type from inline to not inline " + workload[i] + " tuples took: " +
                     (endTime3 - startTime3) + " milliseconds");
             System.out.println("Alter change type from not inline to inline " + workload[i] + " tuples took: " +
@@ -128,33 +169,30 @@ public void test_tuple_number_varies() throws SQLException {
     }
 
     private void NumVarInsertHelper(int insertNum) throws SQLException {
-        String sql = "INSERT INTO tbl VALUES (?, ?, ?);";
-        PreparedStatement pstmt = conn.prepareStatement(sql);
+        String sql;
         for (int i = 0; i < insertNum; i++) {
-            setValues(pstmt, new int [] {i, i+1, i+2});
-            pstmt.addBatch();
+            sql = String.format("INSERT INTO tbl VALUES (%d, %d, %d);", i, i+1, i+2);
+            conn.createStatement().execute(sql);
         }
-        pstmt.executeBatch();
     }
 
     /**
-     * Insert 10000 tuple, and test performance under different
-     * length of the tuple
+     * Insert 'tupleNum' tuple, and test performance under different
+     * length of the tuple defined by workload{}
      * @throws SQLException
      */
     @Test
     public void test_tuple_length_variance() throws SQLException {
+        //define tuple length
         int[] workload = {};
         int tupleNum = 10000;
-        String dropSQL = "DROP TABLE IF EXISTS tbl";
-        String sql = "";
-        conn.createStatement().execute(dropSQL);
+        conn.createStatement().execute(SQL_DROP_TABLE);
         for (int i = 0; i < workload.length; i++) {
-            sql = "CREATE TABLE tbl(id INTEGER PRIMARY KEY, " +
-                    "payload1 VARCHAR(" + workload[i] + ")," +
-                    "payload2 VARCHAR(" + workload[i] + ")," +
-                    "payload3 INTEGER);";
-            conn.createStatement().execute(sql);
+            String createSql = "CREATE TABLE tbl(id INTEGER PRIMARY KEY, " +
+                    "month VARCHAR(" + workload[i] + ")," +
+                    "hour VARCHAR(" + workload[i] + ")," +
+                    "year INTEGER);";
+            conn.createStatement().execute(createSql);
             LengthVarInsertHelper(tupleNum, workload[i]);
 
             try {
@@ -164,25 +202,28 @@ public void test_tuple_length_variance() throws SQLException {
             }
 
             long startTime1 = System.currentTimeMillis();
-            conn.createStatement().execute("ALTER TABLE tbl add payload4 integer;");
+            conn.createStatement().execute(alterAdd);
             long endTime1 = System.currentTimeMillis();
 
             long startTime2 = System.currentTimeMillis();
-            conn.createStatement().execute("ALTER TABLE tbl drop payload1;");
+            conn.createStatement().execute(alterDrop);
             long endTime2 = System.currentTimeMillis();
 
             long startTime3 = System.currentTimeMillis();
-            conn.createStatement().execute("ALTER TABLE tbl alter payload3 type varchar;");
+            conn.createStatement().execute(alterChangeInline);
             long endTime3 = System.currentTimeMillis();
 
-            System.out.println("Alter add column " + workload[i] + " length took: " + (endTime1 - startTime1)
+            System.out.println("Alter add column " + workload[i] * 2 +
+                    " length took: " + (endTime1 - startTime1)
                     + " milliseconds");
-            System.out.println("Alter drop column " + workload[i] + " length took: " + (endTime2 - startTime2)
+            System.out.println("Alter drop column " + workload[i] * 2 +
+                    " length took: " + (endTime2 - startTime2)
                     + " milliseconds");
-            System.out.println("Alter change type from not inline to inline " + workload[i] + " length took: " +
-                    (endTime3 - startTime3) + " milliseconds");
+            System.out.println("Alter change type from not inline to inline " + workload[i] * 2
+                    + " length took: " + (endTime3 - startTime3) +
+                    " milliseconds");
 
-            conn.createStatement().execute(dropSQL);
+            conn.createStatement().execute(SQL_DROP_TABLE);
         }
     }