Create & Init Project...

app/interface/main/app-show/dao/live/BUILD

@@ -0,0 +1,51 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["live_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/app-show/conf: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 = ["live.go"],
+    importpath = "go-common/app/interface/main/app-show/dao/live",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/main/app-card/model/card/live:go_default_library",
+        "//app/interface/main/app-show/conf:go_default_library",
+        "//app/interface/main/app-show/model/live:go_default_library",
+        "//library/ecode:go_default_library",
+        "//library/log:go_default_library",
+        "//library/net/http/blademaster: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-show/dao/live/live.go

@@ -0,0 +1,142 @@
+package live
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/url"
+	"strconv"
+	"time"
+
+	clive "go-common/app/interface/main/app-card/model/card/live"
+	"go-common/app/interface/main/app-show/conf"
+	"go-common/app/interface/main/app-show/model/live"
+	"go-common/library/ecode"
+	"go-common/library/log"
+	httpx "go-common/library/net/http/blademaster"
+
+	"github.com/pkg/errors"
+)
+
+const (
+	_live       = "/appIndex/recommendFeedList"
+	_rec        = "/appIndex/recommendList"
+	_topic      = "/topic/v0/Topic/hots"
+	_dynamichot = "/dynamic_detail/v0/Dynamic/hot"
+)
+
+// Dao is live dao
+type Dao struct {
+	client     *httpx.Client
+	clientAsyn *httpx.Client
+	live       string
+	rec        string
+	topic      string
+	dynamichot string
+}
+
+// New live dao.
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		client:     httpx.NewClient(c.HTTPClient),
+		clientAsyn: httpx.NewClient(c.HTTPClientAsyn),
+		live:       c.Host.ApiLiveCo + _live,
+		rec:        c.Host.ApiLiveCo + _rec,
+		topic:      c.Host.Dynamic + _topic,
+		dynamichot: c.Host.Dynamic + _dynamichot,
+	}
+	return
+}
+
+// Live feed
+func (d *Dao) Feed(c context.Context, mid int64, ak, ip string, now time.Time) (r *live.Feed, err error) {
+	params := url.Values{}
+	params.Set("mid", strconv.FormatInt(mid, 10))
+	params.Set("access_key", ak)
+	var res struct {
+		Code int        `json:"code"`
+		Data *live.Feed `json:"data"`
+	}
+	if err = d.client.Get(c, d.live, ip, params, &res); err != nil {
+		log.Error("Feed url(%s) error(%v)", d.live+"?"+params.Encode(), err)
+		return
+	}
+	if res.Code != 0 {
+		log.Error("Feed url(%s) error(%v)", d.live+"?"+params.Encode(), res.Code)
+		err = fmt.Errorf("feed send failed")
+		return
+	}
+	r = res.Data
+	return
+}
+
+// Recommend get live Recommend data.
+func (d *Dao) Recommend(now time.Time) (r *live.Recommend, err error) {
+	params := url.Values{}
+	params.Set("count", "60")
+	var res struct {
+		Code int             `json:"code"`
+		Data *live.Recommend `json:"data"`
+	}
+	if err = d.clientAsyn.Get(context.TODO(), d.rec, "", params, &res); err != nil { // TODO context arg, service context.TODO
+		log.Error("live recommend url(%s) error(%v)", d.rec+"?"+params.Encode(), err)
+		return
+	}
+	if res.Code != 0 {
+		log.Error("live recommend url(%s) error(%v)", d.rec+"?"+params.Encode(), res.Code)
+		err = fmt.Errorf("recommend send failed")
+		return
+	}
+	r = res.Data
+	return
+}
+
+// TopicHots get live topic hots
+func (d *Dao) TopicHots(c context.Context) (topics []*clive.TopicHot, err error) {
+	var res struct {
+		Code int `json:"code"`
+		Data struct {
+			List []*clive.TopicHot `json:"list"`
+		} `json:"data"`
+	}
+	if err = d.clientAsyn.Get(c, d.topic, "", nil, &res); err != nil {
+		return
+	}
+	b, _ := json.Marshal(&res)
+	log.Info("topichots list url(%v) response(%s)", d.topic, b)
+	if res.Code != ecode.OK.Code() {
+		err = errors.Wrap(err, d.topic)
+		return
+	}
+	for _, t := range res.Data.List {
+		tmp := &clive.TopicHot{}
+		*tmp = *t
+		if err = tmp.TopicJSONChange(); err != nil {
+			log.Error("TopicJSONChange error(%v)", err)
+			return
+		}
+		topics = append(topics, tmp)
+	}
+	return
+}
+
+// DynamicHot get dynamic hot all
+func (d *Dao) DynamicHot(c context.Context) (list []*clive.DynamicHot, err error) {
+	var res struct {
+		Code int `json:"code"`
+		Data struct {
+			List []*clive.DynamicHot `json:"list"`
+		} `json:"data"`
+	}
+	if err = d.clientAsyn.Get(c, d.dynamichot, "", nil, &res); err != nil {
+		return
+	}
+	b, _ := json.Marshal(&res)
+	log.Info("dynamichot list url(%v) response(%s)", d.dynamichot, b)
+	if res.Code != ecode.OK.Code() {
+		err = errors.Wrap(err, d.topic)
+		return
+	}
+	list = res.Data.List
+	return
+}

