hi-server/internal/logic/admin/user/getWithdrawalListLogic.go

package user

import (
	"context"

	usermodel "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/xerr"
	"github.com/pkg/errors"
)

type GetWithdrawalListLogic struct {
	logger.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewGetWithdrawalListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetWithdrawalListLogic {
	return &GetWithdrawalListLogic{
		Logger: logger.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *GetWithdrawalListLogic) GetWithdrawalList(req *types.GetWithdrawalListRequest) (*types.GetWithdrawalListResponse, error) {
	page := req.Page
	size := req.Size
	if page <= 0 {
		page = 1
	}
	if size <= 0 {
		size = 10
	}

	query := l.svcCtx.DB.WithContext(l.ctx).Model(&usermodel.Withdrawal{})
	if req.UserId != nil {
		query = query.Where("user_id = ?", *req.UserId)
	}
	if req.Status != nil {
		query = query.Where("status = ?", *req.Status)
	}
	if req.Method != nil {
		query = query.Where("method = ?", *req.Method)
	}

	var total int64
	if err := query.Count(&total).Error; err != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "count withdrawals failed: %v", err)
	}

	var rows []usermodel.Withdrawal
	if err := query.Order("id DESC").Limit(size).Offset((page - 1) * size).Find(&rows).Error; err != nil {
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "query withdrawals failed: %v", err)
	}

	list := make([]types.WithdrawalLog, 0, len(rows))
	for _, row := range rows {
		list = append(list, types.WithdrawalLog{
			Id:        row.Id,
			UserId:    row.UserId,
			Amount:    row.Amount,
			Content:   row.Content,
			Status:    row.Status,
			Reason:    row.Reason,
			Method:    row.Method,
			Account:   row.Account,
			QrCodeUrl: row.QrCodeUrl,
			CreatedAt: row.CreatedAt.UnixMilli(),
			UpdatedAt: row.UpdatedAt.UnixMilli(),
		})
	}

	return &types.GetWithdrawalListResponse{
		List:  list,
		Total: total,
	}, nil
}