SpringBoot使用Websocket（一）

全栈程序员-用户IM • 2022年7月27日下午3:00 • 未分类

大家好，又见面了，我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE使用 1年只要46元售后保障童叟无欺

webSocket是HTML5的一种新协议，它实现了服务端与客户端的全双工通信，建立在传输层，tcp协议之上，即浏览器与服务端需要先建立tcp协议，再发送webSocket连接建立请求。webSocket的连接：客户端发送请求信息，服务端接受到请求并返回相应的信息。连接建立。客户端发送http请求时，通过 Upgrade:webSocket Connection:Upgrade 告知服务器需要建立的是webSocket连接，并且还会传递webSocket版本号，协议的字版本号，原始地址，主机地址, webSocket相互通信的Header很小，大概只有2Bytes

本文介绍SpringBoot结合WebSocket实现群发消息, 如想学习如何一对一发送可看我第二篇文章springboot使用websocket（二）

一. 消息群发

1. 导入依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
<dependency>
    <groupId>org.projectlombok</groupId>
    <artifactId>lombok</artifactId>
</dependency>

2. 新建WebSocket配置类

/** * @author Gjing **/
@Configuration
public class WebsocketConfiguration { 
   

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

3. 新建WebSocket服务端

/** * @author Gjing **/
@ServerEndpoint("/test")
@Component
@Slf4j
public class MyWebsocketServer { 
   
    /** * 存放所有在线的客户端 */
    private static Map<String, Session> clients = new ConcurrentHashMap<>();

    @OnOpen
    public void onOpen(Session session) { 
   
        log.info("有新的客户端连接了: {}", session.getId());
        //将新用户存入在线的组
        clients.put(session.getId(), session);
    }

    /** * 客户端关闭 * @param session session */
    @OnClose
    public void onClose(Session session) { 
   
        log.info("有用户断开了, id为:{}", session.getId());
        //将掉线的用户移除在线的组里
        clients.remove(session.getId());
    }

    /** * 发生错误 * @param throwable e */
    @OnError
    public void onError(Throwable throwable) { 
   
        throwable.printStackTrace();
    }

    /** * 收到客户端发来消息 * @param message 消息对象 */
    @OnMessage
    public void onMessage(String message) { 
   
        log.info("服务端收到客户端发来的消息: {}", message);
        this.sendAll(message);
    }

    /** * 群发消息 * @param message 消息内容 */
    private void sendAll(String message) { 
   
        for (Map.Entry<String, Session> sessionEntry : clients.entrySet()) { 
   
            sessionEntry.getValue().getAsyncRemote().sendText(message);
        }
    }
}