From 467d44adc6f44339566d2bc26c0e1d6b95d2a71b Mon Sep 17 00:00:00 2001 From: skiffer-git <44203734@qq.com> Date: Tue, 31 Jan 2023 10:03:10 +0800 Subject: [PATCH] Error code standardization --- internal/common/check/access.go | 14 ++++++++++++++ internal/common/check/{check_user.go => user.go} | 0 internal/rpc/friend/friend.go | 2 +- 3 files changed, 15 insertions(+), 1 deletion(-) create mode 100644 internal/common/check/access.go rename internal/common/check/{check_user.go => user.go} (100%) diff --git a/internal/common/check/access.go b/internal/common/check/access.go new file mode 100644 index 000000000..fa6e15ee5 --- /dev/null +++ b/internal/common/check/access.go @@ -0,0 +1,14 @@ +package check + +import ( + "Open_IM/pkg/common/token_verify" + "context" +) + +func Access(ctx context.Context, ownerUserID string) (err error) { + _, err = GetUsersInfo(ctx, ownerUserID) + if err != nil { + return err + } + return token_verify.CheckAccessV3(ctx, ownerUserID) +} diff --git a/internal/common/check/check_user.go b/internal/common/check/user.go similarity index 100% rename from internal/common/check/check_user.go rename to internal/common/check/user.go diff --git a/internal/rpc/friend/friend.go b/internal/rpc/friend/friend.go index 5f75fb29a..90d3b9b8c 100644 --- a/internal/rpc/friend/friend.go +++ b/internal/rpc/friend/friend.go @@ -162,7 +162,7 @@ func (s *friendServer) ImportFriend(ctx context.Context, req *pbFriend.ImportFri if err := token_verify.CheckAdmin(ctx); err != nil { return nil, err } - if _, err := GetUsersInfo(ctx, req.OwnerUserID); err != nil { + if _, err := check.GetUsersInfo(ctx, req.OwnerUserID); err != nil { return nil, err }