Create & Init Project...

app/service/main/assist/dao/account/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/service/main/account/api:go_default_library",
+        "//app/service/main/assist/conf:go_default_library",
+        "//library/ecode:go_default_library",
+        "//vendor/github.com/golang/mock/gomock: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/service/main/assist/dao/account",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/service/main/account/api:go_default_library",
+        "//app/service/main/assist/conf:go_default_library",
+        "//library/ecode:go_default_library",
+        "//library/log: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/service/main/assist/dao/account/dao.go

@@ -0,0 +1,131 @@
+package account
+
+import (
+	"context"
+	accapi "go-common/app/service/main/account/api"
+	"go-common/app/service/main/assist/conf"
+	"go-common/library/ecode"
+	"go-common/library/log"
+)
+
+// Dao is account dao.
+type Dao struct {
+	c   *conf.Config
+	acc accapi.AccountClient
+}
+
+// New new a dao.
+func New(c *conf.Config) (d *Dao) {
+	d = &Dao{
+		c: c,
+	}
+	var err error
+	if d.acc, err = accapi.NewClient(c.AccClient); err != nil {
+		panic(err)
+	}
+	return
+}
+
+// IsFollow check assist follow up.
+func (d *Dao) IsFollow(c context.Context, mid, assistMid int64) (follow bool, err error) {
+	var arg = &accapi.RelationReq{
+		Mid:   assistMid,
+		Owner: mid,
+	}
+	res, err := d.acc.Relation3(c, arg)
+	if err != nil {
+		log.Error("d.acc.Relation2(%d,%d) error(%v)", mid, assistMid, err)
+		return
+	}
+	follow = res.Following
+	return
+}
+
+// IdentifyInfo 获取用户实名认证状态
+func (d *Dao) IdentifyInfo(c context.Context, mid int64, ip string) (err error) {
+	var (
+		arg = &accapi.MidReq{
+			Mid: mid,
+		}
+		rpcRes *accapi.ProfileReply
+		mf     *accapi.Profile
+	)
+	if rpcRes, err = d.acc.Profile3(c, arg); err != nil {
+		log.Error("d.acc.Profile3 error(%v) | mid(%d) ip(%s) arg(%v)", err, mid, ip, arg)
+		err = ecode.CreativeAccServiceErr
+		return
+	}
+	if rpcRes != nil {
+		mf = rpcRes.Profile
+	}
+	if mf.Identification == 1 {
+		return
+	}
+	if err = d.switchIDInfoRet(mf.TelStatus); err != nil {
+		log.Error("switchIDInfoRet res(%v)", mf.TelStatus)
+		return
+	}
+	return
+}
+func (d *Dao) switchIDInfoRet(phoneRet int32) (err error) {
+	switch phoneRet {
+	case 0:
+		err = ecode.UserCheckNoPhone
+	case 1:
+		err = nil
+	case 2:
+		err = ecode.UserCheckInvalidPhone
+	}
+	return
+}
+
+// UserBanned 获取用户封禁状态, disabled when spacesta == 2
+func (d *Dao) UserBanned(c context.Context, mid int64) (err error) {
+	var card *accapi.Card
+	if card, err = d.Card(c, mid, ""); err != nil {
+		log.Error("d.Card() error(%v)", err)
+		err = nil
+		return
+	}
+	if card.Silence == 1 {
+		err = ecode.UserDisabled
+		return
+	}
+	return
+}
+
+// Card get account.
+func (d *Dao) Card(c context.Context, mid int64, ip string) (res *accapi.Card, err error) {
+	var (
+		rpcRes *accapi.CardReply
+		arg    = &accapi.MidReq{
+			Mid: mid,
+		}
+	)
+	if rpcRes, err = d.acc.Card3(c, arg); err != nil {
+		log.Error("s.acc.Card3() error(%v)", err)
+		err = ecode.CreativeAccServiceErr
+	}
+	if rpcRes != nil {
+		res = rpcRes.Card
+	}
+	return
+}
+
+// Cards get infos for space
+func (d *Dao) Cards(c context.Context, mids []int64) (res map[int64]*accapi.Card, err error) {
+	var (
+		arg = &accapi.MidsReq{
+			Mids: mids,
+		}
+		rpcRes *accapi.CardsReply
+	)
+	if rpcRes, err = d.acc.Cards3(c, arg); err != nil {
+		log.Error("s.acc.Cards3() error(%v)", err)
+		err = ecode.CreativeAccServiceErr
+	}
+	if rpcRes != nil {
+		res = rpcRes.Cards
+	}
+	return
+}

