From 3e9c9dbbcd3f6157bf80f1dcbef6fab0fd59924d Mon Sep 17 00:00:00 2001
From: urmsone <31981402+UrmsOne@users.noreply.github.com>
Date: Fri, 27 Dec 2019 10:51:38 +0800
Subject: [PATCH] Get dbUser from Env or default('root') (#985)

* Get dbUser from Env or default('root')

* Get dbUser from Env or default('root')
---
 pkg/db/v1alpha3/common/const.go | 3 +++
 pkg/db/v1alpha3/mysql/mysql.go  | 9 ++++++---
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/pkg/db/v1alpha3/common/const.go b/pkg/db/v1alpha3/common/const.go
index 2c5b03483be..abc42df4ec0 100644
--- a/pkg/db/v1alpha3/common/const.go
+++ b/pkg/db/v1alpha3/common/const.go
@@ -1,6 +1,8 @@
 package common
 
 const (
+	DBUserEnvName = "DB_USER"
+
 	DBNameEnvName = "DB_NAME"
 
 	MySqlDBNameEnvValue = "mysql"
@@ -11,6 +13,7 @@ const (
 	MySQLDBPortEnvName = "KATIB_MYSQL_PORT"
 	MySQLDatabase      = "KATIB_MYSQL_DATABASE"
 
+	DefaultMySQLUser     = "root"
 	DefaultMySQLDatabase = "katib"
 	DefaultMySQLHost     = "katib-db"
 	DefaultMySQLPort     = "3306"
diff --git a/pkg/db/v1alpha3/mysql/mysql.go b/pkg/db/v1alpha3/mysql/mysql.go
index 28378ef52cd..d6a521ec504 100644
--- a/pkg/db/v1alpha3/mysql/mysql.go
+++ b/pkg/db/v1alpha3/mysql/mysql.go
@@ -18,8 +18,9 @@ import (
 )
 
 const (
-	dbDriver     = "mysql"
-	dbNameTmpl   = "root:%s@tcp(%s:%s)/%s?timeout=5s"
+	dbDriver = "mysql"
+	//dbNameTmpl   = "root:%s@tcp(%s:%s)/%s?timeout=5s"
+	dbNameTmpl   = "%s:%s@tcp(%s:%s)/%s?timeout=5s"
 	mysqlTimeFmt = "2006-01-02 15:04:05.999999"
 
 	connectInterval = 5 * time.Second
@@ -33,6 +34,8 @@ type dbConn struct {
 func getDbName() string {
 	dbPassEnvName := common.DBPasswordEnvName
 	dbPass := os.Getenv(dbPassEnvName)
+	dbUser := env.GetEnvOrDefault(
+		common.DBUserEnvName, common.DefaultMySQLUser)
 	dbHost := env.GetEnvOrDefault(
 		common.MySQLDBHostEnvName, common.DefaultMySQLHost)
 	dbPort := env.GetEnvOrDefault(
@@ -40,7 +43,7 @@ func getDbName() string {
 	dbName := env.GetEnvOrDefault(common.MySQLDatabase,
 		common.DefaultMySQLDatabase)
 
-	return fmt.Sprintf(dbNameTmpl, dbPass, dbHost, dbPort, dbName)
+	return fmt.Sprintf(dbNameTmpl, dbUser, dbPass, dbHost, dbPort, dbName)
 }
 
 func openSQLConn(driverName string, dataSourceName string, interval time.Duration,