Create & Init Project...

app/interface/main/app-interface/dao/history/BUILD

@@ -0,0 +1,55 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["dao_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/app-interface/conf:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["dao.go"],
+    importpath = "go-common/app/interface/main/app-interface/dao/history",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/main/app-interface/conf:go_default_library",
+        "//app/interface/main/app-interface/model/history:go_default_library",
+        "//app/interface/main/history/model:go_default_library",
+        "//app/interface/main/history/rpc/client:go_default_library",
+        "//app/interface/openplatform/article/model:go_default_library",
+        "//app/interface/openplatform/article/rpc/client:go_default_library",
+        "//app/service/main/archive/api/gorpc:go_default_library",
+        "//app/service/main/archive/model/archive:go_default_library",
+        "//library/ecode:go_default_library",
+        "//library/net/http/blademaster:go_default_library",
+        "//library/net/metadata:go_default_library",
+        "//vendor/github.com/pkg/errors:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [":package-srcs"],
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+)

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

@@ -0,0 +1,142 @@
+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
+}

app/interface/main/app-interface/dao/history/dao_test.go

@@ -0,0 +1,56 @@
+package history
+
+import (
+	"context"
+	"go-common/app/interface/main/app-interface/conf"
+	"testing"
+
+	"flag"
+	"path/filepath"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var d *Dao
+
+func init() {
+	dir, _ := filepath.Abs("../cmd/app-interface-test.toml")
+	flag.Set("conf", dir)
+	conf.Init()
+	d = New(conf.Conf)
+}
+
+func WithDao(f func(d *Dao)) func() {
+	return func() {
+		Reset(func() {})
+		f(d)
+	}
+}
+
+func TestDao_ArchiveInfo(t *testing.T) {
+	Convey("ArchiveInfo", t, WithDao(func(d *Dao) {
+		_, err := d.Archive(context.TODO(), []int64{1, 2})
+		So(err, ShouldBeNil)
+	}))
+}
+
+func TestDao_ArticleInfo(t *testing.T) {
+	Convey("ArticleInfo", t, WithDao(func(d *Dao) {
+		_, err := d.Article(context.TODO(), []int64{1, 2})
+		So(err, ShouldBeNil)
+	}))
+}
+
+func TestDao_PGCInfo(t *testing.T) {
+	Convey("PGCInfo", t, WithDao(func(d *Dao) {
+		_, err := d.PGC(context.TODO(), "1", "1", 111, 27515256)
+		So(err, ShouldBeNil)
+	}))
+}
+
+func TestDao_GetList(t *testing.T) {
+	Convey("GetList", t, WithDao(func(d *Dao) {
+		_, err := d.History(context.TODO(), 27515256, 1, 20)
+		So(err, ShouldBeNil)
+	}))
+}