70 lines
2.4 KiB
Java
70 lines
2.4 KiB
Java
package com.guwan.backend.netty.chat;
|
||
|
||
import io.netty.bootstrap.ServerBootstrap;
|
||
import io.netty.channel.ChannelFuture;
|
||
import io.netty.channel.ChannelInitializer;
|
||
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.HttpObjectAggregator;
|
||
import io.netty.handler.codec.http.HttpServerCodec;
|
||
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
|
||
import lombok.RequiredArgsConstructor;
|
||
import lombok.extern.slf4j.Slf4j;
|
||
import org.springframework.beans.factory.annotation.Value;
|
||
import org.springframework.stereotype.Component;
|
||
|
||
import javax.annotation.PostConstruct;
|
||
import javax.annotation.PreDestroy;
|
||
|
||
@Slf4j
|
||
@Component
|
||
@RequiredArgsConstructor
|
||
public class ChatServer {
|
||
|
||
@Value("${netty.chat.port}")
|
||
private int port;
|
||
|
||
private final ChatServerHandler chatServerHandler;
|
||
private EventLoopGroup bossGroup;
|
||
private EventLoopGroup workerGroup;
|
||
|
||
@PostConstruct
|
||
public void start() throws Exception {
|
||
bossGroup = new NioEventLoopGroup(1);
|
||
workerGroup = new NioEventLoopGroup();
|
||
|
||
try {
|
||
ServerBootstrap bootstrap = new ServerBootstrap()
|
||
.group(bossGroup, workerGroup)
|
||
.channel(NioServerSocketChannel.class)
|
||
.childHandler(new ChannelInitializer<SocketChannel>() {
|
||
@Override
|
||
protected void initChannel(SocketChannel ch) {
|
||
ch.pipeline()
|
||
.addLast(new HttpServerCodec())
|
||
.addLast(new HttpObjectAggregator(65536))
|
||
.addLast(new WebSocketServerProtocolHandler("/ws/chat"))
|
||
.addLast(chatServerHandler);
|
||
}
|
||
});
|
||
|
||
ChannelFuture future = bootstrap.bind(port).sync();
|
||
log.info("聊天服务器启动成功，WebSocket端口: {}", port);
|
||
} catch (Exception e) {
|
||
log.error("聊天服务器启动失败", e);
|
||
throw e;
|
||
}
|
||
}
|
||
|
||
@PreDestroy
|
||
public void stop() {
|
||
if (bossGroup != null) {
|
||
bossGroup.shutdownGracefully();
|
||
}
|
||
if (workerGroup != null) {
|
||
workerGroup.shutdownGracefully();
|
||
}
|
||
}
|
||
} |