Improve database connection
This commit is contained in:
parent
5a4a87cc9b
commit
6879221996
@ -1,4 +1,4 @@
|
||||
DATABASE_HOST=
|
||||
DATABASE_USER=
|
||||
DATABASE_PASSWORD=
|
||||
DATABASE_NAME=
|
||||
DATABASE_HOST="127.0.0.1:3306"
|
||||
DATABASE_USER="test"
|
||||
DATABASE_PASSWORD="test"
|
||||
DATABASE_NAME="test"
|
||||
|
||||
@ -2,24 +2,22 @@ package database
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"github.com/go-sql-driver/mysql"
|
||||
_ "github.com/go-sql-driver/mysql"
|
||||
)
|
||||
|
||||
func dsn() string {
|
||||
user := os.Getenv("DATABASE_USER")
|
||||
password := os.Getenv("DATABASE_PASSWORD")
|
||||
host := os.Getenv("DATABASE_HOST")
|
||||
port := os.Getenv("DATABASE_PORT")
|
||||
name := os.Getenv("DATABASE_NAME")
|
||||
return fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", user, password, host, port, name)
|
||||
}
|
||||
|
||||
func GetDatabaseConnection() *sql.DB {
|
||||
DB_conn, err := sql.Open("mysql", dsn())
|
||||
cfg := mysql.Config{
|
||||
User: os.Getenv("DATABASE_USER"),
|
||||
Passwd: os.Getenv("DATABASE_PASSWORD"),
|
||||
Net: "tcp",
|
||||
Addr: os.Getenv("DATABASE_HOST"),
|
||||
DBName: os.Getenv("DATABASE_NAME"),
|
||||
}
|
||||
DB_conn, err := sql.Open("mysql", cfg.FormatDSN())
|
||||
if err != nil {
|
||||
log.Println("Database connection error : ", err)
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user