93 lines
2.8 KiB
Go
93 lines
2.8 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"go-common/app/interface/live/push-live/model"
|
|||
|
|
"go-common/library/log"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// liveMessageConsumeproc Live push related message handler
|
|||
|
|
func (s *Service) liveMessageConsumeproc() {
|
|||
|
|
defer func() {
|
|||
|
|
log.Warn("liveMessageConsumeproc exited.")
|
|||
|
|
s.wg.Done()
|
|||
|
|
}()
|
|||
|
|
var (
|
|||
|
|
liveStartMsgs = s.liveStartSub.Messages()
|
|||
|
|
liveCommonMsgs = s.liveCommonSub.Messages()
|
|||
|
|
)
|
|||
|
|
for {
|
|||
|
|
select {
|
|||
|
|
case msg, ok := <-liveStartMsgs:
|
|||
|
|
if !ok {
|
|||
|
|
log.Warn("[service.push|liveMessageConsumeproc] liveStartSub has been closed.")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
log.Info("[service.push|liveMessageConsumeproc] consume liveStartSub key(%s) offset(%d) message(%s)",
|
|||
|
|
msg.Key, msg.Offset, msg.Value)
|
|||
|
|
s.LiveStartMessage(context.TODO(), msg)
|
|||
|
|
case msg, ok := <-liveCommonMsgs:
|
|||
|
|
if !ok {
|
|||
|
|
log.Warn("[service.push|liveMessageConsumeproc] liveCommonSub has been closed.")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
log.Info("[service.push|liveMessageConsumeproc] consume liveCommonSub key(%s) offset(%d) message(%s)",
|
|||
|
|
msg.Key, msg.Offset, msg.Value)
|
|||
|
|
s.LiveCommonMessage(context.TODO(), msg)
|
|||
|
|
default:
|
|||
|
|
time.Sleep(time.Second * 3)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Push 组装业务参数，调用推送平台接口
|
|||
|
|
func (s *Service) Push(task *model.ApPushTask, midMap map[int][]int64) (total int) {
|
|||
|
|
var shouldPushCount int
|
|||
|
|
for t, list := range midMap {
|
|||
|
|
length := len(list)
|
|||
|
|
shouldPushCount += length
|
|||
|
|
if length > 0 {
|
|||
|
|
// 调用批量推送方法，批量推送逻辑会切分mid与出错重试，最后返回实际推送成功数量
|
|||
|
|
task.Group = s.GetPushGroup(t, task.Group)
|
|||
|
|
pushCount := s.dao.BatchPush(&list, task)
|
|||
|
|
log.Info("[service.push|Push] push type(%d), count(%d), target_id(%v)", t, pushCount, task.TargetID)
|
|||
|
|
total += pushCount
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if shouldPushCount == 0 {
|
|||
|
|
log.Info("[service.push|Push] None to push, task(%+v)", task)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
log.Info("[service.push|Push] push done.should(%d), actual(%d), task(%+v).", shouldPushCount, total, task)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CreatePushTask create push task
|
|||
|
|
func (s *Service) CreatePushTask(task *model.ApPushTask, total int) (affected int64, err error) {
|
|||
|
|
task.Total = total
|
|||
|
|
affected, err = s.dao.CreateNewTask(context.TODO(), task)
|
|||
|
|
if err != nil || affected == 0 {
|
|||
|
|
log.Error("[service.push|CreatePushTask] CreateNewTask error(%v), task(%+v)", err, task)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
log.Info("[service.push|CreatePushTask] CreateNewTask success, task(%+v)", task)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetPushGroup 获取不同类型的group
|
|||
|
|
// 兼容逻辑: 开播提醒topic有指定的group(并且单次开播需要区分关注与特别关注两个group)，其余common message topic会传group
|
|||
|
|
func (s *Service) GetPushGroup(t int, g string) string {
|
|||
|
|
var group string
|
|||
|
|
switch t {
|
|||
|
|
case model.RelationAttention:
|
|||
|
|
group = model.AttentionGroup
|
|||
|
|
case model.RelationSpecial:
|
|||
|
|
group = model.SpecialGroup
|
|||
|
|
default:
|
|||
|
|
group = g
|
|||
|
|
}
|
|||
|
|
return group
|
|||
|
|
}
|