78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
package service
|
||
|
||
import (
|
||
"context"
|
||
"time"
|
||
|
||
"go-common/app/interface/main/push-archive/model"
|
||
pb "go-common/app/service/main/push/api/grpc/v1"
|
||
"go-common/library/log"
|
||
)
|
||
|
||
// Setting gets user's archive-result setting.
|
||
func (s *Service) Setting(c context.Context, mid int64) (st *model.Setting, err error) {
|
||
st, err = s.dao.Setting(c, mid)
|
||
if err != nil {
|
||
return
|
||
}
|
||
if st == nil {
|
||
st = &model.Setting{Type: model.PushTypeSpecial} // 如果用户还未上传配置，则默认为特殊关注
|
||
}
|
||
return
|
||
}
|
||
|
||
// SetSetting saves user's archive-result setting.
|
||
func (s *Service) SetSetting(c context.Context, mid int64, st *model.Setting) (err error) {
|
||
err = s.dao.SetSetting(c, mid, st)
|
||
if err == nil {
|
||
set := &pb.SetSettingRequest{
|
||
Mid: mid,
|
||
Type: 1,
|
||
Value: int32(st.Type),
|
||
}
|
||
s.settingCh <- set
|
||
}
|
||
return
|
||
}
|
||
|
||
func (s *Service) setSettingProc() (err error) {
|
||
defer func() {
|
||
if msg := recover(); msg != nil {
|
||
log.Error("setSettingProc got panic(%+v)", msg)
|
||
}
|
||
}()
|
||
|
||
for {
|
||
time.Sleep(time.Millisecond * 200)
|
||
set, open := <-s.settingCh
|
||
if !open {
|
||
log.Error("setSettingProc settingCh is closed")
|
||
return
|
||
}
|
||
|
||
// before send rpc, check db value with new value, if diff, then rpc is later than another update
|
||
dbSet, err := s.dao.Setting(context.TODO(), set.Mid)
|
||
if err != nil {
|
||
log.Error("setSettingProc s.dao.Setting error(%v) set(%+v)", err, set)
|
||
s.settingCh <- set
|
||
continue
|
||
}
|
||
if dbSet == nil || dbSet.Type != int(set.Value) {
|
||
log.Info("setSettingProc push setting value diff, db(%+v) rpc(%+v)", dbSet, set)
|
||
continue
|
||
}
|
||
|
||
// rpc中0-关闭，1-开启
|
||
var tp int32
|
||
if set.Value == model.PushTypeSpecial || set.Value == model.PushTypeAttention {
|
||
tp = 1
|
||
}
|
||
set.Value = tp
|
||
|
||
if _, err := s.pushRPC.SetSetting(context.TODO(), set); err != nil {
|
||
log.Error("s.pushRPC.SetSetting error(%v) set(%+v)", err, set)
|
||
s.settingCh <- set
|
||
}
|
||
}
|
||
}
|