refactor(loader): add a useSSL option for mysql (#650)

---------

Co-authored-by: imbajin <[email protected]>

Author: MuLeiSY2021

hugegraph-loader/src/test/java/org/apache/hugegraph/loader/test/functional/DBUtil.java

@@ -17,13 +17,13 @@
 
 package org.apache.hugegraph.loader.test.functional;
 
+import org.apache.hugegraph.loader.exception.LoadException;
+
 import java.sql.Connection;
 import java.sql.DriverManager;
 import java.sql.SQLException;
 import java.sql.Statement;
 
-import org.apache.hugegraph.loader.exception.LoadException;
-
 public class DBUtil {
 
     private final String driver;
@@ -40,33 +40,28 @@ public DBUtil(String driver, String url, String user, String pass) {
         this.pass = pass;
     }
 
-    public void connect() {
+    public void connect(boolean useSSL) {
         try {
             Class.forName(this.driver);
-            String url = String.format("%s?%s", this.url, "useSSL=false");
+            String url = String.format("%s?useSSL=%s", this.url, useSSL);
             this.conn = DriverManager.getConnection(url, this.user, this.pass);
         } catch (ClassNotFoundException e) {
-            throw new LoadException("Invalid driver class '%s'",
-                                    e, this.driver);
+            throw new LoadException("Invalid driver class '%s'", e, this.driver);
         } catch (SQLException e) {
-            throw new LoadException("Failed to connect database via '%s'",
-                                    e, this.url);
+            throw new LoadException("Failed to connect database via '%s'", e, this.url);
         }
     }
 
-    public void connect(String database) {
+    public void connect(String database, boolean useSSL) {
         this.close();
-        String url = String.format("%s/%s?%s", this.url, database,
-                                   "useSSL=false");
+        String url = String.format("%s/%s?useSSL=%s", this.url, database, useSSL);
         try {
             Class.forName(this.driver);
             this.conn = DriverManager.getConnection(url, this.user, this.pass);
         } catch (ClassNotFoundException e) {
-            throw new LoadException("Invalid driver class '%s'",
-                                    e, this.driver);
+            throw new LoadException("Invalid driver class '%s'", e, this.driver);
         } catch (SQLException e) {
-            throw new LoadException("Failed to connect database via '%s'",
-                                    e, this.url);
+            throw new LoadException("Failed to connect database via '%s'", e, this.url);
         }
     }
 

hugegraph-loader/src/test/java/org/apache/hugegraph/loader/test/functional/JDBCLoadTest.java

@@ -48,16 +48,17 @@ public class JDBCLoadTest extends LoadTest {
     private static final String PASS = "root";
 
     private static final DBUtil dbUtil = new DBUtil(DRIVER, DB_URL, USER, PASS);
+    private static final boolean USE_SSL = false;
 
     @BeforeClass
     public static void setUp() {
         clearServerData();
 
-        dbUtil.connect();
-        // create the database
+        dbUtil.connect(USE_SSL);
+        // create database
         dbUtil.execute(String.format("CREATE DATABASE IF NOT EXISTS `%s`;", DATABASE));
         // create tables
-        dbUtil.connect(DATABASE);
+        dbUtil.connect(DATABASE, USE_SSL);
         // vertex person
         dbUtil.execute("CREATE TABLE IF NOT EXISTS `person` (" +
                        "`id` int(10) unsigned NOT NULL," +