bilibili-go-common/app/interface/main/app-feed/dao/recommend/dao.go

package recommend

import (
	"time"

	"go-common/app/interface/main/app-feed/conf"
	"go-common/library/cache/memcache"
	"go-common/library/cache/redis"
	httpx "go-common/library/net/http/blademaster"
	"go-common/library/queue/databus"
)

// Dao is show dao.
type Dao struct {
	// http client
	client     *httpx.Client
	clientAsyn *httpx.Client
	// hetongzi
	hot string
	// bigdata
	rcmd           string
	group          string
	top            string
	followModeList string
	// redis
	redis     *redis.Pool
	expireRds int
	// databus
	databus *databus.Databus
	// mc
	mc       *memcache.Pool
	expireMc int32
}

// New new a show dao.
func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		// http client
		client:     httpx.NewClient(c.HTTPData),
		clientAsyn: httpx.NewClient(c.HTTPClientAsyn),
		// hetongzi
		hot: c.Host.Hetongzi + _hot,
		// bigdata
		rcmd:           c.Host.Data + _rcmd,
		group:          c.Host.BigData + _group,
		top:            c.Host.Data + _top,
		followModeList: c.Host.Data + _followModeList,
		// redis
		redis:     redis.NewPool(c.Redis.Feed.Config),
		expireRds: int(time.Duration(c.Redis.Feed.ExpireRecommend) / time.Second),
		// databus
		databus: databus.New(c.DislikeDatabus),
		// mc
		mc:       memcache.NewPool(c.Memcache.Cache.Config),
		expireMc: int32(time.Duration(c.Memcache.Cache.ExpireCache) / time.Second),
	}
	return
}

// Close close resource.
func (d *Dao) Close() {
	if d.redis != nil {
		d.redis.Close()
	}
	if d.mc != nil {
		d.mc.Close()
	}
}