openbilibili/app/interface/main/app-feed/dao/bangumi/dao.go

package bangumi

import (
	"context"
	"fmt"
	"net/url"
	"strconv"
	"time"

	"go-common/app/interface/main/app-card/model/card/bangumi"
	"go-common/app/interface/main/app-feed/conf"
	feed "go-common/app/service/main/feed/model"
	episodegrpc "go-common/app/service/openplatform/pgc-season/api/grpc/episode/v1"
	"go-common/library/ecode"
	httpx "go-common/library/net/http/blademaster"
	"go-common/library/net/metadata"
	"go-common/library/xstr"

	"github.com/pkg/errors"
)

const (
	_seasons     = "/api/inner/aid_episodes_v2"
	_updates     = "/internal_api/follow_update"
	_pullSeasons = "/internal_api/follow_seasons"
	_followPull  = "/pgc/internal/moe/2018/follow/pull"
)

// Dao is show dao.
type Dao struct {
	// http client
	client *httpx.Client
	// bangumi
	seasons     string
	updates     string
	pullSeasons string
	followPull  string
	// grpc
	rpcClient episodegrpc.EpisodeClient
}

// New new a bangumi dao.
func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		// http clients
		client:      httpx.NewClient(c.HTTPBangumi),
		seasons:     c.Host.Bangumi + _seasons,
		updates:     c.Host.Bangumi + _updates,
		pullSeasons: c.Host.Bangumi + _pullSeasons,
		followPull:  c.Host.APICo + _followPull,
	}
	var err error
	if d.rpcClient, err = episodegrpc.NewClient(c.PGCRPC); err != nil {
		panic(fmt.Sprintf("episodegrpc NewClientt error (%+v)", err))
	}
	return d
}

func (d *Dao) Seasons(c context.Context, aids []int64, now time.Time) (sm map[int64]*bangumi.Season, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("aids", xstr.JoinInts(aids))
	params.Set("type", "av")
	params.Set("build", "app-feed")
	params.Set("platform", "Golang")
	var res struct {
		Code   int                       `json:"code"`
		Result map[int64]*bangumi.Season `json:"result"`
	}
	if err = d.client.Get(c, d.seasons, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(err, d.seasons+"?"+params.Encode())
		return
	}
	sm = res.Result
	return
}

func (d *Dao) Updates(c context.Context, mid int64, now time.Time) (u *bangumi.Update, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("mid", strconv.FormatInt(mid, 10))
	var res struct {
		Code   int             `json:"code"`
		Result *bangumi.Update `json:"result"`
	}
	if err = d.client.Get(c, d.updates, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(err, d.updates+"?"+params.Encode())
		return
	}
	u = res.Result
	return
}

func (d *Dao) PullSeasons(c context.Context, seasonIDs []int64, now time.Time) (psm map[int64]*feed.Bangumi, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("season_ids", xstr.JoinInts(seasonIDs))
	var res struct {
		Code   int             `json:"code"`
		Result []*feed.Bangumi `json:"result"`
	}
	if err = d.client.Get(c, d.pullSeasons, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(err, d.pullSeasons+"?"+params.Encode())
		return
	}
	psm = make(map[int64]*feed.Bangumi, len(res.Result))
	for _, p := range res.Result {
		psm[p.SeasonID] = p
	}
	return
}

func (d *Dao) FollowPull(c context.Context, mid int64, mobiApp, device string, now time.Time) (moe *bangumi.Moe, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("mid", strconv.FormatInt(mid, 10))
	params.Set("mobi_app", mobiApp)
	params.Set("device", device)
	var res struct {
		Code   int          `json:"code"`
		Result *bangumi.Moe `json:"result"`
	}
	if err = d.client.Get(c, d.followPull, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(err, d.followPull+"?"+params.Encode())
		return
	}
	moe = res.Result
	return
}