openbilibili/go-common
1
0
Fork 0
Code Releases Activity

Create & Init Project...

Browse Source
This commit is contained in:
MiaoWoo
2019-04-22 18:49:16 +08:00
commit fc4fa37393
25440 changed files with 4054998 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
app/interface/main/creative/dao/activity/BUILD Normal file
View File

@@ -0,0 +1,50 @@
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/activity",
tags = ["automanaged"],
visibility = ["//visibility:public"],
deps = [
"//app/interface/main/creative/conf:go_default_library",
"//app/interface/main/creative/model/activity: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/activity:go_default_library",
"//vendor/github.com/smartystreets/goconvey/convey:go_default_library",
"//vendor/gopkg.in/h2non/gock.v1:go_default_library",
],
)

213
app/interface/main/creative/dao/activity/dao.go Normal file
View File

@@ -0,0 +1,213 @@
package activity
import (
"context"
"encoding/json"
"net/url"
"strconv"
"strings"
"go-common/app/interface/main/creative/conf"
"go-common/app/interface/main/creative/model/activity"
"go-common/library/ecode"
"go-common/library/log"
httpx "go-common/library/net/http/blademaster"
)
const (
_activityAllList = "/activity/list/videoall"
_activityUpdate = "/api/likes/upbyaid/%d"
_activitySubject = "/activity/subject/%d"
_activityProtocol = "/activity/protocol/%d"
_actOnlineByTypeURI = "/activity/online/by/type"
_actURLType16 = "https://www.bilibili.com/blackboard/x/activity-tougao-h5/detail?from=snap&id="
_actLike = "/activity/likes/list/%d"
)
// Dao define
type Dao struct {
c *conf.Config
// http
client *httpx.Client
// uri
ActAllListURI string
ActUpdateURI string
ActSubjectURI string
ActProtocolURI string
ActOnlineByTypeURL string
ActLikeURI string
}
// New init dao
func New(c *conf.Config) (d *Dao) {
d = &Dao{
c: c,
client: httpx.NewClient(c.HTTPClient.Normal),
ActAllListURI: c.Host.Activity + _activityAllList,
ActUpdateURI: c.Host.Act + _activityUpdate,
ActSubjectURI: c.Host.Activity + _activitySubject,
ActProtocolURI: c.Host.Act + _activityProtocol,
ActOnlineByTypeURL: c.Host.Activity + _actOnlineByTypeURI,
ActLikeURI: c.Host.Activity + _actLike,
}
return
}
// MissionOnlineByTid fn
func (d *Dao) MissionOnlineByTid(c context.Context, tid, plat int16) (mm []*activity.ActWithTP, err error) {
var res struct {
Code int `json:"code"`
Data []*activity.ActWithTP `json:"data"`
}
mm = make([]*activity.ActWithTP, 0)
params := url.Values{}
params.Set("type", strconv.Itoa(int(tid)))
params.Set("plat", strconv.Itoa(int(plat)))
if err = d.client.Get(c, d.ActOnlineByTypeURL, "", params, &res); err != nil {
log.Error("videoup ActOnlineByTypeURL error(%v) | ActOnlineByTypeURL(%s)", err, d.ActOnlineByTypeURL+"?"+params.Encode())
err = ecode.CreativeActivityErr
return
}
if res.Code != 0 {
log.Error("videoup ActOnlineByTypeURL res.Code nq zero error(%v) | ActOnlineByTypeURL(%s) res(%v)", res.Code, d.ActOnlineByTypeURL+"?"+params.Encode(), res)
err = ecode.CreativeActivityErr
return
}
for _, m := range res.Data {
miss := &activity.ActWithTP{
ID: m.ID,
Name: m.Name,
Hot: m.Hot,
Protocol: m.Protocol,
Types: m.Types,
}
if len(m.Tags) > 0 {
miss.Tags = strings.Split(m.Tags, ",")[0]
} else {
miss.Tags = m.Name
}
if m.Type == 16 && len(m.ActURL) == 0 {
miss.ActURL = _actURLType16 + strconv.FormatInt(miss.ID, 10)
} else {
miss.ActURL = m.ActURL
}
mm = append(mm, miss)
}
return
}
// Activities get activity list.
func (d *Dao) Activities(c context.Context) (act []*activity.Activity, err error) {
var (
url string
res struct {
Code int `json:"code"`
Data []*activity.Activity `json:"data"`
}
)
url = d.ActAllListURI
if err = d.client.Get(c, url, "", nil, &res); err != nil {
log.Error("ActivityList url(%s) response(%v) error(%v)", url, res, err)
err = ecode.CreativeActivityErr
return
}
if res.Code != 0 {
log.Error("ActivityList url(%s) res(%v)", url, res)
err = ecode.CreativeActivityErr
return
}
for _, v := range res.Data {
if v.Type == 16 && len(v.ActURL) == 0 {
v.ActURL = _actURLType16 + strconv.FormatInt(v.ID, 10)
}
}
act = res.Data
return
}
// Unbind update the aid a status of this activity
func (d *Dao) Unbind(c context.Context, aid, missionID int64, ip string) (err error) {
params := url.Values{}
params.Set("mission_id", strconv.FormatInt(missionID, 10))
params.Set("aid", strconv.FormatInt(aid, 10))
params.Set("state", "-1")
ms := json.RawMessage{}
if err = d.client.RESTfulPost(c, d.ActUpdateURI, ip, params, &ms, missionID); err != nil {
log.Error("ActUpdateURI url(%s) missionID(%d) error(%v)", d.ActUpdateURI, missionID, err)
err = ecode.CreativeActivityErr
return
}
log.Info("d.UpdateByAid url(%s) params(%s) res(%s)", d.ActUpdateURI, params.Encode(), string(ms))
return
}
//Subject get any exist activity by missionID
func (d *Dao) Subject(c context.Context, missionID int64) (sub *activity.Activity, err error) {
var res struct {
Code int `json:"code"`
Data *activity.Subject `json:"data"`
}
if err = d.client.RESTfulGet(c, d.ActSubjectURI, "", url.Values{}, &res, missionID); err != nil {
log.Error("ActSubjectURI url(%s) missionID(%d) error(%v)", d.ActSubjectURI, missionID, err)
err = ecode.CreativeActivityErr
return
}
if res.Code != 0 {
log.Error("ActSubjectURI url(%s) res(%v)", d.ActSubjectURI, res)
err = ecode.CreativeActivityErr
return
}
var ID int
if ID, err = strconv.Atoi(res.Data.ID); err != nil {
log.Error("ActSubjectURI url(%s) res(%v)", d.ActSubjectURI, res)
err = ecode.CreativeActivityErr
return
}
sub = &activity.Activity{
Name: res.Data.Name,
ID: int64(ID),
}
return
}
// Protocol fn
func (d *Dao) Protocol(c context.Context, missionID int64) (p *activity.Protocol, err error) {
var res struct {
Code int `json:"code"`
Data *activity.Protocol `json:"data"`
}
if err = d.client.RESTfulGet(c, d.ActProtocolURI, "", url.Values{}, &res, missionID); err != nil {
log.Error("ActProtocolURI url(%s) missionID(%d) error(%v)", d.ActProtocolURI, missionID, err)
err = ecode.CreativeActivityErr
return
}
if res.Code != 0 {
log.Error("ActProtocolURI url(%s) res(%v)", d.ActProtocolURI, res)
err = ecode.CreativeActivityErr
return
}
p = res.Data
return
}
//Likes fn
func (d *Dao) Likes(c context.Context, missionID int64) (likeCnt int, err error) {
var res struct {
Code int `json:"code"`
Data *activity.Like `json:"data"`
}
if err = d.client.RESTfulGet(c, d.ActLikeURI, "", url.Values{}, &res, missionID); err != nil {
log.Error("ActLikeURI url(%s) missionID(%d) error(%v)", d.ActLikeURI, missionID, err)
err = ecode.CreativeActivityErr
return
}
if res.Code != 0 {
log.Error("ActLikeURI url(%s) res(%v)", d.ActLikeURI, res)
err = ecode.CreativeActivityErr
return
}
if res.Data != nil {
likeCnt = res.Data.Count
}
return
}

