feat: 通知公告临时提交

通知公告临时提交

src/main/java/com/youlai/boot/system/handler/MessageHandler.java

@@ -0,0 +1,35 @@
+package com.youlai.boot.system.handler;
+
+import com.youlai.boot.platform.websocket.service.MessageService;
+import com.youlai.boot.system.model.dto.MessageDTO;
+import lombok.RequiredArgsConstructor;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+
+/**
+ * 消息处理器
+ *
+ * @author Theo
+ * @since 2024-9-2 14:32:58
+ */
+@Component
+@RequiredArgsConstructor
+public class MessageHandler {
+
+    private final List<MessageService> messageServices;
+
+
+    /**
+     * 发送消息
+     *
+     * @param messageDTO 消息载体
+     */
+    public void sendMessage(MessageDTO messageDTO) {
+        messageServices.forEach(messageService -> {
+            if (messageService.check(messageDTO.getMessageType())) {
+                messageService.sendMessage(messageDTO);
+            }
+        });
+    }
+}