First commit

2025-12-18 13:50:39 +08:00
parent 569f4882e5
commit 642842c29e
218 changed files with 86902 additions and 0 deletions
--- a/backend/internal/repository/redeem_code_repo.go
+++ b/backend/internal/repository/redeem_code_repo.go
@@ -0,0 +1,133 @@
+package repository
+
+import (
+	"context"
+	"sub2api/internal/model"
+	"time"
+
+	"gorm.io/gorm"
+)
+
+type RedeemCodeRepository struct {
+	db *gorm.DB
+}
+
+func NewRedeemCodeRepository(db *gorm.DB) *RedeemCodeRepository {
+	return &RedeemCodeRepository{db: db}
+}
+
+func (r *RedeemCodeRepository) Create(ctx context.Context, code *model.RedeemCode) error {
+	return r.db.WithContext(ctx).Create(code).Error
+}
+
+func (r *RedeemCodeRepository) CreateBatch(ctx context.Context, codes []model.RedeemCode) error {
+	return r.db.WithContext(ctx).Create(&codes).Error
+}
+
+func (r *RedeemCodeRepository) GetByID(ctx context.Context, id int64) (*model.RedeemCode, error) {
+	var code model.RedeemCode
+	err := r.db.WithContext(ctx).First(&code, id).Error
+	if err != nil {
+		return nil, err
+	}
+	return &code, nil
+}
+
+func (r *RedeemCodeRepository) GetByCode(ctx context.Context, code string) (*model.RedeemCode, error) {
+	var redeemCode model.RedeemCode
+	err := r.db.WithContext(ctx).Where("code = ?", code).First(&redeemCode).Error
+	if err != nil {
+		return nil, err
+	}
+	return &redeemCode, nil
+}
+
+func (r *RedeemCodeRepository) Delete(ctx context.Context, id int64) error {
+	return r.db.WithContext(ctx).Delete(&model.RedeemCode{}, id).Error
+}
+
+func (r *RedeemCodeRepository) List(ctx context.Context, params PaginationParams) ([]model.RedeemCode, *PaginationResult, error) {
+	return r.ListWithFilters(ctx, params, "", "", "")
+}
+
+// ListWithFilters lists redeem codes with optional filtering by type, status, and search query
+func (r *RedeemCodeRepository) ListWithFilters(ctx context.Context, params PaginationParams, codeType, status, search string) ([]model.RedeemCode, *PaginationResult, error) {
+	var codes []model.RedeemCode
+	var total int64
+
+	db := r.db.WithContext(ctx).Model(&model.RedeemCode{})
+
+	// Apply filters
+	if codeType != "" {
+		db = db.Where("type = ?", codeType)
+	}
+	if status != "" {
+		db = db.Where("status = ?", status)
+	}
+	if search != "" {
+		searchPattern := "%" + search + "%"
+		db = db.Where("code ILIKE ?", searchPattern)
+	}
+
+	if err := db.Count(&total).Error; err != nil {
+		return nil, nil, err
+	}
+
+	if err := db.Preload("User").Preload("Group").Offset(params.Offset()).Limit(params.Limit()).Order("id DESC").Find(&codes).Error; err != nil {
+		return nil, nil, err
+	}
+
+	pages := int(total) / params.Limit()
+	if int(total)%params.Limit() > 0 {
+		pages++
+	}
+
+	return codes, &PaginationResult{
+		Total:    total,
+		Page:     params.Page,
+		PageSize: params.Limit(),
+		Pages:    pages,
+	}, nil
+}
+
+func (r *RedeemCodeRepository) Update(ctx context.Context, code *model.RedeemCode) error {
+	return r.db.WithContext(ctx).Save(code).Error
+}
+
+func (r *RedeemCodeRepository) Use(ctx context.Context, id, userID int64) error {
+	now := time.Now()
+	result := r.db.WithContext(ctx).Model(&model.RedeemCode{}).
+		Where("id = ? AND status = ?", id, model.StatusUnused).
+		Updates(map[string]interface{}{
+			"status":  model.StatusUsed,
+			"used_by": userID,
+			"used_at": now,
+		})
+	if result.Error != nil {
+		return result.Error
+	}
+	if result.RowsAffected == 0 {
+		return gorm.ErrRecordNotFound // 兑换码不存在或已被使用
+	}
+	return nil
+}
+
+// ListByUser returns all redeem codes used by a specific user
+func (r *RedeemCodeRepository) ListByUser(ctx context.Context, userID int64, limit int) ([]model.RedeemCode, error) {
+	var codes []model.RedeemCode
+	if limit <= 0 {
+		limit = 10
+	}
+
+	err := r.db.WithContext(ctx).
+		Preload("Group").
+		Where("used_by = ?", userID).
+		Order("used_at DESC").
+		Limit(limit).
+		Find(&codes).Error
+
+	if err != nil {
+		return nil, err
+	}
+	return codes, nil
+}