mirror of
https://github.com/woodchen-ink/aimodels-prices.git
synced 2025-07-18 13:41:59 +08:00
wood chen
aeb05f790a
- 将数据库操作从原生 SQL 迁移到 GORM ORM - 更新模型定义，添加 GORM 标签和关系 - 移除手动创建表的 SQL 方法，改用 AutoMigrate - 更新所有数据库相关处理逻辑以适配 GORM - 升级 Go 版本和依赖库 - 移除数据库和路由中间件，简化项目结构
76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package database
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/logger"
|
||
|
||
"aimodels-prices/config"
|
||
"aimodels-prices/models"
|
||
)
|
||
|
||
// DB 是数据库连接的全局实例
|
||
var DB *gorm.DB
|
||
|
||
// InitDB 初始化数据库连接
|
||
func InitDB(cfg *config.Config) error {
|
||
var err error
|
||
|
||
// 构建MySQL DSN
|
||
dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
||
cfg.DBUser,
|
||
cfg.DBPassword,
|
||
cfg.DBHost,
|
||
cfg.DBPort,
|
||
cfg.DBName,
|
||
)
|
||
|
||
// 连接MySQL
|
||
DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
|
||
Logger: logger.Default.LogMode(logger.Info),
|
||
})
|
||
if err != nil {
|
||
return fmt.Errorf("failed to connect to MySQL: %v", err)
|
||
}
|
||
|
||
// 获取底层的SQL DB
|
||
sqlDB, err := DB.DB()
|
||
if err != nil {
|
||
return fmt.Errorf("failed to get underlying SQL DB: %v", err)
|
||
}
|
||
|
||
// 设置连接池参数
|
||
sqlDB.SetMaxOpenConns(10)
|
||
sqlDB.SetMaxIdleConns(5)
|
||
|
||
// 自动迁移表结构
|
||
if err = migrateModels(); err != nil {
|
||
return fmt.Errorf("failed to migrate models: %v", err)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// migrateModels 自动迁移模型到数据库表
|
||
func migrateModels() error {
|
||
// 自动迁移模型
|
||
if err := DB.AutoMigrate(
|
||
&models.ModelType{},
|
||
&models.Price{},
|
||
&models.Provider{},
|
||
&models.User{},
|
||
&models.Session{},
|
||
); err != nil {
|
||
log.Printf("Failed to migrate tables: %v", err)
|
||
return err
|
||
}
|
||
|
||
// 这里可以添加其他模型的迁移
|
||
// 例如：DB.AutoMigrate(&models.User{})
|
||
|
||
return nil
|
||
}
|