diff --git a/src/main/java/application/controllers/UserController.java b/src/main/java/application/controllers/UserController.java index fd0952e..60349ae 100644 --- a/src/main/java/application/controllers/UserController.java +++ b/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 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() {