增加报警数据websocket 消息推送

ruoyi-modules/ruoyi-resource/src/main/java/org/dromara/resource/dubbo/RemoteWebSocketMessageServiceImpl.java

@@ -0,0 +1,63 @@
+package org.dromara.resource.dubbo;
+
+import com.alibaba.fastjson.JSONObject;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.dubbo.config.annotation.DubboService;
+import org.dromara.common.websocket.dto.WebSocketMessageDto;
+import org.dromara.common.websocket.utils.WebSocketUtils;
+import org.dromara.resource.api.RemoteWebSocketMessageService;
+import org.dromara.resource.api.domain.WebSocketMsgType;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+
+/**
+ * websocket 消息推送实现类
+ *
+ * @author lxj
+ */
+@Slf4j
+@Service
+@DubboService
+public class RemoteWebSocketMessageServiceImpl implements RemoteWebSocketMessageService {
+
+    @Override
+    public void publishMessage(List<Long> sessionKey, WebSocketMsgType webSocketMsgType, Object data) {
+        WebSocketMessageDto dto = new WebSocketMessageDto();
+        dto.setSessionKeys(sessionKey);
+        dto.setMessage(createMsg(webSocketMsgType, data));
+        try {
+            WebSocketUtils.publishMessage(dto);
+        } catch (Exception e) {
+            log.error("发送分布式消息失败,error:{}", e.getMessage());
+        }
+    }
+
+    @Override
+    public void publishAll(WebSocketMsgType webSocketMsgType, Object data) {
+        try {
+            WebSocketUtils.publishAll(createMsg(webSocketMsgType, data));
+        } catch (Exception e) {
+            log.error("群发消息失败,error:{}", e.getMessage());
+        }
+
+    }
+
+    @Override
+    public void sendMessage(Long userId, WebSocketMsgType webSocketMsgType, Object data) {
+        try {
+            WebSocketUtils.sendMessage(userId, createMsg(webSocketMsgType, data));
+        } catch (Exception e) {
+            log.error("发送指定消息失败,error:{}", e.getMessage());
+        }
+
+    }
+
+    private String createMsg(WebSocketMsgType webSocketMsgType, Object data) {
+        JSONObject msg = new JSONObject();
+        msg.put("code", webSocketMsgType.getCode());
+        msg.put("data", data);
+        return msg.toJSONString();
+    }
+
+}