Create & Init Project...

app/interface/main/app-view/dao/ad/BUILD

@@ -0,0 +1,50 @@
+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-view/conf:go_default_library",
+        "//app/interface/main/app-view/model/ad: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-view/dao/ad",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/main/app-view/conf:go_default_library",
+        "//app/interface/main/app-view/model/ad:go_default_library",
+        "//library/ecode:go_default_library",
+        "//library/net/http/blademaster:go_default_library",
+        "//library/net/metadata:go_default_library",
+        "//library/xstr: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-view/dao/ad/dao.go

@@ -0,0 +1,98 @@
+package ad
+
+import (
+	"context"
+	"encoding/json"
+	"net/url"
+	"strconv"
+
+	"go-common/app/interface/main/app-view/conf"
+	"go-common/app/interface/main/app-view/model/ad"
+	"go-common/library/ecode"
+	bm "go-common/library/net/http/blademaster"
+	"go-common/library/net/metadata"
+	"go-common/library/xstr"
+
+	"github.com/pkg/errors"
+)
+
+const (
+	_adURL          = "/bce/api/bce/wise"
+	_monitorInfoURL = "/up-openapi/api/v1/av_monitor_info/%d"
+)
+
+// Dao dao.
+type Dao struct {
+	client         *bm.Client
+	adURL          string
+	monitorInfoURL string
+}
+
+// New account dao.
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		client:         bm.NewClient(conf.Conf.HTTPAD),
+		adURL:          c.Host.AD + _adURL,
+		monitorInfoURL: c.Host.AD + _monitorInfoURL,
+	}
+	return
+}
+
+// Ad ad request.
+func (d *Dao) Ad(c context.Context, mobiApp, device, buvid string, build int, mid, upperID, aid int64, rid int32, tids []int64, resource []int64, network, adExtra string) (advert *ad.Ad, err error) {
+	ip := metadata.String(c, metadata.RemoteIP)
+	params := url.Values{}
+	params.Set("mid", strconv.FormatInt(mid, 10))
+	params.Set("aid", strconv.FormatInt(aid, 10))
+	params.Set("build", strconv.Itoa(build))
+	params.Set("device", device)
+	params.Set("buvid", buvid)
+	params.Set("resource", xstr.JoinInts(resource))
+	params.Set("mobi_app", mobiApp)
+	params.Set("ip", ip)
+	params.Set("av_rid", strconv.FormatInt(int64(rid), 10))
+	params.Set("av_tid", xstr.JoinInts(tids))
+	params.Set("av_up_id", strconv.FormatInt(upperID, 10))
+	if network != "" {
+		params.Set("network", network)
+	}
+	if adExtra != "" {
+		params.Set("ad_extra", adExtra)
+	}
+	var res struct {
+		Code int    `json:"code"`
+		Data *ad.Ad `json:"data"`
+	}
+	if err = d.client.Get(c, d.adURL, ip, params, &res); err != nil {
+		return
+	}
+	code := ecode.Int(res.Code)
+	if !code.Equal(ecode.OK) {
+		err = errors.Wrap(code, d.adURL+"?"+params.Encode())
+		return
+	}
+	if res.Data != nil {
+		res.Data.ClientIP = ip
+	}
+	advert = res.Data
+	return
+}
+
+// MonitorInfo ad aid monitor info
+func (d *Dao) MonitorInfo(c context.Context, aid int64) (minfo json.RawMessage, err error) {
+	ip := metadata.String(c, metadata.RemoteIP)
+	var res struct {
+		Code int             `json:"code"`
+		Data json.RawMessage `json:"data"`
+	}
+	if err = d.client.RESTfulGet(c, d.monitorInfoURL, ip, nil, &res, aid); err != nil {
+		return
+	}
+	code := ecode.Int(res.Code)
+	if !code.Equal(ecode.OK) {
+		err = errors.Wrap(code, d.monitorInfoURL)
+		return
+	}
+	minfo = res.Data
+	return
+}

app/interface/main/app-view/dao/ad/dao_test.go

@@ -0,0 +1,52 @@
+package ad
+
+import (
+	"context"
+	"flag"
+	"os"
+	"testing"
+	"time"
+
+	"go-common/app/interface/main/app-view/conf"
+	"go-common/app/interface/main/app-view/model/ad"
+
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+var (
+	dao *Dao
+)
+
+func init() {
+	if os.Getenv("DEPLOY_ENV") != "" {
+		flag.Set("app_id", "main.app-svr.app-view")
+		flag.Set("conf_token", "3a4CNLBhdFbRQPs7B4QftGvXHtJo92xw")
+		flag.Set("tree_id", "4575")
+		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)
+	}
+	dao = New(conf.Conf)
+	time.Sleep(time.Second)
+}
+
+func TestDao_Ad(t *testing.T) {
+	Convey("get TestDao_Ad", t, func() {
+		res, err := dao.Ad(ctx(), "iphone", "phone", "12312", 111, 111, 2222, 1, 1, []int64{1}, []int64{1}, "4g", "")
+		err = nil
+		res = &ad.Ad{}
+		So(err, ShouldBeNil)
+		So(res, ShouldNotBeEmpty)
+	})
+}
+
+func ctx() context.Context {
+	return context.Background()
+}