bilibili-go-common/app/interface/main/app-interface/dao/history/dao.go

package history

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

	"go-common/app/interface/main/app-interface/conf"
	model "go-common/app/interface/main/app-interface/model/history"
	hismodle "go-common/app/interface/main/history/model"
	hisrpc "go-common/app/interface/main/history/rpc/client"
	artmodle "go-common/app/interface/openplatform/article/model"
	artrpc "go-common/app/interface/openplatform/article/rpc/client"
	arcrpc "go-common/app/service/main/archive/api/gorpc"
	arcmodle "go-common/app/service/main/archive/model/archive"
	"go-common/library/ecode"
	bm "go-common/library/net/http/blademaster"
	"go-common/library/net/metadata"

	"github.com/pkg/errors"
)

const (
	_apiPGC = "/internal_api/get_eps_v2"
)

// Dao is history dao
type Dao struct {
	client     *bm.Client
	historyRPC *hisrpc.Service
	arcRPC     *arcrpc.Service2
	artRPC     *artrpc.Service
	pgcAPI     string
}

// New initial history dao
func New(c *conf.Config) (d *Dao) {
	d = &Dao{
		client:     bm.NewClient(c.HTTPClient),
		historyRPC: hisrpc.New(c.HistoryRPC),
		arcRPC:     arcrpc.New2(c.ArchiveRPC),
		artRPC:     artrpc.New(c.ArticleRPC),
		pgcAPI:     c.Host.Bangumi + _apiPGC,
	}
	return
}

// History get history
func (d *Dao) History(c context.Context, mid int64, pn, ps int) (res []*hismodle.Resource, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	arg := &hismodle.ArgHistories{Mid: mid, Pn: pn, Ps: ps, RealIP: ip}
	if res, err = d.historyRPC.History(c, arg); err != nil {
		err = errors.Wrapf(err, "d.historyRPC.History(%+v)", arg)
	}
	return
}

// Archive get archive info
func (d *Dao) Archive(c context.Context, aids []int64) (info map[int64]*arcmodle.View3, err error) {
	arg := &arcmodle.ArgAids2{Aids: aids}
	if info, err = d.arcRPC.Views3(c, arg); err != nil {
		err = errors.Wrapf(err, "d.arcRPC.Views3(%+v)", arg)
	}
	return
}

// PGC get PGC info
func (d *Dao) PGC(c context.Context, epid, platform string, build, mid int64) (info map[int64]*model.PGCRes, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	params := url.Values{}
	params.Set("ep_ids", epid)
	params.Set("mid", strconv.FormatInt(mid, 10))
	params.Set("build", strconv.FormatInt(build, 10))
	params.Set("platform", platform)
	var res struct {
		Code int             `json:"code"`
		Data []*model.PGCRes `json:"result"`
	}
	if err = d.client.Get(c, d.pgcAPI, ip, params, &res); err != nil {
		return
	}
	if res.Code != ecode.OK.Code() {
		err = errors.Wrap(ecode.Int(res.Code), d.pgcAPI+"?"+params.Encode())
		return
	}
	info = make(map[int64]*model.PGCRes, len(res.Data))
	for _, v := range res.Data {
		v.Title = v.Season.Title
		info[v.EpID] = v
	}
	return
}

// Article get articl info
func (d *Dao) Article(c context.Context, articleIDs []int64) (info map[int64]*artmodle.Meta, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	arg := &artmodle.ArgAids{Aids: articleIDs, RealIP: ip}
	if info, err = d.artRPC.ArticleMetas(c, arg); err != nil {
		err = errors.Wrapf(err, "d.artRPC.ArticleMetas(%+v)", arg)
	}
	return
}

// HistoryByTP histroy by tp
func (d *Dao) HistoryByTP(c context.Context, mid int64, pn, ps int, tp int8) (res []*hismodle.Resource, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	arg := &hismodle.ArgHistories{Mid: mid, Pn: pn, Ps: ps, RealIP: ip, TP: tp}
	if res, err = d.historyRPC.History(c, arg); err != nil {
		err = errors.Wrapf(err, "d.historyRPC.History(%+v)", arg)
	}
	return
}

// Cursor 5.28游标由MaxOid+MaxTP唯一确定 改为 由ViewAt唯一确定（防止客户端改动对客户端仍用max字段）
func (d *Dao) Cursor(c context.Context, mid, max int64, ps int, tp int8, businesses []string) (res []*hismodle.Resource, err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	arg := &hismodle.ArgCursor{Mid: mid, Max: max, Ps: ps, RealIP: ip, TP: tp, ViewAt: max, Businesses: businesses}
	if res, err = d.historyRPC.HistoryCursor(c, arg); err != nil {
		err = errors.Wrapf(err, "d.historyRPC.HistoryCursor(%+v)", arg)
	}
	return
}

// Del for history
func (d *Dao) Del(c context.Context, mid int64, hisRes []*hismodle.Resource) (err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	arg := &hismodle.ArgDelete{Mid: mid, RealIP: ip, Resources: hisRes}
	if err = d.historyRPC.Delete(c, arg); err != nil {
		err = errors.Wrapf(err, "d.historyRPC.Delete(%+v)", arg)
	}
	return
}

// Clear for history
func (d *Dao) Clear(c context.Context, mid int64, businesses []string) (err error) {
	ip := metadata.String(c, metadata.RemoteIP)
	arg := &hismodle.ArgClear{Mid: mid, RealIP: ip, Businesses: businesses}
	if err = d.historyRPC.Clear(c, arg); err != nil {
		err = errors.Wrapf(err, "d.historyRPC.Clear(%+v)", arg)
	}
	return
}