115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package block
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"time"
|
||
|
||
"go-common/app/job/main/member/conf"
|
||
"go-common/app/job/main/member/dao/block"
|
||
"go-common/library/cache"
|
||
"go-common/library/log"
|
||
"go-common/library/queue/databus"
|
||
|
||
"github.com/pkg/errors"
|
||
)
|
||
|
||
// Service struct
|
||
type Service struct {
|
||
dao *block.Dao
|
||
conf *conf.Config
|
||
cache *cache.Cache
|
||
missch chan func()
|
||
|
||
creditSub *databus.Databus
|
||
}
|
||
|
||
// New init
|
||
func New(conf *conf.Config, dao *block.Dao, creditSub *databus.Databus) (s *Service) {
|
||
s = &Service{
|
||
dao: dao,
|
||
conf: conf,
|
||
cache: cache.New(1, 10240),
|
||
missch: make(chan func(), 10240),
|
||
creditSub: creditSub,
|
||
}
|
||
|
||
// 自动解禁检查
|
||
if s.conf.BlockProperty.Flag.ExpireCheck {
|
||
go s.limitcheckproc()
|
||
go s.creditcheckproc()
|
||
}
|
||
|
||
// 小黑屋答题状态订阅
|
||
if s.conf.BlockProperty.Flag.CreditSub {
|
||
go s.creditsubproc()
|
||
}
|
||
|
||
go s.missproc()
|
||
return s
|
||
}
|
||
|
||
// Ping Service
|
||
func (s *Service) Ping(c context.Context) (err error) {
|
||
return s.dao.Ping(c)
|
||
}
|
||
|
||
// Close Service
|
||
func (s *Service) Close() {
|
||
s.dao.Close()
|
||
}
|
||
|
||
func (s *Service) limitcheckproc() {
|
||
defer func() {
|
||
if x := recover(); x != nil {
|
||
log.Error("service.limitcheckproc panic(%v)", x)
|
||
go s.limitcheckproc()
|
||
log.Info("service.limitcheckproc recover")
|
||
}
|
||
}()
|
||
for {
|
||
log.Info("limit check start")
|
||
s.limitExpireHandler(context.TODO())
|
||
log.Info("limit check end")
|
||
time.Sleep(time.Duration(s.conf.BlockProperty.LimitExpireCheckTick))
|
||
}
|
||
}
|
||
|
||
func (s *Service) creditcheckproc() {
|
||
defer func() {
|
||
if x := recover(); x != nil {
|
||
log.Error("%+v", errors.WithStack(fmt.Errorf("service.creditcheckproc panic(%v)", x)))
|
||
go s.creditcheckproc()
|
||
log.Info("service.creditcheckproc recover")
|
||
}
|
||
}()
|
||
for {
|
||
log.Info("black house check start")
|
||
s.creditExpireHandler(context.TODO())
|
||
log.Info("black house check end")
|
||
time.Sleep(time.Duration(s.conf.BlockProperty.CreditExpireCheckTick))
|
||
}
|
||
}
|
||
|
||
func (s *Service) missproc() {
|
||
defer func() {
|
||
if x := recover(); x != nil {
|
||
log.Error("service.missproc panic(%v)", x)
|
||
go s.missproc()
|
||
log.Info("service.missproc recover")
|
||
}
|
||
}()
|
||
for {
|
||
f := <-s.missch
|
||
f()
|
||
}
|
||
}
|
||
|
||
func (s *Service) mission(f func()) {
|
||
select {
|
||
case s.missch <- f:
|
||
default:
|
||
log.Error("s.missch full")
|
||
}
|
||
}
|