Create & Init Project...

app/service/main/assist/rpc/client/BUILD

@@ -0,0 +1,45 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["assist_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/service/main/assist/model/assist:go_default_library",
+        "//vendor/github.com/davecgh/go-spew/spew:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["assist.go"],
+    importpath = "go-common/app/service/main/assist/rpc/client",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/service/main/assist/model/assist: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/assist/rpc/client/assist.go

@@ -0,0 +1,112 @@
+package client
+
+import (
+	"context"
+
+	"go-common/library/net/rpc"
+
+	model "go-common/app/service/main/assist/model/assist"
+)
+
+const (
+	_assists         = "RPC.Assists"
+	_assistIDs       = "RPC.AssistIDs"
+	_assist          = "RPC.Assist"
+	_assistExit      = "RPC.AssistExit"
+	_addAssist       = "RPC.AddAssist"
+	_delAssist       = "RPC.DelAssist"
+	_assistLogs      = "RPC.AssistLogs"
+	_assistLogInfo   = "RPC.AssistLogInfo"
+	_assistLogAdd    = "RPC.AssistLogAdd"
+	_assistLogCancel = "RPC.AssistLogCancel"
+	_assistUps       = "RPC.AssistUps"
+)
+
+const (
+	_appid = "archive.service.assist"
+)
+
+var (
+	_noArg = &struct{}{}
+)
+
+// Service def
+type Service struct {
+	client *rpc.Client2
+}
+
+// New def
+func New(c *rpc.ClientConfig) (s *Service) {
+	s = &Service{}
+	s.client = rpc.NewDiscoveryCli(_appid, c)
+	return s
+}
+
+// Assists def
+func (s *Service) Assists(c context.Context, arg *model.ArgAssists) (res []*model.Assist, err error) {
+	err = s.client.Call(c, _assists, arg, &res)
+	return
+}
+
+// AssistIDs def
+func (s *Service) AssistIDs(c context.Context, arg *model.ArgAssists) (res []int64, err error) {
+	err = s.client.Call(c, _assistIDs, arg, &res)
+	return
+}
+
+// Assist def
+func (s *Service) Assist(c context.Context, arg *model.ArgAssist) (res *model.AssistRes, err error) {
+	res = new(model.AssistRes)
+	err = s.client.Call(c, _assist, arg, &res)
+	return
+}
+
+// AddAssist def
+func (s *Service) AddAssist(c context.Context, arg *model.ArgAssist) (err error) {
+	err = s.client.Call(c, _addAssist, arg, _noArg)
+	return
+}
+
+// DelAssist def
+func (s *Service) DelAssist(c context.Context, arg *model.ArgAssist) (err error) {
+	err = s.client.Call(c, _delAssist, arg, _noArg)
+	return
+}
+
+// AssistLogs def
+func (s *Service) AssistLogs(c context.Context, arg *model.ArgAssistLogs) (res []*model.Log, err error) {
+	err = s.client.Call(c, _assistLogs, arg, &res)
+	return
+}
+
+// AssistLogInfo def
+func (s *Service) AssistLogInfo(c context.Context, arg *model.ArgAssistLog) (res *model.Log, err error) {
+	res = new(model.Log)
+	err = s.client.Call(c, _assistLogInfo, arg, &res)
+	return
+}
+
+// AssistLogAdd def
+func (s *Service) AssistLogAdd(c context.Context, arg *model.ArgAssistLogAdd) (err error) {
+	err = s.client.Call(c, _assistLogAdd, arg, _noArg)
+	return
+}
+
+// AssistLogCancel def
+func (s *Service) AssistLogCancel(c context.Context, arg *model.ArgAssistLog) (err error) {
+	err = s.client.Call(c, _assistLogCancel, arg, _noArg)
+	return
+}
+
+// AssistUps def
+func (s *Service) AssistUps(c context.Context, arg *model.ArgAssistUps) (res *model.AssistUpsPager, err error) {
+	res = new(model.AssistUpsPager)
+	err = s.client.Call(c, _assistUps, arg, &res)
+	return
+}
+
+// AssistExit def
+func (s *Service) AssistExit(c context.Context, arg *model.ArgAssist) (err error) {
+	err = s.client.Call(c, _assistExit, arg, _noArg)
+	return
+}

app/service/main/assist/rpc/client/assist_test.go

@@ -0,0 +1,175 @@
+package client
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	model "go-common/app/service/main/assist/model/assist"
+
+	"github.com/davecgh/go-spew/spew"
+)
+
+const (
+	mid       = 27515256
+	assistMid = 27515255
+	realIP    = "127.0.0.1"
+	logID     = 692
+	subjectID = 111
+	objectID  = "222"
+	detail    = "testing"
+	pn        = 1
+	ps        = 20
+)
+
+func TestAssistRpcService(t *testing.T) {
+	s := New(nil)
+	time.Sleep(1 * time.Second)
+
+	// test assist relation
+	delAssist(t, s)
+	addAssist(t, s)
+	assist(t, s)
+	assists(t, s)
+	assistIDs(t, s)
+	assistUps(t, s)
+	assistExit(t, s)
+	// test assistlog
+	assistLogAdd(t, s)
+	assistLogInfo(t, s)
+	assistLogCancel(t, s)
+	assistLogs(t, s)
+}
+
+func delAssist(t *testing.T, s *Service) {
+	arg := &model.ArgAssist{
+		Mid:       mid,
+		AssistMid: assistMid,
+		RealIP:    realIP,
+	}
+	if err := s.DelAssist(context.TODO(), arg); err != nil {
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func addAssist(t *testing.T, s *Service) {
+	arg := &model.ArgAssist{
+		Mid:       mid,
+		AssistMid: assistMid,
+		RealIP:    realIP,
+	}
+	if err := s.AddAssist(context.TODO(), arg); err != nil {
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func assistLogAdd(t *testing.T, s *Service) {
+	arg := &model.ArgAssistLogAdd{
+		Mid:       mid,
+		AssistMid: assistMid,
+		Type:      model.TypeComment,
+		Action:    model.ActDelete,
+		SubjectID: subjectID,
+		ObjectID:  objectID,
+		Detail:    detail,
+		RealIP:    realIP,
+	}
+	if err := s.AssistLogAdd(context.TODO(), arg); err != nil {
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func assists(t *testing.T, s *Service) {
+	arg := &model.ArgAssists{
+		Mid:    mid,
+		RealIP: realIP,
+	}
+	if res, err := s.Assists(context.TODO(), arg); err != nil && res != nil {
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func assistIDs(t *testing.T, s *Service) {
+	arg := &model.ArgAssists{
+		Mid:    mid,
+		RealIP: realIP,
+	}
+	if res, err := s.AssistIDs(context.TODO(), arg); err != nil && res != nil {
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func assistUps(t *testing.T, s *Service) {
+	arg := &model.ArgAssistUps{
+		AssistMid: assistMid,
+		Ps:        20,
+		Pn:        1,
+		RealIP:    realIP,
+	}
+	if res, err := s.AssistUps(context.TODO(), arg); err != nil && res != nil {
+		spew.Dump(res)
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func assistExit(t *testing.T, s *Service) {
+	arg := &model.ArgAssist{
+		AssistMid: assistMid,
+		Mid:       mid,
+		RealIP:    realIP,
+	}
+	if err := s.AssistExit(context.TODO(), arg); err != nil {
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func assistLogInfo(t *testing.T, s *Service) {
+	arg := &model.ArgAssistLog{
+		Mid:       mid,
+		AssistMid: assistMid,
+		LogID:     logID,
+		RealIP:    realIP,
+	}
+	if res, err := s.AssistLogInfo(context.TODO(), arg); err != nil && res != nil {
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func assist(t *testing.T, s *Service) {
+	arg := &model.ArgAssist{
+		Mid:       mid,
+		AssistMid: assistMid,
+		RealIP:    realIP,
+	}
+	if res, err := s.Assist(context.TODO(), arg); err != nil && res != nil {
+		spew.Dump(res)
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func assistLogs(t *testing.T, s *Service) {
+	arg := &model.ArgAssistLogs{
+		Mid:       mid,
+		AssistMid: assistMid,
+		Stime:     time.Unix(time.Now().Unix(), 0),
+		Etime:     time.Unix(time.Now().Add(48*time.Hour).Unix(), 0),
+		Pn:        ps,
+		Ps:        pn,
+		RealIP:    realIP,
+	}
+	if res, err := s.AssistLogs(context.TODO(), arg); err != nil && res != nil {
+		t.Logf("call error(%v)", err)
+	}
+}
+
+func assistLogCancel(t *testing.T, s *Service) {
+	arg := &model.ArgAssistLog{
+		Mid:       mid,
+		AssistMid: assistMid,
+		LogID:     logID,
+		RealIP:    realIP,
+	}
+	if err := s.AssistLogCancel(context.TODO(), arg); err != nil {
+		t.Logf("call error(%v)", err)
+	}
+}