eta_mini_ht_api/models/merchant/merchant_product.go

package merchant

import (
	"errors"
	"eta/eta_mini_ht_api/models"
	"fmt"
	"github.com/shopspring/decimal"
	"sort"
	"sync/atomic"
	"time"
)

type SaleStatus string
type MerchantProductType string

const (
	detailColumns  = "id,title,price,is_permanent,valid_days,type,sale_status,cover_src,cover_url,description,source_id,created_time,updated_time"
	sourceIdColumn = "id,source_id"
)
const (
	OnSale  SaleStatus          = "on_sale"  //上架
	OffSale SaleStatus          = "off_sale" //下架
	Package MerchantProductType = "package"
	Report  MerchantProductType = "report"
	Video   MerchantProductType = "video"
	Audio   MerchantProductType = "audio"
)

// MerchantProduct 商户产品信息结构体
type MerchantProduct struct {
	ID          int                 `gorm:"column:id;primary_key;autoIncrement;comment:主键"`
	SourceID    int                 `gorm:"column:source_id;type:int(11);comment:单品或者套餐对应的主键"`
	Title       string              `gorm:"column:title;type:varchar(255);comment:标题"`
	CoverSrc    int                 `gorm:"column:cover_src;type:int(11);comment:封面图片资源库id"`
	CoverUrl    string              `gorm:"column:cover_url;type:varchar(255);comment:封面图片url"`
	Description string              `gorm:"column:description;type:varchar(255);comment:描述"`
	Price       decimal.Decimal     `gorm:"column:price;type:decimal(10,2);comment:价格"`
	RiskLevel   string              `gorm:"-"`
	Type        MerchantProductType `gorm:"column:type;type:enum('report','video','audio','package');not null;comment:类型"`
	IsPermanent bool                `gorm:"column:is_permanent;type:int(1);not null;default:0;comment:是否永久"`
	ValidDays   int                 `gorm:"column:valid_days;type:int(11);comment:有效期天数"`
	SaleStatus  SaleStatus          `gorm:"column:sale_status;type:enum('on_sale','off_sale');not null;default:'on_sale';comment:上架/下架状态"`
	Deleted     bool                `gorm:"column:deleted;type:tinyint(1);not null;default:0;comment:是否删除"`
	CreatedTime time.Time           `gorm:"column:created_time;type:datetime;comment:创建时间"`
	UpdatedTime time.Time           `gorm:"column:updated_time;type:datetime;comment:更新时间;default:CURRENT_TIMESTAMP on update CURRENT_TIMESTAMP"`
}

// TableName 指定表名
func (MerchantProduct) TableName() string {
	return "merchant_products"
}

func GetMerchantProductById(id int) (product MerchantProduct, err error) {
	db := models.Main()
	err = db.Select(detailColumns).Where("id = ? and deleted =?", id, false).First(&product).Error
	return
}

func GetMerchantProductBySourceId(sourceId int, productType ...MerchantProductType) (product MerchantProduct, err error) {
	db := models.Main()
	if len(productType) > 0 {
		if len(productType) == 1 {
			err = db.Select(detailColumns).Where("source_id =? and type = ? and deleted =?", sourceId, productType[0], false).First(&product).Error
		} else {
			err = db.Select(detailColumns).Where("source_id =? and type in (?) and deleted =?", sourceId, productType, false).First(&product).Error
		}

	} else {
		err = db.Select(detailColumns).Where("source_id =? and deleted =?", sourceId, false).First(&product).Error
	}

	return
}

func GetProductListBySourceIds(ids []int, detail bool, productType ...MerchantProductType) (productList []MerchantProduct, err error) {
	db := models.Main()
	var columns string
	if productType == nil {
		err = errors.New("productType参数不能为空")
		return
	}
	if detail {
		columns = detailColumns
	} else {
		columns = sourceIdColumn
	}
	if len(productType) > 0 {
		if len(productType) == 1 {
			err = db.Select(columns).Where("source_id in ? and type = ? and deleted =? order by created_time desc", ids, productType[0], false).Find(&productList).Error
		} else {
			err = db.Select(columns).Where("source_id in ? and type in (?) and deleted =? by created_time desc", ids, productType, false).Find(&productList).Error
		}

	} else {
		err = db.Select(columns).Where("source_id in ?  and deleted =? by created_time desc", ids, productType, false).Find(&productList).Error
	}

	return
}

