This commit is contained in:
2025-11-26 11:31:09 +08:00
parent dd38813636
commit 7c87114b30
9 changed files with 65 additions and 46 deletions
+2 -8
View File
@@ -8,20 +8,13 @@ import (
"time"
_ "github.com/go-sql-driver/mysql"
"github.com/joho/godotenv"
)
// DB is the global database connection pool
var DB *sql.DB
func InitDB() (*sql.DB, error) {
// Load environment variables from .env file
err := godotenv.Load()
if err != nil {
return nil, fmt.Errorf("error loading .env file: %v", err)
}
// Get connection details from environment variables
// Get connection details from environment variables (loaded in main)
connStr := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?parseTime=true",
os.Getenv("DB_USER"),
os.Getenv("DB_PASSWORD"),
@@ -31,6 +24,7 @@ func InitDB() (*sql.DB, error) {
)
// Open the database connection
var err error
DB, err = sql.Open("mysql", connStr)
if err != nil {
return nil, fmt.Errorf("error opening database: %v", err)