拆分提现 Repository 文件职责

2026-06-10 14:54:39 +08:00
parent f75174b1cf
commit 8221544aca
6 changed files with 563 additions and 506 deletions
@@ -0,0 +1,193 @@
+package withdrawal
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"time"
+
+	"hfb_sys/backend/internal/model"
+	"hfb_sys/backend/internal/modules/wallet"
+
+	"gorm.io/gorm"
+)
+
+// 管理员查询提现列表
+func (r *Repository) AdminList(ctx context.Context, query AdminListQuery) (*AdminPaginatedResult, error) {
+	db := r.db.WithContext(ctx).Model(&model.WithdrawalRequest{})
+
+	if query.Status != "" {
+		db = db.Where("status = ?", query.Status)
+	}
+	if query.UserID > 0 {
+		db = db.Where("user_id = ?", query.UserID)
+	}
+
+	var total int64
+	if err := db.Count(&total).Error; err != nil {
+		return nil, err
+	}
+
+	offset := (query.Page - 1) * query.Size
+	var withdrawals []model.WithdrawalRequest
+	if err := db.Order("created_at DESC").
+		Offset(offset).Limit(query.Size).
+		Find(&withdrawals).Error; err != nil {
+		return nil, err
+	}
+
+	items := make([]WithdrawalDetailDTO, 0, len(withdrawals))
+	for _, w := range withdrawals {
+		dto, err := r.toDetailDTO(ctx, w)
+		if err != nil {
+			continue
+		}
+		items = append(items, *dto)
+	}
+
+	return &AdminPaginatedResult{
+		Items:    items,
+		Total:    total,
+		Page:     query.Page,
+		PageSize: query.Size,
+	}, nil
+}
+
+// 管理员查询提现详情
+
+// 管理员查询提现详情
+func (r *Repository) AdminFindByID(ctx context.Context, id uint64) (*WithdrawalDetailDTO, error) {
+	var withdrawal model.WithdrawalRequest
+	if err := r.db.WithContext(ctx).First(&withdrawal, id).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, ErrWithdrawalNotFound
+		}
+		return nil, err
+	}
+	return r.toDetailDTO(ctx, withdrawal)
+}
+
+// 管理员审核提现
+
+// 管理员审核提现
+func (r *Repository) Review(ctx context.Context, adminID, id uint64, req ReviewWithdrawalRequest) (*WithdrawalDetailDTO, error) {
+	db := r.db.WithContext(ctx)
+	var withdrawal model.WithdrawalRequest
+	if err := db.First(&withdrawal, id).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, ErrWithdrawalNotFound
+		}
+		return nil, err
+	}
+
+	// 只有待审核状态可以审核
+	if withdrawal.Status != "pending" {
+		return nil, ErrWithdrawalLocked
+	}
+
+	now := time.Now()
+
+	if req.Approved {
+		// 审核通过，进入处理中状态
+		withdrawal.Status = "processing"
+	} else {
+		// 审核拒绝，解冻余额
+		withdrawal.Status = "rejected"
+	}
+
+	withdrawal.ReviewedBy = &adminID
+	withdrawal.ReviewedAt = &now
+	withdrawal.ReviewRemark = req.Remark
+
+	err := db.Transaction(func(tx *gorm.DB) error {
+		if err := tx.Save(&withdrawal).Error; err != nil {
+			return err
+		}
+
+		// 如果拒绝，解冻余额
+		if !req.Approved {
+			if err := wallet.AppendEntries(tx, wallet.Entry{
+				UserID:      withdrawal.UserID,
+				Direction:   "out",
+				AmountCent:  withdrawal.AmountCent,
+				BalanceType: "frozen",
+				BizType:     "withdraw_reject",
+				BizNo:       withdrawal.WithdrawNo,
+				Remark:      fmt.Sprintf("提现被拒绝: %s", req.Remark),
+			}, wallet.Entry{
+				UserID:      withdrawal.UserID,
+				Direction:   "in",
+				AmountCent:  withdrawal.AmountCent,
+				BalanceType: "available",
+				BizType:     "withdraw_reject",
+				BizNo:       withdrawal.WithdrawNo,
+				Remark:      fmt.Sprintf("提现被拒绝: %s", req.Remark),
+			}); err != nil {
+				return err
+			}
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return nil, err
+	}
+
+	return r.AdminFindByID(ctx, id)
+}
+
+// 管理员确认打款
+
+// 管理员确认打款
+func (r *Repository) ConfirmPayment(ctx context.Context, adminID, id uint64, req ConfirmPaymentRequest) (*WithdrawalDetailDTO, error) {
+	db := r.db.WithContext(ctx)
+	var withdrawal model.WithdrawalRequest
+	if err := db.First(&withdrawal, id).Error; err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, ErrWithdrawalNotFound
+		}
+		return nil, err
+	}
+
+	// 只有处理中状态可以确认打款
+	if withdrawal.Status != "processing" {
+		return nil, ErrWithdrawalLocked
+	}
+
+	now := time.Now()
+	withdrawal.Status = "completed"
+	withdrawal.PaidBy = &adminID
+	withdrawal.PaidAt = &now
+	withdrawal.PaymentProofURL = req.PaymentProofURL
+	withdrawal.PaymentRemark = req.Remark
+
+	err := db.Transaction(func(tx *gorm.DB) error {
+		if err := tx.Save(&withdrawal).Error; err != nil {
+			return err
+		}
+
+		// 扣除冻结余额
+		if err := wallet.AppendEntries(tx, wallet.Entry{
+			UserID:      withdrawal.UserID,
+			Direction:   "out",
+			AmountCent:  withdrawal.AmountCent,
+			BalanceType: "frozen",
+			BizType:     "withdraw_complete",
+			BizNo:       withdrawal.WithdrawNo,
+			Remark:      "提现完成",
+		}); err != nil {
+			return err
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return nil, err
+	}
+
+	return r.AdminFindByID(ctx, id)
+}
+
+// 转换为用户DTO