39310d5b9a
- Node group CRUD operations with traffic-based filtering - Three grouping modes: average distribution, subscription-based, and traffic-based - Automatic and manual group recalculation with history tracking - Group assignment preview before applying changes - User subscription group locking to prevent automatic reassignment - Subscribe-to-group mapping configuration - Group calculation history and detailed reports - System configuration for group management (enabled/mode/auto_create) Database: - Add node_group table for group definitions - Add group_history and group_history_detail tables for tracking - Add node_group_ids (JSON) to nodes and subscribe tables - Add node_group_id and group_locked fields to user_subscribe table - Add migration files for schema changes
80 lines
2.6 KiB
Go
80 lines
2.6 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),
|
|
NodeGroupIds: subscribe.JSONInt64Slice(req.NodeGroupIds),
|
|
NodeGroupId: req.NodeGroupId,
|
|
Show: req.Show,
|
|
Sell: req.Sell,
|
|
Sort: req.Sort,
|
|
DeductionRatio: req.DeductionRatio,
|
|
AllowDeduction: req.AllowDeduction,
|
|
ResetCycle: req.ResetCycle,
|
|
RenewalReset: req.RenewalReset,
|
|
ShowOriginalPrice: req.ShowOriginalPrice,
|
|
}
|
|
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
|
|
}
|