|
@@ -89,21 +89,29 @@ public class MessageController {
|
|
|
@ApiOperation(value="未读消息数量")
|
|
|
@GetMapping(value = "unReadCount")
|
|
|
public MessageResult<Long> unReadCount(HttpServletRequest request){
|
|
|
+
|
|
|
AttributePrincipal principal = (AttributePrincipal) request.getUserPrincipal();
|
|
|
- Company company = companyService.findByCreateBy(principal.getName());
|
|
|
MessageResult<Long> msgResult = new MessageResult<>();
|
|
|
|
|
|
- Map<String,Object> searchParams = new HashMap<>();
|
|
|
- searchParams.put("recipientId",company.getId());
|
|
|
+ Company company = companyService.findByCreateBy(principal.getName());
|
|
|
|
|
|
- List<Sort> sortList = new ArrayList<>();
|
|
|
- sortList.add(new Sort("create_time","desc"));
|
|
|
- searchParams.put("status",false);
|
|
|
+ if(company!=null) {
|
|
|
+ Map<String, Object> searchParams = new HashMap<>();
|
|
|
+ searchParams.put("recipientId", company.getId());
|
|
|
+
|
|
|
+ List<Sort> sortList = new ArrayList<>();
|
|
|
+ sortList.add(new Sort("create_time", "desc"));
|
|
|
+ searchParams.put("status", false);
|
|
|
|
|
|
- Page<Message> page = messageService.pageSearch(searchParams,1,1,sortList);
|
|
|
+ Page<Message> page = messageService.pageSearch(searchParams, 1, 1, sortList);
|
|
|
+
|
|
|
+ msgResult.setData(page.getTotal());
|
|
|
+ }
|
|
|
+ else{
|
|
|
+ msgResult.setData(0L);
|
|
|
+ }
|
|
|
|
|
|
msgResult.setResult(true);
|
|
|
- msgResult.setData(page.getTotal());
|
|
|
|
|
|
return msgResult;
|
|
|
}
|