package com.ycl.websocket;
|
|
import com.ycl.websocket.handler.HeartBeatHandler;
|
import com.ycl.websocket.handler.WebSocketHandler;
|
import io.netty.bootstrap.ServerBootstrap;
|
import io.netty.channel.ChannelFuture;
|
import io.netty.channel.ChannelInitializer;
|
import io.netty.channel.ChannelOption;
|
import io.netty.channel.EventLoopGroup;
|
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.HttpServerCodec;
|
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
|
import io.netty.handler.codec.serialization.ObjectEncoder;
|
import io.netty.handler.stream.ChunkedWriteHandler;
|
import io.netty.handler.timeout.IdleStateHandler;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Component;
|
|
/**
|
* webscoket 服务
|
*
|
* @author：xp
|
* @date：2024/4/11 17:47
|
*/
|
@Slf4j
|
@Component
|
@RequiredArgsConstructor
|
public class WebsocketServer {
|
|
private final WebSocketHandler webSocketHandler;
|
|
public void runWebsocket() throws Exception {
|
// 处理 I/O 操作的多线程事件循环组（线程池）。bossGroup用于接收传入的连接，workerGroup用于处理IO操作
|
EventLoopGroup bossGroup = new NioEventLoopGroup();
|
EventLoopGroup workerGroup = new NioEventLoopGroup();
|
try {
|
// 服务器引导(启动)类，提供了一些方法使开发者能够简单的启动服务端
|
ServerBootstrap b = new ServerBootstrap();
|
b.group(bossGroup, workerGroup)
|
// 指定服务器通道(主、子通道)的实现类
|
.channel(NioServerSocketChannel.class)
|
// 设置通道初始化器(ChannelInitializer)，每次创建的通道都按这个初始化
|
.childHandler(new ChannelInitializer<SocketChannel>() {
|
@Override
|
public void initChannel(SocketChannel ch) throws Exception {
|
// websocket的握手阶段是使用的Http，所以需要添处理http请求：
|
// 用于将 HTTP 请求和响应转换为字节流以及将字节流转换为 HTTP 请求和响应
|
ch.pipeline().addLast(new HttpServerCodec());
|
ch.pipeline().addLast(new ObjectEncoder());
|
// 以块的方式来写的处理器
|
ch.pipeline().addLast(new ChunkedWriteHandler());
|
// ch.pipeline().addLast(new NettyWebSocketParamHandler(secret));
|
// 针对客户端，若10s内无读事件则触发心跳处理方法HeartBeatHandler#userEventTriggered
|
ch.pipeline().addLast(new IdleStateHandler(60 , 60 , 60));
|
// 自定义空闲状态检测(自定义心跳检测handler)
|
ch.pipeline().addLast(new HeartBeatHandler());
|
ch.pipeline().addLast(new WebSocketServerProtocolHandler("/ws", "WebSocket", true, 65536 * 10));
|
ch.pipeline().addLast(webSocketHandler);
|
}
|
})
|
// 设置服务器通道(主通道)的选项，此处是设置连接请求队列的最大长度是128
|
.option(ChannelOption.SO_BACKLOG, 128);
|
|
// 绑定服务器到指定的端口，并且等待绑定操作完成。
|
ChannelFuture f = b.bind(8044).sync();
|
log.info("websocket启动成功");
|
log.info("程序启动成功");
|
// 等待服务器的通道关闭。
|
f.channel().closeFuture().sync();
|
} finally {
|
workerGroup.shutdownGracefully();
|
bossGroup.shutdownGracefully();
|
log.error("websocket关闭");
|
}
|
}
|
|
}
|
