Create & Init Project...

app/interface/main/tv/dao/pgc/BUILD

@@ -0,0 +1,58 @@
+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",
+        "media_test.go",
+        "style_cache_test.go",
+    ],
+    embed = [":go_default_library"],
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/tv/conf:go_default_library",
+        "//app/interface/main/tv/model:go_default_library",
+        "//library/ecode:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+        "//vendor/gopkg.in/h2non/gock.v1:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "dao.go",
+        "media.go",
+        "style_cache.go",
+    ],
+    importpath = "go-common/app/interface/main/tv/dao/pgc",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/main/tv/conf:go_default_library",
+        "//app/interface/main/tv/model:go_default_library",
+        "//library/cache/memcache:go_default_library",
+        "//library/log:go_default_library",
+        "//library/net/http/blademaster: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/pgc/dao.go

@@ -0,0 +1,24 @@
+package pgc
+
+import (
+	"go-common/app/interface/main/tv/conf"
+	"go-common/library/cache/memcache"
+	bm "go-common/library/net/http/blademaster"
+)
+
+// Dao is account dao.
+type Dao struct {
+	conf   *conf.Config
+	client *bm.Client
+	mc     *memcache.Pool
+}
+
+// New account dao.
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		conf:   c,
+		client: bm.NewClient(c.HTTPClient),
+		mc:     memcache.NewPool(c.Memcache.Config),
+	}
+	return
+}

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

@@ -0,0 +1,45 @@
+package pgc
+
+import (
+	"context"
+	"flag"
+	"os"
+	"strings"
+
+	"go-common/app/interface/main/tv/conf"
+
+	"gopkg.in/h2non/gock.v1"
+)
+
+var (
+	d   *Dao
+	ctx = context.Background()
+)
+
+func init() {
+	// dir, _ := filepath.Abs("../../cmd/tv-interface.toml")
+	// flag.Set("conf", dir)
+	if os.Getenv("DEPLOY_ENV") != "" {
+		flag.Set("app_id", "main.web-svr.tv-interface")
+		flag.Set("conf_token", "07c1826c1f39df02a1411cdd6f455879")
+		flag.Set("tree_id", "15326")
+		flag.Set("conf_version", "docker-1")
+		flag.Set("deploy_env", "uat")
+		flag.Set("conf_host", "config.bilibili.co")
+		flag.Set("conf_path", "/tmp")
+		flag.Set("region", "sh")
+		flag.Set("zone", "sh001")
+	}
+	flag.Parse()
+	if err := conf.Init(); err != nil {
+		panic(err)
+	}
+	d = New(conf.Conf)
+}
+
+func httpMock(method, url string) *gock.Request {
+	r := gock.New(url)
+	r.Method = strings.ToUpper(method)
+	d.client.SetTransport(gock.DefaultTransport)
+	return r
+}

app/interface/main/tv/dao/pgc/media.go

@@ -0,0 +1,38 @@
+package pgc
+
+import (
+	"context"
+
+	"go-common/app/interface/main/tv/model"
+	"go-common/library/log"
+)
+
+// Media gets the media detail data from PGC API
+func (d *Dao) Media(ctx context.Context, tvParam *model.MediaParam) (detail *model.SeasonDetail, err error) {
+	var result model.MediaResp
+	if err = d.client.Get(ctx, d.conf.Host.APIMedia, "", tvParam.GenerateUrl(), &result); err != nil {
+		log.Error("ClientGet Sid %d, error[%v]", tvParam.SeasonID, err)
+		return
+	}
+	if err = result.CodeErr(); err != nil {
+		log.Error("PGC API MediaResp: [CODE:(%d),MESSAGE:(%s)]", result.Code, result.Message)
+		return
+	}
+	detail = result.Result
+	return
+}
+
+// MediaV2 gets the media detail data from PGC API V2
+func (d *Dao) MediaV2(ctx context.Context, tvParam *model.MediaParam) (detail *model.SnDetailV2, err error) {
+	var result model.MediaRespV2
+	if err = d.client.Get(ctx, d.conf.Host.APIMediaV2, "", tvParam.GenerateUrl(), &result); err != nil {
+		log.Error("ClientGet Sid %d, error[%v]", tvParam.SeasonID, err)
+		return
+	}
+	if err = result.CodeErr(); err != nil {
+		log.Error("PGC API MediaResp: [CODE:(%d),MESSAGE:(%s)]", result.Code, result.Message)
+		return
+	}
+	detail = result.Result
+	return
+}

app/interface/main/tv/dao/pgc/style_cache.go

@@ -0,0 +1,36 @@
+package pgc
+
+import (
+	"context"
+
+	"go-common/app/interface/main/tv/model"
+	"go-common/library/cache/memcache"
+	"go-common/library/log"
+)
+
+const (
+	_style = "style_label"
+)
+
+// GetLabelCache .
+func (d *Dao) GetLabelCache(ctx context.Context) (res map[int64][]*model.ParamStyle, err error) {
+	var (
+		conn = d.mc.Get(ctx)
+		key  = _style
+		rp   *memcache.Item
+	)
+	res = make(map[int64][]*model.ParamStyle)
+	defer conn.Close()
+	if rp, err = conn.Get(key); err != nil {
+		if err == memcache.ErrNotFound {
+			err = nil
+		} else {
+			log.Error("mc.Get(%s) error(%v)", key, err)
+		}
+		return
+	}
+	if err = conn.Scan(rp, &res); err != nil {
+		log.Error("conn.Scan error(%v)", err)
+	}
+	return
+}

app/interface/main/tv/dao/pgc/style_cache_test.go

@@ -0,0 +1,19 @@
+package pgc
+
+import (
+	"testing"
+
+	"github.com/smartystreets/goconvey/convey"
+)
+
+func TestDaoGetLabelCache(t *testing.T) {
+	convey.Convey("GetLabelCache", t, func(cx convey.C) {
+		cx.Convey("When everything goes positive", func(cx convey.C) {
+			res, err := d.GetLabelCache(ctx)
+			cx.Convey("Then err should be nil.res should not be nil.", func(cx convey.C) {
+				cx.So(err, convey.ShouldBeNil)
+				cx.So(res, convey.ShouldNotBeNil)
+			})
+		})
+	})
+}