27 lines
435 B
Go
27 lines
435 B
Go
package database
|
||
|
||
import (
|
||
"log"
|
||
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
var db *gorm.DB
|
||
|
||
// Init 使用 DSN 连接 MySQL，供 handler 通过 DB() 使用
|
||
func Init(dsn string) error {
|
||
var err error
|
||
db, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
log.Println("database: connected")
|
||
return nil
|
||
}
|
||
|
||
// DB 返回全局 *gorm.DB，仅在 Init 成功后调用
|
||
func DB() *gorm.DB {
|
||
return db
|
||
}
|