Create & Init Project...

app/service/main/resource/rpc/client/BUILD

@@ -0,0 +1,42 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["client_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = ["//app/service/main/resource/model:go_default_library"],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["client.go"],
+    importpath = "go-common/app/service/main/resource/rpc/client",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/service/main/resource/model:go_default_library",
+        "//library/net/rpc: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/resource/rpc/client/client.go

@@ -0,0 +1,126 @@
+package client
+
+import (
+	"context"
+
+	"go-common/app/service/main/resource/model"
+	"go-common/library/net/rpc"
+)
+
+var (
+	_noArg = &struct{}{}
+)
+
+const (
+	_resourceAll   = "RPC.ResourceAll"
+	_assignmentAll = "RPC.AssignmentAll"
+	_defBanner     = "RPC.DefBanner"
+	_resource      = "RPC.Resource"
+	_resources     = "RPC.Resources"
+	_banners       = "RPC.Banners"
+	_pasterAPP     = "RPC.PasterAPP"
+	_indexIcon     = "RPC.IndexIcon"
+	_playerIcon    = "RPC.PlayerIcon"
+	_cmtbox        = "RPC.Cmtbox"
+	_sidebars      = "RPC.SideBars"
+	_abtest        = "RPC.AbTest"
+	_pasterCID     = "RPC.PasterCID"
+
+	// app id
+	_appid = "resource.service"
+)
+
+// Service is resource rpc client.
+type Service struct {
+	client *rpc.Client2
+}
+
+// New new a resource rpc client.
+func New(c *rpc.ClientConfig) (s *Service) {
+	s = &Service{}
+	s.client = rpc.NewDiscoveryCli(_appid, c)
+	return
+}
+
+// ResourceAll get all resource.
+func (s *Service) ResourceAll(c context.Context) (res []*model.Resource, err error) {
+	err = s.client.Call(c, _resourceAll, _noArg, &res)
+	return
+}
+
+// AssignmentAll get all assignment.
+func (s *Service) AssignmentAll(c context.Context) (res []*model.Assignment, err error) {
+	err = s.client.Call(c, _assignmentAll, _noArg, &res)
+	return
+}
+
+// DefBanner get default banner.
+func (s *Service) DefBanner(c context.Context) (res *model.Assignment, err error) {
+	res = new(model.Assignment)
+	err = s.client.Call(c, _defBanner, _noArg, res)
+	return
+}
+
+// Resource get resource.
+func (s *Service) Resource(c context.Context, arg *model.ArgRes) (res *model.Resource, err error) {
+	res = new(model.Resource)
+	err = s.client.Call(c, _resource, arg, res)
+	return
+}
+
+// Resources get resource.
+func (s *Service) Resources(c context.Context, arg *model.ArgRess) (res map[int]*model.Resource, err error) {
+	err = s.client.Call(c, _resources, arg, &res)
+	return
+}
+
+// Banners get banners.
+func (s *Service) Banners(c context.Context, arg *model.ArgBanner) (res *model.Banners, err error) {
+	err = s.client.Call(c, _banners, arg, &res)
+	return
+}
+
+// PasterAPP get paster.
+func (s *Service) PasterAPP(c context.Context, arg *model.ArgPaster) (res *model.Paster, err error) {
+	res = new(model.Paster)
+	err = s.client.Call(c, _pasterAPP, arg, res)
+	return
+}
+
+// IndexIcon get index icons.
+func (s *Service) IndexIcon(c context.Context) (res map[string][]*model.IndexIcon, err error) {
+	err = s.client.Call(c, _indexIcon, _noArg, &res)
+	return
+}
+
+// PlayerIcon get palyer config.
+func (s *Service) PlayerIcon(c context.Context) (res *model.PlayerIcon, err error) {
+	res = new(model.PlayerIcon)
+	err = s.client.Call(c, _playerIcon, _noArg, res)
+	return
+}
+
+// Cmtbox get live box
+func (s *Service) Cmtbox(c context.Context, arg *model.ArgCmtbox) (res *model.Cmtbox, err error) {
+	err = s.client.Call(c, _cmtbox, arg, &res)
+	return
+}
+
+// SideBars get side bar.
+func (s *Service) SideBars(c context.Context) (res *model.SideBars, err error) {
+	res = new(model.SideBars)
+	err = s.client.Call(c, _sidebars, _noArg, res)
+	return
+}
+
+// AbTest get abtest.
+func (s *Service) AbTest(c context.Context, arg *model.ArgAbTest) (res map[string]*model.AbTest, err error) {
+	err = s.client.Call(c, _abtest, arg, &res)
+	return
+}
+
+// PasterCID get all Paster's cid.
+func (s *Service) PasterCID(c context.Context) (res map[int64]int64, err error) {
+	err = s.client.Call(c, _pasterCID, _noArg, &res)
+	return
+}

app/service/main/resource/rpc/client/client_test.go

@@ -0,0 +1,191 @@
+package client
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"go-common/app/service/main/resource/model"
+)
+
+func TestResource(t *testing.T) {
+	s := New(nil)
+	time.Sleep(2 * time.Second)
+	testResourceAll(t, s)
+	testAssignmentAll(t, s)
+	testDefBanner(t, s)
+	testResource(t, s)
+	testResources(t, s)
+	testBanners(t, s)
+	testPasterAPP(t, s)
+	testIndexIcon(t, s)
+	testPlayerIcon(t, s)
+	testCmtbox(t, s)
+	testSideBars(t, s)
+	testAbTest(t, s)
+	testPasterCID(t, s)
+}
+
+// testResourceAll test rpc ResourceAll.
+func testResourceAll(t *testing.T, s *Service) {
+	res, err := s.ResourceAll(context.TODO())
+	if err != nil {
+		t.Logf("testResourceAll error(%v) \n", err)
+		return
+	}
+	t.Logf("testResourceAll res: %+v \n", res)
+}
+
+// testAssignmentAll test rpc AssignmentAll.
+func testAssignmentAll(t *testing.T, s *Service) {
+	res, err := s.AssignmentAll(context.TODO())
+	if err != nil {
+		t.Logf("testAssignmentAll error(%v) \n", err)
+		return
+	}
+	t.Logf("testAssignmentAll res: %+v \n", res)
+}
+
+// testDefBanner test rpc DefBanner.
+func testDefBanner(t *testing.T, s *Service) {
+	res, err := s.DefBanner(context.TODO())
+	if err != nil {
+		t.Logf("testDefBanner error(%v) \n", err)
+		return
+	}
+	t.Logf("testDefBanner res: %+v \n", res)
+}
+
+// testResource test rpc Resource.
+func testResource(t *testing.T, s *Service) {
+	p := &model.ArgRes{
+		ResID: 1233,
+	}
+	res, err := s.Resource(context.TODO(), p)
+	if err != nil {
+		t.Logf("testResource error(%v) \n", err)
+		return
+	}
+	t.Logf("testResource res: %+v \n", res)
+}
+
+// testResources test rpc Resources.
+func testResources(t *testing.T, s *Service) {
+	p := &model.ArgRess{
+		ResIDs: []int{1187, 1639},
+	}
+	res, err := s.Resources(context.TODO(), p)
+	if err != nil {
+		t.Logf("testResources error(%v) \n", err)
+		return
+	}
+	t.Logf("testResources res: %+v \n", res)
+}
+
+// testBanners test rpc Banners.
+func testBanners(t *testing.T, s *Service) {
+	ab := &model.ArgBanner{
+		Plat:      1,
+		ResIDs:    "454,467",
+		Build:     508000,
+		MID:       1493031,
+		Channel:   "abc",
+		IP:        "211.139.80.6",
+		Buvid:     "123",
+		Network:   "wifi",
+		MobiApp:   "iphone",
+		Device:    "test",
+		IsAd:      true,
+		OpenEvent: "abc",
+	}
+	res, err := s.Banners(context.TODO(), ab)
+	if err != nil {
+		t.Logf("testBanners error(%v) \n", err)
+		return
+	}
+	t.Logf("testBanners res: %+v \n", res)
+}
+
+// testPasterAPP test rpc Paster.
+func testPasterAPP(t *testing.T, s *Service) {
+	p := &model.ArgPaster{
+		Platform: int8(4),
+		AdType:   int8(1),
+		Aid:      "10097274",
+		TypeId:   "11",
+		Buvid:    "666666",
+	}
+	res, err := s.PasterAPP(context.TODO(), p)
+	if err != nil {
+		t.Logf("testPasterAPP error(%v) \n", err)
+		return
+	}
+	t.Logf("testPaster res: %+v \n", res)
+}
+
+// testIndexIcon test rpc IndexIcon.
+func testIndexIcon(t *testing.T, s *Service) {
+	res, err := s.IndexIcon(context.TODO())
+	if err != nil {
+		t.Logf("testIndexIcon error(%v) \n", err)
+		return
+	}
+	t.Logf("testIndexIcon res: %+v \n", res)
+}
+
+// testPlayerIcon test rpc PlayerIcon.
+func testPlayerIcon(t *testing.T, s *Service) {
+	res, err := s.PlayerIcon(context.TODO())
+	if err != nil {
+		t.Logf("testPlayerIcon error(%v) \n", err)
+		return
+	}
+	t.Logf("testPlayerIcon res: %+v \n", res)
+}
+
+// testCmtbox test rpc Resource.
+func testCmtbox(t *testing.T, s *Service) {
+	p := &model.ArgCmtbox{
+		ID: 1,
+	}
+	res, err := s.Cmtbox(context.TODO(), p)
+	if err != nil {
+		t.Logf("testCmtbox error(%v) \n", err)
+		return
+	}
+	t.Logf("testCmtbox res: %+v \n", res)
+}
+
+// testSideBars test rpc SideBars.
+func testSideBars(t *testing.T, s *Service) {
+	res, err := s.SideBars(context.TODO())
+	if err != nil {
+		t.Logf("testSideBars error(%v) \n", err)
+		return
+	}
+	t.Logf("testSideBars res: %+v \n", res)
+}
+
+// testAbTest test rpc abtest.
+func testAbTest(t *testing.T, s *Service) {
+	p := &model.ArgAbTest{
+		Groups: "不显示热门tab,显示热门tab",
+		IP:     "127.0.0.1",
+	}
+	res, err := s.AbTest(context.TODO(), p)
+	if err != nil {
+		t.Logf("testAbTest error(%v) \n", err)
+		return
+	}
+	t.Logf("testAbTest res: %+v \n", res)
+}
+
+// testPasterCID test rpc PasterCID
+func testPasterCID(t *testing.T, s *Service) {
+	res, err := s.PasterCID(context.TODO())
+	if err != nil {
+		t.Logf("testPasterCID error(%v) \n", err)
+		return
+	}
+	t.Logf("testPasterCID res: %+v \n", res)
+}