app/service/main/assist/dao/account/dao_test.go

@@ -0,0 +1,131 @@
+package account
+
+import (
+	"context"
+	"flag"
+	accapi "go-common/app/service/main/account/api"
+	"go-common/app/service/main/assist/conf"
+	"go-common/library/ecode"
+	"os"
+	"testing"
+
+	"github.com/golang/mock/gomock"
+	"github.com/smartystreets/goconvey/convey"
+)
+
+var (
+	d *Dao
+)
+
+func TestMain(m *testing.M) {
+	if os.Getenv("DEPLOY_ENV") != "" {
+		flag.Set("app_id", "main.archive.assist-service")
+		flag.Set("conf_token", "6e0dae2c95d90ff8d0da53460ef11ae8")
+		flag.Set("tree_id", "2084")
+		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/assist-service.toml")
+	}
+	flag.Parse()
+	if err := conf.Init(); err != nil {
+		panic(err)
+	}
+	d = New(conf.Conf)
+	m.Run()
+	os.Exit(0)
+}
+
+func WithMock(t *testing.T, f func(mock *gomock.Controller)) func() {
+	return func() {
+		mockCtrl := gomock.NewController(t)
+		defer mockCtrl.Finish()
+		f(mockCtrl)
+	}
+}
+
+func TestIdentifyInfo(t *testing.T) {
+	convey.Convey("1", t, WithMock(t, func(mockCtrl *gomock.Controller) {
+		var (
+			c   = context.Background()
+			mid = int64(2089809)
+			ip  = "127.0.0.1"
+			err error
+		)
+		mock := accapi.NewMockAccountClient(mockCtrl)
+		d.acc = mock
+		mockReq := &accapi.MidReq{
+			Mid: mid,
+		}
+		mock.EXPECT().Profile3(gomock.Any(), mockReq).Return(nil, ecode.CreativeAccServiceErr)
+		err = d.IdentifyInfo(c, mid, ip)
+		convey.So(err, convey.ShouldNotBeNil)
+	}))
+	convey.Convey("2", t, WithMock(t, func(mockCtrl *gomock.Controller) {
+		var (
+			c   = context.Background()
+			mid = int64(2089809)
+			ip  = "127.0.0.1"
+			err error
+		)
+		mock := accapi.NewMockAccountClient(mockCtrl)
+		d.acc = mock
+		mockReq := &accapi.MidReq{
+			Mid: mid,
+		}
+		rpcRes := &accapi.ProfileReply{
+			Profile: &accapi.Profile{
+				Identification: 0,
+				TelStatus:      2,
+			},
+		}
+		mock.EXPECT().Profile3(gomock.Any(), mockReq).Return(rpcRes, nil)
+		err = d.IdentifyInfo(c, mid, ip)
+		convey.So(err, convey.ShouldNotBeNil)
+	}))
+}
+
+func TestIsFollow(t *testing.T) {
+	convey.Convey("1", t, WithMock(t, func(mockCtrl *gomock.Controller) {
+		var (
+			c         = context.Background()
+			mid       = int64(2089809)
+			assistMid = int64(11)
+			err       error
+			follow    bool
+		)
+		mock := accapi.NewMockAccountClient(mockCtrl)
+		d.acc = mock
+		mockReq := &accapi.RelationReq{
+			Mid:   assistMid,
+			Owner: mid,
+		}
+		mock.EXPECT().Relation3(gomock.Any(), mockReq).Return(nil, ecode.CreativeAccServiceErr)
+		follow, err = d.IsFollow(c, mid, assistMid)
+		convey.So(err, convey.ShouldNotBeNil)
+		convey.So(follow, convey.ShouldBeFalse)
+	}))
+}
+func TestCard(t *testing.T) {
+	convey.Convey("TestCard", t, WithMock(t, func(mockCtrl *gomock.Controller) {
+		var (
+			c   = context.Background()
+			mid = int64(2089809)
+			err error
+			res *accapi.Card
+		)
+		mock := accapi.NewMockAccountClient(mockCtrl)
+		d.acc = mock
+		mockReq := &accapi.MidReq{
+			Mid: mid,
+		}
+		mock.EXPECT().Card3(gomock.Any(), mockReq).Return(nil, ecode.CreativeAccServiceErr)
+		res, err = d.Card(c, mid, "")
+		convey.So(err, convey.ShouldNotBeNil)
+		convey.So(res, convey.ShouldBeNil)
+	}))
+}