shanshanzhong
c582087c0f
feat: 添加版本和构建时间变量 fix: 修正短信队列类型注释错误 style: 清理未使用的代码和测试文件 docs: 更新安装文档中的下载链接 chore: 迁移数据库脚本添加日志和订阅配置
77 lines
2.4 KiB
Go
77 lines
2.4 KiB
Go
package subscribe
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"github.com/perfect-panel/server/pkg/device"
|
|
|
|
"github.com/perfect-panel/server/internal/model/subscribe"
|
|
"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"
|
|
)
|
|
|
|
type UpdateSubscribeLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// Update subscribe
|
|
func NewUpdateSubscribeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateSubscribeLogic {
|
|
return &UpdateSubscribeLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *UpdateSubscribeLogic) UpdateSubscribe(req *types.UpdateSubscribeRequest) error {
|
|
// Query the database to get the subscribe information
|
|
_, err := l.svcCtx.SubscribeModel.FindOne(l.ctx, req.Id)
|
|
if err != nil {
|
|
l.Logger.Error("[UpdateSubscribe] Database query error", logger.Field("error", err.Error()), logger.Field("subscribe_id", req.Id))
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "get subscribe error: %v", err.Error())
|
|
}
|
|
discount := ""
|
|
if len(req.Discount) > 0 {
|
|
val, _ := json.Marshal(req.Discount)
|
|
discount = string(val)
|
|
}
|
|
sub := &subscribe.Subscribe{
|
|
Id: req.Id,
|
|
Name: req.Name,
|
|
Language: req.Language,
|
|
Description: req.Description,
|
|
UnitPrice: req.UnitPrice,
|
|
UnitTime: req.UnitTime,
|
|
Discount: discount,
|
|
Replacement: req.Replacement,
|
|
Inventory: req.Inventory,
|
|
Traffic: req.Traffic,
|
|
SpeedLimit: req.SpeedLimit,
|
|
DeviceLimit: req.DeviceLimit,
|
|
Quota: req.Quota,
|
|
Nodes: tool.Int64SliceToString(req.Nodes),
|
|
NodeTags: tool.StringSliceToString(req.NodeTags),
|
|
Show: req.Show,
|
|
Sell: req.Sell,
|
|
Sort: req.Sort,
|
|
DeductionRatio: req.DeductionRatio,
|
|
AllowDeduction: req.AllowDeduction,
|
|
ResetCycle: req.ResetCycle,
|
|
RenewalReset: req.RenewalReset,
|
|
}
|
|
err = l.svcCtx.SubscribeModel.Update(l.ctx, sub)
|
|
if err != nil {
|
|
l.Logger.Error("[UpdateSubscribe] update subscribe failed", logger.Field("error", err.Error()), logger.Field("subscribe", sub))
|
|
return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseUpdateError), "update subscribe error: %v", err.Error())
|
|
}
|
|
l.svcCtx.DeviceManager.Broadcast(device.SubscribeUpdate)
|
|
return nil
|
|
}
|