app/interface/main/app-show/dao/live/live_test.go

@@ -0,0 +1,175 @@
+package live
+
+import (
+	"context"
+	"flag"
+	"os"
+	"strings"
+	"testing"
+	"time"
+
+	"go-common/app/interface/main/app-show/conf"
+
+	. "github.com/smartystreets/goconvey/convey"
+	gock "gopkg.in/h2non/gock.v1"
+)
+
+var (
+	d *Dao
+)
+
+func ctx() context.Context {
+	return context.Background()
+}
+
+func init() {
+	if os.Getenv("DEPLOY_ENV") != "" {
+		flag.Set("app_id", "main.app-svr.app-show")
+		flag.Set("conf_token", "Pae4IDOeht4cHXCdOkay7sKeQwHxKOLA")
+		flag.Set("tree_id", "2687")
+		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)
+	time.Sleep(time.Second)
+}
+
+func httpMock(method, url string) *gock.Request {
+	r := gock.New(url)
+	r.Method = strings.ToUpper(method)
+	return r
+}
+
+func TestFeed(t *testing.T) {
+	Convey("Feed", t, func() {
+		d.client.SetTransport(gock.DefaultTransport)
+		httpMock("GET", d.live).Reply(200).JSON(`{"code":0,"count":1,"lives":[{"owner":{"face":"xxx","mid":1,"name":"xxxx"}}]}`)
+		res, err := d.Feed(ctx(), 1, "", "", time.Now())
+		So(res, ShouldNotBeEmpty)
+		So(err, ShouldBeNil)
+	})
+}
+
+func TestRecommend(t *testing.T) {
+	Convey("Recommend", t, func() {
+		d.clientAsyn.SetTransport(gock.DefaultTransport)
+		httpMock("GET", d.rec).Reply(200).JSON(`{
+			"code": 0,
+			"data": {
+				"count": 1,
+				"lives": {
+					"subject": [{
+						"owner": {
+							"face": "xxx",
+							"mid": 1,
+							"name": "xxxx"
+						}
+					}],
+					"hot": [{
+						"owner": {
+							"face": "xxx",
+							"mid": 1,
+							"name": "xxxx"
+						}
+					}]
+				}
+			}
+		}`)
+		res, err := d.Recommend(time.Now())
+		So(res, ShouldNotBeEmpty)
+		So(err, ShouldBeNil)
+	})
+}
+
+func TestTopicHots(t *testing.T) {
+	Convey("TopicHots", t, func() {
+		d.clientAsyn.SetTransport(gock.DefaultTransport)
+		httpMock("GET", d.topic).Reply(200).JSON(`{
+			"code": 0,
+			"data": {
+				"list": [{
+					"topic_id": 7279615,
+					"topic_name": "CP23",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/b46cda4c7e953764c0fcda49c0f06639e7092792.jpg\",\"image_width\":800,\"image_height\":500}"
+				}, {
+					"topic_id": 9029281,
+					"topic_name": "2018COS总结",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/6e703453de103b5b5b9e00640ff9e9d9156950e6.jpg\",\"image_width\":800,\"image_height\":500}"
+				}, {
+					"topic_id": 2838293,
+					"topic_name": "陪你过冬天",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/716e12155e1f2150418c9f215de6bb3c9b38f516.png\",\"image_width\":800,\"image_height\":500}"
+				}, {
+					"topic_id": 2525230,
+					"topic_name": "故事王StoryMan",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/5adb6388723c321d99c5b04fe9839cdaacf6da0d.jpg\",\"image_width\":800,\"image_height\":500}"
+				}, {
+					"topic_id": 8977836,
+					"topic_name": "萌宠暖宝宝",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/adba2d41863e5bf1d269cdf2d4803fac099f8288.jpg\",\"image_width\":800,\"image_height\":500}"
+				}, {
+					"topic_id": 105286,
+					"topic_name": "国家宝藏",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/9650de6e22d0742b63db0feac5bd891faf050b54.png\",\"image_width\":800,\"image_height\":500}"
+				}, {
+					"topic_id": 8948501,
+					"topic_name": "2018绘画总结",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/94c6edd82202ffdfa62c91d0d9d62dcb5b40a0b8.png\",\"image_width\":800,\"image_height\":500}"
+				}, {
+					"topic_id": 8977910,
+					"topic_name": "冬天喝奶茶",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/c0f2533e71732eb700a6ec5a37b633fd5ae707f3.png\",\"image_width\":800,\"image_height\":500}"
+				}, {
+					"topic_id": 2872407,
+					"topic_name": "冬日必备",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/54e1efc022377e209ef98a680f1fb7777e242ca4.png\",\"image_width\":800,\"image_height\":500}"
+				}, {
+					"topic_id": 2953953,
+					"topic_name": "冬日穿搭",
+					"picture": "{\"image_src\":\"https:\\/\\/i0.hdslb.com\\/bfs\\/album\\/6522e801f6882c171622a47eb3087d50af42fe7b.jpg\",\"image_width\":800,\"image_height\":500}"
+				}]
+			}
+		}`)
+		res, err := d.TopicHots(ctx())
+		So(res, ShouldNotBeEmpty)
+		So(err, ShouldBeNil)
+	})
+}
+
+func TestDynamicHot(t *testing.T) {
+	Convey("DynamicHot", t, func() {
+		d.clientAsyn.SetTransport(gock.DefaultTransport)
+		httpMock("GET", d.dynamichot).Reply(200).JSON(`{
+			"code": 0,
+			"data": {
+				"list": [{
+					"dynamic_id": 198340924012294846,
+					"audit_status": 0,
+					"delete_status": 0,
+					"mid": 440290,
+					"nick_name": "音乐鱼",
+					"face_img": "http://i2.hdslb.com/bfs/face/d5e81352871fbe33e7c1aed68ad237c1e0588db3.jpg",
+					"rid_type": 2,
+					"rid": 10362627,
+					"view_count": 5641,
+					"comment_count": 10,
+					"rcmd_reason": "",
+					"dynamic_text": "#CP23##COSPLAY#这两天腿和腰都要断了，光顾着逛没怎么拍 ，但还是满足了😂昨晚上还吃了老上海味道，现在急需一顿火锅！",
+					"img_count": 9,
+					"imgs": ["https://i0.hdslb.com/bfs/album/615e9efe438a78f25b4ee399fec1e168eaac8c6e.jpg", "https://i0.hdslb.com/bfs/album/1a31a0e93c383183b7b7b02e777248689cac1368.jpg", "https://i0.hdslb.com/bfs/album/270f4b3f84337fa8d2755cff46f7aef6d6f633de.jpg", "https://i0.hdslb.com/bfs/album/25e2fc36946d3f40e242c7619898e2ad5564fd1c.jpg", "https://i0.hdslb.com/bfs/album/2897ee217c455502cde345666ea66086d9f16fc3.jpg", "https://i0.hdslb.com/bfs/album/7c5948d362e17a5118c9cdf095d04c193e50de37.jpg", "https://i0.hdslb.com/bfs/album/473d201e4756fca5b38bc0b535a1086db44d2173.jpg", "https://i0.hdslb.com/bfs/album/91fc858bba009c338d4e1a13ba1953378bd19b15.jpg", "https://i0.hdslb.com/bfs/album/d34307d2e9006847b7e974351ed89c0fe3ad282a.jpg"]
+				}]
+			}
+		}`)
+		res, err := d.DynamicHot(ctx())
+		So(res, ShouldNotBeEmpty)
+		So(err, ShouldBeNil)
+	})
+}