lizixuan
/
m2pool_web_backend
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update 修改消息存储规则

This commit is contained in:
yyb 2025-04-21 10:15:16 +08:00
parent de4788319b
commit 1d6fc57d27
1 changed files with 24 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
m2pool-modules/m2pool-chat/src/main/java/com/m2pool/chat/service/impl/StompServiceImpl.java
View File

@ -14,6 +14,9 @@ import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.messaging.simp.SimpMessagingTemplate; import org.springframework.messaging.simp.SimpMessagingTemplate;
import org.springframework.messaging.simp.user.SimpUserRegistry; import org.springframework.messaging.simp.user.SimpUserRegistry;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.TransactionCallbackWithoutResult;
import org.springframework.transaction.support.TransactionTemplate;
import javax.annotation.Resource; import javax.annotation.Resource;
import java.time.LocalDateTime; import java.time.LocalDateTime;
@ -36,6 +39,9 @@ public class StompServiceImpl implements StompService {
@Resource @Resource
private ChatRoomMapper chatRoomMapper; private ChatRoomMapper chatRoomMapper;
@Resource
private TransactionTemplate transactionTemplate;
@Override @Override
public AjaxResult sendMessageToUser(UserMessageVo userMessageVo) { public AjaxResult sendMessageToUser(UserMessageVo userMessageVo) {
WebsocketMessageDto websocketMessageDto = new WebsocketMessageDto(); WebsocketMessageDto websocketMessageDto = new WebsocketMessageDto();
@ -47,11 +53,24 @@ public class StompServiceImpl implements StompService {
} }
//当前用户发送其他人, 发送给指定用户. 否则默认发送给当前发送者 //当前用户发送其他人, 发送给指定用户. 否则默认发送给当前发送者
messagingTemplate.convertAndSendToUser(userMessageVo.getEmail(), Destination.QUEUE + "/" + userMessageVo.getEmail(), userMessageVo.getContent()); messagingTemplate.convertAndSendToUser(userMessageVo.getEmail(), Destination.QUEUE + "/" + userMessageVo.getEmail(), userMessageVo.getContent());
// 接收者和发送者 都不是游客才能存储消息 和修改聊天室信息 // 接收者和发送者 都不是游客才能存储消息 和修改聊天室信息。后续可改为消息中间件解耦形式。
if (!TOURIST.equals(userMessageVo.getReceiveUserType()) && TOURIST.equals(userMessageVo.getSendUserType())){ if (!TOURIST.equals(userMessageVo.getReceiveUserType()) && !TOURIST.equals(userMessageVo.getSendUserType())){
transactionTemplate.execute(new TransactionCallbackWithoutResult() {
@Override
protected void doInTransactionWithoutResult(TransactionStatus status) {
try {
// 插入消息并获取 ID
Long id = insertMessage(userMessageVo); Long id = insertMessage(userMessageVo);
websocketMessageDto.setId(id); websocketMessageDto.setId(id);
// 更新房间信息
updateRoom(userMessageVo); updateRoom(userMessageVo);
} catch (Exception e) {
// 回滚事务
status.setRollbackOnly();
throw e;
}
}
});
} }
return AjaxResult.success(websocketMessageDto); return AjaxResult.success(websocketMessageDto);
} }