mark message as read

src/main/java/application/controllers/UserController.java

@@ -103,6 +103,22 @@ public class UserController {
         return loginResponse.toString();
     }
 
+    @RequestMapping(value = "/markMessageAsRead/{userId}/{messageId}",  method = RequestMethod.POST)
+    @ResponseBody
+    public String markMessageAsRead(@PathVariable("userId") long userId, @PathVariable("messageId") long messageId) throws JSONException{
+        User user = userRepository.findOne(userId);
+        Set<Message> messages = user.getMessages();
+
+        for(Message message : messages){
+            if(message.getMessageId() == messageId){
+                message.setRead(true);
+            }
+        }
+
+        userRepository.save(user);
+        return (new JSONObject()).put("success", true).toString();
+    }
+
     @RequestMapping("/addMe")
     @ResponseBody
     public String addMe() {