Create & Init Project...

app/service/main/feed/rpc/server/BUILD

@@ -0,0 +1,48 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_test",
+    "go_library",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = ["rpc_test.go"],
+    embed = [":go_default_library"],
+    rundir = ".",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/openplatform/article/model:go_default_library",
+        "//app/service/main/feed/model:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["rpc.go"],
+    importpath = "go-common/app/service/main/feed/rpc/server",
+    tags = ["automanaged"],
+    deps = [
+        "//app/interface/openplatform/article/model:go_default_library",
+        "//app/service/main/feed/conf:go_default_library",
+        "//app/service/main/feed/model:go_default_library",
+        "//app/service/main/feed/service:go_default_library",
+        "//library/net/rpc:go_default_library",
+        "//library/net/rpc/context: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/feed/rpc/server/rpc.go

@@ -0,0 +1,109 @@
+package server
+
+import (
+	artmdl "go-common/app/interface/openplatform/article/model"
+	"go-common/app/service/main/feed/conf"
+	feedmdl "go-common/app/service/main/feed/model"
+	"go-common/app/service/main/feed/service"
+	"go-common/library/net/rpc"
+	"go-common/library/net/rpc/context"
+)
+
+// RPC struct info
+type RPC struct {
+	s *service.Service
+}
+
+// New new rpc server.
+func New(c *conf.Config, s *service.Service) (svr *rpc.Server) {
+	r := &RPC{s: s}
+	svr = rpc.NewServer(c.RPCServer)
+	if err := svr.Register(r); err != nil {
+		panic(err)
+	}
+	return
+}
+
+// Ping check connection success.
+func (r *RPC) Ping(c context.Context, arg *struct{}, res *struct{}) (err error) {
+	return
+}
+
+// AddArc add archive when archive passed. purge cache.
+func (r *RPC) AddArc(c context.Context, arg *feedmdl.ArgArc, res *struct{}) (err error) {
+	err = r.s.AddArc(c, arg.Mid, arg.Aid, arg.PubDate, arg.RealIP)
+	return
+}
+
+// DelArc delete archive when archive not passed. purge cache.
+func (r *RPC) DelArc(c context.Context, arg *feedmdl.ArgAidMid, res *struct{}) (err error) {
+	err = r.s.DelArc(c, arg.Mid, arg.Aid, arg.RealIP)
+	return
+}
+
+// PurgeFeedCache purge cache when attention/unattention upper
+func (r *RPC) PurgeFeedCache(c context.Context, arg *feedmdl.ArgMid, res *struct{}) (err error) {
+	err = r.s.PurgeFeedCache(c, arg.Mid, arg.RealIP)
+	return
+}
+
+// AppFeed receive ArgMid contains mid and real ip, then return app feed.
+func (r *RPC) AppFeed(c context.Context, arg *feedmdl.ArgFeed, res *[]*feedmdl.Feed) (err error) {
+	*res, err = r.s.Feed(c, true, arg.Mid, arg.Pn, arg.Ps, arg.RealIP)
+	return
+}
+
+// WebFeed receive ArgMid contains mid and real ip, then return app feed.
+func (r *RPC) WebFeed(c context.Context, arg *feedmdl.ArgFeed, res *[]*feedmdl.Feed) (err error) {
+	*res, err = r.s.Feed(c, false, arg.Mid, arg.Pn, arg.Ps, arg.RealIP)
+	return
+}
+
+// Fold receive ArgFold contains mid, then return upper's fold archives.
+func (r *RPC) Fold(c context.Context, arg *feedmdl.ArgFold, res *[]*feedmdl.Feed) (err error) {
+	*res, err = r.s.Fold(c, arg.Mid, arg.Aid, arg.RealIP)
+	return
+}
+
+// AppUnreadCount receive ArgUnreadCount contains mid, then return unread count.
+func (r *RPC) AppUnreadCount(c context.Context, arg *feedmdl.ArgUnreadCount, res *int) (err error) {
+	*res, err = r.s.UnreadCount(c, true, arg.WithoutBangumi, arg.Mid, arg.RealIP)
+	return
+}
+
+// WebUnreadCount receive ArgUnreadCount contains mid, then return unread count.
+func (r *RPC) WebUnreadCount(c context.Context, arg *feedmdl.ArgMid, res *int) (err error) {
+	withoutBangumi := false
+	*res, err = r.s.UnreadCount(c, false, withoutBangumi, arg.Mid, arg.RealIP)
+	return
+}
+
+// ArchiveFeed receive ArgFeed contains mid and real ip
+func (r *RPC) ArchiveFeed(c context.Context, arg *feedmdl.ArgFeed, res *[]*feedmdl.Feed) (err error) {
+	*res, err = r.s.ArchiveFeed(c, arg.Mid, arg.Pn, arg.Ps, arg.RealIP)
+	return
+}
+
+// BangumiFeed receive ArgFeed contains mid and real ip
+func (r *RPC) BangumiFeed(c context.Context, arg *feedmdl.ArgFeed, res *[]*feedmdl.Feed) (err error) {
+	*res, err = r.s.BangumiFeed(c, arg.Mid, arg.Pn, arg.Ps, arg.RealIP)
+	return
+}
+
+// ChangeArcUpper refresh feed cache when change archive's author
+func (r *RPC) ChangeArcUpper(c context.Context, arg *feedmdl.ArgChangeUpper, res *struct{}) (err error) {
+	err = r.s.ChangeAuthor(c, arg.Aid, arg.OldMid, arg.NewMid, arg.RealIP)
+	return
+}
+
+// ArticleFeed receive ArgFeed contains mid and real ip
+func (r *RPC) ArticleFeed(c context.Context, arg *feedmdl.ArgFeed, res *[]*artmdl.Meta) (err error) {
+	*res, err = r.s.ArticleFeed(c, arg.Mid, arg.Pn, arg.Ps, arg.RealIP)
+	return
+}
+
+// ArticleUnreadCount receive ArgUnreadCount contains mid, then return unread count.
+func (r *RPC) ArticleUnreadCount(c context.Context, arg *feedmdl.ArgMid, res *int) (err error) {
+	*res, err = r.s.ArticleUnreadCount(c, arg.Mid, arg.RealIP)
+	return
+}

app/service/main/feed/rpc/server/rpc_test.go

@@ -0,0 +1,42 @@
+package server
+
+import (
+	artmdl "go-common/app/interface/openplatform/article/model"
+	feed "go-common/app/service/main/feed/model"
+	"net/rpc"
+	"testing"
+)
+
+const (
+	addr = "172.16.33.57:6361"
+
+	_testArticleFeed = "RPC.ArticleFeed"
+)
+
+func TestFeedRpc(t *testing.T) {
+	client, err := rpc.Dial("tcp", addr)
+	defer client.Close()
+	if err != nil {
+		t.Errorf("rpc.Dial(tcp, \"%s\") error(%v)", addr, err)
+		t.FailNow()
+	}
+	feedRPC(client, t)
+}
+
+func feedRPC(client *rpc.Client, t *testing.T) {
+	arg := &feed.ArgFeed{}
+	arg.Mid = 88888929
+	res := &[]*artmdl.Meta{}
+	if err := client.Call(_testArticleFeed, arg, &res); err != nil {
+		t.Errorf("client.Call(%s) error(%v)", _testArticleFeed, err)
+		t.FailNow()
+	} else {
+		result("article", t, res)
+	}
+}
+
+func result(name string, t *testing.T, res interface{}) {
+	t.Log("[==========" + name + "单元测试结果==========]")
+	t.Log(res)
+	t.Log("[↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑↑]\r\n")
+}