eta_mini_ht_api/service/product/subscribe_servcie.go

package product

import (
	logger "eta/eta_mini_ht_api/common/component/log"
	"eta/eta_mini_ht_api/common/utils/page"
	"eta/eta_mini_ht_api/domian/merchant"
	"sync"
	"time"
)

type SubscribeDTO struct {
	Title           string
	Abstract        string
	SourceId        int
	SourceTitle     string
	SourceAbstract  string
	SourceSrc       string
	Type            string
	RiskLevel       string
	CoverSrc        string
	PermissionNames string
	CreatedDate     string
}

func SubscribeList(templateUserId int, productType string, pageInfo page.PageInfo) (list []SubscribeDTO, err error) {
	var subscribeList []merchant.UserAccessDTO
	subscribeList, err = merchant.SubscribeList(templateUserId, productType, pageInfo)
	var wg sync.WaitGroup
	wg.Add(len(subscribeList))
	for _, Subscribe := range subscribeList {
		go func(Subscribe merchant.UserAccessDTO) {
			defer wg.Done()
			product, _ := merchant.GetMerchantProductById(Subscribe.ProductID)
			subscribe := convertToSubscribeDTO(product)
			if err != nil {
				logger.Error("获取风险等级失败[productId:%d]", product.Id)
			}
			switch product.Type {
			case "audio", "video":
				subscribe.RiskLevel, subscribe.SourceTitle, _, subscribe.SourceSrc, subscribe.PermissionNames, err = GetProductRiskLevel(product)
			case "report":
				subscribe.RiskLevel, subscribe.SourceTitle, subscribe.SourceAbstract, _, subscribe.PermissionNames, err = GetProductRiskLevel(product)
			case "package":
				subscribe.RiskLevel, _, _, _, subscribe.PermissionNames, err = GetProductRiskLevel(product)
			default:
				logger.Error("未知产品类型[productType:%s]", product.Type)
				return
			}
			list = append(list, subscribe)
		}(Subscribe)
	}
	wg.Wait()
	return
}

func convertToSubscribeDTO(product merchant.MerchantProductDTO) SubscribeDTO {
	return SubscribeDTO{
		SourceId:    product.SourceId,
		Type:        product.Type,
		Title:       product.Title,
		Abstract:    product.Description,
		CoverSrc:    product.CoverSrc,
		CreatedDate: product.CreatedTime.Format(time.DateOnly),
	}
}
func GetTotalPageCountByProductType(productType string, templateUserId int) (int64, int64) {
	return merchant.GetTotalPageCountByProductType(productType, templateUserId)
}