48 lines
913 B
Go
48 lines
913 B
Go
package config
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
_ "github.com/go-sql-driver/mysql"
|
|
)
|
|
|
|
var DB *sql.DB
|
|
|
|
// JWTSecret is the secret key used for signing JWT tokens
|
|
var JWTSecret = "your-secret-key-change-this-in-production" // Default value
|
|
|
|
func InitDB() {
|
|
var err error
|
|
|
|
// Try to get DSN from environment variable, otherwise use default
|
|
dsn := os.Getenv("DB_DSN")
|
|
if dsn == "" {
|
|
dsn = "root:root@tcp(127.0.0.1:3306)/nl_blog?charset=utf8mb4&parseTime=True&loc=Local"
|
|
}
|
|
|
|
// Try to get JWT secret from environment variable
|
|
if secret := os.Getenv("JWT_SECRET"); secret != "" {
|
|
JWTSecret = secret
|
|
}
|
|
|
|
DB, err = sql.Open("mysql", dsn)
|
|
if err != nil {
|
|
log.Fatal("Failed to connect to database:", err)
|
|
}
|
|
|
|
if err = DB.Ping(); err != nil {
|
|
log.Fatal("Failed to ping database:", err)
|
|
}
|
|
|
|
fmt.Println("Database connected successfully")
|
|
}
|
|
|
|
func CloseDB() {
|
|
if DB != nil {
|
|
DB.Close()
|
|
}
|
|
}
|