openbilibili/app/interface/main/mcn/service/service.go

package service

import (
	"context"
	"time"

	"go-common/app/admin/main/mcn/model"
	"go-common/app/interface/main/mcn/conf"
	"go-common/app/interface/main/mcn/dao/bfs"
	"go-common/app/interface/main/mcn/dao/cache"
	"go-common/app/interface/main/mcn/dao/global"
	"go-common/app/interface/main/mcn/dao/mcndao"
	"go-common/app/interface/main/mcn/dao/msg"
	"go-common/app/interface/main/mcn/tool/worker"

	"go-common/app/interface/main/mcn/dao/datadao"

	"github.com/bluele/gcache"
)

// Service struct
type Service struct {
	c             *conf.Config
	mcndao        *mcndao.Dao
	bfsdao        *bfs.Dao
	notifych      chan func()
	msg           *msg.Dao
	msgMap        map[model.MSGType]*model.MSG
	worker        *worker.Pool
	uniqueChecker *UniqueCheck
	datadao       *datadao.Dao
}

// New init
func New(c *conf.Config) (s *Service) {
	var localcache = gcache.New(c.RankCache.Size).Simple().Build()
	global.Init(c)
	s = &Service{
		c:             c,
		mcndao:        mcndao.New(c, localcache),
		bfsdao:        bfs.New(c),
		notifych:      make(chan func(), 10240),
		msg:           msg.New(c),
		worker:        worker.New(nil),
		uniqueChecker: NewUniqueCheck(),
		datadao:       datadao.New(c),
	}
	s.datadao.Client.Debug = true
	s.refreshCache()
	s.setMsgTypeMap()
	go s.cacheproc()
	return s
}

// Ping Service
func (s *Service) Ping(c context.Context) (err error) {
	return nil
}

// Close Service
func (s *Service) Close() {
	s.worker.Close()
	s.worker.Wait()
	s.mcndao.Close()
}

func (s *Service) refreshCache() {
	cache.LoadCache()
	s.loadMcnUniqueCache()
}

func (s *Service) cacheproc() {
	for {
		time.Sleep(5 * time.Minute)
		s.refreshCache()
	}
}