71 lines
2.1 KiB
Go
71 lines
2.1 KiB
Go
package system
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
|
|
"github.com/perfect-panel/server/initialize"
|
|
"github.com/perfect-panel/server/internal/config"
|
|
"github.com/perfect-panel/server/internal/model/system"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
"github.com/perfect-panel/server/pkg/tool"
|
|
"github.com/perfect-panel/server/pkg/xerr"
|
|
"github.com/pkg/errors"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UpdateSubscribeConfigLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewUpdateSubscribeConfigLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateSubscribeConfigLogic {
|
|
return &UpdateSubscribeConfigLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *UpdateSubscribeConfigLogic) UpdateSubscribeConfig(req *types.SubscribeConfig) error {
|
|
v := reflect.ValueOf(*req)
|
|
// Get the reflection type of the structure
|
|
t := v.Type()
|
|
err := l.svcCtx.SystemModel.Transaction(l.ctx, func(db *gorm.DB) error {
|
|
var err error
|
|
for i := 0; i < v.NumField(); i++ {
|
|
// Get the field name
|
|
fieldName := t.Field(i).Name
|
|
// Get the field value to string
|
|
fieldValue := tool.ConvertValueToString(v.Field(i))
|
|
// Update the site config
|
|
err = db.Model(&system.System{}).Where("`category` = 'subscribe' and `key` = ?", fieldName).Update("value", fieldValue).Error
|
|
if err != nil {
|
|
break
|
|
}
|
|
}
|
|
return l.svcCtx.Redis.Del(l.ctx, config.SubscribeConfigKey, config.GlobalConfigKey).Err()
|
|
})
|
|
|
|
if err != nil {
|
|
l.Errorw("[UpdateSubscribeConfigLogic] update subscribe config error: ", logger.Field("error", err.Error()))
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseUpdateError), "update subscribe config error: %v", err)
|
|
}
|
|
|
|
if l.svcCtx.Config.Subscribe.SubscribePath != req.SubscribePath {
|
|
go func(svc *svc.ServiceContext) {
|
|
err = svc.Restart()
|
|
if err != nil {
|
|
l.Errorw("[UpdateSubscribeConfigLogic] restart error: ", logger.Field("error", err.Error()))
|
|
}
|
|
}(l.svcCtx)
|
|
return nil
|
|
}
|
|
|
|
initialize.Subscribe(l.svcCtx)
|
|
return nil
|
|
}
|