springboot整合netty做websocket服务器

本文通过示例代码展示了如何使用SpringBoot结合Netty搭建WebSocket服务器,提供了关键代码片段,并提到了完整代码可在作者的GitHub上查看。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

下面直接贴代码,一下是主要代码,还有一些部分代码可以到我的github上看,github地址点击打开链接

NettyServer

package com.youxiong.netty.server;

import com.youxiong.netty.handler.MyChannelHandler;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.stream.ChunkedWriteHandler;
import io.netty.handler.timeout.IdleStateHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.util.concurrent.TimeUnit;

@Component
public class NettyServer {

    private static final Logger LOGGER = LoggerFactory.getLogger(NettyServer.class);

    @Value("${netty.server.port}")
    public Integer port;


    public Integer getPort() {
        return port;
    }

    public void setPort(Integer port) {
        this.port = port;
    }

    private void startServer(){
        //服务端需要2个线程组  boss处理客户端连接  work进行客服端连接之后的处理
        EventLoopGroup boss = new NioEventLoopGroup();
        EventLoopGroup work = new NioEventLoopGroup();
        try {
            ServerBootstrap bootstrap = new ServerBootstrap();
            //服务器 配置
            bootstrap.group(boss,work).channel(NioServerSocketChannel.class)
            .childHandler(new ChannelInitializer<SocketChannel>() {
                protected void initChannel(SocketChannel socketChannel) throws Exception {
                    // HttpServerCodec:将请求和应答消息解码为HTTP消息
                    socketChannel.pipeline().addLast("http-codec",new HttpServerCodec());
                    // HttpObjectAggregator:将HTTP消息的多个部分合成一条完整的HTTP消息
                    socketChannel.pipeline().addLast("aggregator",new HttpObjectAggregator(65536));
                    // ChunkedWriteHandler:向客户端发送HTML5文件
                    socketChannel.pipeline().addLast("http-chunked",new ChunkedWriteHandler());
                    // 进行设置心跳检测
                    socketChannel.pipeline().addLast(new IdleStateHandler(60,30,60*30, TimeUnit.SECONDS));
                    //
### Spring Boot 整合 NettyWebSocket 实现方案 #### 1. 添加依赖项 要在Spring Boot项目中集成NettyWebSocket,首先需要在`pom.xml`文件中添加必要的依赖项。这通常包括Spring WebFlux、Netty以及WebSocket支持的相关库。 ```xml <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-webflux</artifactId> </dependency> <dependency> <groupId>io.projectreactor.netty</groupId> <artifactId>reactor-netty</artifactId> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> ``` 这些依赖项提供了构建基于反应式的Web应用程序所需的基础结构和支持[^4]。 #### 2. 创建WebSocket配置类 创建一个Java配置类来定义WebSocket服务器的行为。通过继承`WebSocketServerEndpointConfigurer`并重写相应的方法可以自定义握手处理器和其他设置。 ```java @Configuration public class WebSocketConfig implements WebSocketServerCustomizer { @Override public void customize(ReactiveWebSocketService service) { // 自定义WebSocket服务行为 } } ``` 此部分代码用于初始化WebSocket的服务端逻辑,并允许开发者根据需求调整其工作方式[^2]。 #### 3. 编写处理程序 编写具体的WebSocket事件处理程序,负责监听来自客户端的消息并与之交互。这里展示了一个简单的例子: ```java @Component @ServerEndpoint("/ws/{userId}") public class MyWebSocketHandler extends TextWebSocketHandler { private final Logger logger = LoggerFactory.getLogger(MyWebSocketHandler.class); @Autowired private PushMsgService pushMsgService; @OnOpen public void handleConnection(ServerHttpRequest request, ServerHttpResponse response, WebSocketSession session, EndpointConfig config) throws IOException { String userId = (String)session.getAttributes().get("USER_ID"); this.pushMsgService.addUser(userId, session); logger.info("New connection from user {}", userId); } @OnMessage public void handleMessage(TextMessage message, WebSocketSession session) throws IOException { // 处理接收到的信息 } @OnError public void handleError(Throwable error){ // 错误处理机制 } @OnClose public void handleClose(CloseReason reason, WebSocketSession session)throws Exception{ // 断开连接后的清理操作 } } ``` 上述代码片段展示了如何利用注解驱动的方式捕获不同类型的WebSocket生命周期事件,并执行相应的业务逻辑[^5]。 #### 4. 构建消息推送服务 为了能够向已建立的WebSocket会话发送通知或更新数据给前端页面,还需要开发一套有效的消息广播系统。下面给出了一种可能的设计模式——即封装好的`PushMsgService`接口及其默认实现: ```java @Service public class SimplePushMsgServiceImpl implements PushMsgService { private Map<String, List<WebSocketSession>> usersSessionsMap = new ConcurrentHashMap<>(); @Override public synchronized void addUser(String userId, WebSocketSession session) { if (!usersSessionsMap.containsKey(userId)) { usersSessionsMap.putIfAbsent(userId, Collections.synchronizedList(new ArrayList<>())); } usersSessionsMap.get(userId).add(session); } @Override public void pushMsgToOne(String userId, String msg) { Optional.ofNullable(usersSessionsMap.get(userId)) .ifPresent(sessions -> sessions.forEach(session -> sendToSession(msg, session))); } @Override public void pushMsgToAll(String msg) { usersSessionsMap.values() .stream() .flatMap(Collection::stream) .forEach(session -> sendToSession(msg, session)); } private void sendToSession(String msg, WebSocketSession session) { try { if (session.isOpen()) { session.sendMessage(new TextMessage(msg)); } } catch (IOException e) { log.error("Failed to send message", e); } } } ``` 这段代码实现了基本的一对一及一对多的消息分发能力,确保任何时刻都能及时有效地传递信息到目标用户那里。 #### 5. 测试与调试 最后,在完成以上步骤之后就可以启动应用并通过工具如Postman或者专门设计用来测试WebSocket的应用来进行验证了。确保一切正常运行后再考虑部署上线等问题[^3]。
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值