refactor: reorganize packages into a domain-based layered structure

This commit is contained in:
Leon
2025-12-05 09:59:57 +08:00
parent 34b05ad83f
commit c1862b5475
8 changed files with 83 additions and 94 deletions

34
internal/db/db.go Normal file
View File

@@ -0,0 +1,34 @@
package db
import (
"feedsystem_video_go/internal/account"
"feedsystem_video_go/internal/config"
"fmt"
"gorm.io/driver/mysql"
"gorm.io/gorm"
)
func NewDB(dbcfg config.DatabaseConfig) (*gorm.DB, error) {
dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
dbcfg.User, dbcfg.Password, dbcfg.Host, dbcfg.Port, dbcfg.DBName)
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
if err != nil {
return nil, err
}
return db, nil
}
func AutoMigrate(db *gorm.DB) error {
return db.AutoMigrate(&account.User{})
}
func CloseDB(db *gorm.DB) error {
sqlDB, err := db.DB()
if err != nil {
return err
}
return sqlDB.Close()
}