Create & Init Project...

app/interface/main/tv/dao/favorite/BUILD

@@ -0,0 +1,52 @@
+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/tv/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/tv/dao/favorite",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/main/tv/conf:go_default_library",
+        "//app/interface/main/tv/model:go_default_library",
+        "//app/service/main/favorite/api/gorpc:go_default_library",
+        "//app/service/main/favorite/model:go_default_library",
+        "//library/ecode:go_default_library",
+        "//library/log: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/tv/dao/favorite/dao.go

@@ -0,0 +1,103 @@
+package favorite
+
+import (
+	"context"
+	"fmt"
+	"net/url"
+
+	"go-common/app/interface/main/tv/conf"
+	"go-common/app/interface/main/tv/model"
+	favrpc "go-common/app/service/main/favorite/api/gorpc"
+	favmdl "go-common/app/service/main/favorite/model"
+	"go-common/library/ecode"
+	"go-common/library/log"
+	bm "go-common/library/net/http/blademaster"
+	"go-common/library/net/metadata"
+
+	"github.com/pkg/errors"
+)
+
+// Dao is account dao.
+type Dao struct {
+	favRPC *favrpc.Service // rpc
+	conf   *conf.Config
+	client *bm.Client
+}
+
+// New account dao.
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		favRPC: favrpc.New2(c.FavoriteRPC),
+		conf:   c,
+		client: bm.NewClient(c.HTTPClient),
+	}
+	return
+}
+
+const (
+	_FavBusiness = 2
+	_DefaultFav  = 0
+)
+
+// FavoriteV3 picks favorite info from rpc
+func (d *Dao) FavoriteV3(ctx context.Context, mid int64, pn int) (res *favmdl.Favorites, err error) {
+	var ip = metadata.String(ctx, metadata.RemoteIP)
+	arg := &favmdl.ArgFavs{
+		Type:   _FavBusiness,
+		Mid:    mid,
+		Fid:    _DefaultFav,
+		Tv:     1,
+		Pn:     pn,
+		Ps:     d.conf.Cfg.FavPs,
+		RealIP: ip,
+	}
+	if res, err = d.favRPC.Favorites(ctx, arg); err != nil {
+		err = errors.Wrapf(err, "%v", arg)
+	}
+	return
+}
+
+// favAct adds/deletes favorite into/from the default folder
+func (d *Dao) favAct(ctx context.Context, mid int64, aid int64, host string) (err error) {
+	var (
+		ip     = metadata.String(ctx, metadata.RemoteIP)
+		params = url.Values{}
+		res    = model.RespFavAct{}
+	)
+	params.Set("mid", fmt.Sprintf("%d", mid))
+	params.Set("aid", fmt.Sprintf("%d", aid))
+	if err = d.client.Post(ctx, host, ip, params, &res); err != nil {
+		log.Error("FavAdd Aid %d, Mid %d, Err %v", aid, mid, err)
+		return
+	}
+	if res.Code != ecode.OK.Code() {
+		err = errors.Wrap(ecode.Int(res.Code), fmt.Sprintf("Fav AID %d, Mid %d, API Error %s", aid, mid, res.Message))
+		log.Error("FavAdd ERROR:%v, URL: %s", err, host+"?"+params.Encode())
+		return
+	}
+	return
+}
+
+// FavAdd def.
+func (d *Dao) FavAdd(ctx context.Context, mid, aid int64) (err error) {
+	host := d.conf.Host.FavAdd
+	return d.favAct(ctx, mid, aid, host)
+}
+
+// FavDel deletes favorite from the default folder
+func (d *Dao) FavDel(ctx context.Context, mid int64, aid int64) (err error) {
+	host := d.conf.Host.FavDel
+	return d.favAct(ctx, mid, aid, host)
+}
+
+// InDefault returns whether the aid is in Default of Mid
+func (d *Dao) InDefault(ctx context.Context, mid int64, aid int64) (bool, error) {
+	var ip = metadata.String(ctx, metadata.RemoteIP)
+	arg := &favmdl.ArgInDefaultFolder{
+		Type:   _FavBusiness,
+		Mid:    mid,
+		RealIP: ip,
+		Oid:    aid,
+	}
+	return d.favRPC.InDefault(ctx, arg)
+}

app/interface/main/tv/dao/favorite/dao_test.go

@@ -0,0 +1,76 @@
+package favorite
+
+import (
+	"context"
+	"encoding/json"
+	"flag"
+	"path/filepath"
+	"testing"
+	"time"
+
+	"go-common/app/interface/main/tv/conf"
+
+	"fmt"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var d *Dao
+
+func init() {
+	dir, _ := filepath.Abs("../../cmd/tv-interface.toml")
+	flag.Set("conf", dir)
+	conf.Init()
+	d = New(conf.Conf)
+	time.Sleep(5 * time.Second)
+}
+
+func WithDao(f func(d *Dao)) func() {
+	return func() {
+		Reset(func() {})
+		f(d)
+	}
+}
+
+func TestDao_FavoriteV3(t *testing.T) {
+	Convey("TestDao_FavoriteV3", t, func() {
+		res, err := d.FavoriteV3(context.Background(), 88894921, 1)
+		So(err, ShouldBeNil)
+		data, _ := json.Marshal(res)
+		Println(string(data))
+	})
+}
+
+func TestDao_FavAdd(t *testing.T) {
+	Convey("TestDao_FavAdd", t, func() {
+		err := d.FavAdd(context.Background(), 88894921, 10098813)
+		So(err, ShouldBeNil)
+		err = d.FavAdd(context.Background(), 88894921, 28417042)
+		So(err, ShouldBeNil)
+	})
+}
+
+func TestDao_FavDel(t *testing.T) {
+	Convey("TestDao_FavDel", t, func() {
+		err := d.FavDel(context.Background(), 88894921, 28417042)
+		So(err, ShouldBeNil)
+	})
+}
+
+func TestDao_InDefault(t *testing.T) {
+	Convey("TestDao_InDefault", t, WithDao(func(d *Dao) {
+		var mid = int64(27515418)
+		res, err := d.FavoriteV3(context.Background(), mid, 1)
+		So(err, ShouldBeNil)
+		if res == nil || len(res.List) == 0 {
+			fmt.Println("empty Fav")
+			return
+		}
+		exist, err2 := d.InDefault(context.Background(), mid, res.List[0].Oid)
+		So(err2, ShouldBeNil)
+		So(exist, ShouldBeTrue)
+		exist, err2 = d.InDefault(context.Background(), mid, 888888888888)
+		So(err2, ShouldBeNil)
+		So(exist, ShouldBeFalse)
+	}))
+}