check userId exists and userId is type uuid

Monify-Dev · Jun 29, 2024 · 304a98a · 304a98a
1 parent 3807478
commit 304a98a
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 9 deletions.
diff --git a/services/api/controllers/friend/accept_invitation.go b/services/api/controllers/friend/accept_invitation.go
@@ -14,8 +14,8 @@ import (
 
 func (s Service) AcceptInvitation(ctx context.Context, req *monify.AcceptInvitationRequest) (*monify.FriendEmpty, error) {
 	logger := ctx.Value(lib.LoggerContextKey{}).(*zap.Logger)
-	userId := ctx.Value(lib.UserIdContextKey{}).(uuid.UUID)
-	if userId == uuid.Nil {
+	_, ok := ctx.Value(lib.UserIdContextKey{}).(uuid.UUID)
+	if !ok {
 		return nil, status.Error(codes.Unauthenticated, "Unauthorized.")
 	}
 

diff --git a/services/api/controllers/friend/friend_invite.go b/services/api/controllers/friend/friend_invite.go
@@ -16,7 +16,8 @@ func (s Service) InviteFriend(ctx context.Context, req *monify.InviteFriendReque
 
 	logger := ctx.Value(lib.LoggerContextKey{}).(*zap.Logger)
 	userId := ctx.Value(lib.UserIdContextKey{}).(uuid.UUID)
-	if userId == uuid.Nil {
+	userId, ok := ctx.Value(lib.UserIdContextKey{}).(uuid.UUID)
+	if !ok {
 		return nil, status.Error(codes.Unauthenticated, "Unauthorized.")
 	}
 	receiver_nickId := req.GetReceiverNickId()

diff --git a/services/api/controllers/friend/list_friend.go b/services/api/controllers/friend/list_friend.go
@@ -14,8 +14,8 @@ import (
 
 func (s Service) ListFriend(ctx context.Context, req *monify.FriendEmpty) (*monify.ListFriendResponse, error) {
 	logger := ctx.Value(lib.LoggerContextKey{}).(*zap.Logger)
-	userId := ctx.Value(lib.UserIdContextKey{})
-	if userId == nil {
+	userId, ok := ctx.Value(lib.UserIdContextKey{}).(uuid.UUID)
+	if !ok {
 		return nil, status.Error(codes.Unauthenticated, "Unauthorized.")
 	}
 	db := ctx.Value(lib.DatabaseContextKey{}).(*sql.DB)

diff --git a/services/api/controllers/friend/list_friend_invitation.go b/services/api/controllers/friend/list_friend_invitation.go
@@ -6,15 +6,16 @@ import (
 	"monify/lib"
 	monify "monify/protobuf/gen/go"
 
+	"github.com/google/uuid"
 	"go.uber.org/zap"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/status"
 )
 
 func (s Service) ListFriendInvitation(ctx context.Context, req *monify.FriendEmpty) (*monify.ListFriendInvitationResponse, error) {
 	logger := ctx.Value(lib.LoggerContextKey{}).(*zap.Logger)
-	userId := ctx.Value(lib.UserIdContextKey{})
-	if userId == nil {
+	userId, ok := ctx.Value(lib.UserIdContextKey{}).(uuid.UUID)
+	if !ok {
 		return nil, status.Error(codes.Unauthenticated, "Unauthorized.")
 	}
 	db := ctx.Value(lib.DatabaseContextKey{}).(*sql.DB)

diff --git a/services/api/controllers/friend/reject_invitation.go b/services/api/controllers/friend/reject_invitation.go
@@ -14,8 +14,8 @@ import (
 
 func (s Service) RejectInvitation(ctx context.Context, req *monify.RejectInvitationRequest) (*monify.FriendEmpty, error) {
 	logger := ctx.Value(lib.LoggerContextKey{}).(*zap.Logger)
-	userId := ctx.Value(lib.UserIdContextKey{}).(uuid.UUID)
-	if userId == uuid.Nil {
+	_, ok := ctx.Value(lib.UserIdContextKey{}).(uuid.UUID)
+	if !ok {
 		return nil, status.Error(codes.Unauthenticated, "Unauthorized.")
 	}
 	db := ctx.Value(lib.DatabaseContextKey{}).(*sql.DB)