forked from SteamWar/SteamWar
Improve network code
This commit is contained in:
@@ -1,47 +0,0 @@
|
|||||||
/*
|
|
||||||
* This file is a part of the SteamWar software.
|
|
||||||
*
|
|
||||||
* Copyright (C) 2020 SteamWar.de-Serverteam
|
|
||||||
*
|
|
||||||
* This program is free software: you can redistribute it and/or modify
|
|
||||||
* it under the terms of the GNU Affero General Public License as published by
|
|
||||||
* the Free Software Foundation, either version 3 of the License, or
|
|
||||||
* (at your option) any later version.
|
|
||||||
*
|
|
||||||
* This program is distributed in the hope that it will be useful,
|
|
||||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
||||||
* GNU Affero General Public License for more details.
|
|
||||||
*
|
|
||||||
* You should have received a copy of the GNU Affero General Public License
|
|
||||||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package de.steamwar.core;
|
|
||||||
|
|
||||||
import lombok.NoArgsConstructor;
|
|
||||||
|
|
||||||
import java.util.concurrent.atomic.AtomicBoolean;
|
|
||||||
|
|
||||||
@NoArgsConstructor
|
|
||||||
public class Promise<E> {
|
|
||||||
|
|
||||||
private AtomicBoolean hasValue = new AtomicBoolean(false);
|
|
||||||
private E value;
|
|
||||||
|
|
||||||
public void setValue(E value) {
|
|
||||||
this.value = value;
|
|
||||||
hasValue.set(true);
|
|
||||||
}
|
|
||||||
|
|
||||||
public E getValue() {
|
|
||||||
if (hasValue.get()) {
|
|
||||||
return value;
|
|
||||||
}
|
|
||||||
|
|
||||||
while (hasValue.get()) {
|
|
||||||
Thread.yield();
|
|
||||||
}
|
|
||||||
return value;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -22,7 +22,10 @@ package de.steamwar.entity;
|
|||||||
import com.mojang.authlib.GameProfile;
|
import com.mojang.authlib.GameProfile;
|
||||||
import com.mojang.authlib.properties.Property;
|
import com.mojang.authlib.properties.Property;
|
||||||
import de.steamwar.Reflection;
|
import de.steamwar.Reflection;
|
||||||
import de.steamwar.core.*;
|
import de.steamwar.core.BountifulWrapper;
|
||||||
|
import de.steamwar.core.Core;
|
||||||
|
import de.steamwar.core.FlatteningWrapper;
|
||||||
|
import de.steamwar.core.ProtocolWrapper;
|
||||||
import de.steamwar.network.NetworkSender;
|
import de.steamwar.network.NetworkSender;
|
||||||
import de.steamwar.network.packets.common.PlayerSkinRequestPacket;
|
import de.steamwar.network.packets.common.PlayerSkinRequestPacket;
|
||||||
import lombok.Getter;
|
import lombok.Getter;
|
||||||
@@ -72,22 +75,21 @@ public class RPlayer extends REntity {
|
|||||||
server.addEntity(this);
|
server.addEntity(this);
|
||||||
}
|
}
|
||||||
|
|
||||||
public static final Map<UUID, Promise<Property>> SKIN_DATA_PROMISES = new LinkedHashMap<UUID, Promise<Property>>() {
|
public static final Map<UUID, Property> SKIN_DATA_PROMISES = new LinkedHashMap<UUID, Property>() {
|
||||||
@Override
|
@Override
|
||||||
protected boolean removeEldestEntry(Map.Entry<UUID, Promise<Property>> eldest) {
|
protected boolean removeEldestEntry(Map.Entry<UUID, Property> eldest) {
|
||||||
return size() > 100;
|
return size() > 100;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
private GameProfile getGameProfile() {
|
private GameProfile getGameProfile() {
|
||||||
Property property = SKIN_DATA_PROMISES.computeIfAbsent(uuid, __ -> {
|
Property skinData = SKIN_DATA_PROMISES.computeIfAbsent(uuid, __ -> {
|
||||||
Promise<Property> future = new Promise<>();
|
|
||||||
NetworkSender.sendOrQueue(new PlayerSkinRequestPacket(uuid));
|
NetworkSender.sendOrQueue(new PlayerSkinRequestPacket(uuid));
|
||||||
return future;
|
return new Property("textures", null, null);
|
||||||
}).getValue();
|
});
|
||||||
if (property != null) {
|
if (skinData.getValue() != null) {
|
||||||
GameProfile gameProfile = new GameProfile(uuid, name);
|
GameProfile gameProfile = new GameProfile(uuid, name);
|
||||||
gameProfile.getProperties().put("textures", property);
|
gameProfile.getProperties().put("textures", skinData);
|
||||||
return gameProfile;
|
return gameProfile;
|
||||||
} else {
|
} else {
|
||||||
return new GameProfile(actualUUID, name);
|
return new GameProfile(actualUUID, name);
|
||||||
|
|||||||
@@ -21,7 +21,6 @@ package de.steamwar.network;
|
|||||||
|
|
||||||
import com.mojang.authlib.properties.Property;
|
import com.mojang.authlib.properties.Property;
|
||||||
import de.steamwar.core.BountifulWrapper;
|
import de.steamwar.core.BountifulWrapper;
|
||||||
import de.steamwar.core.Promise;
|
|
||||||
import de.steamwar.entity.RPlayer;
|
import de.steamwar.entity.RPlayer;
|
||||||
import de.steamwar.network.handlers.InventoryHandler;
|
import de.steamwar.network.handlers.InventoryHandler;
|
||||||
import de.steamwar.network.packets.PacketHandler;
|
import de.steamwar.network.packets.PacketHandler;
|
||||||
@@ -75,8 +74,8 @@ public class CoreNetworkHandler extends PacketHandler {
|
|||||||
|
|
||||||
@Handler
|
@Handler
|
||||||
public void handlePlayerSkinResponse(PlayerSkinResponsePacket packet) {
|
public void handlePlayerSkinResponse(PlayerSkinResponsePacket packet) {
|
||||||
Promise<Property> propertyPromise = RPlayer.SKIN_DATA_PROMISES.get(packet.getUuid());
|
Property property = RPlayer.SKIN_DATA_PROMISES.get(packet.getUuid());
|
||||||
if (propertyPromise == null) return;
|
if (property == null) return;
|
||||||
propertyPromise.setValue(new Property("textures", packet.getSkin(), packet.getSignature()));
|
RPlayer.SKIN_DATA_PROMISES.put(packet.getUuid(), new Property("textures", packet.getSkin(), packet.getSignature()));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -27,15 +27,12 @@ import org.bukkit.entity.Player;
|
|||||||
import org.bukkit.event.EventHandler;
|
import org.bukkit.event.EventHandler;
|
||||||
import org.bukkit.event.Listener;
|
import org.bukkit.event.Listener;
|
||||||
import org.bukkit.event.player.PlayerJoinEvent;
|
import org.bukkit.event.player.PlayerJoinEvent;
|
||||||
import org.bukkit.event.player.PlayerQuitEvent;
|
|
||||||
|
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
import java.util.concurrent.atomic.AtomicInteger;
|
|
||||||
|
|
||||||
public class NetworkSender implements Listener {
|
public class NetworkSender implements Listener {
|
||||||
|
|
||||||
private static AtomicInteger numberOfPlayers = new AtomicInteger(0);
|
|
||||||
private static List<NetworkPacket> queued = new ArrayList<>();
|
private static List<NetworkPacket> queued = new ArrayList<>();
|
||||||
|
|
||||||
static {
|
static {
|
||||||
@@ -47,21 +44,17 @@ public class NetworkSender implements Listener {
|
|||||||
|
|
||||||
@EventHandler
|
@EventHandler
|
||||||
public void onPlayerJoin(PlayerJoinEvent event) {
|
public void onPlayerJoin(PlayerJoinEvent event) {
|
||||||
numberOfPlayers.incrementAndGet();
|
if (!Bukkit.getOnlinePlayers().isEmpty()) {
|
||||||
if (numberOfPlayers.get() > 1) return;
|
return;
|
||||||
|
}
|
||||||
Bukkit.getScheduler().runTaskLater(Core.getInstance(), () -> {
|
Bukkit.getScheduler().runTaskLater(Core.getInstance(), () -> {
|
||||||
queued.forEach(NetworkSender::send);
|
queued.forEach(NetworkSender::send);
|
||||||
queued.clear();
|
queued.clear();
|
||||||
}, 1);
|
}, 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
@EventHandler
|
|
||||||
public void onPlayerQuit(PlayerQuitEvent event) {
|
|
||||||
numberOfPlayers.decrementAndGet();
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void sendOrQueue(NetworkPacket packet) {
|
public static void sendOrQueue(NetworkPacket packet) {
|
||||||
if (numberOfPlayers.get() > 0) {
|
if (!Bukkit.getOnlinePlayers().isEmpty()) {
|
||||||
send(packet);
|
send(packet);
|
||||||
} else {
|
} else {
|
||||||
queued.add(packet);
|
queued.add(packet);
|
||||||
|
|||||||
@@ -30,10 +30,11 @@ import com.velocitypowered.api.util.GameProfile;
|
|||||||
import de.steamwar.network.packets.PacketHandler;
|
import de.steamwar.network.packets.PacketHandler;
|
||||||
import de.steamwar.network.packets.common.PlayerSkinRequestPacket;
|
import de.steamwar.network.packets.common.PlayerSkinRequestPacket;
|
||||||
import de.steamwar.network.packets.common.PlayerSkinResponsePacket;
|
import de.steamwar.network.packets.common.PlayerSkinResponsePacket;
|
||||||
import de.steamwar.persistent.Storage;
|
|
||||||
import de.steamwar.velocitycore.VelocityCore;
|
import de.steamwar.velocitycore.VelocityCore;
|
||||||
import de.steamwar.velocitycore.network.NetworkSender;
|
import de.steamwar.velocitycore.network.NetworkSender;
|
||||||
import de.steamwar.velocitycore.network.ServerMetaInfo;
|
import de.steamwar.velocitycore.network.ServerMetaInfo;
|
||||||
|
import lombok.AllArgsConstructor;
|
||||||
|
import lombok.Data;
|
||||||
import lombok.SneakyThrows;
|
import lombok.SneakyThrows;
|
||||||
|
|
||||||
import java.io.BufferedReader;
|
import java.io.BufferedReader;
|
||||||
@@ -55,15 +56,9 @@ public class PlayerSkinHandler extends PacketHandler {
|
|||||||
VelocityCore.getProxy().getEventManager().register(VelocityCore.get(), this);
|
VelocityCore.getProxy().getEventManager().register(VelocityCore.get(), this);
|
||||||
}
|
}
|
||||||
|
|
||||||
private Map<UUID, String> skins = new LinkedHashMap<>() {
|
private Map<UUID, SkinData> skins = new LinkedHashMap<>() {
|
||||||
@Override
|
@Override
|
||||||
protected boolean removeEldestEntry(Map.Entry<UUID, String> eldest) {
|
protected boolean removeEldestEntry(Map.Entry<UUID, SkinData> eldest) {
|
||||||
return size() > maxCacheSize;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
private Map<UUID, String> signatures = new LinkedHashMap<>() {
|
|
||||||
@Override
|
|
||||||
protected boolean removeEldestEntry(Map.Entry<UUID, String> eldest) {
|
|
||||||
return size() > maxCacheSize;
|
return size() > maxCacheSize;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -71,8 +66,9 @@ public class PlayerSkinHandler extends PacketHandler {
|
|||||||
@Handler
|
@Handler
|
||||||
@SneakyThrows
|
@SneakyThrows
|
||||||
public void handle(PlayerSkinRequestPacket packet) {
|
public void handle(PlayerSkinRequestPacket packet) {
|
||||||
if (skins.containsKey(packet.getUuid()) && signatures.containsKey(packet.getUuid())) {
|
if (skins.containsKey(packet.getUuid())) {
|
||||||
NetworkSender.send(((ServerMetaInfo) packet.getMetaInfos()).sender().getServer(), new PlayerSkinResponsePacket(packet.getUuid(), skins.get(packet.getUuid()), signatures.get(packet.getUuid())));
|
SkinData skinData = skins.get(packet.getUuid());
|
||||||
|
NetworkSender.send(((ServerMetaInfo) packet.getMetaInfos()).sender().getServer(), new PlayerSkinResponsePacket(packet.getUuid(), skinData.skin, skinData.signature));
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -98,8 +94,7 @@ public class PlayerSkinHandler extends PacketHandler {
|
|||||||
if (prop.get("name").getAsString().equals("textures")) {
|
if (prop.get("name").getAsString().equals("textures")) {
|
||||||
String skin = prop.get("value").getAsString();
|
String skin = prop.get("value").getAsString();
|
||||||
String signature = prop.get("signature").getAsString();
|
String signature = prop.get("signature").getAsString();
|
||||||
skins.put(packet.getUuid(), skin);
|
skins.put(packet.getUuid(), new SkinData(skin, signature));
|
||||||
signatures.put(packet.getUuid(), signature);
|
|
||||||
NetworkSender.send(((ServerMetaInfo) packet.getMetaInfos()).sender().getServer(), new PlayerSkinResponsePacket(packet.getUuid(), skin, signature));
|
NetworkSender.send(((ServerMetaInfo) packet.getMetaInfos()).sender().getServer(), new PlayerSkinResponsePacket(packet.getUuid(), skin, signature));
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@@ -112,14 +107,15 @@ public class PlayerSkinHandler extends PacketHandler {
|
|||||||
GameProfile gameProfile = player.getGameProfile();
|
GameProfile gameProfile = player.getGameProfile();
|
||||||
GameProfile.Property property = gameProfile.getProperties().stream().filter(p -> p.getName().equals("textures")).findFirst().orElse(null);
|
GameProfile.Property property = gameProfile.getProperties().stream().filter(p -> p.getName().equals("textures")).findFirst().orElse(null);
|
||||||
if (property == null) return;
|
if (property == null) return;
|
||||||
skins.put(player.getUniqueId(), property.getValue());
|
skins.put(player.getUniqueId(), new SkinData(property.getValue(), property.getSignature()));
|
||||||
signatures.put(player.getUniqueId(), property.getSignature());
|
|
||||||
|
|
||||||
Set<UUID> uuidSet = skins.keySet();
|
Set<UUID> uuidSet = skins.keySet();
|
||||||
VelocityCore.getProxy().getAllServers().forEach(server -> {
|
VelocityCore.getProxy().getAllServers().forEach(server -> {
|
||||||
for (UUID uuid : uuidSet) {
|
for (UUID uuid : uuidSet) {
|
||||||
NetworkSender.send(server, new PlayerSkinResponsePacket(uuid, skins.get(uuid), signatures.get(uuid)));
|
NetworkSender.send(server, new PlayerSkinResponsePacket(uuid, property.getValue(), property.getSignature()));
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public record SkinData(String skin, String signature) {}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user