21 lines
380 B
Go
21 lines
380 B
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"gorm.io/driver/sqlite"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
// DB 全局数据库实例，Init 后可用。
|
||
|
|
var DB *gorm.DB
|
||
|
|
|
||
|
|
// Init 初始化 SQLite 数据库连接并执行自动迁移。
|
||
|
|
func Init(dsn string, models ...any) error {
|
||
|
|
var err error
|
||
|
|
DB, err = gorm.Open(sqlite.Open(dsn), &gorm.Config{})
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return DB.AutoMigrate(models...)
|
||
|
|
}
|