openbilibili/app/service/main/up/dao/global/resource.go

package global

import (
	"runtime"

	accgrpc "go-common/app/service/main/account/api"
	arcgrpc "go-common/app/service/main/archive/api"
	"go-common/app/service/main/up/conf"
	"go-common/library/database/sql"
	"go-common/library/sync/pipeline/fanout"

	"github.com/pkg/errors"
)

var (
	upCrmDB *sql.DB
	worker  *fanout.Fanout
	arcCli  arcgrpc.ArchiveClient
	accCli  accgrpc.AccountClient
)

// GetArcClient .
func GetArcClient() arcgrpc.ArchiveClient {
	return arcCli
}

// GetAccClient .
func GetAccClient() accgrpc.AccountClient {
	return accCli
}

// GetWorker .
func GetWorker() *fanout.Fanout {
	return worker
}

// GetUpCrmDB .
func GetUpCrmDB() *sql.DB {
	return upCrmDB
}

//Init init global
func Init(c *conf.Config) {
	var err error
	if arcCli, err = arcgrpc.NewClient(c.GRPCClient.Archive); err != nil {
		panic(errors.WithMessage(err, "Failed to dial archive service"))
	}
	if accCli, err = accgrpc.NewClient(c.GRPCClient.Account); err != nil {
		panic(errors.WithMessage(err, "Failed to dial account service"))
	}
	upCrmDB = sql.NewMySQL(c.DB.UpCRM)
	worker = fanout.New("cache", fanout.Worker(runtime.NumCPU()), fanout.Buffer(1024))
}

// Close .
func Close() {
	upCrmDB.Close()
	worker.Close()
}