mirrors/riiablo
1
0
Fork 0
mirror of https://github.com/collinsmith/riiablo.git synced 2025-02-02 04:13:50 +07:00
Code Issues Actions Packages Projects Releases Wiki Activity

Parameterized ConnectionLimiter max connections

Browse Source
This commit is contained in:
Collin Smith 2020-06-27 15:26:11 -07:00
parent 413ebc8f6c
commit 5f81d4d53a
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
server/netty/src/com/riiablo/server/d2gs/Main.java
View File

@ -55,7 +55,7 @@ public class Main extends ApplicationAdapter implements PacketProcessor {
private EventLoopGroup bossGroup;
private EventLoopGroup workerGroup;
private final ConnectionLimiter connectionLimiter = new ConnectionLimiter();
private final ConnectionLimiter connectionLimiter = new ConnectionLimiter(MAX_CLIENTS);
private final ConcurrentHashMap<SocketAddress, Integer> connectionIds = new ConcurrentHashMap<>(32);
private final ConcurrentHashMap<SocketAddress, ClientData> clientDatas = new ConcurrentHashMap<>(32);
@ -191,17 +191,22 @@ public class Main extends ApplicationAdapter implements PacketProcessor {
static class ConnectionLimiter extends ChannelInboundHandlerAdapter {
static final String TAG = "ConnectionLimiter";
final int maxClients;
final AtomicInteger connections = new AtomicInteger();
ConnectionLimiter(int maxClients) {
this.maxClients = maxClients;
}
@Override
public void channelActive(ChannelHandlerContext ctx) throws Exception {
int count = connections.incrementAndGet();
if (count <= MAX_CLIENTS) {
Gdx.app.debug(TAG, String.format("connection accepted. %d / %d", count, MAX_CLIENTS));
Gdx.app.debug(TAG, String.format("connection accepted. %d / %d", count, maxClients));
super.channelActive(ctx);
} else {
ctx.close();
Gdx.app.debug(TAG, String.format("closing connection. maximum concurrent connections reached %d / %d", count, MAX_CLIENTS));
Gdx.app.debug(TAG, String.format("closing connection. maximum concurrent connections reached %d / %d", count, maxClients));
}
}