From 6516ddc56b7ddf971bb8faef2500ee951f7aa835 Mon Sep 17 00:00:00 2001
From: zhaowenyu <35589726@qq.com>
Date: Fri, 17 Oct 2025 16:22:17 +0800
Subject: [PATCH] 2025-10-17 zwy 机场状态信息WebSocket配置与服务类
---
ard-sample/src/main/java/com/ard/sample/component/websocket/config/WebSocketDockConfig.java | 18 ++++++
ard-sample/src/main/java/com/ard/sample/component/websocket/service/impl/WebSocketDockServer.java | 125 +++++++++++++++++++++++++++++++++++++++++
2 files changed, 143 insertions(+), 0 deletions(-)
diff --git a/ard-sample/src/main/java/com/ard/sample/component/websocket/config/WebSocketDockConfig.java b/ard-sample/src/main/java/com/ard/sample/component/websocket/config/WebSocketDockConfig.java
new file mode 100644
index 0000000..da048be
--- /dev/null
+++ b/ard-sample/src/main/java/com/ard/sample/component/websocket/config/WebSocketDockConfig.java
@@ -0,0 +1,18 @@
+package com.ard.sample.component.websocket.config;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.socket.server.standard.ServerEndpointExporter;
+
+/**
+ * 启用WebSocket支持，注册端点处理器
+ */
+@Configuration
+public class WebSocketDockConfig {
+
+ // 自动注册@ServerEndpoint注解的类
+ @Bean
+ public ServerEndpointExporter serverEndpointExporter() {
+ return new ServerEndpointExporter();
+ }
+}
\ No newline at end of file
diff --git a/ard-sample/src/main/java/com/ard/sample/component/websocket/service/impl/WebSocketDockServer.java b/ard-sample/src/main/java/com/ard/sample/component/websocket/service/impl/WebSocketDockServer.java
new file mode 100644
index 0000000..f9ca321
--- /dev/null
+++ b/ard-sample/src/main/java/com/ard/sample/component/websocket/service/impl/WebSocketDockServer.java
@@ -0,0 +1,125 @@
+package com.ard.sample.component.websocket.service.impl;
+
+import org.springframework.stereotype.Component;
+
+import javax.websocket.*;
+import javax.websocket.server.PathParam;
+import javax.websocket.server.ServerEndpoint;
+import java.io.IOException;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * WebSocket端点类：处理连接、消息收发、断开等事件
+ * 端点路径：ws://localhost:8080/ws/{username}（username为客户端标识）
+ */
+@ServerEndpoint("/ws/Dock/{username}")
+@Component
+public class WebSocketDockServer {
+
+ // 静态变量：存储所有在线客户端的会话（线程安全）
+ private static final ConcurrentHashMap<String, Session> ONLINE_USERS = new ConcurrentHashMap<>();
+
+ // 当前连接的用户名
+ private String username;
+
+
+ /**
+ * 客户端连接成功时触发
+ */
+ @OnOpen
+ public void onOpen(Session session, @PathParam("username") String username) {
+ this.username = username;
+ // 保存会话
+ ONLINE_USERS.put(username, session);
+ // 群发"用户上线"通知
+ sendToAll("系统通知：用户【" + username + "】已上线，当前在线人数：" + ONLINE_USERS.size());
+ System.out.println("用户【" + username + "】连接成功，在线人数：" + ONLINE_USERS.size());
+ }
+
+
+ /**
+ * 客户端断开连接时触发
+ */
+ @OnClose
+ public void onClose() {
+ // 移除会话
+ ONLINE_USERS.remove(username);
+ // 群发"用户下线"通知
+ sendToAll("系统通知：用户【" + username + "】已下线，当前在线人数：" + ONLINE_USERS.size());
+ System.out.println("用户【" + username + "】断开连接，在线人数：" + ONLINE_USERS.size());
+ }
+
+
+ /**
+ * 收到客户端消息时触发
+ * 消息格式支持："all:消息内容"（群发）、"username:消息内容"（指定用户发送）
+ */
+ @OnMessage
+ public void onMessage(String message) {
+ System.out.println("收到【" + username + "】的消息：" + message);
+
+ if (message.startsWith("all:")) {
+ // 群发消息（格式：all:你好大家）
+ String content = message.substring(4);
+ sendToAll("【" + username + "】：" + content);
+ } else if (message.contains(":")) {
+ // 给指定用户发消息（格式：目标用户名:消息内容）
+ String[] parts = message.split(":", 2);
+ String targetUser = parts[0];
+ String content = parts[1];
+ sendToUser(targetUser, "【" + username + "】对你说：" + content);
+ } else {
+ // 默认群发
+ sendToAll("【" + username + "】：" + message);
+ }
+ }
+
+
+ /**
+ * 连接异常时触发
+ */
+ @OnError
+ public void onError(Session session, Throwable error) {
+ System.err.println("WebSocket错误（用户【" + username + "】）：");
+ error.printStackTrace();
+ }
+
+
+ // -------------------------- 消息发送工具方法 --------------------------
+
+ /**
+ * 向单个用户发送消息
+ * @param targetUser 目标用户名
+ * @param message 消息内容
+ */
+ public static void sendToUser(String targetUser, String message) {
+ Session session = ONLINE_USERS.get(targetUser);
+ if (session != null && session.isOpen()) { // 检查会话是否有效
+ try {
+ // 同步发送消息
+ session.getBasicRemote().sendText(message);
+ } catch (IOException e) {
+ System.err.println("向【" + targetUser + "】发送消息失败：" + e.getMessage());
+ }
+ } else {
+ System.out.println("用户【" + targetUser + "】不在线");
+ }
+ }
+
+
+ /**
+ * 向所有在线用户群发消息
+ * @param message 消息内容
+ */
+ public static void sendToAll(String message) {
+ for (Session session : ONLINE_USERS.values()) {
+ if (session.isOpen()) { // 只给活跃会话发送
+ try {
+ session.getBasicRemote().sendText(message);
+ } catch (IOException e) {
+ System.err.println("群发消息失败：" + e.getMessage());
+ }
+ }
+ }
+ }
+}
\ No newline at end of file
--
Gitblit v1.9.3