add configurable options for ping check and sslmode

README.md

@@ -100,11 +100,13 @@ port="5432"
 user="postgres"
 dbname="postgres"
 password="xxxxx" 
+sslmode="disable"
 maxIdleConn = 2
 maxOpenConn = 2
 
 [app]
 debug = true
+pingCheck = true
 
 ```
 

kshieldconfig_example.toml

@@ -5,6 +5,7 @@
 # port = "5432"
 # user = "postgres"
 # password = "password123"
+# sslmode = "disable"
 # dbname = "mydb"
 # maxIdleConn = 10
 # maxOpenConn = 100

pkg/config/config.go

@@ -235,6 +235,7 @@ type GeneratePassword struct {
 type App struct {
 	Debug              bool   `toml:"debug"`
 	Hostname           string `toml:"hostname"`
+	PingCheck					 bool   `toml:"pingCheck"`
 	Run                bool
 	RunPostgres        bool
 	RunMySql           bool

pkg/mysqldb/connect.go

@@ -28,10 +28,13 @@ func Open(conf config.MySQL) (*sql.DB, string, error) {
 			Msg("Failed to connect to database")
 		return nil, "", err
 	}
-	err = db.Ping()
-	if err != nil {
-		// fmt.Printf("Failed to connect to database. Error:	%s", err.Error())
-		return nil, "", err
+
+	if cnf.App.PingCheck {
+		err = db.Ping()
+		if err != nil {
+			// fmt.Printf("Failed to connect to database. Error:	%s", err.Error())
+			return nil, "", err
+		}
 	}
 	if conf.MaxIdleConn > 0 {
 		db.SetMaxIdleConns(conf.MaxIdleConn)

pkg/postgresdb/connect.go

@@ -15,7 +15,7 @@ type Postgres struct {
 	User     string `toml:"user"`
 	Password string `toml:"password"`
 	DBName   string `toml:"dbname"`
-	// SSLmode     string `toml:"sslmode"`
+	SSLmode     string `toml:"sslmode"`
 	MaxIdleConn int `toml:"maxIdleConn"`
 	MaxOpenConn int `toml:"maxOpenConn"`
 }
@@ -35,7 +35,7 @@ var re = regexp.MustCompile(`(?m)(?:host=)([^\s]+)`)
 
 func Open(conf Postgres) (*sql.DB, string, error) {
 	// "host=localhost port=5432 user=postgres password=postgres dbname=postgres sslmode=disable"
-	url := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=disable", conf.Host, conf.Port, conf.User, conf.Password, conf.DBName)
+	url := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=%s", conf.Host, conf.Port, conf.User, conf.Password, conf.DBName)
 
 	db, err := ConnectDatabaseUsingConnectionString(url)
 	if err != nil {
@@ -78,14 +78,16 @@ func ConnectDatabaseUsingConnectionString(url string) (*sql.DB, error) {
 		return nil, err
 	}
 
-	err = db.Ping()
-	if err != nil {
-		log.Error().
-			Err(err).
-			Str("conn", url).
-			Msg("Failed to ping database")
-		db.Close()
-		return nil, err
+	if cnf.App.PingCheck {
+		err = db.Ping()
+		if err != nil {
+			log.Error().
+				Err(err).
+				Str("conn", url).
+				Msg("Failed to ping database")
+			db.Close()
+			return nil, err
+		}
 	}
 
 	return db, nil