113 lines
2.7 KiB
Go
113 lines
2.7 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"go-common/app/job/bbq/video/dao"
|
||
"go-common/library/log"
|
||
"go-common/library/sync/errgroup"
|
||
"sync/atomic"
|
||
"time"
|
||
|
||
notice "go-common/app/service/bbq/notice-service/api/v1"
|
||
)
|
||
|
||
// 信号量，为了避免task执行超过周期，连续执行两个任务会出现问题
|
||
var i32 int32
|
||
|
||
// SysMsgTask 推送消息
|
||
func (s *Service) SysMsgTask() {
|
||
ctx := context.Background()
|
||
newValue := atomic.AddInt32(&i32, 1)
|
||
defer atomic.AddInt32(&i32, -1)
|
||
if newValue > 1 {
|
||
log.Errorv(ctx, log.KV("log", "sysMsgTask pending"))
|
||
return
|
||
}
|
||
|
||
res, err := s.dao.RawCheckTask(ctx, "checkSysMsg")
|
||
if err != nil {
|
||
log.Errorv(ctx, log.KV("log", "get last sysMsgTask id fail"))
|
||
return
|
||
}
|
||
|
||
lastSysMsgID := res.LastCheck
|
||
curSysMsgID := res.LastCheck
|
||
list, err := s.dao.GetNewSysMsg(ctx, curSysMsgID)
|
||
if err != nil {
|
||
log.Errorv(ctx, log.KV("log", "get new sysMsg fail"))
|
||
return
|
||
}
|
||
if len(list) == 0 {
|
||
log.Infov(ctx, log.KV("log", "no new sys msg to be sync to notice"))
|
||
return
|
||
}
|
||
|
||
var mids []int64
|
||
for _, item := range list {
|
||
curSysMsgID = item.Id
|
||
|
||
notice := notice.NoticeBase{
|
||
Mid: item.Receiver,
|
||
ActionMid: item.Sender,
|
||
SvId: 0,
|
||
NoticeType: 4,
|
||
Text: item.Text,
|
||
JumpUrl: item.JumpUrl,
|
||
BizType: dao.NoticeBizTypeSysMsg,
|
||
BizId: item.Id,
|
||
}
|
||
// 全量系统消息
|
||
if item.Receiver == 0 {
|
||
lastUserID := int64(0)
|
||
if len(mids) == 0 {
|
||
for {
|
||
if userBases, err := s.dao.UsersByLast(ctx, lastUserID); err != nil {
|
||
log.Errorv(ctx, log.KV("log", "sys msg task: get user base fail"))
|
||
break
|
||
} else {
|
||
for _, userBase := range userBases {
|
||
mids = append(mids, userBase.MID)
|
||
}
|
||
if len(userBases) > 0 {
|
||
lastUserID = userBases[len(userBases)-1].ID
|
||
} else {
|
||
break
|
||
}
|
||
}
|
||
|
||
}
|
||
}
|
||
midChan := make(chan int64, 20)
|
||
go func() {
|
||
for _, mid := range mids {
|
||
midChan <- mid
|
||
}
|
||
close(midChan)
|
||
}()
|
||
startTime := time.Now()
|
||
g := errgroup.Group{}
|
||
for i := 0; i < 10; i++ {
|
||
g.Go(func() error {
|
||
subNotice := notice
|
||
for mid := range midChan {
|
||
subNotice.Mid = mid
|
||
s.dao.CreateNotice(ctx, &subNotice)
|
||
}
|
||
return nil
|
||
})
|
||
}
|
||
g.Wait()
|
||
log.Info("total sys msg notice push: cost_time=%f, mid_len=%d", time.Since(startTime).Seconds(), len(mids))
|
||
} else {
|
||
s.dao.CreateNotice(ctx, ¬ice)
|
||
}
|
||
}
|
||
|
||
if _, err := s.dao.UpdateTaskLastCheck(ctx, "checkSysMsg", curSysMsgID); err != nil {
|
||
log.Errorv(ctx, log.KV("log", "update check_task mysql fail"))
|
||
return
|
||
}
|
||
|
||
log.Infov(ctx, log.KV("log", "no new sys msg to be sync to notice"), log.KV("last_sys_id", lastSysMsgID), log.KV("cur_sys_id", curSysMsgID))
|
||
}
|