kirby/infra-dashboard
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve database connection

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
.env.example
View File

@ -1,4 +1,4 @@
DATABASE_HOST= DATABASE_HOST="127.0.0.1:3306"
DATABASE_USER= DATABASE_USER="test"
DATABASE_PASSWORD= DATABASE_PASSWORD="test"
DATABASE_NAME= DATABASE_NAME="test"

20
Database/utils.go
View File

@ -2,24 +2,22 @@ package database
import ( import (
"database/sql" "database/sql"
"fmt"
"log" "log"
"os" "os"
"github.com/go-sql-driver/mysql"
_ "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 { 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 { if err != nil {
log.Println("Database connection error : ", err) log.Println("Database connection error : ", err)
} }