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

package service

import (
	"context"

	"go-common/app/interface/main/playlist/conf"
	"go-common/app/interface/main/playlist/dao"
	accclient "go-common/app/service/main/account/api"
	accwarden "go-common/app/service/main/account/api"
	accrpc "go-common/app/service/main/account/rpc/client"
	arcrpc "go-common/app/service/main/archive/api/gorpc"
	arcclient "go-common/app/service/main/archive/api"
	favrpc "go-common/app/service/main/favorite/api/gorpc"
	"go-common/app/service/main/filter/rpc/client"
	"go-common/library/cache"
	"go-common/library/log"
)

// Service service struct.
type Service struct {
	c   *conf.Config
	dao *dao.Dao
	// rpc
	fav    *favrpc.Service
	arc    *arcrpc.Service2
	acc    *accrpc.Service3
	filter *filter.Service
	// cache proc
	cache *cache.Cache
	// playlist power mids
	allowMids map[int64]struct{}
	maxSort   int64
	arcClient arcclient.ArchiveClient
	accClient accwarden.AccountClient
}

// New new service.
func New(c *conf.Config) *Service {
	s := &Service{
		c:       c,
		dao:     dao.New(c),
		fav:     favrpc.New2(c.FavoriteRPC),
		arc:     arcrpc.New2(c.ArchiveRPC),
		acc:     accrpc.New3(c.AccountRPC),
		filter:  filter.New(c.FilterRPC),
		cache:   cache.New(1, 1024),
		maxSort: c.Rule.MinSort + 4*c.Rule.SortStep*int64(c.Rule.MaxVideoCnt),
	}
	var err error
	if s.arcClient, err = arcclient.NewClient(c.ArcClient); err != nil {
		panic(err)
	}
	if s.accClient, err = accclient.NewClient(c.AccClient); err != nil {
		panic(err)
	}

	s.initMids()
	return s
}

func (s *Service) initMids() {
	tmp := make(map[int64]struct{}, len(s.c.Rule.PowerMids))
	for _, id := range s.c.Rule.PowerMids {
		tmp[id] = struct{}{}
	}
	s.allowMids = tmp
}

// Ping ping service.
func (s *Service) Ping(c context.Context) (err error) {
	if err = s.dao.Ping(c); err != nil {
		log.Error("s.dao.Ping error(%v)", err)
	}
	return
}