增加消息确认接口

周峰
1 parent 9ade224a
Showing 2 changed files with 15 additions and 0 deletions
src/main/java/com/huaheng/pc/stompwebsocket/controller/WebSocketController.java
src/main/java/com/huaheng/pc/stompwebsocket/service/StompWebSocketServiceImpl.java
@@ -3,9 +3,11 @@ package com.huaheng.pc.stompwebsocket.controller;
 import com.huaheng.pc.stompwebsocket.WebsocketConstants;
 import com.huaheng.pc.stompwebsocket.domain.StompPayload;
 import com.huaheng.pc.stompwebsocket.service.StompWebSocketServiceImpl;
+import org.springframework.messaging.Message;
 import org.springframework.messaging.handler.annotation.MessageMapping;
 import org.springframework.messaging.handler.annotation.SendTo;
 import org.springframework.stereotype.Controller;
+import org.springframework.util.LinkedMultiValueMap;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
@@ -42,6 +44,15 @@ public class WebSocketController {
     }
     /**
+     * 服务器接收confirm消息
+     */
+    @MessageMapping("/websocket/message/confirm")
+    public void handleConfirmMsgByWebsocket(Message headers){
+        String msg_id = ((LinkedMultiValueMap<String, String>) headers.getHeaders().get("nativeHeaders")).get("msg-id").get(0);
+        stompWebSocketService.confirmMsg(msg_id);
+    }
+
+    /**
      * 服务器接收httpget上行的普通文本消息，再通过websocket广播出去
      * http://127.0.0.1:8888/wms/websocket/message/string?msg=huaheng_robot
      */
@@ -76,6 +76,10 @@ public class StompWebSocketServiceImpl extends ServiceImpl&lt;StompMessageMapper, S
         messaging.convertAndSendToUser(user, topic, payload.getData(), headers);
     }
+    public void confirmMsg(String msg_id){
+        removeById(msg_id);
+    }
+
     /**
      * 将离线后的消息发送给用户
      * @param user