Improve database connection

This commit is contained in:
kirby 2024-03-18 15:54:01 +01:00
parent 5a4a87cc9b
commit 6879221996
2 changed files with 13 additions and 15 deletions

View File

@ -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"

View File

@ -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)
}