db/db.go

@@ -41,17 +41,23 @@ func NewDB(c Config) (*DB, error) {
 		c.port = defaultPort
 	}
 
-	if len(c.sslMode) == 0 {
-		c.sslMode = defaultSslMode
+	var source string
+
+	switch c.driver {
+	case "mysql":
+		source = c.username +
+			":" + c.password +
+			"@tcp(" + c.host + ":" + strconv.Itoa(int(c.port)) + ")/" +
+			c.dbname +
+			"?parseTime=true"
+	case "postgres":
+		source = "user=" + c.username +
+			" password=" + c.password +
+			" dbname=/" + c.dbname +
+			" host=" + c.host +
+			" port=" + strconv.Itoa(int(c.port))
 	}
 
-	source := "user=" + c.username +
-		" password=" + c.password +
-		" dbname=" + c.dbname +
-		" host=" + c.host +
-		" port=" + strconv.Itoa(int(c.port)) +
-		" sslmode=" + c.sslMode
-
 	sqlConn, err := sql.Open(c.driver, source)
 	if err != nil {
 		return nil, fmt.Errorf("open DB connection error: %w", err)

db/db_config.go

@@ -5,10 +5,9 @@ import (
 )
 
 const (
-	defaultDriver  = "mysql"
-	defaultHost    = "localhost"
-	defaultPort    = 3306
-	defaultSslMode = "disable"
+	defaultDriver = "mysql"
+	defaultHost   = "localhost"
+	defaultPort   = 3306
 )
 
 type Config struct {
@@ -19,7 +18,6 @@ type Config struct {
 	dbname   string
 	host     string
 	port     uint16
-	sslMode  string
 
 	maxOpenConns uint16
 	maxIdleConns uint16
@@ -81,12 +79,6 @@ func WithPort(port uint16) ConfigOption {
 	}
 }
 
-func WithSSLMode(sslMode string) ConfigOption {
-	return func(s *Config) {
-		s.sslMode = sslMode
-	}
-}
-
 func WithMaxOpenConns(maxOpenConns uint16) ConfigOption {
 	return func(s *Config) {
 		s.maxOpenConns = maxOpenConns