feat: 添加 db(gorm) 支持

This commit is contained in:
zhaoyupeng
2025-06-17 14:02:19 +08:00
parent 1a35c30630
commit b3fdecd62f
6 changed files with 222 additions and 7 deletions

22
database/db/db.go Normal file
View File

@ -0,0 +1,22 @@
package db
import (
"context"
"gorm.io/gorm"
)
type DB interface {
Session(ctx context.Context) *gorm.DB
}
type db struct {
tx *gorm.DB
}
var (
Default DB
)
func (db *db) Session(ctx context.Context) *gorm.DB {
return db.tx.Session(&gorm.Session{Context: ctx})
}