added connection string to config

2021-11-04 20:28:37 +01:00
parent 9be7b6c18f
commit 8725def3a1
5 changed files with 11 additions and 9 deletions
--- a/internal/db/db.go
+++ b/internal/db/db.go
@@ -19,18 +19,16 @@ type DB struct {
 	dbConn *sql.DB
 }

-func NewDB(driver, dataSourceName string) (*DB, error) {
-	db, err := sql.Open(driver, dataSourceName)
+func NewDB(dataSourceName string) (*DB, error) {
+	db, err := sql.Open("mysql", dataSourceName)

 	if err != nil {
 		return nil, err
 	}

-	if driver == "mysql" {
-		db.SetConnMaxLifetime(time.Minute * 3)
-		db.SetMaxOpenConns(10)
-		db.SetMaxIdleConns(10)
-	}
+	db.SetConnMaxLifetime(time.Minute * 3)
+	db.SetMaxOpenConns(10)
+	db.SetMaxIdleConns(10)

 	return &DB{
 		dbConn: db,