压缩历史至 95081aff

internal/apps/redenvelope/tasks.go

@@ -0,0 +1,133 @@
+/*
+Copyright 2025 linux.do
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package redenvelope
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/hibiken/asynq"
+	"github.com/linux-do/credit/internal/db"
+	"github.com/linux-do/credit/internal/logger"
+	"github.com/linux-do/credit/internal/model"
+	"github.com/linux-do/credit/internal/service"
+	"gorm.io/gorm"
+)
+
+// HandleRefundExpiredRedEnvelopes 处理过期红包退款的定时任务
+func HandleRefundExpiredRedEnvelopes(ctx context.Context, t *asynq.Task) error {
+	logger.InfoF(ctx, "开始处理过期红包退款任务")
+	refundExpiredRedEnvelopes(ctx)
+	logger.InfoF(ctx, "过期红包退款任务完成")
+	return nil
+}
+
+// refundExpiredRedEnvelopes 退款过期红包
+func refundExpiredRedEnvelopes(ctx context.Context) {
+	const batchSize = 100 // 每批处理100个红包
+	var lastID uint64 = 0
+	var totalProcessed int = 0
+
+	for {
+		// 使用游标分页查询过期红包
+		var expiredEnvelopes []model.RedEnvelope
+		if err := db.DB(ctx).
+			Where("id > ? AND status = ? AND expires_at < ? AND remaining_amount > 0", lastID, model.RedEnvelopeStatusActive, time.Now()).
+			Order("id ASC").
+			Limit(batchSize).
+			Find(&expiredEnvelopes).Error; err != nil {
+			logger.ErrorF(ctx, "查询过期红包失败: %v", err)
+			return
+		}
+
+		// 没有更多数据，退出循环
+		if len(expiredEnvelopes) == 0 {
+			break
+		}
+
+		logger.InfoF(ctx, "本批次找到 %d 个需要退款的过期红包", len(expiredEnvelopes))
+
+		// 处理每个过期红包
+		for _, envelope := range expiredEnvelopes {
+			if err := db.DB(ctx).Transaction(func(tx *gorm.DB) error {
+				// 更新红包状态为已过期
+				if err := tx.Model(&model.RedEnvelope{}).
+					Where("id = ? AND status = ?", envelope.ID, model.RedEnvelopeStatusActive).
+					Updates(map[string]interface{}{
+						"status":           model.RedEnvelopeStatusExpired,
+						"remaining_amount": 0,
+						"remaining_count":  0,
+					}).Error; err != nil {
+					return err
+				}
+
+				// 退还剩余金额给创建者
+				if envelope.RemainingAmount.IsPositive() {
+					// 增加余额并减少total_payment
+					if err := service.UpdateBalance(tx, service.BalanceUpdateOptions{
+						UserID:     envelope.CreatorID,
+						Amount:     envelope.RemainingAmount.Neg(),
+						Operation:  service.BalanceDeduct,
+						TotalField: "total_payment",
+					}); err != nil {
+						return err
+					}
+
+					// 创建退款订单记录
+					remarkMsg := fmt.Sprintf("红包过期退款，红包ID:%d", envelope.ID)
+					if envelope.Greeting != "" {
+						remarkMsg = fmt.Sprintf("%s，祝福语: %s", remarkMsg, envelope.Greeting)
+					}
+
+					order := model.Order{
+						OrderName:   "红包退款",
+						PayerUserID: 0,
+						PayeeUserID: envelope.CreatorID,
+						Amount:      envelope.RemainingAmount,
+						Status:      model.OrderStatusSuccess,
+						Type:        model.OrderTypeRedEnvelopeRefund,
+						Remark:      remarkMsg,
+						TradeTime:   time.Now(),
+						ExpiresAt:   time.Now().Add(24 * time.Hour),
+					}
+
+					if err := tx.Create(&order).Error; err != nil {
+						return err
+					}
+
+					logger.InfoF(ctx, "红包ID:%d 退款成功，金额:%s", envelope.ID, envelope.RemainingAmount.String())
+				}
+
+				return nil
+			}); err != nil {
+				logger.ErrorF(ctx, "红包ID:%d 退款失败: %v", envelope.ID, err)
+			} else {
+				totalProcessed++
+			}
+
+			// 更新游标
+			lastID = envelope.ID
+		}
+	}
+
+	if totalProcessed > 0 {
+		logger.InfoF(ctx, "退款任务完成，共处理 %d 个过期红包", totalProcessed)
+	} else {
+		logger.InfoF(ctx, "没有需要退款的过期红包")
+	}
+}