mirror/Velocity
2
0
Fork 0
mirror of https://github.com/PaperMC/Velocity.git synced 2025-03-31 16:30:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Avoid duplicate logins.

Browse Source
This commit is contained in:
Andrew Steinborn 2018-08-05 01:30:40 -04:00
parent 22d1398f73
commit 4e64b04464
2 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
proxy/src/main/java/com/velocitypowered/proxy/VelocityServer.java
View File

@ -118,9 +118,15 @@ public class VelocityServer implements ProxyServer {
return httpClient;
}
public void registerConnection(ConnectedPlayer connection) {
connectionsByName.put(connection.getUsername(), connection);
connectionsByUuid.put(connection.getUniqueId(), connection);
public boolean registerConnection(ConnectedPlayer connection) {
if (connectionsByName.putIfAbsent(connection.getUsername(), connection) != null) {
return false;
}
if (connectionsByUuid.putIfAbsent(connection.getUniqueId(), connection) != null) {
connectionsByName.remove(connection.getUsername(), connection);
return false;
}
return true;
}
public void unregisterConnection(ConnectedPlayer connection) {

8
proxy/src/main/java/com/velocitypowered/proxy/connection/client/LoginSessionHandler.java
View File

@ -136,11 +136,15 @@ public class LoginSessionHandler implements MinecraftSessionHandler {
success.setUuid(profile.idAsUuid());
inbound.write(success);
logger.info("{} has connected", player);
inbound.setAssociation(player);
inbound.setState(StateRegistry.PLAY);
if (!VelocityServer.getServer().registerConnection(player)) {
inbound.closeWith(Disconnect.create(TextComponent.of("You are already on this proxy!", TextColor.RED)));
}
logger.info("{} has connected", player);
inbound.setSessionHandler(new InitialConnectSessionHandler(player));
VelocityServer.getServer().registerConnection(player);
player.createConnectionRequest(toTry.get()).fireAndForget();
}
}