eta_mini_crm_ht/services/product.go

package services

import (
	"errors"
	"eta/eta_mini_crm_ht/models"
	"eta/eta_mini_crm_ht/utils"
	"fmt"
	"strconv"
	"time"
)

var (
	tranMediaMap = map[models.MediaType]string{
		models.Video: "视频",
		models.Audio: "音频",
	}
)

func GetUnsetProductCountByCondition(productType string, permissionIds []int, condition string, pars []interface{}) (total int, ids []int, err error) {
	setIds, err := models.GetProductSourceIdsByProductType(productType)
	if err != nil {
		return
	}
	switch productType {
	case "report":
		return getReportIdsByCondition(setIds, permissionIds, condition, pars)
	case "media":
		return getMediaIdsByCondition(setIds, permissionIds, condition, pars)
	default:
		err = errors.New("不支持的产品类型查询")
		return
	}
}

func getReportIdsByCondition(setIds []int, permissionIds []int, condition string, pars []interface{}) (total int, filterIds []int, err error) {
	permissionNames, err := models.GetPermissionNamesWithRiskLevel(permissionIds)
	var conditionHT string
	var conditionETA string
	if err != nil {
		utils.FileLog.Error("查询品种名称失败", err.Error)
		return
	}
	if len(permissionNames) > 0 {
		conditionHT = "(source ='HT' and plate_name in (" + utils.GetOrmReplaceHolder(len(permissionNames)) + "))"
	}
	filterPermissionIds, err := models.FilterPermissionIdsWithRiskLevel(permissionIds)
	if err != nil {
		utils.FileLog.Error("过滤品种Id失败", err.Error)
		return
	}
	classifyCondition := utils.GetOrmReplaceHolder(len(filterPermissionIds))
	var classifyPars []interface{}
	classifyPars = append(classifyPars, filterPermissionIds)
	classifyIds, err := models.GetClassifyIdsByPermissionIds(classifyCondition, classifyPars)
	if len(classifyIds) > 0 {
		conditionETA = "(source ='ETA' and classify_id in (" + utils.GetOrmReplaceHolder(len(classifyIds)) + "))"
	}
	if conditionHT != "" {
		if conditionETA != "" {
			condition += "and ( " + conditionHT + " or " + conditionETA + ")"
			pars = append(pars, permissionNames)
			pars = append(pars, classifyIds)
		} else {
			condition += "and " + conditionHT
			pars = append(pars, permissionNames)
		}
	} else {
		if conditionETA != "" {
			condition += "and " + conditionETA
			pars = append(pars, classifyIds)
		}
	}
	var ids []int
	ids, err = models.GetReportIdsByCondition(condition, pars)
	if err != nil {
		utils.FileLog.Error("查询报告Id失败", err.Error)
		return
	}
	filterMap := make(map[int]bool)
	for _, filterId := range setIds {
		filterMap[filterId] = true
	}
	for _, id := range ids {
		if !filterMap[id] {
			filterIds = append(filterIds, id)
			total++
		}
	}
	return
}

func getMediaIdsByCondition(setIds []int, permissionIds []int, condition string, pars []interface{}) (total int, filterIds []int, err error) {
	filterPermissionIds, err := models.FilterPermissionIdsWithRiskLevel(permissionIds)
	if err != nil {
		utils.FileLog.Error("过滤品种id失败", err.Error)
		return
	}
	mediaIds, err := models.GetMediaIdIdsByPermissionId(filterPermissionIds)
	if err != nil {
		utils.FileLog.Error("过滤媒体Id失败", err.Error)
		return
	}
	if condition != "" {
		condition += "and id in (" + utils.GetOrmReplaceHolder(len(mediaIds)) + ")"
		pars = append(pars, mediaIds)
		mediaIds, err = models.GetMediaIdsByCondition(condition, pars)
	}
	filterMap := make(map[int]bool)
	for _, filterId := range setIds {
		filterMap[filterId] = true
	}
	for _, id := range mediaIds {
		if !filterMap[id] {
			total++
			filterIds = append(filterIds, id)
		}
	}
	return
}

func GetUnsetProductByCondition(productType string, ids []int, sortCondition string, startSize int, pageSize int) (products []*ProductView, err error) {
	switch productType {
	case "report":
		var reports []models.Report
		reports, err = models.GetReportByIds(sortCondition, ids, startSize, pageSize)
		if err != nil {
			utils.FileLog.Error("查询报告产品失败", err.Error)
			return
		}
		for _, report := range reports {
			products = append(products, &ProductView{
				ProductName:   report.Title,
				SourceId:      report.Id,
				ProductType:   "report",
				PublishedTime: report.PublishedTime,
			})
		}
		return
	case "media":
		var medias []models.Media
		medias, err = models.GetMediasById(ids, sortCondition, startSize, pageSize)
		if err != nil {
			utils.FileLog.Error("查询报告产品失败", err.Error)
			return
		}
		for _, media := range medias {
			products = append(products, &ProductView{
				ProductName:   media.MediaName,
				SourceId:      media.Id,
				ProductType:   string(media.MediaType),
				PublishedTime: media.PublishedTime.Format(time.DateTime),
			})
		}
		return
	default:
		err = errors.New("不支持的产品类型查询")
		return
	}
}

