fix get implementation user returned 404 always

SwimResults · Dec 5, 2024 · 74e8a67 · 74e8a67
1 parent 98d044a
commit 74e8a67
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/controller/notification_user_controller.go b/controller/notification_user_controller.go
@@ -57,7 +57,7 @@ func getNotificationUser(c *gin.Context) {
 		return
 	}
 
-	notificationUser, err2 := service.GetNotificationUserById(user.Identifier)
+	notificationUser, err2 := service.GetNotificationUserByUserId(user.Identifier)
 	if err2 != nil {
 		c.IndentedJSON(http.StatusNotFound, gin.H{"message": err2.Error()})
 		return

diff --git a/service/notification_user_service.go b/service/notification_user_service.go
@@ -62,6 +62,10 @@ func GetNotificationUserById(id primitive.ObjectID) (model.NotificationUser, err
 	return getNotificationUserByBsonDocument(bson.D{{"_id", id}})
 }
 
+func GetNotificationUserByUserId(id primitive.ObjectID) (model.NotificationUser, error) {
+	return getNotificationUserByBsonDocument(bson.D{{"user_id", id}})
+}
+
 func GetNotificationUserByToken(token string) (model.NotificationUser, error) {
 	return getNotificationUserByBsonDocument(bson.D{{"token", token}})
 }