springboot整合websocket如何实现群聊

这篇文章将为大家详细讲解有关SpringBoot整合websocket如何实现群聊，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。

实现思路

发送者向服务器发送大家早上好。其它客户端可以收到对应消息。

项目展示

通过springboot引入WEBSocket，实现群聊，通过在线websocket测试进行展示。

核心代码

pom引入jar

<parent>        <groupId>org.springframework.boot</groupId>        <artifactId>spring-boot-starter-parent</artifactId>        <version>2.3.10.RELEASE</version>        <relativePath/> <!-- lookup parent from repository -->    </parent>    <properties>        <java.version>1.8</java.version>    </properties>    <dependencies>        <dependency>            <groupId>org.springframework.boot</groupId>            <artifactId>spring-boot-starter-web</artifactId>        </dependency>        <dependency>            <groupId>org.apache.commons</groupId>            <artifactId>commons-lang3</artifactId>        </dependency>        <dependency>            <groupId>org.projectlombok</groupId>            <artifactId>lombok</artifactId>            <version>1.18.12</version>        </dependency>        <dependency>            <groupId>org.springframework.boot</groupId>            <artifactId>spring-boot-starter-websocket</artifactId>        </dependency>        <dependency>            <groupId>com.alibaba</groupId>            <artifactId>fastJSON</artifactId>            <version>1.2.62</version>        </dependency>    </dependencies>

使用springboot 2.3.10.RELEASE版本；java8

WebSocketConfig配置类

@Configurationpublic class WebSocketConfig{   @Bean   public ServerEndpointExporter serverEndpointExporter() {      return new ServerEndpointExporter();   }}

websocket有4个注解，解释如下

• @OnOpen 有新连接

• @OnMessage 有新消息时

• @OnClose 关闭连接

• @OnError 连接异常

对于这个4个注解，配置在方法上即可。

WebsocketServerEndpoint接收类

@Slf4j@Controller@ServerEndpoint ("/")public class WebsocketServerEndpoint{   @OnOpen   public void onOpen(Session session, EndpointConfig config) {      log.info("[onOpen][session({}) 接入]", session);      WebSocketUtil.broadcast (session.getId ()+"上线了");      WebSocketUtil.addSession (session);   }   @OnMessage   public void onMessage(Session session, String message) {      WebSocketUtil.broadcast (message);      log.info("[onOpen][session({}) 接收到一条消息({})]", session, message); // 生产环境下，请设置成 debug 级别   }   @OnClose   public void onClose(Session session, CloseReason closeReason) {      WebSocketUtil.removeSession (session);      WebSocketUtil.broadcast (session.getId ()+"下线了");      log.info("[onClose][session({}) 连接关闭。关闭原因是({})}]", session, closeReason);   }   @OnError   public void onError(Session session, Throwable throwable) {      log.info("[onClose][session({}) 发生异常]", session, throwable);   }}

WebSocketUtil工具类

@Slf4jpublic class WebSocketUtil{    // ========== 会话相关 ==========        private static final Map<String, Session> SESSION_USER_MAP = new ConcurrentHashMap<> ();        public static void addSession(Session session) {        // 更新 SESSION_USER_MAP        SESSION_USER_MAP.put(session.getId (), session);    }        public static void removeSession(Session session) {        // 从 SESSION_USER_MAP 中移除          SESSION_USER_MAP.remove(session.getId ());    }    // ========== 消息相关 ==========        public static <T extends Message> void broadcast(String message) {        // 创建消息//        String messageText = buildTextMessage(type, message);        // 遍历 SESSION_USER_MAP ，进行逐个发送        for (String sessionId : SESSION_USER_MAP.keySet()) {            sendTextMessage(SESSION_USER_MAP.get (sessionId), message);        }    }        public static <T extends Message> void send(Session session, String type, T message) {        // 创建消息        String messageText = buildTextMessage(type, message);        // 遍历给单个 Session ，进行逐个发送        sendTextMessage(session, messageText);    }        private static <T extends Message> String buildTextMessage(String type, T message) {        jsONObject messageObject = new JSONObject();        messageObject.put("type", type);        messageObject.put("body", message);        return messageObject.toString();    }        private static void sendTextMessage(Session session, String messageText) {        if (session == null) {            log.error("[sendTextMessage][session 为 null]");            return;        }        RemoteEndpoint.Basic basic = session.getBasicRemote();        if (basic == null) {            log.error("[sendTextMessage][session 的  为 null]");            return;        }        try {            basic.sendText(messageText);        } catch (IOException e) {            log.error("[sendTextMessage][session({}) 发送消息{}) 发生异常",                    session, messageText, e);        }    }}

在线websocket调试输入ws://localhost:8080/即可。

关于“springboot整合websocket如何实现群聊”这篇文章就分享到这里了，希望以上内容可以对大家有一定的帮助，使各位可以学到更多知识，如果觉得文章不错，请把它分享出去让更多的人看到。