Blizzard
59 lines
1.6 KiB
Go
59 lines
1.6 KiB
Go
// Code scaffolded by goctl. Safe to edit.
|
|
// goctl 1.10.1
|
|
|
|
package svc
|
|
|
|
import (
|
|
"sundynix-micro-go/app/file/rpc/fileservice"
|
|
"sundynix-micro-go/app/radio/api/internal/config"
|
|
radioModel "sundynix-micro-go/app/radio/model"
|
|
"sundynix-micro-go/app/user/rpc/userservice"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"github.com/zeromicro/go-zero/zrpc"
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type ServiceContext struct {
|
|
Config config.Config
|
|
DB *gorm.DB
|
|
UserRpc userservice.UserService
|
|
FileRpc fileservice.FileService
|
|
}
|
|
|
|
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(
|
|
&radioModel.SundynixRadioUserProfile{},
|
|
&radioModel.SundynixRadioCategory{},
|
|
&radioModel.SundynixRadioChannel{},
|
|
&radioModel.SundynixRadioProgram{},
|
|
&radioModel.SundynixRadioVoice{},
|
|
&radioModel.SundynixRadioLike{},
|
|
&radioModel.SundynixRadioFavorite{},
|
|
&radioModel.SundynixRadioComment{},
|
|
&radioModel.SundynixRadioHistory{},
|
|
&radioModel.SundynixRadioSubscription{},
|
|
&radioModel.SundynixRadioSubscriptionOrder{},
|
|
&radioModel.SundynixRadioPayNotify{},
|
|
&radioModel.SundynixRadioVipConfig{},
|
|
&radioModel.SundynixRadioListenLog{},
|
|
); err != nil {
|
|
logx.Errorf("数据库迁移失败: %v", err)
|
|
}
|
|
|
|
return &ServiceContext{
|
|
Config: c,
|
|
DB: db,
|
|
UserRpc: userservice.NewUserService(zrpc.MustNewClient(c.UserRpc)),
|
|
FileRpc: fileservice.NewFileService(zrpc.MustNewClient(c.FileRpc)),
|
|
}
|
|
}
|