106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
|
package account
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"go-common/app/interface/main/creative/conf"
|
|||
|
|
"go-common/app/interface/main/creative/dao/account"
|
|||
|
|
"go-common/app/interface/main/creative/dao/archive"
|
|||
|
|
"go-common/app/interface/main/creative/dao/article"
|
|||
|
|
"go-common/app/interface/main/creative/dao/up"
|
|||
|
|
"go-common/app/interface/main/creative/service"
|
|||
|
|
"go-common/library/log"
|
|||
|
|
"go-common/library/stat/prom"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
//Service struct.
|
|||
|
|
type Service struct {
|
|||
|
|
c *conf.Config
|
|||
|
|
acc *account.Dao
|
|||
|
|
article *article.Dao
|
|||
|
|
archive *archive.Dao
|
|||
|
|
up *up.Dao
|
|||
|
|
exemptIDCheckUps map[int64]int64
|
|||
|
|
exemptZeroLevelAndAnswerUps map[int64]int64
|
|||
|
|
uploadTopSizeUps map[int64]int64
|
|||
|
|
missch chan func()
|
|||
|
|
pCacheHit *prom.Prom
|
|||
|
|
pCacheMiss *prom.Prom
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//New get service.
|
|||
|
|
func New(c *conf.Config, rpcdaos *service.RPCDaos) *Service {
|
|||
|
|
s := &Service{
|
|||
|
|
c: c,
|
|||
|
|
acc: rpcdaos.Acc,
|
|||
|
|
article: rpcdaos.Art,
|
|||
|
|
archive: rpcdaos.Arc,
|
|||
|
|
up: rpcdaos.Up,
|
|||
|
|
missch: make(chan func(), 1024),
|
|||
|
|
pCacheHit: prom.CacheHit,
|
|||
|
|
pCacheMiss: prom.CacheMiss,
|
|||
|
|
}
|
|||
|
|
s.loadExemptIDCheckUps()
|
|||
|
|
s.loadExemptZeroLevelAndAnswer()
|
|||
|
|
s.loadUpSpecialUploadTopSize()
|
|||
|
|
go s.loadproc()
|
|||
|
|
go s.cacheproc()
|
|||
|
|
return s
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// loadproc
|
|||
|
|
func (s *Service) loadproc() {
|
|||
|
|
for {
|
|||
|
|
time.Sleep(5 * time.Minute)
|
|||
|
|
s.loadExemptIDCheckUps()
|
|||
|
|
s.loadExemptZeroLevelAndAnswer()
|
|||
|
|
s.loadUpSpecialUploadTopSize()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// loadExemptIDCheckUps 白名单人员，免实名认证 type=8.
|
|||
|
|
func (s *Service) loadExemptIDCheckUps() {
|
|||
|
|
ups, err := s.up.UpSpecial(context.TODO(), 8)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
s.exemptIDCheckUps = ups
|
|||
|
|
log.Info("exemptIDCheckUps ups: (%v)", len(ups))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// loadExemptIDCheckUps 白名单人员，免账号激活，免账号升级到1级 type=12
|
|||
|
|
func (s *Service) loadExemptZeroLevelAndAnswer() {
|
|||
|
|
ups, err := s.up.UpSpecial(context.TODO(), 12)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
s.exemptZeroLevelAndAnswerUps = ups
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// loadUpSpecialUploadTopSize 投稿名单限制升级 上传视频可以超过4G,但是在8G以下 type=16
|
|||
|
|
func (s *Service) loadUpSpecialUploadTopSize() {
|
|||
|
|
ups, err := s.up.UpSpecial(context.TODO(), 16)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
s.uploadTopSizeUps = ups
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AddCache add to chan for cache
|
|||
|
|
func (s *Service) addCache(f func()) {
|
|||
|
|
select {
|
|||
|
|
case s.missch <- f:
|
|||
|
|
default:
|
|||
|
|
log.Warn("cacheproc chan full")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// cacheproc is a routine for execute closure.
|
|||
|
|
func (s *Service) cacheproc() {
|
|||
|
|
for {
|
|||
|
|
f := <-s.missch
|
|||
|
|
f()
|
|||
|
|
}
|
|||
|
|
}
|