Blizzard
de2b72221c
- UpdatePet 的 fields map 之前漏了 birthday 和 arrived_at， 导致编辑档案改到家日期不落库（首页天数一直显示「-」） - DisableForeignKeyConstraintWhenMigrating：外键列自动索引和字段 index tag 撞成同名索引，父子表同迁时报 1061，关掉库级外键约束创建 Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
"gorm.io/gorm/schema"
|
|
|
|
"github.com/sundynix/pets-be/internal/config"
|
|
"github.com/sundynix/pets-be/internal/model"
|
|
)
|
|
|
|
// New 建立 GORM 连接，所有表名统一加 sundynix_ 前缀
|
|
func New(cfg *config.Config) (*gorm.DB, error) {
|
|
logLevel := logger.Info
|
|
if cfg.Server.Mode == "release" {
|
|
logLevel = logger.Warn
|
|
}
|
|
|
|
db, err := gorm.Open(mysql.Open(cfg.MySQL.DSN()), &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{
|
|
TablePrefix: "sundynix_",
|
|
SingularTable: false,
|
|
},
|
|
// 关联在 Go 层管，不靠库级外键。迁移期建外键时 MySQL 会为外键列自动
|
|
// 补一个索引，和字段上的 `index` tag 撞成同名索引，父子表同迁时第二次
|
|
// 建索引报 1061 duplicate。关掉外键约束创建，索引就只由 index tag 建一次
|
|
DisableForeignKeyConstraintWhenMigrating: true,
|
|
Logger: logger.Default.LogMode(logLevel),
|
|
})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("gorm open: %w", err)
|
|
}
|
|
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
sqlDB.SetMaxOpenConns(50)
|
|
sqlDB.SetMaxIdleConns(10)
|
|
sqlDB.SetConnMaxLifetime(time.Hour)
|
|
|
|
return db, nil
|
|
}
|
|
|
|
// Migrate 自动建表
|
|
func Migrate(db *gorm.DB) error {
|
|
return db.AutoMigrate(model.AllModels()...)
|
|
}
|