type ProductView struct {
	ProductName   string
	SourceId      int
	ProductType   string
	RiskLevel     string
	PublishedTime string
}

func GetRiskLevel(productType string, id int) (riskLevel string, productName string, err error) {
	switch productType {
	case "report":
		var report *models.Report
		report, err = models.GetReportById(id)
		if err != nil {
			utils.FileLog.Error("获取报告失败,无法确认风险等级", err.Error())
			return
		}
		switch report.Source {
		case models.SourceHT:
			var permission models.Permission
			permission, err = models.GetPermissionByName(report.PlateName)
			if err != nil {
				utils.FileLog.Error("获取品种信息,无法确认风险等级", err.Error())
				return
			}
			return permission.RiskLevel, report.Title, nil
		case models.SourceETA:
			var permissionIds []int
			permissionIds, err = models.GetByPermissionIdsByClassifyId(report.ClassifyId)
			if err != nil {
				utils.FileLog.Error("获取品种信息,无法确认风险等级", err.Error())
				return
			}
			var permissions []models.Permission
			permissions, err = models.GetPermissionsWithRiskLevel(permissionIds)
			if err != nil {
				utils.FileLog.Error("获取品种信息,无法确认风险等级", err.Error())
				return
			}
			var riskNum int
			for _, permission := range permissions {
				if permission.RiskLevel == "" {
					continue
				}
				var parseErr error
				currentNum, parseErr := strconv.Atoi(permission.RiskLevel[1:])
				if parseErr != nil {
					continue
				}
				if currentNum > riskNum {
					riskNum = currentNum
				}
			}
			if riskNum > 0 {
				return fmt.Sprintf("R%d", riskNum), report.Title, nil
			}
			return
		}
	case "media":
		var media *models.Media
		media, err = models.GetMediaById("", id)
		if err != nil {
			utils.FileLog.Error("获取媒体信息,无法确认风险等级", err.Error())
			return
		}
		var permissionsIds []int
		permissionsIds, err = models.GetPermissionIdsByMediaId(media.MediaType, id)
		if err != nil {
			utils.FileLog.Error("获取媒体信息,无法确认风险等级", err.Error())
			return
		}
		var permissions []models.Permission
		permissions, err = models.GetPermissionsWithRiskLevel(permissionsIds)
		if err != nil {
			utils.FileLog.Error("获取媒体信息,无法确认风险等级", err.Error())
			return
		}
		var riskNum int
		for _, permission := range permissions {

			if permission.RiskLevel == "" {
				continue
			}
			var parseErr error
			currentNum, parseErr := strconv.Atoi(permission.RiskLevel[1:])
			if parseErr != nil {
				continue
			}
			if currentNum > riskNum {
				riskNum = currentNum
			}
		}
		if riskNum > 0 {
			return fmt.Sprintf("R%d", riskNum), media.MediaName, nil
		}
		return
	case "package":
		var permission *models.Permission
		permission, err = models.GetPermissionByPermissionId(id)
		if err != nil {
			utils.FileLog.Error("获取套餐品种信息,无法确认风险等级", err.Error())
			return
		}
		if permission.RiskLevel == "" {
			return "", "", errors.New("当前品种未设置风测等级")
		}
		return permission.RiskLevel, "", nil
	default:
		var media *models.Media
		media, err = models.GetMediaById(models.MediaType(productType), id)
		if err != nil {
			utils.FileLog.Error("获取媒体信息,无法确认风险等级", err.Error())
			return
		}
		var permissionsIds []int
		permissionsIds, err = models.GetPermissionIdsByMediaId(media.MediaType, id)
		if err != nil {
			utils.FileLog.Error("获取媒体信息,无法确认风险等级", err.Error())
			return
		}
		var permissions []models.Permission
		permissions, err = models.GetPermissionsWithRiskLevel(permissionsIds)
		if err != nil {
			utils.FileLog.Error("获取媒体信息,无法确认风险等级", err.Error())
			return
		}
		var riskNum int
		for _, permission := range permissions {

			if permission.RiskLevel == "" {
				continue
			}
			var parseErr error
			currentNum, parseErr := strconv.Atoi(permission.RiskLevel[1:])
			if parseErr != nil {
				continue
			}
			if currentNum > riskNum {
				riskNum = currentNum
			}
		}
		if riskNum > 0 {
			return fmt.Sprintf("R%d", riskNum), media.MediaName, nil
		}
		return
	}
	return
}