95 lines
2.9 KiB
Go
95 lines
2.9 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"go-common/app/interface/live/push-live/dao"
|
|||
|
|
"go-common/app/interface/live/push-live/model"
|
|||
|
|
"go-common/library/cache/redis"
|
|||
|
|
"go-common/library/log"
|
|||
|
|
"go-common/library/queue/databus"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// LiveCommonMessage 直播通用消息
|
|||
|
|
func (s *Service) LiveCommonMessage(ctx context.Context, msg *databus.Message) (err error) {
|
|||
|
|
defer msg.Commit()
|
|||
|
|
var (
|
|||
|
|
mids []int64
|
|||
|
|
mMap = make(map[int64]bool) // mid去重
|
|||
|
|
midMap = make(map[int][]int64) // 最终格式化后的mid map
|
|||
|
|
)
|
|||
|
|
m := new(model.LiveCommonMessage)
|
|||
|
|
if err = json.Unmarshal(msg.Value, &m); err != nil {
|
|||
|
|
log.Error("[service.common_message|LiveCommonMessage] json Unmarshal error(%v), model(%v)", err, m)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
task := s.InitCommonTask(m)
|
|||
|
|
if mids, err = s.convertStrToInt64(m.MsgContent.Mids); err != nil {
|
|||
|
|
log.Error("[service.push|LiveCommonMessage] format Mids error(%v), task(%v), model(%v)", err, task, m)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// remove duplicated mid
|
|||
|
|
for _, mid := range mids {
|
|||
|
|
mMap[mid] = true
|
|||
|
|
}
|
|||
|
|
// mid filter
|
|||
|
|
business := m.MsgContent.Business
|
|||
|
|
filteredMids := s.midFilter(mMap, business, task)
|
|||
|
|
midMap[business] = filteredMids
|
|||
|
|
log.Info("[service.push|LiveCommonMessage] message info: before(%d), after(%d), model(%v), task(%v)",
|
|||
|
|
len(mMap), len(midMap[business]), m, task)
|
|||
|
|
total := s.Push(task, midMap)
|
|||
|
|
// create push task
|
|||
|
|
go s.CreatePushTask(task, total)
|
|||
|
|
go s.setPushInterval(business, s.safeGetExpired(), filteredMids, task)
|
|||
|
|
log.Info("[service.push|LiveCommonMessage] common message push done, total(%d), err(%v)", total, err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// InitCommonTask Init push task by common message model
|
|||
|
|
func (s *Service) InitCommonTask(m *model.LiveCommonMessage) (task *model.ApPushTask) {
|
|||
|
|
task = &model.ApPushTask{
|
|||
|
|
Type: model.LivePushType,
|
|||
|
|
TargetID: 0,
|
|||
|
|
AlertTitle: m.MsgContent.AlertTitle,
|
|||
|
|
AlertBody: m.MsgContent.AlertBody,
|
|||
|
|
MidSource: m.MsgContent.Business,
|
|||
|
|
LinkType: m.MsgContent.LinkType,
|
|||
|
|
LinkValue: m.MsgContent.LinkValue,
|
|||
|
|
ExpireTime: m.MsgContent.ExpireTime,
|
|||
|
|
Group: m.MsgContent.Group,
|
|||
|
|
}
|
|||
|
|
return task
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// setPushInterval 活动预约，对每个mid设置推送平滑key
|
|||
|
|
func (s *Service) setPushInterval(business int, expired int32, mids []int64, task *model.ApPushTask) (total int, err error) {
|
|||
|
|
if business != 111 {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
var conn redis.Conn
|
|||
|
|
defer func() {
|
|||
|
|
if conn != nil {
|
|||
|
|
conn.Close()
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
// redis conn
|
|||
|
|
conn, err = redis.Dial(s.c.Redis.PushInterval.Proto, s.c.Redis.PushInterval.Addr, s.dao.RedisOption()...)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("[service.common_message|setPushInterval] redis.Dial error(%v), task(%v), mids(%d)",
|
|||
|
|
err, task, len(mids))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
for _, mid := range mids {
|
|||
|
|
key := dao.GetIntervalKey(mid)
|
|||
|
|
_, err = conn.Do("SET", key, task.LinkValue, "EX", expired)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Error("[service.common_message|setPushInterval] set redis error(%v), task(%v), mid(%d)",
|
|||
|
|
err, task, mid)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
total++
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|