Create & Init Project...

app/service/main/relation/rpc/client/BUILD

@@ -0,0 +1,32 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["relation.go"],
+    importpath = "go-common/app/service/main/relation/rpc/client",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/service/main/relation/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/relation/rpc/client/relation.go

@@ -0,0 +1,299 @@
+package relation
+
+import (
+	"context"
+
+	"go-common/app/service/main/relation/model"
+	"go-common/library/net/rpc"
+)
+
+const (
+	// relation
+	_relation       = "RPC.Relation"
+	_relations      = "RPC.Relations"
+	_stat           = "RPC.Stat"
+	_stats          = "RPC.Stats"
+	_followings     = "RPC.Followings"
+	_sameFollowings = "RPC.SameFollowings"
+	_whispers       = "RPC.Whispers"
+	_blacks         = "RPC.Blacks"
+	_followers      = "RPC.Followers"
+	_addFollowing   = "RPC.AddFollowing"
+	_delFollowing   = "RPC.DelFollowing"
+	_addWhisper     = "RPC.AddWhisper"
+	_delWhisper     = "RPC.DelWhisper"
+	_addBlack       = "RPC.AddBlack"
+	_delBlack       = "RPC.DelBlack"
+	_delFollower    = "RPC.DelFollower"
+
+	// relation tag
+	_tag           = "RPC.Tag"
+	_tags          = "RPC.Tags"
+	_userTag       = "RPC.UserTag"
+	_createTag     = "RPC.CreateTag"
+	_updateTag     = "RPC.UpdateTag"
+	_delTag        = "RPC.DelTag"
+	_tagsAddUsers  = "RPC.TagsAddUsers"
+	_tagsCopyUsers = "RPC.TagsCopyUsers"
+	_tagsMoveUsers = "RPC.TagsMoveUsers"
+	_AddSpecial    = "RPC.AddSpecial"
+	_DelSpecial    = "RPC.DelSpecial"
+	_Special       = "RPC.Special"
+
+	// prompt
+	_prompt      = "RPC.Prompt"
+	_closePrompt = "RPC.ClosePrompt"
+
+	// followers incr notify
+	_FollowersUnread           = "RPC.FollowersUnread"
+	_FollowersUnreadCount      = "RPC.FollowersUnreadCount"
+	_ResetFollowersUnread      = "RPC.ResetFollowersUnread"
+	_ResetFollowersUnreadCount = "RPC.ResetFollowersUnreadCount"
+	_DisableFollowerNotify     = "RPC.DisableFollowerNotify"
+	_EnableFollowerNotify      = "RPC.EnableFollowerNotify"
+	_FollowerNotifySetting     = "RPC.FollowerNotifySetting"
+
+	// achieve
+	_AchieveGet = "RPC.AchieveGet"
+	_Achieve    = "RPC.Achieve"
+)
+
+var (
+	_noRes = &struct{}{}
+)
+
+const (
+	_appid = "account.service.relation"
+)
+
+// Service struct info.
+type Service struct {
+	client *rpc.Client2
+}
+
+// New new service instance and return.
+func New(c *rpc.ClientConfig) (s *Service) {
+	s = &Service{}
+	s.client = rpc.NewDiscoveryCli(_appid, c)
+	return
+}
+
+// Relation get relation info.
+func (s *Service) Relation(c context.Context, arg *model.ArgRelation) (res *model.Following, err error) {
+	res = new(model.Following)
+	err = s.client.Call(c, _relation, arg, res)
+	return
+}
+
+// Relations get relation infos.
+func (s *Service) Relations(c context.Context, arg *model.ArgRelations) (res map[int64]*model.Following, err error) {
+	err = s.client.Call(c, _relations, arg, &res)
+	return
+}
+
+// Followings get followings infos.
+func (s *Service) Followings(c context.Context, arg *model.ArgMid) (res []*model.Following, err error) {
+	err = s.client.Call(c, _followings, arg, &res)
+	return
+}
+
+// Whispers get whispers infos.
+func (s *Service) Whispers(c context.Context, arg *model.ArgMid) (res []*model.Following, err error) {
+	err = s.client.Call(c, _whispers, arg, &res)
+	return
+}
+
+// Blacks get black list.
+func (s *Service) Blacks(c context.Context, arg *model.ArgMid) (res []*model.Following, err error) {
+	err = s.client.Call(c, _blacks, arg, &res)
+	return
+}
+
+// Followers get followers list.
+func (s *Service) Followers(c context.Context, arg *model.ArgMid) (res []*model.Following, err error) {
+	err = s.client.Call(c, _followers, arg, &res)
+	return
+}
+
+// Stat get user relation stat.
+func (s *Service) Stat(c context.Context, arg *model.ArgMid) (res *model.Stat, err error) {
+	res = new(model.Stat)
+	err = s.client.Call(c, _stat, arg, &res)
+	return
+}
+
+// Stats get users relation stat.
+func (s *Service) Stats(c context.Context, arg *model.ArgMids) (res map[int64]*model.Stat, err error) {
+	err = s.client.Call(c, _stats, arg, &res)
+	return
+}
+
+// ModifyRelation modify user relation.
+func (s *Service) ModifyRelation(c context.Context, arg *model.ArgFollowing) (err error) {
+	switch arg.Action {
+	case model.ActAddBlack:
+		err = s.client.Call(c, _addBlack, arg, _noRes)
+	case model.ActAddFollowing:
+		err = s.client.Call(c, _addFollowing, arg, _noRes)
+	case model.ActAddWhisper:
+		err = s.client.Call(c, _addWhisper, arg, _noRes)
+	case model.ActDelBalck:
+		err = s.client.Call(c, _delBlack, arg, _noRes)
+	case model.ActDelFollower:
+		err = s.client.Call(c, _delFollower, arg, _noRes)
+	case model.ActDelFollowing:
+		err = s.client.Call(c, _delFollowing, arg, _noRes)
+	case model.ActDelWhisper:
+		err = s.client.Call(c, _delWhisper, arg, _noRes)
+	default:
+
+	}
+	return
+}
+
+// Tag tag
+func (s *Service) Tag(c context.Context, arg *model.ArgTagId) (res []int64, err error) {
+	err = s.client.Call(c, _tag, arg, &res)
+	return
+}
+
+// Tags tags
+func (s *Service) Tags(c context.Context, arg *model.ArgMid) (res []*model.TagCount, err error) {
+	err = s.client.Call(c, _tags, arg, &res)
+	return
+}
+
+// UserTag user tag
+func (s *Service) UserTag(c context.Context, arg *model.ArgRelation) (res map[int64]string, err error) {
+	err = s.client.Call(c, _userTag, arg, &res)
+	return
+}
+
+// CreateTag create tag
+func (s *Service) CreateTag(c context.Context, arg *model.ArgTag) (res int64, err error) {
+	err = s.client.Call(c, _createTag, arg, &res)
+	return
+}
+
+// UpdateTag update tag
+func (s *Service) UpdateTag(c context.Context, arg *model.ArgTagUpdate) (err error) {
+	err = s.client.Call(c, _updateTag, arg, _noRes)
+	return
+}
+
+// DelTag del tag
+func (s *Service) DelTag(c context.Context, arg *model.ArgTagDel) (err error) {
+	err = s.client.Call(c, _delTag, arg, _noRes)
+	return
+}
+
+// TagsAddUsers tags add users
+func (s *Service) TagsAddUsers(c context.Context, arg *model.ArgTagsMoveUsers) (err error) {
+	err = s.client.Call(c, _tagsAddUsers, arg, _noRes)
+	return
+}
+
+// TagsCopyUsers tags copy users
+func (s *Service) TagsCopyUsers(c context.Context, arg *model.ArgTagsMoveUsers) (err error) {
+	err = s.client.Call(c, _tagsCopyUsers, arg, _noRes)
+	return
+}
+
+// TagsMoveUsers tags move users
+func (s *Service) TagsMoveUsers(c context.Context, arg *model.ArgTagsMoveUsers) (err error) {
+	err = s.client.Call(c, _tagsMoveUsers, arg, _noRes)
+	return
+}
+
+// Prompt rpc rpompt client
+func (s *Service) Prompt(c context.Context, arg *model.ArgPrompt) (b bool, err error) {
+	err = s.client.Call(c, _prompt, arg, &b)
+	return
+}
+
+// ClosePrompt close prompt client.
+func (s *Service) ClosePrompt(c context.Context, arg *model.ArgPrompt) (err error) {
+	err = s.client.Call(c, _closePrompt, arg, _noRes)
+	return
+}
+
+// AddSpecial add specail.
+func (s *Service) AddSpecial(c context.Context, arg *model.ArgFollowing) (err error) {
+	err = s.client.Call(c, _AddSpecial, arg, &_noRes)
+	return
+}
+
+// DelSpecial del special.
+func (s *Service) DelSpecial(c context.Context, arg *model.ArgFollowing) (err error) {
+	err = s.client.Call(c, _DelSpecial, arg, &_noRes)
+	return
+}
+
+// Special get special.
+func (s *Service) Special(c context.Context, arg *model.ArgMid) (res []int64, err error) {
+	err = s.client.Call(c, _Special, arg, &res)
+	return
+}
+
+// FollowersUnread check unread status, for the 'show red point' function.
+func (s *Service) FollowersUnread(c context.Context, arg *model.ArgMid) (show bool, err error) {
+	err = s.client.Call(c, _FollowersUnread, arg, &show)
+	return
+}
+
+// FollowersUnreadCount unread count.
+func (s *Service) FollowersUnreadCount(c context.Context, arg *model.ArgMid) (count int64, err error) {
+	err = s.client.Call(c, _FollowersUnreadCount, arg, &count)
+	return
+}
+
+// AchieveGet is
+func (s *Service) AchieveGet(c context.Context, arg *model.ArgAchieveGet) (*model.AchieveGetReply, error) {
+	reply := &model.AchieveGetReply{}
+	err := s.client.Call(c, _AchieveGet, arg, &reply)
+	return reply, err
+}
+
+// Achieve is
+func (s *Service) Achieve(c context.Context, arg *model.ArgAchieve) (*model.Achieve, error) {
+	reply := &model.Achieve{}
+	err := s.client.Call(c, _Achieve, arg, &reply)
+	return reply, err
+}
+
+// ResetFollowersUnread is
+func (s *Service) ResetFollowersUnread(c context.Context, arg *model.ArgMid) (err error) {
+	err = s.client.Call(c, _ResetFollowersUnread, arg, &_noRes)
+	return
+}
+
+// ResetFollowersUnreadCount is
+func (s *Service) ResetFollowersUnreadCount(c context.Context, arg *model.ArgMid) (err error) {
+	err = s.client.Call(c, _ResetFollowersUnreadCount, arg, &_noRes)
+	return
+}
+
+// DisableFollowerNotify set followerNotify as disabled.
+func (s *Service) DisableFollowerNotify(c context.Context, arg *model.ArgMid) (err error) {
+	err = s.client.Call(c, _DisableFollowerNotify, arg, &_noRes)
+	return
+}
+
+// EnableFollowerNotify set followerNotify as disabled.
+func (s *Service) EnableFollowerNotify(c context.Context, arg *model.ArgMid) (err error) {
+	err = s.client.Call(c, _EnableFollowerNotify, arg, &_noRes)
+	return
+}
+
+// FollowerNotifySetting get followerNotify setting.
+func (s *Service) FollowerNotifySetting(c context.Context, arg *model.ArgMid) (followerNotify *model.FollowerNotifySetting, err error) {
+	followerNotify = &model.FollowerNotifySetting{}
+	err = s.client.Call(c, _FollowerNotifySetting, arg, followerNotify)
+	return
+}
+
+// SameFollowings is
+func (s *Service) SameFollowings(c context.Context, arg *model.ArgSameFollowing) (res []*model.Following, err error) {
+	err = s.client.Call(c, _sameFollowings, arg, &res)
+	return
+}

