Create & Init Project...

This commit is contained in:
2019-04-22 18:49:16 +08:00
commit fc4fa37393
25440 changed files with 4054998 additions and 0 deletions

View File

@@ -0,0 +1,52 @@
package(default_visibility = ["//visibility:public"])
load(
"@io_bazel_rules_go//go:def.bzl",
"go_library",
"go_test",
)
go_library(
name = "go_default_library",
srcs = ["dao.go"],
importpath = "go-common/app/interface/main/creative/dao/medal",
tags = ["automanaged"],
visibility = ["//visibility:public"],
deps = [
"//app/interface/main/creative/conf:go_default_library",
"//app/interface/main/creative/model/medal: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"],
)
go_test(
name = "go_default_test",
srcs = ["dao_test.go"],
embed = [":go_default_library"],
rundir = ".",
tags = ["automanaged"],
deps = [
"//app/interface/main/creative/conf:go_default_library",
"//app/interface/main/creative/model/medal:go_default_library",
"//library/ecode:go_default_library",
"//vendor/github.com/bouk/monkey:go_default_library",
"//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
"//vendor/gopkg.in/h2non/gock.v1:go_default_library",
],
)

View File

@@ -0,0 +1,294 @@
package medal
import (
"context"
"net/http"
"net/url"
"strconv"
"strings"
"go-common/app/interface/main/creative/conf"
"go-common/app/interface/main/creative/model/medal"
"go-common/library/ecode"
"go-common/library/log"
httpx "go-common/library/net/http/blademaster"
)
const (
_checkStatus = "/fans_medal/v1/medal/if_can_open"
_openMedal = "/fans_medal/v1/medal/open"
_getMedal = "/fans_medal/v1/medal/get"
_recentFans = "/fans_medal/v1/fans_medal/get_recent_fans_list"
_checkMedal = "/fans_medal/v1/medal/check_open"
_fansRank = "/fans_medal/v1/medal/top_medal_fans_list"
_rename = "/fans_medal/v1/medal/rename"
)
// Dao define
type Dao struct {
c *conf.Config
// http
client *httpx.Client
// uri
checkStatusURI string
openMedalURI string
getMedalURI string
recentFansURI string
checkMedalURI string
fansRankURI string
renameURI string
}
// New init dao
func New(c *conf.Config) (d *Dao) {
d = &Dao{
c: c,
client: httpx.NewClient(c.HTTPClient.Slow),
checkStatusURI: c.Host.Live + _checkStatus,
openMedalURI: c.Host.Live + _openMedal,
getMedalURI: c.Host.Live + _getMedal,
recentFansURI: c.Host.Live + _recentFans,
checkMedalURI: c.Host.Live + _checkMedal,
fansRankURI: c.Host.Live + _fansRank,
renameURI: c.Host.Live + _rename,
}
return
}
// Status check if up can open medal
func (d *Dao) Status(c context.Context, mid int64) (st *medal.Status, err error) {
var (
req *http.Request
params = url.Values{}
)
params.Set("uid", strconv.FormatInt(mid, 10))
params.Set("source", "2")
if req, err = http.NewRequest("GET", d.checkStatusURI+"?"+params.Encode(), nil); err != nil {
log.Error("Status url(%s) error(%v)", d.checkStatusURI+"?"+params.Encode(), err)
err = ecode.CreativeFansMedalErr
return
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
var res struct {
Code int `json:"code"`
Data *medal.Status `json:"data"`
}
if err = d.client.Do(c, req, &res); err != nil {
log.Error("d.Status url(%s) res(%+v) err(%v)", d.checkStatusURI+"?"+params.Encode(), res, err)
err = ecode.CreativeFansMedalErr
return
}
if res.Code != 0 {
log.Error("Status url(%s) res(%v)", d.checkStatusURI+"?"+params.Encode(), res)
err = ecode.Int(res.Code)
return
}
st = res.Data
return
}
// OpenMedal open medal for up
func (d *Dao) OpenMedal(c context.Context, mid int64, name string) (err error) {
var (
req *http.Request
params = url.Values{}
)
params.Set("uid", strconv.FormatInt(mid, 10))
params.Set("medal_name", name)
params.Set("source", "2")
if req, err = http.NewRequest("POST", d.openMedalURI, strings.NewReader(params.Encode())); err != nil {
log.Error("OpenMedal url(%s) error(%v)", d.openMedalURI+"?"+params.Encode(), err)
err = ecode.CreativeFansMedalErr
return
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
var res struct {
Code int `json:"code"`
Msg string `json:"msg"`
}
if err = d.client.Do(c, req, &res); err != nil {
log.Error("d.OpenMedal url(%s) res(%+v) err(%v)", d.openMedalURI+"?"+params.Encode(), res, err)
err = ecode.CreativeFansMedalErr
return
}
if res.Code != 0 {
log.Error("OpenMedal url(%s) res(%v)", d.openMedalURI+"?"+params.Encode(), res)
err = ecode.Int(res.Code)
return
}
return
}
// CheckMedal check medal name for up
func (d *Dao) CheckMedal(c context.Context, mid int64, name string) (valid int, err error) {
var (
req *http.Request
params = url.Values{}
)
params.Set("uid", strconv.FormatInt(mid, 10))
params.Set("medal_name", name)
params.Set("source", "2")
if req, err = http.NewRequest("POST", d.checkMedalURI, strings.NewReader(params.Encode())); err != nil {
log.Error("checkMedalURI url(%s) error(%v)", d.checkMedalURI+"?"+params.Encode(), err)
err = ecode.CreativeFansMedalErr
return
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
var res struct {
Code int `json:"code"`
Data struct {
Enable bool `json:"enable"`
} `json:"data"`
}
if err = d.client.Do(c, req, &res); err != nil {
log.Error("d.checkMedalURI url(%s) res(%+v) err(%v)", d.checkMedalURI+"?"+params.Encode(), res, err)
err = ecode.CreativeFansMedalErr
return
}
if res.Code != 0 {
log.Error("checkMedalURI url(%s) res(%v)", d.checkMedalURI+"?"+params.Encode(), res)
err = ecode.Int(res.Code)
return
}
if !res.Data.Enable {
valid = 1
}
return
}
//Medal get medal
func (d *Dao) Medal(c context.Context, mid int64) (mdl *medal.Medal, err error) {
var (
req *http.Request
params = url.Values{}
)
params.Set("uid", strconv.FormatInt(mid, 10))
params.Set("source", "2")
if req, err = http.NewRequest("GET", d.getMedalURI+"?"+params.Encode(), nil); err != nil {
log.Error("Medal url(%s) error(%v)", d.getMedalURI+"?"+params.Encode(), err)
err = ecode.CreativeFansMedalErr
return
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
var res struct {
Code int `json:"code"`
Data *medal.Medal `json:"data"`
}
if err = d.client.Do(c, req, &res); err != nil {
log.Error("Medal url(%s) response(%+v) error(%v)", d.getMedalURI+"?"+params.Encode(), res, err)
err = ecode.CreativeFansMedalErr
return
}
if res.Code != 0 {
log.Error("Medal url(%s) res(%v)", d.getMedalURI+"?"+params.Encode(), res)
err = ecode.Int(res.Code)
return
}
mdl = res.Data
return
}
//RecentFans get a list of recent fans
func (d *Dao) RecentFans(c context.Context, mid int64) (fans []*medal.RecentFans, err error) {
var (
req *http.Request
params = url.Values{}
)
params.Set("uid", strconv.FormatInt(mid, 10))
params.Set("source", "2")
params.Set("size", "10")
if req, err = http.NewRequest("GET", d.recentFansURI+"?"+params.Encode(), nil); err != nil {
log.Error("RecentFans url(%s) error(%v)", d.recentFansURI+"?"+params.Encode(), err)
err = ecode.CreativeFansMedalErr
return
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
var res struct {
Code int `json:"code"`
Data []*medal.RecentFans `json:"data"`
}
if err = d.client.Do(c, req, &res); err != nil {
log.Error("RecentFans url(%s) response(%+v) error(%v)", d.recentFansURI+"?"+params.Encode(), res, err)
err = ecode.CreativeFansMedalErr
return
}
if res.Code != 0 {
log.Error("RecentFans url(%s) res(%v)", d.recentFansURI+"?"+params.Encode(), res)
err = ecode.Int(res.Code)
return
}
fans = res.Data
return
}
//Rank get fans rank list
func (d *Dao) Rank(c context.Context, mid int64) (rank []*medal.FansRank, err error) {
var (
req *http.Request
params = url.Values{}
)
params.Set("uid", strconv.FormatInt(mid, 10))
if req, err = http.NewRequest("GET", d.fansRankURI+"?"+params.Encode(), nil); err != nil {
log.Error("Rank url(%s) error(%v)", d.fansRankURI+"?"+params.Encode(), err)
err = ecode.CreativeFansMedalErr
return
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
var res struct {
Code int `json:"code"`
Data []*medal.FansRank `json:"data"`
}
if err = d.client.Do(c, req, &res); err != nil {
log.Error("Rank url(%s) response(%+v) error(%v)", d.fansRankURI+"?"+params.Encode(), res, err)
err = ecode.CreativeFansMedalErr
return
}
if res.Code != 0 {
log.Error("Rank url(%s) res(%v)", d.fansRankURI+"?"+params.Encode(), res)
err = ecode.Int(res.Code)
return
}
rank = res.Data
return
}
//Rename rename medal name
func (d *Dao) Rename(c context.Context, mid int64, name, ak, ck string) (err error) {
var (
req *http.Request
params = url.Values{}
)
params.Set("uid", strconv.FormatInt(mid, 10))
params.Set("medal_name", name)
params.Set("source", "2")
if ak != "" {
params.Set("access_key", ak)
params.Set("platform", "ios")
} else {
params.Set("platform", "pc")
}
if req, err = http.NewRequest("POST", d.renameURI, strings.NewReader(params.Encode())); err != nil {
log.Error("Rename url(%s) error(%v)", d.renameURI+"?"+params.Encode(), err)
err = ecode.CreativeFansMedalErr
return
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
req.Header.Set("Cookie", ck)
var res struct {
Code int `json:"code"`
}
if err = d.client.Do(c, req, &res); err != nil {
log.Error("Rename url(%s) response(%+v) error(%v)", d.renameURI+"?"+params.Encode(), res, err)
err = ecode.CreativeFansMedalErr
return
}
if res.Code != 0 {
log.Error("Rename url(%s) res(%v)", d.renameURI+"?"+params.Encode(), res)
err = ecode.Int(res.Code)
return
}
return
}

View File

@@ -0,0 +1,339 @@
package medal
import (
"context"
"encoding/json"
"flag"
"go-common/app/interface/main/creative/conf"
"go-common/app/interface/main/creative/model/medal"
"go-common/library/ecode"
"io"
"net/http"
"net/url"
"os"
"strconv"
"strings"
"testing"
"github.com/bouk/monkey"
"github.com/smartystreets/goconvey/convey"
gock "gopkg.in/h2non/gock.v1"
)
var (
d *Dao
)
func TestMain(m *testing.M) {
if os.Getenv("DEPLOY_ENV") != "" {
flag.Set("app_id", "main.archive.creative")
flag.Set("conf_token", "96b6a6c10bb311e894c14a552f48fef8")
flag.Set("tree_id", "2305")
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")
} else {
flag.Set("conf", "../../cmd/creative.toml")
}
flag.Parse()
if err := conf.Init(); err != nil {
panic(err)
}
d = New(conf.Conf)
m.Run()
os.Exit(0)
}
func httpMock(method, url string) *gock.Request {
r := gock.New(url)
r.Method = strings.ToUpper(method)
d.client.SetTransport(gock.DefaultTransport)
return r
}
func TestRank(t *testing.T) {
var (
c = context.TODO()
err error
mid int64
data = make([]*medal.FansRank, 0)
res = &struct {
Code int `json:"code"`
Data []*medal.FansRank `json:"data"`
}{}
)
convey.Convey("1", t, func(ctx convey.C) {
defer gock.OffAll()
httpMock("Do", d.fansRankURI).Reply(-502)
data, err = d.Rank(c, mid)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(data, convey.ShouldBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
defer gock.OffAll()
res.Code = 0
res.Data = append(res.Data, &medal.FansRank{})
js, _ := json.Marshal(res)
httpMock("Do", d.fansRankURI).Reply(200).JSON(string(js))
data, err = d.Rank(c, mid)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(data, convey.ShouldBeNil)
})
})
}
func TestRename(t *testing.T) {
var (
c = context.TODO()
err error
mid int64
name, ck, ak string
res = &struct {
Code int `json:"code"`
}{}
)
convey.Convey("1", t, func(ctx convey.C) {
defer gock.OffAll()
httpMock("Do", d.fansRankURI).Reply(-502)
err = d.Rename(c, mid, name, ak, ck)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
defer gock.OffAll()
res.Code = 0
ak = "iamak"
js, _ := json.Marshal(res)
httpMock("Do", d.fansRankURI).Reply(200).JSON(string(js))
err = d.Rename(c, mid, name, ak, ck)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
}
func TestRecentFans(t *testing.T) {
var (
c = context.TODO()
err error
mid = int64(2089809)
data = make([]*medal.RecentFans, 0)
res = &struct {
Code int `json:"code"`
Data []*medal.RecentFans `json:"data"`
}{}
)
convey.Convey("1", t, func(ctx convey.C) {
defer gock.OffAll()
httpMock("Do", d.recentFansURI).Reply(-502)
data, err = d.RecentFans(c, mid)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(data, convey.ShouldBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
defer gock.OffAll()
res.Code = 0
res.Data = append(res.Data, &medal.RecentFans{})
js, _ := json.Marshal(res)
httpMock("Do", d.recentFansURI).Reply(200).JSON(string(js))
data, err = d.RecentFans(c, mid)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(data, convey.ShouldBeNil)
})
})
convey.Convey("3", t, func(ctx convey.C) {
req := &http.Request{
Header: make(map[string][]string),
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
monkeyNewGetRequest(req, ecode.CreativeFansMedalErr)
data, err = d.RecentFans(c, mid)
ctx.Convey("3", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(data, convey.ShouldBeNil)
})
})
}
func monkeyNewGetRequest(req *http.Request, err error) {
monkey.Patch(http.NewRequest, func(_, _ string, _ io.Reader) (*http.Request, error) {
return req, err
})
}
func TestMedal(t *testing.T) {
var (
c = context.TODO()
err error
mid = int64(2089809)
data = &medal.Medal{}
res = &struct {
Code int `json:"code"`
Data *medal.Medal `json:"data"`
}{}
)
convey.Convey("1", t, func(ctx convey.C) {
defer gock.OffAll()
httpMock("Do", d.getMedalURI).Reply(-502)
data, err = d.Medal(c, mid)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(data, convey.ShouldBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
defer gock.OffAll()
res.Code = 0
res.Data = &medal.Medal{
UID: "2089809",
}
js, _ := json.Marshal(res)
httpMock("Do", d.getMedalURI).Reply(200).JSON(string(js))
data, err = d.Medal(c, mid)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(data, convey.ShouldBeNil)
})
})
}
func TestStatus(t *testing.T) {
var (
c = context.TODO()
err error
mid = int64(2089809)
data *medal.Status
res = &struct {
Code int `json:"code"`
Data *medal.Status `json:"data"`
}{}
)
convey.Convey("1", t, func(ctx convey.C) {
defer gock.OffAll()
httpMock("Do", d.checkStatusURI).Reply(-502)
data, err = d.Status(c, mid)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(data, convey.ShouldBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
defer gock.OffAll()
res.Code = 20032
js, _ := json.Marshal(res)
httpMock("Do", d.checkStatusURI).Reply(200).JSON(string(js))
data, err = d.Status(c, mid)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(data, convey.ShouldBeNil)
ctx.So(res, convey.ShouldNotBeNil)
})
})
}
func TestOpenMedal(t *testing.T) {
var (
c = context.TODO()
err error
mid int64
name string
res = &struct {
Code int `json:"code"`
Msg string `json:"msg"`
}{}
)
convey.Convey("1", t, func(ctx convey.C) {
defer gock.OffAll()
httpMock("Do", d.openMedalURI).Reply(-502)
err = d.OpenMedal(c, mid, name)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
convey.Convey("2", t, func(ctx convey.C) {
defer gock.OffAll()
res.Code = 0
res.Msg = "ok"
js, _ := json.Marshal(res)
httpMock("Do", d.openMedalURI).Reply(200).JSON(string(js))
err = d.OpenMedal(c, mid, name)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
convey.Convey("3", t, func(ctx convey.C) {
res.Code = 0
res.Msg = "ok"
req := &http.Request{
URL: &url.URL{},
Header: make(map[string][]string),
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
monkeyNewGetRequest(req, nil)
err = d.OpenMedal(c, mid, name)
ctx.Convey("3", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
})
})
}
func TestCheckMedal(t *testing.T) {
var (
c = context.TODO()
err error
mid int64
valid int
name string
res = &struct {
Code int `json:"code"`
Data struct {
Enable bool `json:"enable"`
} `json:"data"`
}{}
)
convey.Convey("1", t, func(ctx convey.C) {
defer gock.OffAll()
httpMock("Do", d.checkMedalURI).Reply(-502)
valid, err = d.CheckMedal(c, mid, name)
ctx.Convey("1", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(valid, convey.ShouldBeGreaterThanOrEqualTo, 0)
})
})
convey.Convey("2", t, func(ctx convey.C) {
defer gock.OffAll()
res.Code = 0
js, _ := json.Marshal(res)
httpMock("Do", d.checkMedalURI).Reply(200).JSON(string(js))
valid, err = d.CheckMedal(c, mid, name)
ctx.Convey("2", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(valid, convey.ShouldBeGreaterThanOrEqualTo, 0)
})
})
convey.Convey("3", t, func(ctx convey.C) {
res.Code = 0
req := &http.Request{
URL: &url.URL{},
Header: make(map[string][]string),
}
req.Header.Set("X-BiliLive-UID", strconv.FormatInt(mid, 10))
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
monkeyNewGetRequest(req, ecode.CreativeFansMedalErr)
valid, err = d.CheckMedal(c, mid, name)
ctx.Convey("3", func(ctx convey.C) {
ctx.So(err, convey.ShouldNotBeNil)
ctx.So(valid, convey.ShouldBeGreaterThanOrEqualTo, 0)
})
})
}