435 lines
12 KiB
Go
435 lines
12 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"sync/atomic"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"go-common/app/admin/main/activity/model"
|
||
|
|
"go-common/app/job/main/activity/model/bnj"
|
||
|
|
"go-common/app/service/main/account/api"
|
||
|
|
"go-common/library/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
_preScore = 20000
|
||
|
|
_stepOne = 1
|
||
|
|
_stepTwo = 2
|
||
|
|
_stepThree = 3
|
||
|
|
_stepFour = 4
|
||
|
|
_stepFlagValue = "1"
|
||
|
|
)
|
||
|
|
|
||
|
|
var bnjSteps = []int{_stepOne, _stepTwo, _stepThree, _stepFour}
|
||
|
|
|
||
|
|
func (s *Service) bnjproc() {
|
||
|
|
defer s.waiter.Done()
|
||
|
|
var (
|
||
|
|
c = context.Background()
|
||
|
|
lastTs int64
|
||
|
|
)
|
||
|
|
for {
|
||
|
|
if s.closed {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if s.bnjTimeFinish == 1 {
|
||
|
|
log.Warn("bnjproc bnjTimeFinish")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
msg, ok := <-s.bnjSub.Messages()
|
||
|
|
if !ok {
|
||
|
|
log.Info("bnjproc: databus consumer exit!")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
msg.Commit()
|
||
|
|
m := &bnj.ResetMsg{}
|
||
|
|
if err := json.Unmarshal(msg.Value, m); err != nil {
|
||
|
|
log.Error("bnjproc json.Unmarshal(%s) error(%+v)", msg.Value, err)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if m.Mid <= 0 {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
// broadcast max every 1s
|
||
|
|
if m.Ts-lastTs < 1 {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
lastTs = m.Ts
|
||
|
|
atomic.StoreInt64(&s.bnjLessSecond, s.bnjMaxSecond)
|
||
|
|
// default name
|
||
|
|
pushMsg := &bnj.Push{Second: s.bnjLessSecond, Name: ""}
|
||
|
|
if info, err := s.accClient.Info3(c, &api.MidReq{Mid: m.Mid}); err != nil || info == nil {
|
||
|
|
log.Error("bnjproc s.accClient.Info3(%d) error(%v)", m.Mid, err)
|
||
|
|
} else {
|
||
|
|
var name []rune
|
||
|
|
runes := []rune(info.Info.Name)
|
||
|
|
nameLen := len(runes)
|
||
|
|
if nameLen == 2 {
|
||
|
|
name = append(runes[0:1], []rune("*")...)
|
||
|
|
} else if nameLen > 2 {
|
||
|
|
for i, v := range runes {
|
||
|
|
if i == 0 {
|
||
|
|
name = append(name, v)
|
||
|
|
} else if i == nameLen-1 {
|
||
|
|
name = append(name, runes[nameLen-1:]...)
|
||
|
|
} else if i == 1 {
|
||
|
|
name = append(name, []rune("*")...)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
name = runes
|
||
|
|
}
|
||
|
|
pushMsg.Name = string(name)
|
||
|
|
if pushStr, err := json.Marshal(pushMsg); err != nil {
|
||
|
|
log.Error("bnjproc json.Marshal(%+v) error(%v)", pushMsg, err)
|
||
|
|
} else {
|
||
|
|
atomic.StoreInt64(&s.bnjLessSecond, s.bnjMaxSecond)
|
||
|
|
log.Info("bnjproc mid(%d) reset lessTime(%d) maxTime(%d)", m.Mid, s.bnjLessSecond, s.bnjMaxSecond)
|
||
|
|
if err := s.retryPushAll(context.Background(), string(pushStr), _retryTimes); err != nil {
|
||
|
|
log.Error("bnjproc s.bnj.PushAll(%s) error(%v)", string(pushStr), err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
log.Info("bnjproc key:%s partition:%d offset:%d", msg.Key, msg.Partition, msg.Offset)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func bnjWxFlagKey(lid int64, step int) string {
|
||
|
|
return fmt.Sprintf("bnj_wx_%d_%d", lid, step)
|
||
|
|
}
|
||
|
|
|
||
|
|
func bnjMsgFlagKey(lid int64, step int) string {
|
||
|
|
return fmt.Sprintf("bnj_msg_%d_%d", lid, step)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) initBnjSecond() {
|
||
|
|
for {
|
||
|
|
time.Sleep(time.Second)
|
||
|
|
if time.Now().Unix() < s.c.Bnj2019.StartTime.Unix() {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
break
|
||
|
|
}
|
||
|
|
if value, err := s.retryCacheTimeFinish(context.Background(), _retryTimes); err != nil {
|
||
|
|
log.Error("initBnjSecond s.dao.retryCacheTimeFinish error(%v)", err)
|
||
|
|
return
|
||
|
|
} else if value > 0 {
|
||
|
|
log.Warn("initBnjSecond time finish")
|
||
|
|
atomic.StoreInt64(&s.bnjTimeFinish, value)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
// init step flag
|
||
|
|
for _, v := range s.c.Bnj2019.Time {
|
||
|
|
if v.Step > 0 {
|
||
|
|
if value, err := s.retryRsGet(context.Background(), bnjMsgFlagKey(s.c.Bnj2019.LID, v.Step), _retryTimes); err != nil {
|
||
|
|
log.Error("initBnjSecond msg s.dao.retryRsGet error(%v)")
|
||
|
|
} else if value != "" {
|
||
|
|
log.Info("initBnjSecond msg bnjMsgFlagMap[step:%d]", v.Step)
|
||
|
|
s.bnjMsgFlagMap[v.Step] = 1
|
||
|
|
}
|
||
|
|
if value, err := s.retryRsGet(context.Background(), bnjWxFlagKey(s.c.Bnj2019.LID, v.Step), _retryTimes); err != nil {
|
||
|
|
log.Error("initBnjSecond wx s.dao.retryRsGet error(%v)")
|
||
|
|
} else if value != "" {
|
||
|
|
log.Info("initBnjSecond wx bnjWxMsgFlagMap[step:%d]", v.Step)
|
||
|
|
s.bnjWxMsgFlagMap[v.Step] = 1
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
scores, err := s.retryBatchLikeActSum(context.Background(), []int64{s.c.Bnj2019.LID}, _retryTimes)
|
||
|
|
if err != nil {
|
||
|
|
// TODO need to restart
|
||
|
|
log.Error("initBnjSecond failed s.dao.BatchLikeActSum(%d) error(%v)", s.c.Bnj2019.LID, err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if score, ok := scores[s.c.Bnj2019.LID]; ok {
|
||
|
|
for i, v := range s.c.Bnj2019.Time {
|
||
|
|
if score >= v.Score {
|
||
|
|
atomic.StoreInt64(&s.bnjMaxSecond, v.Second)
|
||
|
|
atomic.StoreInt64(&s.bnjLessSecond, v.Second)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
if i == len(s.c.Bnj2019.Time)-1 {
|
||
|
|
atomic.StoreInt64(&s.bnjMaxSecond, v.Second)
|
||
|
|
atomic.StoreInt64(&s.bnjLessSecond, v.Second)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// max second
|
||
|
|
atomic.StoreInt64(&s.bnjMaxSecond, s.c.Bnj2019.Time[len(s.c.Bnj2019.Time)-1].Second)
|
||
|
|
atomic.StoreInt64(&s.bnjLessSecond, s.c.Bnj2019.Time[len(s.c.Bnj2019.Time)-1].Second)
|
||
|
|
}
|
||
|
|
if lessSecond, err := s.bnj.CacheLessTime(context.Background()); err != nil {
|
||
|
|
log.Error("initBnjSecond s.dao.CacheLessTime error(%v)", err)
|
||
|
|
} else if lessSecond > 0 {
|
||
|
|
atomic.StoreInt64(&s.bnjLessSecond, lessSecond)
|
||
|
|
}
|
||
|
|
log.Warn("initBnjSecond maxSecond(%d) lessSecond(%d)", s.bnjMaxSecond, s.bnjLessSecond)
|
||
|
|
go s.maxSecondproc()
|
||
|
|
go s.lessSecondproc()
|
||
|
|
s.waiter.Add(1)
|
||
|
|
go s.bnjproc()
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) maxSecondproc() {
|
||
|
|
ctx := context.Background()
|
||
|
|
for {
|
||
|
|
if s.closed {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
time.Sleep(time.Second)
|
||
|
|
if scores, err := s.dao.BatchLikeActSum(context.Background(), []int64{s.c.Bnj2019.LID}); err != nil {
|
||
|
|
log.Error("maxSecondproc s.dao.BatchLikeActSum(%d) error(%v)", s.c.Bnj2019.LID, err)
|
||
|
|
} else {
|
||
|
|
if score, ok := scores[s.c.Bnj2019.LID]; ok {
|
||
|
|
for _, v := range s.c.Bnj2019.Time {
|
||
|
|
if score >= v.Score {
|
||
|
|
atomic.StoreInt64(&s.bnjMaxSecond, v.Second)
|
||
|
|
if s.bnjLessSecond > s.bnjMaxSecond {
|
||
|
|
atomic.StoreInt64(&s.bnjLessSecond, s.bnjMaxSecond)
|
||
|
|
}
|
||
|
|
msg := v.Msg
|
||
|
|
mc := v.MsgMc
|
||
|
|
msgTitle := v.MsgTitle
|
||
|
|
step := v.Step
|
||
|
|
if step > 0 && s.bnjMsgFlagMap[step] == 0 {
|
||
|
|
if err = s.retryRsSet(ctx, bnjMsgFlagKey(s.c.Bnj2019.LID, step), _stepFlagValue, _retryTimes); err != nil {
|
||
|
|
log.Error("s.retryRsSet(%d,%d) error(%v)", s.c.Bnj2019.LID, step, err)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
if msg != "" && msgTitle != "" && mc != "" {
|
||
|
|
go s.sendMessageToSubs(ctx, s.c.Bnj2019.LID, mc, msgTitle, msg, _retryTimes)
|
||
|
|
} else {
|
||
|
|
log.Error("bnj msg conf step(%d) error", step)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
log.Info("bnj send msg step:%d finish", step)
|
||
|
|
s.bnjMsgFlagMu.Lock()
|
||
|
|
s.bnjMsgFlagMap[step] = 1
|
||
|
|
s.bnjMsgFlagMu.Unlock()
|
||
|
|
}
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
for _, v := range s.c.Bnj2019.Time {
|
||
|
|
if score+_preScore >= v.Score {
|
||
|
|
wxMsg := v.WxMsg
|
||
|
|
step := v.Step
|
||
|
|
if step > 0 && s.bnjWxMsgFlagMap[step] == 0 {
|
||
|
|
if err = s.retryRsSet(ctx, bnjWxFlagKey(s.c.Bnj2019.LID, step), _stepFlagValue, _retryTimes); err != nil {
|
||
|
|
log.Error("s.retryRsSet(%d,%d) error(%v)", s.c.Bnj2019.LID, step, err)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
if wxMsg != "" && s.c.Bnj2019.WxUser != "" {
|
||
|
|
if err = s.retrySendWechat(ctx, s.c.Bnj2019.WxTitle, wxMsg, s.c.Bnj2019.WxUser, _retryTimes); err != nil {
|
||
|
|
log.Error("s.retrySendWechat(%s,%s) error(%v)", s.c.Bnj2019.WxTitle, wxMsg, err)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
log.Error("bnj wx msg conf step(%d) error", step)
|
||
|
|
break
|
||
|
|
}
|
||
|
|
log.Info("bnj send wx step:%d finish", step)
|
||
|
|
s.bnjWxMsgFlagMu.Lock()
|
||
|
|
s.bnjWxMsgFlagMap[step] = 1
|
||
|
|
s.bnjWxMsgFlagMu.Unlock()
|
||
|
|
}
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
log.Warn("maxSecondproc lid not found")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) lessSecondproc() {
|
||
|
|
for {
|
||
|
|
if s.closed {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
time.Sleep(time.Second)
|
||
|
|
atomic.AddInt64(&s.bnjLessSecond, -1)
|
||
|
|
if s.c.Bnj2019.GameCancel != 0 {
|
||
|
|
log.Warn("lessSecondproc bnj game cancel")
|
||
|
|
atomic.StoreInt64(&s.bnjLessSecond, 0)
|
||
|
|
}
|
||
|
|
if s.bnjLessSecond <= 0 {
|
||
|
|
if err := s.retryAddCacheTimeFinish(context.Background(), 1, _retryTimes); err != nil {
|
||
|
|
log.Error("lessSecondproc s.bnj.AddCacheTimeFinish error(%v)", err)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
log.Warn("lessSecondproc bnj time Finish")
|
||
|
|
atomic.StoreInt64(&s.bnjTimeFinish, 1)
|
||
|
|
pushMsg := &bnj.Push{Second: 0, Name: "", TimelinePic: s.c.Bnj2019.TimelinePic, H5TimelinePic: s.c.Bnj2019.H5TimelinePic}
|
||
|
|
if pushStr, err := json.Marshal(pushMsg); err != nil {
|
||
|
|
log.Error("lessSecondproc json.Marshal(%+v) error(%v)", pushMsg, err)
|
||
|
|
} else {
|
||
|
|
atomic.StoreInt64(&s.bnjLessSecond, s.bnjMaxSecond)
|
||
|
|
if err := s.retryPushAll(context.Background(), string(pushStr), _retryTimes); err != nil {
|
||
|
|
log.Error("lessSecondproc s.bnj.PushAll error(%v)", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
pushMsg := &bnj.Push{Second: s.bnjLessSecond, Name: ""}
|
||
|
|
if pushStr, err := json.Marshal(pushMsg); err != nil {
|
||
|
|
log.Error("lessSecondproc json.Marshal(%+v) error(%v)", pushMsg, err)
|
||
|
|
} else {
|
||
|
|
if err := s.retryPushAll(context.Background(), string(pushStr), 1); err != nil {
|
||
|
|
log.Error("lessSecondproc s.bnj.PushAll error(%v)", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) retryBatchLikeActSum(c context.Context, lids []int64, retryCnt int) (res map[int64]int64, err error) {
|
||
|
|
for i := 0; i < retryCnt; i++ {
|
||
|
|
if res, err = s.dao.BatchLikeActSum(c, lids); err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
time.Sleep(100 * time.Millisecond)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) retryAddCacheTimeFinish(c context.Context, value int64, retryCnt int) (err error) {
|
||
|
|
for i := 0; i < retryCnt; i++ {
|
||
|
|
if err = s.bnj.AddCacheTimeFinish(c, value); err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
time.Sleep(100 * time.Millisecond)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) retryCacheTimeFinish(c context.Context, retryCnt int) (value int64, err error) {
|
||
|
|
for i := 0; i < retryCnt; i++ {
|
||
|
|
if value, err = s.bnj.CacheTimeFinish(c); err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
time.Sleep(100 * time.Millisecond)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) retryPushAll(c context.Context, msg string, retryCnt int) (err error) {
|
||
|
|
for i := 0; i < retryCnt; i++ {
|
||
|
|
if err = s.bnj.PushAll(c, msg); err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) cronInformationMessage() {
|
||
|
|
log.Info("cronInformationMessage start cron")
|
||
|
|
if s.c.Bnj2019.LID == 0 || s.c.Bnj2019.MidLimit == 0 {
|
||
|
|
log.Error("cronInformationMessage conf error")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
var (
|
||
|
|
c = context.Background()
|
||
|
|
title, msg, mc string
|
||
|
|
)
|
||
|
|
for _, v := range s.c.Bnj2019.Message {
|
||
|
|
if time.Now().Unix() >= v.Start.Unix() {
|
||
|
|
title = v.Title
|
||
|
|
msg = v.Content
|
||
|
|
mc = v.Mc
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if title == "" || msg == "" || mc == "" {
|
||
|
|
log.Error("cronInformationMessage message conf error")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
s.sendMessageToSubs(c, s.c.Bnj2019.LID, mc, title, msg, _retryTimes)
|
||
|
|
log.Info("cronInformationMessage finish title(%s)", title)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) sendMessageToSubs(c context.Context, lid int64, mc, title, msg string, retryCnt int) {
|
||
|
|
var minID int64
|
||
|
|
log.Info("sendMessageToSubs mc:%s title:%s start", mc, title)
|
||
|
|
for {
|
||
|
|
time.Sleep(100 * time.Millisecond)
|
||
|
|
actions, err := s.retryLikeActList(c, lid, minID, s.c.Bnj2019.MidLimit, retryCnt)
|
||
|
|
if err != nil {
|
||
|
|
log.Error("sendMessageToSubs s.dao.LikeActList(%d,%d,%d) error(%v)", lid, minID, s.c.Bnj2019.MidLimit, err)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if len(actions) == 0 {
|
||
|
|
log.Info("sendMessageToSubs finish")
|
||
|
|
break
|
||
|
|
}
|
||
|
|
var mids []int64
|
||
|
|
for i, v := range actions {
|
||
|
|
if v.Mid > 0 {
|
||
|
|
mids = append(mids, v.Mid)
|
||
|
|
}
|
||
|
|
if i == len(actions)-1 {
|
||
|
|
minID = v.ID
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if len(mids) == 0 {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if err = s.retrySendMessage(c, mids, mc, title, msg, _retryTimes); err != nil {
|
||
|
|
log.Error("sendMessageToSubs s.dao.SendMessage(mids:%v) error(%v)", mids, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) retryLikeActList(c context.Context, lid, minID, limit int64, retryCnt int) (list []*model.LikeAction, err error) {
|
||
|
|
for i := 0; i < retryCnt; i++ {
|
||
|
|
if list, err = s.dao.LikeActList(c, lid, minID, limit); err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) retrySendMessage(c context.Context, mids []int64, mc, title, msg string, retryCnt int) (err error) {
|
||
|
|
for i := 0; i < retryCnt; i++ {
|
||
|
|
if err = s.bnj.SendMessage(c, mids, mc, title, msg); err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) retryRsSet(c context.Context, key, value string, retryCnt int) (err error) {
|
||
|
|
for i := 0; i < retryCnt; i++ {
|
||
|
|
if err = s.dao.RsSet(c, key, value); err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) retryRsGet(c context.Context, key string, retryCnt int) (value string, err error) {
|
||
|
|
for i := 0; i < retryCnt; i++ {
|
||
|
|
if value, err = s.dao.RsGet(c, key); err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *Service) retrySendWechat(c context.Context, title, msg, user string, retryCnt int) (err error) {
|
||
|
|
for i := 0; i < retryCnt; i++ {
|
||
|
|
if err = s.bnj.SendWechat(c, title, msg, user); err == nil {
|
||
|
|
break
|
||
|
|
}
|
||
|
|
time.Sleep(10 * time.Millisecond)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|