update 修改消息存储规则

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

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