Create & Init Project...

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

@@ -0,0 +1,47 @@
+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-wall/conf:go_default_library",
+        "//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["live.go"],
+    importpath = "go-common/app/interface/main/app-wall/dao/live",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/interface/main/app-wall/conf:go_default_library",
+        "//library/ecode: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/app-wall/dao/live/live.go

@@ -0,0 +1,94 @@
+package live
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"net/http"
+	"net/url"
+	"strconv"
+
+	"go-common/app/interface/main/app-wall/conf"
+	"go-common/library/ecode"
+	"go-common/library/log"
+	httpx "go-common/library/net/http/blademaster"
+)
+
+const (
+	_liveURL   = "/gift/simGift"
+	_addVipURL = "/user/v0/Vip/addVip"
+)
+
+// Dao is live dao
+type Dao struct {
+	client    *httpx.Client
+	liveURL   string
+	addVipURL string
+}
+
+// New live dao.
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		client:    httpx.NewClient(c.HTTPClient),
+		liveURL:   c.Host.Live + _liveURL,
+		addVipURL: c.Host.APILive + _addVipURL,
+	}
+	return
+}
+
+// Pack
+func (d *Dao) Pack(c context.Context, mid int64, cardType int) (err error) {
+	var res struct {
+		Code int `json:"code"`
+	}
+	upack := map[string]map[string]interface{}{
+		"header": map[string]interface{}{},
+		"body": map[string]interface{}{
+			"uid":       mid,
+			"card_type": cardType,
+		},
+	}
+	bytesData, err := json.Marshal(upack)
+	if err != nil {
+		log.Error("json.Marshal error(%v)", err)
+		return
+	}
+	req, err := http.NewRequest("POST", d.liveURL, bytes.NewReader(bytesData))
+	if err != nil {
+		log.Error("http.NewRequest error(%v)", err)
+		return
+	}
+	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
+	req.Header.Set("X-BACKEND-BILI-REAL-IP", "")
+	log.Info("unicom pack mid(%d) card_type(%d)", mid, cardType)
+	if err = d.client.Do(c, req, &res); err != nil || res.Code != 0 {
+		err = ecode.Int(res.Code)
+		log.Error("unicom pack d.client.Do(%s) mid(%d) card_type(%d) error(%v)", d.liveURL, mid, cardType, err)
+		return
+	}
+	return
+}
+
+// AddVip add live vip
+func (d *Dao) AddVip(c context.Context, mid int64, day int) (msg string, err error) {
+	params := url.Values{}
+	params.Set("vip_type", "1")
+	params.Set("day", strconv.Itoa(day))
+	params.Set("uid", strconv.FormatInt(mid, 10))
+	params.Set("platform", "main")
+	var res struct {
+		Code int    `json:"code"`
+		Msg  string `json:"message"`
+	}
+	if err = d.client.Post(c, d.addVipURL, "", params, &res); err != nil {
+		log.Error("live add vip url(%v) error(%v)", d.addVipURL+"?"+params.Encode(), err)
+		return
+	}
+	msg = res.Msg
+	if res.Code != 0 {
+		err = ecode.Int(res.Code)
+		log.Error("live add vip url(%v) res code(%d)", d.addVipURL+"?"+params.Encode(), res.Code)
+		return
+	}
+	return
+}

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

@@ -0,0 +1,43 @@
+package live
+
+import (
+	"context"
+	"flag"
+	"path/filepath"
+	"testing"
+	"time"
+
+	"go-common/app/interface/main/app-wall/conf"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var (
+	d *Dao
+)
+
+func ctx() context.Context {
+	return context.Background()
+}
+
+func init() {
+	dir, _ := filepath.Abs("../../cmd/app-wall-test.toml")
+	flag.Set("conf", dir)
+	conf.Init()
+	d = New(conf.Conf)
+	time.Sleep(time.Second)
+}
+
+func TestPack(t *testing.T) {
+	Convey("Pack", t, func() {
+		err := d.Pack(ctx(), 1, 1)
+		So(err, ShouldBeNil)
+	})
+}
+
+func TestAddVip(t *testing.T) {
+	Convey("AddVip", t, func() {
+		_, err := d.AddVip(ctx(), 1, 1)
+		So(err, ShouldBeNil)
+	})
+}