Blizzard
60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package svc
|
|
|
|
import (
|
|
plantModel "sundynix-micro-go/app/plant/model"
|
|
"sundynix-micro-go/app/plant/rpc/internal/config"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type ServiceContext struct {
|
|
Config config.Config
|
|
DB *gorm.DB
|
|
}
|
|
|
|
func NewServiceContext(c config.Config) *ServiceContext {
|
|
db, err := gorm.Open(mysql.Open(c.DB.DataSource), &gorm.Config{})
|
|
if err != nil {
|
|
logx.Errorf("连接数据库失败: %v", err)
|
|
panic(err)
|
|
}
|
|
|
|
if err := db.AutoMigrate(
|
|
&plantModel.UserProfile{},
|
|
&plantModel.MyPlant{},
|
|
&plantModel.MyPlantOss{},
|
|
&plantModel.CarePlan{},
|
|
&plantModel.CareRecord{},
|
|
&plantModel.CareTask{},
|
|
&plantModel.GrowthRecord{},
|
|
&plantModel.Wiki{},
|
|
&plantModel.WikiClass{},
|
|
&plantModel.WikiOss{},
|
|
&plantModel.WikiClassRelation{},
|
|
&plantModel.WikiRelated{},
|
|
&plantModel.UserStar{},
|
|
&plantModel.Post{},
|
|
&plantModel.PostComment{},
|
|
&plantModel.PostLike{},
|
|
&plantModel.PostOss{},
|
|
&plantModel.Topic{},
|
|
&plantModel.MediaCheckResult{},
|
|
&plantModel.ExchangeItem{},
|
|
&plantModel.ExchangeOrder{},
|
|
&plantModel.LevelConfig{},
|
|
&plantModel.BadgeConfig{},
|
|
&plantModel.UserBadge{},
|
|
&plantModel.AiChatHistory{},
|
|
&plantModel.GrowthRecordOss{},
|
|
&plantModel.Banner{},
|
|
&plantModel.SysAiConfig{},
|
|
&plantModel.ClassifyRecord{},
|
|
); err != nil {
|
|
logx.Errorf("数据库迁移失败: %v", err)
|
|
}
|
|
|
|
return &ServiceContext{Config: c, DB: db}
|
|
}
|