Code optimization

reginald-chand · Jun 28, 2024 · 677ea51 · 677ea51
1 parent 6c87c07
commit 677ea51
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/server/controllers/comment/comment.controller.mjs b/server/controllers/comment/comment.controller.mjs
@@ -30,6 +30,14 @@ export const commentController = async (request, response) => {
       return response.status(404).json({ responseMessage: "Post not found." });
     }
 
+    const existingPostIds = new Set(
+      existingPosts.map((post) => post._id.toString())
+    );
+
+    if (!existingPostIds.has(postId.toString())) {
+      return response.status(404).json({ responseMessage: "Post not found." });
+    }
+
     for (const post of existingPosts) {
       const existingCommentDocument = await CommentModel.findOne({
         _id: { $eq: post._id },