115 lines
2.3 KiB
Go
115 lines
2.3 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"go-common/app/job/main/block/conf"
|
|||
|
|
"go-common/app/job/main/block/dao"
|
|||
|
|
"go-common/library/cache"
|
|||
|
|
"go-common/library/log"
|
|||
|
|
"go-common/library/queue/databus"
|
|||
|
|
|
|||
|
|
"github.com/pkg/errors"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Service struct
|
|||
|
|
type Service struct {
|
|||
|
|
dao *dao.Dao
|
|||
|
|
cache *cache.Cache
|
|||
|
|
missch chan func()
|
|||
|
|
|
|||
|
|
creditSub *databus.Databus
|
|||
|
|
accountNotifyPub *databus.Databus
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// New init
|
|||
|
|
func New() (s *Service) {
|
|||
|
|
s = &Service{
|
|||
|
|
dao: dao.New(),
|
|||
|
|
cache: cache.New(1, 10240),
|
|||
|
|
missch: make(chan func(), 10240),
|
|||
|
|
creditSub: databus.New(conf.Conf.Databus.Credit),
|
|||
|
|
accountNotifyPub: databus.New(conf.Conf.AccountNotify),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 自动解禁检查
|
|||
|
|
if conf.Conf.Property.Flag.ExpireCheck {
|
|||
|
|
go s.limitcheckproc()
|
|||
|
|
go s.creditcheckproc()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 小黑屋答题状态订阅
|
|||
|
|
if conf.Conf.Property.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(conf.Conf.Property.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(conf.Conf.Property.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")
|
|||
|
|
}
|
|||
|
|
}
|