203
app/interface/main/creative/dao/activity/dao_test.go Normal file
View File

@@ -0,0 +1,203 @@
package activity
import (
"context"
"flag"
"fmt"
"go-common/app/interface/main/creative/conf"
"go-common/app/interface/main/creative/model/activity"
"os"
"strings"
"testing"
"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 Test_Likes(t *testing.T) {
var (
c = context.TODO()
err error
missionID = int64(10364)
likeCnt int
)
convey.Convey("Likes", t, func(ctx convey.C) {
defer gock.OffAll()
url := fmt.Sprintf(d.ActLikeURI, 10364)
httpMock("GET", url).Reply(200).JSON(`{"code":0,"data":{"id":53,"mid":27515310,"uname":"1qs314567","state":2,"type":2,"position":4,"url":"http://i0.hdslb.com/bfs/article/578a61e7caf47b5deaa80940b2806f1d9ce53dde.png","md5":"79d02f08c2b7b0d2b30a6b6a4f61c97e","info":"{\"width\":325,\"height\":50}","ctime":1499764110,"mtime":1499828335},"message":"","ttl":1}`)
likeCnt, err = d.Likes(c, missionID)
ctx.Convey("Then err should be nil.has should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
ctx.So(likeCnt, convey.ShouldBeGreaterThanOrEqualTo, 0)
})
})
}
func Test_Protocol(t *testing.T) {
var (
c = context.TODO()
err error
missionID = int64(10364)
p *activity.Protocol
)
convey.Convey("Protocol", t, func(ctx convey.C) {
defer gock.OffAll()
url := fmt.Sprintf(d.ActProtocolURI, 10364)
httpMock("GET", url).Reply(200).JSON(`{"code":0,"data":{"id":"231","sid":"10364","protocol":"说明不超过201234567890123","types":"","tags":"","hot":"0","bgm_id":"0","paster_id":"0","oids":"","screen_set":"1"}}`)
p, err = d.Protocol(c, missionID)
ctx.Convey("Then err should be nil.has should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
ctx.So(p, convey.ShouldNotBeNil)
})
})
}
func Test_Subject(t *testing.T) {
var (
c = context.TODO()
err error
missionID = int64(10364)
act *activity.Activity
)
convey.Convey("Subject", t, func(ctx convey.C) {
defer gock.OffAll()
url := fmt.Sprintf(d.ActSubjectURI, 10364)
httpMock("GET", url).Reply(200).JSON(`{"code":0,"data":{"id":"231","sid":"10364","Subject":"说明不超过201234567890123","types":"","tags":"","hot":"0","bgm_id":"0","paster_id":"0","oids":"","screen_set":"1"}}`)
act, err = d.Subject(c, missionID)
ctx.Convey("Then err should be nil.has should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
ctx.So(act, convey.ShouldNotBeNil)
})
})
}
func Test_Activities(t *testing.T) {
var (
c = context.TODO()
err error
acts []*activity.Activity
)
convey.Convey("Activities", t, func(ctx convey.C) {
defer gock.OffAll()
url := d.ActAllListURI
httpMock("GET", url).Reply(200).JSON(`{"code":0,"data":[{"id":10329,"oid":0,"type":4,"state":1,"stime":"2018-10-14 14:32:00","etime":"2018-12-17 00:00:00","ctime":"2018-08-30 18:03:30","mtime":"2018-10-17 18:27:44","name":"这是一个标题，123kkkk有点长有点长，可能展示不下，只能展示13个字","author":"jinchenchen","act_url":"","lstime":"2018-09-08 00:00:00","letime":"2018-12-01 00:00:00","cover":"//uat-i0.hdslb.com/bfs/test/static/20181017/fb8f33d1a41042b9a1ebb515fdc19d94/nBljdwyCo.jpg","dic":"sdf","flag":"33","uetime":"0000-00-00 00:00:00","ustime":"0000-00-00 00:00:00","level":"0","h5_cover":"","rank":"123","like_limit":"1","android_url":"","ios_url":"","fan_limit_max":"0","fan_limit_min":"0","tags":"","hot":0,"bgm_id":123,"paster_id":123,"oids":"10110549|10110164|10110536|10110546","screen_set":1,"protocol":"活动说明，123MMMMM展示展示，只能展示20个，展示不下，显示"}]}`)
acts, err = d.Activities(c)
ctx.Convey("Then err should be nil.has should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
ctx.So(acts, convey.ShouldNotBeNil)
})
})
}
func Test_MissionOnlineByTid(t *testing.T) {
var (
c = context.TODO()
err error
tid = int16(160)
plat = int16(1)
mm []*activity.ActWithTP
)
convey.Convey("MissionOnlineByTid", t, func(ctx convey.C) {
defer gock.OffAll()
url := d.ActOnlineByTypeURL
httpMock("GET", url).Reply(200).JSON(`
{
"code": 0,
"data": [
{
"id": 10329,
"oid": 0,
"type": 4,
"state": 1,
"stime": "2018-10-14 14:32:00",
"etime": "2018-12-17 00:00:00",
"ctime": "2018-08-30 18:03:30",
"mtime": "2018-10-17 18:27:44",
"name": "这是一个",
"author": "jinchenchen",
"act_url": "",
"lstime": "2018-09-08 00:00:00",
"letime": "2018-12-01 00:00:00",
"cover": "//uat-i0.hdslb.com/bfs/test/static/20181017/fb8f33d1a41042b9a1ebb515fdc19d94/nBljdwyCo.jpg",
"dic": "sdf",
"flag": "33",
"uetime": "0000-00-00 00:00:00",
"ustime": "0000-00-00 00:00:00",
"level": "0",
"h5_cover": "",
"rank": "123",
"like_limit": "1",
"android_url": "",
"ios_url": "",
"fan_limit_max": "0",
"fan_limit_min": "0",
"tags": "",
"types": "",
"hot": 0,
"bgm_id": 123,
"paster_id": 123,
"oids": "10110549|10110164|10110536|10110546",
"screen_set": 1,
"protocol": "活动说明"
}
]
}
`)
mm, err = d.MissionOnlineByTid(c, tid, plat)
ctx.Convey("Then err should be nil.has should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
ctx.So(mm, convey.ShouldNotBeNil)
})
})
}
func Test_Unbind(t *testing.T) {
var (
c = context.TODO()
err error
missionID = int64(10364)
aid = int64(10110788)
ip = "127.0.0.1"
)
convey.Convey("Unbind", t, func(ctx convey.C) {
defer gock.OffAll()
url := fmt.Sprintf(d.ActUpdateURI, 10364)
httpMock("POST", url).Reply(200).JSON(`{"code":0,"data":""}`)
err = d.Unbind(c, aid, missionID, ip)
ctx.Convey("Then err should be nil.has should not be nil.", func(ctx convey.C) {
ctx.So(err, convey.ShouldBeNil)
})
})
}