BUG修复,返回结构统一
This commit is contained in:
47
server/config/config.go
Normal file
47
server/config/config.go
Normal file
@@ -0,0 +1,47 @@
|
||||
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()
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user