403 lines
12 KiB
Go
403 lines
12 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
|
||
"go-common/app/job/live/xroom-feed/internal/model"
|
||
daoAnchorV1 "go-common/app/service/live/dao-anchor/api/grpc/v1"
|
||
"go-common/library/log"
|
||
)
|
||
|
||
const (
|
||
_isAll = -1
|
||
_roomStatusTagId = 3
|
||
_onlineCurrent = 3
|
||
)
|
||
|
||
type attrFilter struct {
|
||
attrId int64
|
||
attrSubId int64
|
||
max int64
|
||
min int64
|
||
top int64
|
||
}
|
||
|
||
var attrType = map[string]int64{
|
||
_onlineType: 1,
|
||
_incomeType: 2,
|
||
_dmsType: 3,
|
||
_hourRankType: 4,
|
||
_liveDaysType: 5,
|
||
}
|
||
|
||
var tagType = map[string]bool{
|
||
_areaType: true,
|
||
_roomStatusType: true,
|
||
_anchorCateType: true,
|
||
}
|
||
|
||
type filterItem struct {
|
||
isTagHit bool
|
||
item *daoAnchorV1.AttrResp
|
||
}
|
||
|
||
// ParentAreaIds ...
|
||
var ParentAreaIds []int64
|
||
|
||
// AreaIds ...
|
||
var AreaIds [][]int64
|
||
|
||
// getConditionTypeIndex ...
|
||
func (s *Service) getConditionTypeIndex(conds []*model.RuleProtocol, sType string) (i int64) {
|
||
i = -1
|
||
for index, cond := range conds {
|
||
if cond == nil {
|
||
continue
|
||
}
|
||
if cond.ConfType == sType {
|
||
i = int64(index)
|
||
break
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
// genCondConfRoomList ... recId for log
|
||
func (s *Service) genCondConfRoomList(ctx context.Context, Condition []*model.RuleProtocol, Cond string, recId int) (roomIds []int64) {
|
||
ParentAreaIds := make([]int64, 0)
|
||
AreaIds := make([][]int64, 0)
|
||
areaFilter := make(map[int64]map[int64]bool)
|
||
roomStatus := make(map[int64]bool)
|
||
anchorCate := make(map[int64]bool)
|
||
attrs := make([]*daoAnchorV1.AttrReq, 0)
|
||
attrsFilter := make(map[int64]map[int64]*attrFilter)
|
||
for _, cond := range Condition {
|
||
// 统计型标签
|
||
if attrTypeV, ok := attrType[cond.Key]; ok {
|
||
isAppend := true
|
||
param := &daoAnchorV1.AttrReq{
|
||
AttrId: attrTypeV,
|
||
}
|
||
if len(cond.Condition) <= 0 || cond.Condition[0] == nil {
|
||
continue
|
||
}
|
||
|
||
if cond.Key == _onlineType {
|
||
switch cond.Condition[0].StringV {
|
||
case "current":
|
||
param.AttrSubId = 1
|
||
case "last7day":
|
||
param.AttrSubId = 2
|
||
case "last30day":
|
||
param.AttrSubId = 3
|
||
}
|
||
}
|
||
|
||
if cond.Key == _incomeType || cond.Key == _dmsType {
|
||
switch cond.Condition[0].StringV {
|
||
case "range15min":
|
||
param.AttrSubId = 1
|
||
case "range30min":
|
||
param.AttrSubId = 2
|
||
case "range45min":
|
||
param.AttrSubId = 3
|
||
case "range60min":
|
||
param.AttrSubId = 4
|
||
}
|
||
}
|
||
|
||
if cond.Key == _liveDaysType {
|
||
switch cond.Condition[0].StringV {
|
||
case "last7day":
|
||
param.AttrSubId = 3
|
||
case "last30day":
|
||
param.AttrSubId = 4
|
||
}
|
||
}
|
||
|
||
if cond.Key == _hourRankType {
|
||
if cond.Condition[0].TopV <= 0 {
|
||
isAppend = false
|
||
}
|
||
param.AttrSubId = 1 // 目前小时榜固定是1 @orca
|
||
}
|
||
|
||
// 排除后台选择了小时榜，但是top写了0，导致dao anchor不返回这个attr，导致下面and or 过滤出错
|
||
if isAppend {
|
||
attrsFilter[attrTypeV] = make(map[int64]*attrFilter)
|
||
attrsFilter[attrTypeV][param.AttrSubId] = &attrFilter{}
|
||
if i := s.getConditionTypeIndex(cond.Condition, _confTypeRange); i >= 0 && cond.Condition[i] != nil {
|
||
attrsFilter[attrTypeV][param.AttrSubId].max = cond.Condition[i].Max
|
||
attrsFilter[attrTypeV][param.AttrSubId].min = cond.Condition[i].Min
|
||
}
|
||
if i := s.getConditionTypeIndex(cond.Condition, _confTypeTop); i >= 0 && cond.Condition[i] != nil {
|
||
attrsFilter[attrTypeV][param.AttrSubId].top = cond.Condition[i].TopV
|
||
}
|
||
attrs = append(attrs, param)
|
||
}
|
||
}
|
||
// 展示型标签
|
||
if _, ok := tagType[cond.Key]; ok {
|
||
if cond.Key == _areaType {
|
||
for index, areaCond := range cond.Condition {
|
||
// parent area id dimension
|
||
if index == 0 {
|
||
err := json.Unmarshal([]byte(areaCond.StringV), &ParentAreaIds)
|
||
if err != nil {
|
||
log.Error("[genCondConfRoomList]recId:%d, unmarshalParentAreaIdErr:%+v", recId, err)
|
||
continue
|
||
}
|
||
if len(ParentAreaIds) == 1 && ParentAreaIds[0] == _isAll {
|
||
areaFilter[_isAll] = make(map[int64]bool)
|
||
break
|
||
}
|
||
|
||
for _, pId := range ParentAreaIds {
|
||
if _, ok := areaFilter[pId]; !ok {
|
||
areaFilter[pId] = make(map[int64]bool)
|
||
}
|
||
}
|
||
}
|
||
// area id dimension
|
||
if index == 1 {
|
||
err := json.Unmarshal([]byte(areaCond.StringV), &AreaIds)
|
||
if err != nil {
|
||
log.Error("[genCondConfRoomList]recId:%d, unmarshalAreaIdErr:%+v", recId, err)
|
||
continue
|
||
}
|
||
for pIdIndex, ids := range AreaIds {
|
||
// 后台保证长度对应 ParentAreaId[pIdIndex]
|
||
if len(ParentAreaIds) <= pIdIndex {
|
||
continue
|
||
}
|
||
if _, ok := areaFilter[ParentAreaIds[pIdIndex]]; !ok {
|
||
continue
|
||
}
|
||
for _, id := range ids {
|
||
if _, ok := areaFilter[ParentAreaIds[pIdIndex]][id]; !ok {
|
||
areaFilter[ParentAreaIds[pIdIndex]][id] = true
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
if cond.Key == _roomStatusType && len(cond.Condition) >= 0 && cond.Condition[0] != nil {
|
||
switch cond.Condition[0].StringV {
|
||
case "lottery":
|
||
roomStatus[2] = true
|
||
case "pk":
|
||
roomStatus[1] = true
|
||
}
|
||
}
|
||
|
||
if cond.Key == _anchorCateType && len(cond.Condition) >= 0 && cond.Condition[0] != nil {
|
||
switch cond.Condition[0].StringV {
|
||
case "normal":
|
||
anchorCate[0] = true
|
||
case "sign":
|
||
anchorCate[1] = true
|
||
case "union":
|
||
anchorCate[2] = true
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
log.Info("[genCondConfRoomList]recId:%d, attrs: %+v", recId, attrs)
|
||
roomList, err := s.getOnlineListByAttrs(ctx, attrs)
|
||
if err != nil {
|
||
log.Error("[getOnlineListByAttrs]recId:%d, getOnlineListByAttrsErr:%+v, resp:%+v", recId, err, roomList)
|
||
return
|
||
}
|
||
|
||
filterRoomData := make([]*filterItem, 0)
|
||
for _, roomData := range roomList {
|
||
isHit := false
|
||
isTagHit := false
|
||
if Cond == _condAnd {
|
||
isHit = s.andFilter(attrsFilter, areaFilter, roomStatus, anchorCate, roomData, recId)
|
||
}
|
||
|
||
if Cond == _condOr {
|
||
isHit, isTagHit = s.orFilter(attrsFilter, areaFilter, roomStatus, anchorCate, roomData, recId)
|
||
}
|
||
|
||
if !isHit {
|
||
continue
|
||
}
|
||
|
||
// 命中逻辑
|
||
filterRoomData = append(filterRoomData, &filterItem{
|
||
isTagHit: isTagHit,
|
||
item: roomData,
|
||
})
|
||
}
|
||
|
||
return s.sort(attrsFilter, filterRoomData, Cond, recId)
|
||
}
|
||
|
||
func (s *Service) andFilter(attrsFilters map[int64]map[int64]*attrFilter, areaFilter map[int64]map[int64]bool, roomStatus, anchorCate map[int64]bool, roomData *daoAnchorV1.AttrResp, recId int) (isHit bool) {
|
||
// area filter
|
||
if len(areaFilter) > 0 {
|
||
_, isParentAll := areaFilter[_isAll]
|
||
// 不是全选继续判断
|
||
if !isParentAll {
|
||
if _, ok := areaFilter[roomData.ParentAreaId]; !ok {
|
||
log.Info("[andFilter]recId:%d, ParentAreaIdNotMatch:areaFilter:%+v, parentId:%d, roomId:%d", recId, areaFilter, roomData.ParentAreaId, roomData.RoomId)
|
||
return false
|
||
}
|
||
|
||
_, isAreaAll := areaFilter[roomData.ParentAreaId][_isAll]
|
||
_, isAreaIdExist := areaFilter[roomData.ParentAreaId][roomData.AreaId]
|
||
if !isAreaAll && !isAreaIdExist {
|
||
log.Info("[andFilter]recId:%d, AreaIdNotMatch:areaFilter:%+v, Id:%d, roomId:%d", recId, areaFilter, roomData.AreaId, roomData.RoomId)
|
||
return false
|
||
}
|
||
}
|
||
}
|
||
|
||
//如果配置了房间状态筛选
|
||
if len(roomStatus) > 0 {
|
||
roomStatusTag := &daoAnchorV1.TagData{}
|
||
for _, tag := range roomData.TagList {
|
||
if tag.TagId == _roomStatusTagId {
|
||
roomStatusTag = tag
|
||
break
|
||
}
|
||
}
|
||
if _, ok := roomStatus[roomStatusTag.TagSubId]; !ok {
|
||
log.Info("[andFilter]recId:%d, roomStatusNotMatch:roomStatus:%+v, tagSubId:%d, roomId:%d", recId, roomStatus, roomStatusTag.TagSubId, roomData.RoomId)
|
||
return false
|
||
}
|
||
}
|
||
|
||
//如果配置了主播类型筛选
|
||
if len(anchorCate) > 0 {
|
||
if _, ok := anchorCate[roomData.AnchorProfileType]; !ok {
|
||
log.Info("[andFilter]recId:%d, anchorCateNotMatch:anchorCate:%+v, anchorProfileType:%d, roomId:%d", recId, anchorCate, roomData.AnchorProfileType, roomData.RoomId)
|
||
return false
|
||
}
|
||
}
|
||
|
||
// attr(统计类) filter
|
||
attrsMap := make(map[int64]*daoAnchorV1.AttrData)
|
||
for _, attr := range roomData.AttrList {
|
||
attrsMap[attr.AttrId] = attr
|
||
}
|
||
for attrId, attrFilter := range attrsFilters {
|
||
if _, ok := attrsMap[attrId]; !ok {
|
||
log.Info("[andFilter]recId:%d, attrNotExist:attrId:%d, attrFilter:%d, roomID:%d", recId, attrId, attrsMap, roomData.RoomId)
|
||
return false
|
||
}
|
||
for attrSubId, attrInfo := range attrFilter {
|
||
if attrInfo == nil {
|
||
continue
|
||
}
|
||
if attrsMap[attrId].AttrSubId == attrSubId {
|
||
if attrInfo.min > 0 && attrsMap[attrId].AttrValue < attrInfo.min {
|
||
log.Info("[andFilter]recId:%d, attrMinNotMatch:attrId:%d, value:%d, max:%d, min:%d, roomID:%d", recId, attrId, attrsMap[attrId].AttrValue, attrInfo.max, attrInfo.min, roomData.RoomId)
|
||
return false
|
||
}
|
||
|
||
if attrInfo.max > 0 && attrsMap[attrId].AttrValue > attrInfo.max {
|
||
log.Info("[andFilter]recId:%d, attrMaxNotMatch:attrId:%d, value:%d, max:%d, min:%d, roomID:%d", recId, attrId, attrsMap[attrId].AttrValue, attrInfo.max, attrInfo.min, roomData.RoomId)
|
||
return false
|
||
}
|
||
}
|
||
}
|
||
}
|
||
log.Info("[andFilter]recId:%d, successMatch:roomData:%+v,attrs:%+v, areaFilter:%+v, roomStatus:%+v, anchorState:%+v, roomID:%d", recId, roomData, attrsFilters, areaFilter, roomStatus, anchorCate, roomData.RoomId)
|
||
|
||
return true
|
||
}
|
||
|
||
func (s *Service) orFilter(attrsFilters map[int64]map[int64]*attrFilter, areaFilter map[int64]map[int64]bool, roomStatus, anchorCate map[int64]bool, roomData *daoAnchorV1.AttrResp, recId int) (isHit bool, isTagHit bool) {
|
||
// area filter
|
||
isTagHit = false
|
||
if len(areaFilter) > 0 {
|
||
if _, ok := areaFilter[_isAll]; ok {
|
||
log.Info("[orFilter]recId:%d, ParentAreaIdAllMatch:areaFilter:%+v, parentId:%d, roomID:%d", recId, areaFilter, roomData.ParentAreaId, roomData.RoomId)
|
||
isTagHit = true
|
||
return true, isTagHit
|
||
}
|
||
|
||
if _, ok := areaFilter[roomData.ParentAreaId]; ok {
|
||
log.Info("[orFilter]recId:%d, ParentAreaIdMatch:areaFilter:%+v, parentId:%d, roomID:%d", recId, areaFilter, roomData.ParentAreaId, roomData.RoomId)
|
||
isTagHit = true
|
||
return true, isTagHit
|
||
}
|
||
|
||
_, isAll := areaFilter[roomData.ParentAreaId][_isAll]
|
||
_, isAreaIdExist := areaFilter[roomData.ParentAreaId][roomData.AreaId]
|
||
if isAll || isAreaIdExist {
|
||
log.Info("[orFilter]recId:%d, AreaIdMatch:areaFilter:%+v, Id:%d, roomID:%d", recId, areaFilter, roomData.AreaId, roomData.RoomId)
|
||
isTagHit = true
|
||
return true, isTagHit
|
||
}
|
||
}
|
||
|
||
//如果配置了房间状态筛选
|
||
if len(roomStatus) > 0 {
|
||
roomStatusTag := &daoAnchorV1.TagData{}
|
||
for _, tag := range roomData.TagList {
|
||
if tag.TagId == _roomStatusTagId {
|
||
roomStatusTag = tag
|
||
break
|
||
}
|
||
}
|
||
if _, ok := roomStatus[roomStatusTag.TagSubId]; ok {
|
||
log.Info("[orFilter]recId:%d, roomStatusMatch:roomStatus:%+v, tagSubId:%d, roomID:%d", recId, roomStatus, roomStatusTag.TagSubId, roomData.RoomId)
|
||
isTagHit = true
|
||
return true, isTagHit
|
||
}
|
||
}
|
||
|
||
//如果配置了主播类型筛选
|
||
if len(anchorCate) > 0 {
|
||
if _, ok := anchorCate[roomData.AnchorProfileType]; ok {
|
||
log.Info("[orFilter]recId:%d, anchorCateMatch:anchorCate:%+v, anchorProfileType:%d, roomID:%d", recId, anchorCate, roomData.AnchorProfileType, roomData.RoomId)
|
||
isTagHit = true
|
||
return true, isTagHit
|
||
}
|
||
}
|
||
|
||
// attr filter
|
||
attrDataMap := make(map[int64]*daoAnchorV1.AttrData)
|
||
for _, attr := range roomData.AttrList {
|
||
attrDataMap[attr.AttrId] = attr
|
||
}
|
||
for attrId, attrFilter := range attrsFilters {
|
||
if _, ok := attrDataMap[attrId]; !ok {
|
||
continue
|
||
}
|
||
|
||
//小时榜不需要判断max min
|
||
if attrId == attrType[_hourRankType] {
|
||
return true, isTagHit
|
||
}
|
||
for attrSubId, attrFilterInfo := range attrFilter {
|
||
attrData, ok := attrDataMap[attrId]
|
||
if !ok {
|
||
continue
|
||
}
|
||
if attrData.AttrSubId == attrSubId {
|
||
if attrFilterInfo.min > 0 && attrData.AttrValue < attrFilterInfo.min {
|
||
continue
|
||
}
|
||
if attrFilterInfo.max > 0 && attrData.AttrValue > attrFilterInfo.max {
|
||
continue
|
||
}
|
||
log.Info("[orFilter]recId:%d, attrMaxMinMatch:attrId:%d, value:%d, max:%d, min:%d, roomID:%d", recId, attrId, attrDataMap[attrId].AttrValue, attrFilterInfo.max, attrFilterInfo.min, roomData.RoomId)
|
||
|
||
return true, isTagHit
|
||
}
|
||
}
|
||
}
|
||
|
||
log.Info("[orFilter]recId:%d, failMatch:roomData:%+v,attrs:%+v, areaFilter:%+v, roomStatus:%+v, anchorState:%+v, roomId:%d", recId, roomData, attrsFilters, areaFilter, roomStatus, anchorCate, roomData.RoomId)
|
||
return false, isTagHit
|
||
}
|