Add 'backend/' from commit '4b5069d17945252abee2a826818098906af56334'
git-subtree-dir: backend git-subtree-mainline:9c93274511git-subtree-split:4b5069d179
This commit is contained in:
@@ -0,0 +1,37 @@
|
||||
package bd
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"log"
|
||||
"os"
|
||||
"github.com/go-sql-driver/mysql"
|
||||
"github.com/joho/godotenv"
|
||||
)
|
||||
|
||||
var db *sql.DB
|
||||
|
||||
func GetDB() *sql.DB {
|
||||
|
||||
if db == nil {
|
||||
errenv := godotenv.Load()
|
||||
if errenv != nil {
|
||||
log.Println("No .env file found; using environment variables")
|
||||
}
|
||||
// Capture connection properties.
|
||||
cfg := mysql.NewConfig()
|
||||
cfg.User = os.Getenv("DB_USER")
|
||||
cfg.Passwd = os.Getenv("DB_PASS")
|
||||
cfg.Net = "tcp"
|
||||
cfg.Addr = os.Getenv("DB_HOST")
|
||||
cfg.DBName = os.Getenv("DB_NAME")
|
||||
|
||||
// Get a database handle.
|
||||
var err error
|
||||
db, err = sql.Open("mysql", cfg.FormatDSN())
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
}
|
||||
return db
|
||||
}
|
||||
Reference in New Issue
Block a user