app/service/main/relation/rpc/server/BUILD

@@ -0,0 +1,35 @@
+package(default_visibility = ["//visibility:public"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = ["rpc.go"],
+    importpath = "go-common/app/service/main/relation/rpc/server",
+    tags = ["automanaged"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//app/service/main/relation/conf:go_default_library",
+        "//app/service/main/relation/model:go_default_library",
+        "//app/service/main/relation/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/relation/rpc/server/rpc.go

@@ -0,0 +1,288 @@
+package rpc
+
+import (
+	"go-common/app/service/main/relation/conf"
+	"go-common/app/service/main/relation/model"
+	"go-common/app/service/main/relation/service"
+	"go-common/library/net/rpc"
+	"go-common/library/net/rpc/context"
+)
+
+// RPC rpc
+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
+}
+
+// Relation relation
+func (r *RPC) Relation(c context.Context, a *model.ArgRelation, res *model.Following) (err error) {
+	var f *model.Following
+	if f, err = r.s.Relation(c, a.Mid, a.Fid); err == nil && f != nil {
+		*res = *f
+	}
+	return
+}
+
+// Relations relations
+func (r *RPC) Relations(c context.Context, a *model.ArgRelations, res *map[int64]*model.Following) (err error) {
+	*res, err = r.s.Relations(c, a.Mid, a.Fids)
+	return
+}
+
+// Stat stat
+func (r *RPC) Stat(c context.Context, a *model.ArgMid, res *model.Stat) (err error) {
+	var st *model.Stat
+	if st, err = r.s.Stat(c, a.Mid); err == nil && st != nil {
+		*res = *st
+	}
+	return
+}
+
+// Stats stats
+func (r *RPC) Stats(c context.Context, a *model.ArgMids, res *map[int64]*model.Stat) (err error) {
+	*res, err = r.s.Stats(c, a.Mids)
+	return
+}
+
+// Attentions attentions
+func (r *RPC) Attentions(c context.Context, a *model.ArgMid, res *[]*model.Following) (err error) {
+	*res, err = r.s.Attentions(c, a.Mid)
+	return
+}
+
+// Followings followings
+func (r *RPC) Followings(c context.Context, a *model.ArgMid, res *[]*model.Following) (err error) {
+	*res, err = r.s.Followings(c, a.Mid)
+	return
+}
+
+// AddFollowing add following
+func (r *RPC) AddFollowing(c context.Context, a *model.ArgFollowing, res *struct{}) (err error) {
+	err = r.s.AddFollowing(c, a.Mid, a.Fid, a.Source, a.Infoc)
+	return
+}
+
+// DelFollowing del following
+func (r *RPC) DelFollowing(c context.Context, a *model.ArgFollowing, res *struct{}) (err error) {
+	err = r.s.DelFollowing(c, a.Mid, a.Fid, a.Source, a.Infoc)
+	return
+}
+
+// Whispers whispers
+func (r *RPC) Whispers(c context.Context, a *model.ArgMid, res *[]*model.Following) (err error) {
+	*res, err = r.s.Whispers(c, a.Mid)
+	return
+}
+
+// AddWhisper add whisper
+func (r *RPC) AddWhisper(c context.Context, a *model.ArgFollowing, res *struct{}) (err error) {
+	err = r.s.AddWhisper(c, a.Mid, a.Fid, a.Source, a.Infoc)
+	return
+}
+
+// DelWhisper del whisper
+func (r *RPC) DelWhisper(c context.Context, a *model.ArgFollowing, res *struct{}) (err error) {
+	err = r.s.DelWhisper(c, a.Mid, a.Fid, a.Source, a.Infoc)
+	return
+}
+
+// Blacks blacks
+func (r *RPC) Blacks(c context.Context, a *model.ArgMid, res *[]*model.Following) (err error) {
+	*res, err = r.s.Blacks(c, a.Mid)
+	return
+}
+
+// AddBlack add black
+func (r *RPC) AddBlack(c context.Context, a *model.ArgFollowing, res *struct{}) (err error) {
+	err = r.s.AddBlack(c, a.Mid, a.Fid, a.Source, a.Infoc)
+	return
+}
+
+// DelBlack del black
+func (r *RPC) DelBlack(c context.Context, a *model.ArgFollowing, res *struct{}) (err error) {
+	err = r.s.DelBlack(c, a.Mid, a.Fid, a.Source, a.Infoc)
+	return
+}
+
+// Followers followers
+func (r *RPC) Followers(c context.Context, a *model.ArgMid, res *[]*model.Following) (err error) {
+	*res, err = r.s.Followers(c, a.Mid)
+	return
+}
+
+// DelFollower del Follower
+func (r *RPC) DelFollower(c context.Context, a *model.ArgFollowing, res *struct{}) (err error) {
+	err = r.s.DelFollowing(c, a.Fid, a.Mid, a.Source, a.Infoc)
+	return
+}
+
+// Tag tag
+func (r *RPC) Tag(c context.Context, a *model.ArgTagId, res *[]int64) (err error) {
+	*res, err = r.s.Tag(c, a.Mid, a.TagId, a.RealIP)
+	return
+}
+
+// Tags tags
+func (r *RPC) Tags(c context.Context, a *model.ArgMid, res *[]*model.TagCount) (err error) {
+	*res, err = r.s.Tags(c, a.Mid, a.RealIP)
+	return
+}
+
+// UserTag user tag
+func (r *RPC) UserTag(c context.Context, a *model.ArgRelation, res *map[int64]string) (err error) {
+	*res, err = r.s.UserTag(c, a.Mid, a.Fid, a.RealIP)
+	return
+}
+
+// CreateTag create tag
+func (r *RPC) CreateTag(c context.Context, a *model.ArgTag, res *int64) (err error) {
+	*res, err = r.s.CreateTag(c, a.Mid, a.Tag, a.RealIP)
+	return
+}
+
+// UpdateTag update tag
+func (r *RPC) UpdateTag(c context.Context, a *model.ArgTagUpdate, res *struct{}) (err error) {
+	err = r.s.UpdateTag(c, a.Mid, a.TagId, a.New, a.RealIP)
+	return
+}
+
+// DelTag del tag
+func (r *RPC) DelTag(c context.Context, a *model.ArgTagDel, res *struct{}) (err error) {
+	err = r.s.DelTag(c, a.Mid, a.TagId, a.RealIP)
+	return
+}
+
+// TagsAddUsers tags add users
+func (r *RPC) TagsAddUsers(c context.Context, a *model.ArgTagsMoveUsers, res *struct{}) (err error) {
+	err = r.s.TagsAddUsers(c, a.Mid, a.AfterTagIds, a.Fids, a.RealIP)
+	return
+}
+
+// TagsCopyUsers tags copy users
+func (r *RPC) TagsCopyUsers(c context.Context, a *model.ArgTagsMoveUsers, res *struct{}) (err error) {
+	err = r.s.TagsMoveUsers(c, a.Mid, a.BeforeID, a.AfterTagIds, a.Fids, a.RealIP)
+	return
+}
+
+// TagsMoveUsers tags move users
+func (r *RPC) TagsMoveUsers(c context.Context, a *model.ArgTagsMoveUsers, res *struct{}) (err error) {
+	err = r.s.TagsMoveUsers(c, a.Mid, a.BeforeID, a.AfterTagIds, a.Fids, a.RealIP)
+	return
+}
+
+// Prompt rpc prompt.
+func (r *RPC) Prompt(c context.Context, m *model.ArgPrompt, res *bool) (err error) {
+	*res, err = r.s.Prompt(c, m)
+	return
+}
+
+// ClosePrompt close prompt.
+func (r *RPC) ClosePrompt(c context.Context, m *model.ArgPrompt, res *struct{}) (err error) {
+	return r.s.ClosePrompt(c, m)
+}
+
+// AddSpecial add user to special.
+func (r *RPC) AddSpecial(c context.Context, m *model.ArgFollowing, res *struct{}) (err error) {
+	return r.s.AddSpecial(c, m.Mid, m.Fid)
+}
+
+// DelSpecial del user from sepcial.
+func (r *RPC) DelSpecial(c context.Context, m *model.ArgFollowing, res *struct{}) (err error) {
+	return r.s.DelSpecial(c, m.Mid, m.Fid)
+}
+
+// Special get user specail list.
+func (r *RPC) Special(c context.Context, m *model.ArgMid, res *[]int64) (err error) {
+	*res, err = r.s.Special(c, m.Mid)
+	return
+}
+
+// FollowersUnread is
+func (r *RPC) FollowersUnread(c context.Context, arg *model.ArgMid, res *bool) (err error) {
+	*res, err = r.s.Unread(c, arg.Mid)
+	return
+}
+
+// FollowersUnreadCount is
+func (r *RPC) FollowersUnreadCount(c context.Context, arg *model.ArgMid, res *int64) (err error) {
+	*res, err = r.s.UnreadCount(c, arg.Mid)
+	return
+}
+
+// AchieveGet is
+func (r *RPC) AchieveGet(c context.Context, arg *model.ArgAchieveGet, res *model.AchieveGetReply) error {
+	reply, err := r.s.AchieveGet(c, arg)
+	if err != nil {
+		return err
+	}
+	*res = *reply
+	return nil
+}
+
+// Achieve is
+func (r *RPC) Achieve(c context.Context, arg *model.ArgAchieve, res *model.Achieve) error {
+	reply, err := r.s.Achieve(c, arg)
+	if err != nil {
+		return err
+	}
+	*res = *reply
+	return nil
+}
+
+// ResetFollowersUnread is
+func (r *RPC) ResetFollowersUnread(c context.Context, arg *model.ArgMid, res *struct{}) (err error) {
+	err = r.s.ResetUnread(c, arg.Mid)
+	return
+}
+
+// ResetFollowersUnreadCount is
+func (r *RPC) ResetFollowersUnreadCount(c context.Context, arg *model.ArgMid, res *struct{}) (err error) {
+	err = r.s.ResetUnreadCount(c, arg.Mid)
+	return
+}
+
+// DisableFollowerNotify set followerNotify as disabled.
+func (r *RPC) DisableFollowerNotify(c context.Context, arg *model.ArgMid, res *struct{}) (err error) {
+	err = r.s.DisableFollowerNotify(c, arg)
+	return
+}
+
+// EnableFollowerNotify set followerNotify as enabled.
+func (r *RPC) EnableFollowerNotify(c context.Context, arg *model.ArgMid, res *struct{}) (err error) {
+	err = r.s.EnableFollowerNotify(c, arg)
+	return
+}
+
+// FollowerNotifySetting get member follower notify setting
+func (r *RPC) FollowerNotifySetting(c context.Context, arg *model.ArgMid, res *model.FollowerNotifySetting) (err error) {
+	rely, err := r.s.FollowerNotifySetting(c, arg)
+	if err != nil {
+		return
+	}
+	*res = *rely
+	return
+}
+
+// SameFollowings is
+func (r *RPC) SameFollowings(c context.Context, arg *model.ArgSameFollowing, res *[]*model.Following) error {
+	reply, err := r.s.SameFollowings(c, arg)
+	if err != nil {
+		return err
+	}
+	*res = reply
+	return nil
+}