Add command line option to override default connect param

database.go

@@ -62,6 +62,7 @@ type ConnectionConfig struct {
 	Host     string
 	Port     int
 	Database string
+	Params   string
 }
 
 /*

dbbench.go

@@ -92,6 +92,8 @@ func init() {
 		"Database connection port")
 	flag.StringVar(&GlobalConfig.Database, "database", "",
 		"Database connection database")
+	flag.StringVar(&GlobalConfig.Params, "params", "",
+		"Override default connection parameters")
 }
 
 func main() {

sql_database.go

@@ -195,28 +195,31 @@ func checkSQLQuery(q string) error {
 }
 
 func mySQLDataSourceName(cc *ConnectionConfig) string {
-	return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?allowAllFiles=true&interpolateParams=true",
+	return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?%s",
 		firstString(cc.Username, "root"),
 		firstString(cc.Password, ""),
 		firstString(cc.Host, "localhost"),
 		firstInt(cc.Port, 3306),
-		firstString(cc.Database, ""))
+		firstString(cc.Database, ""),
+		firstString(cc.Params, "allowAllFiles=true&interpolateParams=true"))
 }
 
 func postgresDataSourceName(cc *ConnectionConfig) string {
-	return fmt.Sprintf("postgres://%s:%s@%s:%d/%s",
+	return fmt.Sprintf("postgres://%s:%s@%s:%d/%s?%s",
 		firstString(cc.Username, "root"),
 		firstString(cc.Password, ""),
 		firstString(cc.Host, "localhost"),
 		firstInt(cc.Port, 5432),
-		firstString(cc.Database, ""))
+		firstString(cc.Database, ""),
+		firstString(cc.Params, "sslmode=disable"))
 }
 
 func sqlServerDataSourceName(cc *ConnectionConfig) string {
-	return fmt.Sprintf("user id=%s;password=%s;server=%s;port=%d;database=%s",
+	return fmt.Sprintf("user id=%s;password=%s;server=%s;port=%d;database=%s;%s",
 		firstString(cc.Username, "root"),
 		firstString(cc.Password, ""),
 		firstString(cc.Host, "localhost"),
 		firstInt(cc.Port, 1433),
-		firstString(cc.Database, ""))
+		firstString(cc.Database, ""),
+		firstString(cc.Params, ""))
 }