单聊群聊功能提交

This commit is contained in:
zhongzb
2023-07-30 01:20:40 +08:00
parent 05cef16bef
commit 6dd2a27df7
4 changed files with 13 additions and 5 deletions

View File

@@ -25,5 +25,5 @@ public interface ContactService {
Integer getMsgUnReadCount(Message message);
Map<Long, MsgReadInfoDTO> getMsgReadInfo(List<Message> messages);
Map<Long, MsgReadInfoDTO> getMsgReadInfo(List<Message> messages, boolean needUnread);
}

View File

@@ -51,17 +51,23 @@ public class ContactServiceImpl implements ContactService {
}
@Override
public Map<Long, MsgReadInfoDTO> getMsgReadInfo(List<Message> messages) {
public Map<Long, MsgReadInfoDTO> getMsgReadInfo(List<Message> messages, boolean needUnread) {
Map<Long, List<Message>> roomGroup = messages.stream().collect(Collectors.groupingBy(Message::getRoomId));
AssertUtil.equal(roomGroup.size(), 1, "只能查相同房间下的消息");
Long roomId = roomGroup.keySet().iterator().next();
Integer totalCount = contactDao.getTotalCount(roomId);
Integer totalCount = null;
if (needUnread) {
totalCount = contactDao.getTotalCount(roomId);
}
Integer finalTotalCount = totalCount;
return messages.stream().map(message -> {
MsgReadInfoDTO readInfoDTO = new MsgReadInfoDTO();
readInfoDTO.setMsgId(message.getId());
Integer readCount = contactDao.getReadCount(message);
readInfoDTO.setReadCount(readCount);
readInfoDTO.setUnReadCount(totalCount - readCount);
if (needUnread) {
readInfoDTO.setUnReadCount(finalTotalCount - readCount);
}
return readInfoDTO;
}).collect(Collectors.toMap(MsgReadInfoDTO::getMsgId, Function.identity()));
}