143 lines
4.6 KiB
Go
143 lines
4.6 KiB
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
|
||
}
|