package order import ( "context" "encoding/json" "time" "github.com/perfect-panel/server/internal/model/log" "github.com/perfect-panel/server/pkg/constant" "gorm.io/gorm" "github.com/hibiken/asynq" "github.com/perfect-panel/server/internal/model/order" "github.com/perfect-panel/server/internal/model/user" "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" queue "github.com/perfect-panel/server/queue/types" "github.com/pkg/errors" ) type RenewalLogic struct { logger.Logger ctx context.Context svcCtx *svc.ServiceContext } // NewRenewalLogic creates a new renewal logic instance for subscription renewal operations func NewRenewalLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RenewalLogic { return &RenewalLogic{ Logger: logger.WithContext(ctx), ctx: ctx, svcCtx: svcCtx, } } // Renewal processes subscription renewal orders including discount calculation, // coupon validation, gift amount deduction, fee calculation, and order creation func (l *RenewalLogic) Renewal(req *types.RenewalOrderRequest) (resp *types.RenewalOrderResponse, err error) { u, ok := l.ctx.Value(constant.CtxKeyUser).(*user.User) if !ok { logger.Error("current user is not found in context") return nil, errors.Wrapf(xerr.NewErrCode(xerr.InvalidAccess), "Invalid Access") } if req.Quantity <= 0 { l.Debugf("[Renewal] Quantity is less than or equal to 0, setting to 1") req.Quantity = 1 } orderNo := tool.GenerateTradeNo() // find user subscribe userSubscribe, err := l.svcCtx.UserModel.FindOneUserSubscribe(l.ctx, req.UserSubscribeID) if err != nil { return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "find user subscribe error: %v", err.Error()) } // find subscription sub, err := l.svcCtx.SubscribeModel.FindOne(l.ctx, userSubscribe.SubscribeId) if err != nil { l.Errorw("[Renewal] Database query error", logger.Field("error", err.Error()), logger.Field("subscribe_id", userSubscribe.SubscribeId)) return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "find subscribe error: %v", err.Error()) } // check subscribe plan status if !*sub.Sell { return nil, errors.Wrapf(xerr.NewErrCode(xerr.ERROR), "subscribe not sell") } var discount float64 = 1 if sub.Discount != "" { var dis []types.SubscribeDiscount _ = json.Unmarshal([]byte(sub.Discount), &dis) discount = getDiscount(dis, req.Quantity) } price := sub.UnitPrice * req.Quantity amount := int64(float64(price) * discount) discountAmount := price - amount var coupon int64 = 0 if req.Coupon != "" { couponInfo, err := l.svcCtx.CouponModel.FindOneByCode(l.ctx, req.Coupon) if err != nil { if errors.Is(err, gorm.ErrRecordNotFound) { return nil, errors.Wrapf(xerr.NewErrCode(xerr.CouponNotExist), "coupon not found") } return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "find coupon error: %v", err.Error()) } if couponInfo.Count != 0 && couponInfo.Count <= couponInfo.UsedCount { return nil, errors.Wrapf(xerr.NewErrCode(xerr.CouponInsufficientUsage), "coupon used") } couponSub := tool.StringToInt64Slice(couponInfo.Subscribe) if len(couponSub) > 0 && !tool.Contains(couponSub, sub.Id) { return nil, errors.Wrapf(xerr.NewErrCode(xerr.CouponNotApplicable), "coupon not match") } var count int64 err = l.svcCtx.DB.Transaction(func(tx *gorm.DB) error { return tx.Model(&order.Order{}).Where("user_id = ? and coupon = ?", u.Id, req.Coupon).Count(&count).Error }) if err != nil { l.Errorw("[Renewal] Database query error", logger.Field("error", err.Error()), logger.Field("user_id", u.Id), logger.Field("coupon", req.Coupon)) return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "find coupon error: %v", err.Error()) } if count >= couponInfo.UserLimit { return nil, errors.Wrapf(xerr.NewErrCode(xerr.CouponInsufficientUsage), "coupon limit exceeded") } coupon = calculateCoupon(amount, couponInfo) } payment, err := l.svcCtx.PaymentModel.FindOne(l.ctx, req.Payment) if err != nil { l.Errorw("[Renewal] Database query error", logger.Field("error", err.Error()), logger.Field("payment", req.Payment)) return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "find payment error: %v", err.Error()) } amount -= coupon var deductionAmount int64 // Check user deduction amount if u.GiftAmount > 0 { if u.GiftAmount >= amount { deductionAmount = amount u.GiftAmount -= deductionAmount amount = 0 } else { deductionAmount = u.GiftAmount amount -= u.GiftAmount u.GiftAmount = 0 } } var feeAmount int64 // Calculate the handling fee if amount > 0 { feeAmount = calculateFee(amount, payment) } amount += feeAmount // create order orderInfo := order.Order{ UserId: u.Id, ParentId: userSubscribe.OrderId, OrderNo: orderNo, Type: 2, Quantity: req.Quantity, Price: price, Amount: amount, GiftAmount: deductionAmount, Discount: discountAmount, Coupon: req.Coupon, CouponDiscount: coupon, PaymentId: payment.Id, Method: payment.Platform, FeeAmount: feeAmount, Status: 1, SubscribeId: userSubscribe.SubscribeId, SubscribeToken: userSubscribe.Token, } // Database transaction err = l.svcCtx.DB.Transaction(func(db *gorm.DB) error { // update user deduction && Pre deduction ,Return after canceling the order if orderInfo.GiftAmount > 0 { // update user deduction && Pre deduction ,Return after canceling the order if err := l.svcCtx.UserModel.Update(l.ctx, u, db); err != nil { l.Errorw("[Renewal] Database update error", logger.Field("error", err.Error()), logger.Field("user", u)) return err } // create deduction record giftLog := log.Gift{ Type: log.GiftTypeReduce, OrderNo: orderInfo.OrderNo, SubscribeId: 0, Amount: orderInfo.GiftAmount, Balance: u.GiftAmount, Remark: "Renewal order deduction", Timestamp: time.Now().UnixMilli(), } content, _ := giftLog.Marshal() if err := db.Model(&log.SystemLog{}).Create(&log.SystemLog{ Type: log.TypeGift.Uint8(), Date: time.Now().Format(time.DateOnly), ObjectID: u.Id, Content: string(content), }).Error; err != nil { l.Errorw("[Renewal] Database insert error", logger.Field("error", err.Error()), logger.Field("deductionLog", giftLog)) return err } } // insert order return db.Model(&order.Order{}).Create(&orderInfo).Error }) if err != nil { l.Errorw("[Renewal] Database insert error", logger.Field("error", err.Error()), logger.Field("order", orderInfo)) return nil, errors.Wrapf(err, "insert order error: %v", err.Error()) } // Deferred task payload := queue.DeferCloseOrderPayload{ OrderNo: orderInfo.OrderNo, } val, err := json.Marshal(payload) if err != nil { l.Errorw("[Renewal] Marshal payload error", logger.Field("error", err.Error()), logger.Field("payload", payload)) } task := asynq.NewTask(queue.DeferCloseOrder, val, asynq.MaxRetry(3)) taskInfo, err := l.svcCtx.Queue.Enqueue(task, asynq.ProcessIn(CloseOrderTimeMinutes*time.Minute)) if err != nil { l.Errorw("[Renewal] Enqueue task error", logger.Field("error", err.Error()), logger.Field("task", task)) } else { l.Infow("[Renewal] Enqueue task success", logger.Field("TaskID", taskInfo.ID)) } return &types.RenewalOrderResponse{ OrderNo: orderInfo.OrderNo, }, nil }