func GetTotalPageCountByProductType(productType MerchantProductType) (total int64, latestId int64) {
	db := models.Main()
	_ = db.Model(&MerchantProduct{}).Select("count(*)").Where("type=? and deleted =?", productType, false).Scan(&total).Error
	_ = db.Model(&MerchantProduct{}).Select("max(id)").Where("type=? and deleted =?", productType, false).Scan(&latestId).Error
	return
}

func GetProductPageByProductType(productIds []int, id int64, offset int, limit int) (list []MerchantProduct, err error) {
	db := models.Main()
	err = db.Select(detailColumns).Where("id <= ?  and deleted =?  and id in ? order by created_time desc limit ?,? ", id, false, productIds, offset, limit).Find(&list).Error
	return
}

func GetProductListByProductType(productType MerchantProductType, detail bool) (list []MerchantProduct, err error) {
	db := models.Main()
	var columns string
	if detail {
		columns = detailColumns
	} else {
		columns = sourceIdColumn
	}
	if productType == "" {
		err = errors.New("productType参数不能为空")
		return
	}
	err = db.Select(columns).Where(" type = ? and deleted =? order by created_time desc", productType, false).Find(&list).Error
	return
}
func CountProductList(isSignal bool) (total, latestId int64, ids []int) {
	db := models.Main()
	var condition string
	if isSignal {
		condition = "type !=?"
	} else {
		condition = "type =?"
	}
	_ = db.Select("count(*)").Where(" ? and deleted =? order by id desc ", condition, Package, false).Scan(&total).Error
	_ = db.Select("max(id)").Where(" ? and deleted =? order by id desc ", condition, Package, false).Scan(&total).Error
	_ = db.Select("id").Where(" ? and deleted =? order by id desc ", condition, Package, false).Scan(&ids).Error
	return
}

type MerchantProductIdMap struct {
	Type      string
	ProductId string
}

func GetProductByProductType() (productIds map[string]string, err error) {
	db := models.Main()
	var productIdMap []MerchantProductIdMap
	sql := `SELECT type,GROUP_CONCAT(source_id ORDER BY source_id SEPARATOR ',') as source_id FROM merchant_products GROUP BY type`
	err = db.Raw(sql).Find(&productIdMap).Error
	productIds = make(map[string]string, len(productIdMap))
	for _, v := range productIdMap {
		productIds[v.Type] = v.ProductId
	}
	return
}

func generateSignalPdSql(idMap map[string][]int) (condition string, pars []interface{}) {
	for k, v := range idMap {
		if condition == "" {
			condition = "(" + fmt.Sprintf("(type =? and source_id in ?)")
		} else {
			condition = condition + fmt.Sprintf(" or (type =? and source_id in ?)")
		}
		pars = append(pars, k, v)
	}
	condition = condition + ")"
	return
}

func LatestId() (latestId int64) {
	db := models.Main()
	_ = db.Select("max(id)").Where("deleted=?", false).Scan(&latestId).Error
	return
}

type ProductDTO struct {
	SourceId   int
	SourceType string
	Score      float64
}

func ProductListBySort(list []ProductDTO, id int64, offset int, size int) (productList []MerchantProduct, err error) {
	var productMap map[string][]int
	sort.Slice(list, func(i, j int) bool {
		return list[i].Score > list[j].Score
	})
	var idSort []int
	for _, v := range list {
		idSort = append(idSort, v.SourceId)
	}
	for _, v := range list {
		ids := productMap[v.SourceType]
		ids = append(ids, v.SourceId)
		productMap[v.SourceType] = ids
	}
	sql, pars := generateSignalPdSql(productMap)
	db := models.Main()
	err = db.Select(detailColumns).Where(fmt.Sprintf("id <= ? and deleted =? and  %s order by Field(id,?) desc limit ?,? ", sql), id, false, pars, idSort, offset, size).Find(&productList).Error
	return
}

type packageStruct struct {
	SourceId int
	Weight   int32
}

func PackageListBySort(weightMap map[int]*atomic.Int32, id int64, offset int, size int) (productList []MerchantProduct, err error) {
	var packageList []packageStruct
	for k, v := range weightMap {
		packageList = append(packageList, packageStruct{
			SourceId: k,
			Weight:   v.Load(),
		})
	}
	sort.Slice(packageList, func(i, j int) bool {
		return packageList[i].Weight > packageList[j].Weight
	})
	var idSort []int
	for _, v := range packageList {
		idSort = append(idSort, v.SourceId)
	}
	db := models.Main()
	err = db.Select(detailColumns).Where("id <= ?  and type =? and deleted =? and source_id in ? order by Field(id,?) desc limit ?,? ", id, Package, false, idSort, offset, size).Find(&productList).Error
	return
}