MiniDay/hamster-ball
1
0
Fork 0
Code Issues Pull Requests Actions Releases 9 Activity

perf: 优化连接稳定性

Browse Source
This commit is contained in:
叁只仓鼠
2022-10-29 22:19:44 +08:00
parent 70647e8e71
commit e2f4c06c0c
10 changed files with 142 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
hamster-ball-server/src/main/java/cn/hamster3/mc/plugin/ball/server/Bootstrap.java
View File

@@ -2,7 +2,7 @@ package cn.hamster3.mc.plugin.ball.server;
import cn.hamster3.mc.plugin.ball.server.command.CommandHandler;
import cn.hamster3.mc.plugin.ball.server.config.ServerConfig;
import cn.hamster3.mc.plugin.ball.server.connector.BallChannelInitializer;
import cn.hamster3.mc.plugin.ball.server.connector.BallServerChannelInitializer;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelOption;
@@ -33,8 +33,7 @@ public class Bootstrap {
.group(loopGroup)
.channel(NioServerSocketChannel.class)
.childOption(ChannelOption.TCP_NODELAY, true)
.childOption(ChannelOption.SO_KEEPALIVE, true)
.childHandler(BallChannelInitializer.INSTANCE);
.childHandler(BallServerChannelInitializer.INSTANCE);
ChannelFuture channelFuture = bootstrap.bind(ServerConfig.getHost(), ServerConfig.getPort());
channelFuture.addListener(future -> {
if (future.isSuccess()) {

30
hamster-ball-server/src/main/java/cn/hamster3/mc/plugin/ball/server/connector/BallChannelHandler.java → hamster-ball-server/src/main/java/cn/hamster3/mc/plugin/ball/server/connector/BallServerChannelHandler.java
View File

@@ -2,33 +2,49 @@ package cn.hamster3.mc.plugin.ball.server.connector;
import cn.hamster3.mc.plugin.ball.common.data.BallMessageInfo;
import cn.hamster3.mc.plugin.ball.server.constant.ConstantObjects;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import org.jetbrains.annotations.NotNull;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class BallChannelHandler extends SimpleChannelInboundHandler<String> {
private static final Logger LOGGER = LoggerFactory.getLogger("ChannelHandler");
@ChannelHandler.Sharable
public class BallServerChannelHandler extends SimpleChannelInboundHandler<String> {
public static final BallServerChannelHandler INSTANCE = new BallServerChannelHandler();
private static final Logger LOGGER = LoggerFactory.getLogger("BallServerChannelHandler");
public BallChannelHandler() {
private BallServerChannelHandler() {
super(true);
}
@Override
protected void channelRead0(ChannelHandlerContext context, String message) {
if ("ping".equals(message)) {
LOGGER.info("从服务器 {} 上收到一条 ping 消息.", context.channel().remoteAddress());
context.writeAndFlush("pong");
return;
}
try {
BallMessageInfo messageInfo = ConstantObjects.GSON.fromJson(message, BallMessageInfo.class);
LOGGER.info("从服务器 {} 上收到一条消息: \n {}", messageInfo.getSenderID(), messageInfo);
BallChannelInitializer.broadcastMessage(messageInfo);
LOGGER.info("从服务器 {} 上收到一条消息: {}", context.channel().remoteAddress(), messageInfo);
BallServerChannelInitializer.broadcastMessage(messageInfo);
} catch (Exception e) {
LOGGER.error(String.format("处理消息 %s 时出现错误: ", message), e);
}
}
@Override
public void channelActive(@NotNull ChannelHandlerContext context) {
LOGGER.warn("与服务器 {} 的连接已可用.", context.channel().remoteAddress());
}
@Override
public void channelInactive(ChannelHandlerContext context) {
context.close();
BallChannelInitializer.CHANNELS.remove(context.channel());
synchronized (BallServerChannelInitializer.CHANNELS) {
BallServerChannelInitializer.CHANNELS.remove(context.channel());
}
LOGGER.warn("与服务器 {} 的连接已断开.", context.channel().remoteAddress());
}
@@ -36,4 +52,4 @@ public class BallChannelHandler extends SimpleChannelInboundHandler<String> {
public void exceptionCaught(ChannelHandlerContext context, Throwable cause) {
LOGGER.warn("与服务器 {} 通信时出现了一个错误: ", context.channel().remoteAddress(), cause);
}
}
}

25
hamster-ball-server/src/main/java/cn/hamster3/mc/plugin/ball/server/connector/BallChannelInitializer.java → hamster-ball-server/src/main/java/cn/hamster3/mc/plugin/ball/server/connector/BallServerChannelInitializer.java
View File

@@ -9,6 +9,7 @@ import io.netty.handler.codec.LengthFieldBasedFrameDecoder;
import io.netty.handler.codec.LengthFieldPrepender;
import io.netty.handler.codec.string.StringDecoder;
import io.netty.handler.codec.string.StringEncoder;
import io.netty.handler.timeout.IdleStateHandler;
import org.jetbrains.annotations.NotNull;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@@ -16,20 +17,23 @@ import org.slf4j.LoggerFactory;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;
public class BallChannelInitializer extends ChannelInitializer<NioSocketChannel> {
public static final BallChannelInitializer INSTANCE = new BallChannelInitializer();
public class BallServerChannelInitializer extends ChannelInitializer<NioSocketChannel> {
public static final BallServerChannelInitializer INSTANCE = new BallServerChannelInitializer();
public static final List<Channel> CHANNELS = new ArrayList<>();
private static final Logger LOGGER = LoggerFactory.getLogger("BallServerCentre");
private static final Logger LOGGER = LoggerFactory.getLogger("BallServerChannelInitializer");
private BallChannelInitializer() {
private BallServerChannelInitializer() {
}
public static void broadcastMessage(BallMessageInfo messageInfo) {
String string = messageInfo.toString();
for (Channel channel : CHANNELS) {
channel.writeAndFlush(string);
synchronized (CHANNELS) {
for (Channel channel : CHANNELS) {
channel.writeAndFlush(string);
}
}
}
@@ -45,13 +49,16 @@ public class BallChannelInitializer extends ChannelInitializer<NioSocketChannel>
}
channel.pipeline()
.addLast(new IdleStateHandler(10, 0, 0, TimeUnit.SECONDS))
.addLast(BallServerKeepAliveHandler.INSTANCE)
.addLast(new LengthFieldPrepender(8))
.addLast(new LengthFieldBasedFrameDecoder(Integer.MAX_VALUE, 0, 8, 0, 8))
.addLast(new StringDecoder(StandardCharsets.UTF_8))
.addLast(new StringEncoder(StandardCharsets.UTF_8))
.addLast(new BallChannelHandler());
.addLast(BallServerChannelHandler.INSTANCE);
CHANNELS.add(channel);
synchronized (CHANNELS) {
CHANNELS.add(channel);
}
}
}

28
hamster-ball-server/src/main/java/cn/hamster3/mc/plugin/ball/server/connector/BallServerKeepAliveHandler.java Normal file
View File

@@ -0,0 +1,28 @@
package cn.hamster3.mc.plugin.ball.server.connector;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleUserEventChannelHandler;
import io.netty.handler.timeout.IdleStateEvent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@ChannelHandler.Sharable
public class BallServerKeepAliveHandler extends SimpleUserEventChannelHandler<IdleStateEvent> {
public static final BallServerKeepAliveHandler INSTANCE = new BallServerKeepAliveHandler();
private static final Logger LOGGER = LoggerFactory.getLogger("BallServerKeepAliveHandler");
private BallServerKeepAliveHandler() {
super(true);
}
@Override
protected void eventReceived(ChannelHandlerContext context, IdleStateEvent event) {
context.close();
synchronized (BallServerChannelInitializer.CHANNELS) {
BallServerChannelInitializer.CHANNELS.remove(context.channel());
}
LOGGER.warn("由于无法验证连接存活，与服务器 {} 的连接已断开.", context.channel().remoteAddress());
}
}

2
hamster-ball-server/src/main/resources/log4j2.component.properties Normal file
View File

@@ -0,0 +1,2 @@
log4j2.loggerContextFactory=org.apache.logging.log4j.core.impl.Log4jContextFactory
log4j.configurationFile=log4j2.xml

4
hamster-ball-server/src/main/resources/log4j2.xml
View File

@@ -2,10 +2,10 @@
<Configuration status="WARN" packages="com.mojang.util">
<Appenders>
<Console name="SysOut" target="SYSTEM_OUT">
<PatternLayout pattern="[%d{HH:mm:ss}] [%t/%level]: %msg%n"/>
<PatternLayout pattern="[%d{HH:mm:ss}] [%logger/%level]: %msg%n"/>
</Console>
<RollingRandomAccessFile name="File" fileName="logs/latest.log" filePattern="logs/%d{yyyy-MM-dd}-%i.log.gz">
<PatternLayout pattern="[%d{HH:mm:ss}] [%t/%level]: %msg%n"/>
<PatternLayout charset="UTF-8" pattern="[%d{HH:mm:ss}] [%logger/%level]: %msg%n"/>
<Policies>
<TimeBasedTriggeringPolicy/>
<OnStartupTriggeringPolicy/>