addui

2025-12-19 17:05:09 +08:00
parent cc3472ff40
commit 39c46937ea
307 changed files with 87686 additions and 129 deletions
--- a/internal/infrastructure/database/repositories/product/gorm_component_report_repository.go
+++ b/internal/infrastructure/database/repositories/product/gorm_component_report_repository.go
@@ -0,0 +1,130 @@
+package repositories
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+
+	"tyapi-server/internal/domains/product/entities"
+	"tyapi-server/internal/domains/product/repositories"
+	"tyapi-server/internal/shared/database"
+
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+)
+
+const (
+	ComponentReportDownloadsTable = "component_report_downloads"
+)
+
+type GormComponentReportRepository struct {
+	*database.CachedBaseRepositoryImpl
+}
+
+var _ repositories.ComponentReportRepository = (*GormComponentReportRepository)(nil)
+
+func NewGormComponentReportRepository(db *gorm.DB, logger *zap.Logger) repositories.ComponentReportRepository {
+	return &GormComponentReportRepository{
+		CachedBaseRepositoryImpl: database.NewCachedBaseRepositoryImpl(db, logger, ComponentReportDownloadsTable),
+	}
+}
+
+func (r *GormComponentReportRepository) CreateDownload(ctx context.Context, download *entities.ComponentReportDownload) (*entities.ComponentReportDownload, error) {
+	err := r.CreateEntity(ctx, download)
+	if err != nil {
+		return nil, err
+	}
+	return download, nil
+}
+
+func (r *GormComponentReportRepository) UpdateDownload(ctx context.Context, download *entities.ComponentReportDownload) error {
+	return r.UpdateEntity(ctx, download)
+}
+
+func (r *GormComponentReportRepository) GetDownloadByID(ctx context.Context, id string) (*entities.ComponentReportDownload, error) {
+	var download entities.ComponentReportDownload
+	err := r.SmartGetByID(ctx, id, &download)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, gorm.ErrRecordNotFound
+		}
+		return nil, err
+	}
+	return &download, nil
+}
+
+func (r *GormComponentReportRepository) GetUserDownloads(ctx context.Context, userID string, productID *string) ([]*entities.ComponentReportDownload, error) {
+	var downloads []entities.ComponentReportDownload
+	query := r.GetDB(ctx).Where("user_id = ? AND payment_status = ?", userID, "success")
+
+	if productID != nil && *productID != "" {
+		query = query.Where("product_id = ?", *productID)
+	}
+
+	err := query.Order("created_at DESC").Find(&downloads).Error
+	if err != nil {
+		return nil, err
+	}
+
+	result := make([]*entities.ComponentReportDownload, len(downloads))
+	for i := range downloads {
+		result[i] = &downloads[i]
+	}
+	return result, nil
+}
+
+func (r *GormComponentReportRepository) HasUserDownloaded(ctx context.Context, userID string, productCode string) (bool, error) {
+	var count int64
+	err := r.GetDB(ctx).Model(&entities.ComponentReportDownload{}).
+		Where("user_id = ? AND product_code = ? AND payment_status = ?", userID, productCode, "success").
+		Count(&count).Error
+	if err != nil {
+		return false, err
+	}
+	return count > 0, nil
+}
+
+func (r *GormComponentReportRepository) GetUserDownloadedProductCodes(ctx context.Context, userID string) ([]string, error) {
+	var downloads []entities.ComponentReportDownload
+	err := r.GetDB(ctx).
+		Select("DISTINCT sub_product_codes").
+		Where("user_id = ? AND payment_status = ?", userID, "success").
+		Find(&downloads).Error
+	if err != nil {
+		return nil, err
+	}
+
+	codesMap := make(map[string]bool)
+	for _, download := range downloads {
+		if download.SubProductCodes != "" {
+			var codes []string
+			if err := json.Unmarshal([]byte(download.SubProductCodes), &codes); err == nil {
+				for _, code := range codes {
+					codesMap[code] = true
+				}
+			}
+		}
+		// 也添加主产品编号
+		if download.ProductCode != "" {
+			codesMap[download.ProductCode] = true
+		}
+	}
+
+	codes := make([]string, 0, len(codesMap))
+	for code := range codesMap {
+		codes = append(codes, code)
+	}
+	return codes, nil
+}
+
+func (r *GormComponentReportRepository) GetDownloadByPaymentOrderID(ctx context.Context, orderID string) (*entities.ComponentReportDownload, error) {
+	var download entities.ComponentReportDownload
+	err := r.GetDB(ctx).Where("payment_order_id = ?", orderID).First(&download).Error
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, gorm.ErrRecordNotFound
+		}
+		return nil, err
+	}
+	return &download, nil
+}
--- a/internal/infrastructure/database/repositories/product/gorm_product_ui_component_repository.go
+++ b/internal/infrastructure/database/repositories/product/gorm_product_ui_component_repository.go
@@ -0,0 +1,80 @@
+package repositories
+
+import (
+	"context"
+	"fmt"
+	"tyapi-server/internal/domains/product/entities"
+	"tyapi-server/internal/domains/product/repositories"
+
+	"gorm.io/gorm"
+)
+
+// GormProductUIComponentRepository 产品UI组件关联仓储实现
+type GormProductUIComponentRepository struct {
+	db *gorm.DB
+}
+
+// NewGormProductUIComponentRepository 创建产品UI组件关联仓储实例
+func NewGormProductUIComponentRepository(db *gorm.DB) repositories.ProductUIComponentRepository {
+	return &GormProductUIComponentRepository{db: db}
+}
+
+// Create 创建产品UI组件关联
+func (r *GormProductUIComponentRepository) Create(ctx context.Context, relation entities.ProductUIComponent) (entities.ProductUIComponent, error) {
+	if err := r.db.WithContext(ctx).Create(&relation).Error; err != nil {
+		return entities.ProductUIComponent{}, fmt.Errorf("创建产品UI组件关联失败: %w", err)
+	}
+	return relation, nil
+}
+
+// GetByProductID 根据产品ID获取UI组件关联列表
+func (r *GormProductUIComponentRepository) GetByProductID(ctx context.Context, productID string) ([]entities.ProductUIComponent, error) {
+	var relations []entities.ProductUIComponent
+	if err := r.db.WithContext(ctx).
+		Preload("UIComponent").
+		Where("product_id = ?", productID).
+		Find(&relations).Error; err != nil {
+		return nil, fmt.Errorf("获取产品UI组件关联列表失败: %w", err)
+	}
+	return relations, nil
+}
+
+// GetByUIComponentID 根据UI组件ID获取产品关联列表
+func (r *GormProductUIComponentRepository) GetByUIComponentID(ctx context.Context, componentID string) ([]entities.ProductUIComponent, error) {
+	var relations []entities.ProductUIComponent
+	if err := r.db.WithContext(ctx).
+		Preload("Product").
+		Where("ui_component_id = ?", componentID).
+		Find(&relations).Error; err != nil {
+		return nil, fmt.Errorf("获取UI组件产品关联列表失败: %w", err)
+	}
+	return relations, nil
+}
+
+// Delete 删除产品UI组件关联
+func (r *GormProductUIComponentRepository) Delete(ctx context.Context, id string) error {
+	if err := r.db.WithContext(ctx).Delete(&entities.ProductUIComponent{}, id).Error; err != nil {
+		return fmt.Errorf("删除产品UI组件关联失败: %w", err)
+	}
+	return nil
+}
+
+// DeleteByProductID 根据产品ID删除所有关联
+func (r *GormProductUIComponentRepository) DeleteByProductID(ctx context.Context, productID string) error {
+	if err := r.db.WithContext(ctx).Where("product_id = ?", productID).Delete(&entities.ProductUIComponent{}).Error; err != nil {
+		return fmt.Errorf("根据产品ID删除UI组件关联失败: %w", err)
+	}
+	return nil
+}
+
+// BatchCreate 批量创建产品UI组件关联
+func (r *GormProductUIComponentRepository) BatchCreate(ctx context.Context, relations []entities.ProductUIComponent) error {
+	if len(relations) == 0 {
+		return nil
+	}
+
+	if err := r.db.WithContext(ctx).CreateInBatches(relations, 100).Error; err != nil {
+		return fmt.Errorf("批量创建产品UI组件关联失败: %w", err)
+	}
+	return nil
+}
--- a/internal/infrastructure/database/repositories/product/gorm_ui_component_repository.go
+++ b/internal/infrastructure/database/repositories/product/gorm_ui_component_repository.go
@@ -0,0 +1,129 @@
+package repositories
+
+import (
+	"context"
+	"fmt"
+	"tyapi-server/internal/domains/product/entities"
+	"tyapi-server/internal/domains/product/repositories"
+
+	"gorm.io/gorm"
+)
+
+// GormUIComponentRepository UI组件仓储实现
+type GormUIComponentRepository struct {
+	db *gorm.DB
+}
+
+// NewGormUIComponentRepository 创建UI组件仓储实例
+func NewGormUIComponentRepository(db *gorm.DB) repositories.UIComponentRepository {
+	return &GormUIComponentRepository{db: db}
+}
+
+// Create 创建UI组件
+func (r *GormUIComponentRepository) Create(ctx context.Context, component entities.UIComponent) (entities.UIComponent, error) {
+	if err := r.db.WithContext(ctx).Create(&component).Error; err != nil {
+		return entities.UIComponent{}, fmt.Errorf("创建UI组件失败: %w", err)
+	}
+	return component, nil
+}
+
+// GetByID 根据ID获取UI组件
+func (r *GormUIComponentRepository) GetByID(ctx context.Context, id string) (*entities.UIComponent, error) {
+	var component entities.UIComponent
+	if err := r.db.WithContext(ctx).Where("id = ?", id).First(&component).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return nil, nil
+		}
+		return nil, fmt.Errorf("获取UI组件失败: %w", err)
+	}
+	return &component, nil
+}
+
+// GetByCode 根据编码获取UI组件
+func (r *GormUIComponentRepository) GetByCode(ctx context.Context, code string) (*entities.UIComponent, error) {
+	var component entities.UIComponent
+	if err := r.db.WithContext(ctx).Where("component_code = ?", code).First(&component).Error; err != nil {
+		if err == gorm.ErrRecordNotFound {
+			return nil, nil
+		}
+		return nil, fmt.Errorf("获取UI组件失败: %w", err)
+	}
+	return &component, nil
+}
+
+// List 获取UI组件列表
+func (r *GormUIComponentRepository) List(ctx context.Context, filters map[string]interface{}) ([]entities.UIComponent, int64, error) {
+	var components []entities.UIComponent
+	var total int64
+
+	query := r.db.WithContext(ctx).Model(&entities.UIComponent{})
+
+	// 应用过滤条件
+	if isActive, ok := filters["is_active"]; ok {
+		query = query.Where("is_active = ?", isActive)
+	}
+
+	if keyword, ok := filters["keyword"]; ok && keyword != "" {
+		query = query.Where("component_name LIKE ? OR component_code LIKE ? OR description LIKE ?",
+			"%"+keyword.(string)+"%", "%"+keyword.(string)+"%", "%"+keyword.(string)+"%")
+	}
+
+	// 获取总数
+	if err := query.Count(&total).Error; err != nil {
+		return nil, 0, fmt.Errorf("获取UI组件总数失败: %w", err)
+	}
+
+	// 分页
+	if page, ok := filters["page"]; ok {
+		if pageSize, ok := filters["page_size"]; ok {
+			offset := (page.(int) - 1) * pageSize.(int)
+			query = query.Offset(offset).Limit(pageSize.(int))
+		}
+	}
+
+	// 排序
+	if sortBy, ok := filters["sort_by"]; ok {
+		if sortOrder, ok := filters["sort_order"]; ok {
+			query = query.Order(fmt.Sprintf("%s %s", sortBy, sortOrder))
+		}
+	} else {
+		query = query.Order("sort_order ASC, created_at DESC")
+	}
+
+	// 获取数据
+	if err := query.Find(&components).Error; err != nil {
+		return nil, 0, fmt.Errorf("获取UI组件列表失败: %w", err)
+	}
+
+	return components, total, nil
+}
+
+// Update 更新UI组件
+func (r *GormUIComponentRepository) Update(ctx context.Context, component entities.UIComponent) error {
+	if err := r.db.WithContext(ctx).Save(&component).Error; err != nil {
+		return fmt.Errorf("更新UI组件失败: %w", err)
+	}
+	return nil
+}
+
+// Delete 删除UI组件
+func (r *GormUIComponentRepository) Delete(ctx context.Context, id string) error {
+	if err := r.db.WithContext(ctx).Delete(&entities.UIComponent{}, id).Error; err != nil {
+		return fmt.Errorf("删除UI组件失败: %w", err)
+	}
+	return nil
+}
+
+// GetByCodes 根据编码列表获取UI组件
+func (r *GormUIComponentRepository) GetByCodes(ctx context.Context, codes []string) ([]entities.UIComponent, error) {
+	var components []entities.UIComponent
+	if len(codes) == 0 {
+		return components, nil
+	}
+
+	if err := r.db.WithContext(ctx).Where("component_code IN ?", codes).Find(&components).Error; err != nil {
+		return nil, fmt.Errorf("根据编码列表获取UI组件失败: %w", err)
+	}
+
+	return components, nil
+}