gen2d/backend/internal/db/db.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...)
}

// GetDB 获取全局数据库实例。
func GetDB() *gorm.DB {
	return DB
}