v0.1

internal/domains/product/services/product_subscription_service.go

@@ -5,10 +5,13 @@ import (
 	"errors"
 	"fmt"
 
+	"gorm.io/gorm"
+
 	"go.uber.org/zap"
 
 	"tyapi-server/internal/domains/product/entities"
 	"tyapi-server/internal/domains/product/repositories"
+	"tyapi-server/internal/domains/product/repositories/queries"
 )
 
 // ProductSubscriptionService 产品订阅领域服务
@@ -32,6 +35,31 @@ func NewProductSubscriptionService(
 	}
 }
 
+// UserSubscribedProductByCode 查找用户已订阅的产品
+func (s *ProductSubscriptionService) UserSubscribedProductByCode(ctx context.Context, userID string, productCode string) (*entities.Subscription, error) {
+	product, err := s.productRepo.FindByCode(ctx, productCode)
+	if err != nil {
+		return nil, err
+	}
+	subscription, err := s.subscriptionRepo.FindByUserAndProduct(ctx, userID, product.ID)
+	if err != nil {
+		return nil, err
+	}
+	return subscription, nil
+}
+
+// GetUserSubscribedProduct 查找用户已订阅的产品
+func (s *ProductSubscriptionService) GetUserSubscribedProduct(ctx context.Context, userID string, productID string) (*entities.Subscription, error) {
+	subscription, err := s.subscriptionRepo.FindByUserAndProduct(ctx, userID, productID)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, nil
+		}
+		return nil, err
+	}
+	return subscription, nil
+}
+
 // CanUserSubscribeProduct 检查用户是否可以订阅产品
 func (s *ProductSubscriptionService) CanUserSubscribeProduct(ctx context.Context, userID string, productID string) (bool, error) {
 	// 检查产品是否存在且可订阅
@@ -92,6 +120,11 @@ func (s *ProductSubscriptionService) CreateSubscription(ctx context.Context, use
 	return &createdSubscription, nil
 }
 
+// ListSubscriptions 获取订阅列表
+func (s *ProductSubscriptionService) ListSubscriptions(ctx context.Context, query *queries.ListSubscriptionsQuery) ([]*entities.Subscription, int64, error) {
+	return s.subscriptionRepo.ListSubscriptions(ctx, query)
+}
+
 // GetUserSubscriptions 获取用户订阅列表
 func (s *ProductSubscriptionService) GetUserSubscriptions(ctx context.Context, userID string) ([]*entities.Subscription, error) {
 	return s.subscriptionRepo.FindByUserID(ctx, userID)
@@ -141,4 +174,20 @@ func (s *ProductSubscriptionService) GetProductStats(ctx context.Context) (map[s
 	}
 
 	return stats, nil
-} 
+}
+
+func (s *ProductSubscriptionService) SaveSubscription(ctx context.Context, subscription *entities.Subscription) error {
+	exists, err := s.subscriptionRepo.Exists(ctx, subscription.ID)
+	if err != nil {
+		return fmt.Errorf("检查订阅是否存在失败: %w", err)
+	}
+	if exists {
+		return s.subscriptionRepo.Update(ctx, *subscription)
+	} else {
+		_, err := s.subscriptionRepo.Create(ctx, *subscription)
+		if err != nil {
+			return fmt.Errorf("创建订阅失败: %w", err)
+		}
+		return nil
+	}
+}