|
|
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") |
|
|
} |
|
|
}
|
|
|
|
