Upstream/PlayerBalancer
1
0
Fork 0
mirror of https://github.com/BGHDDevelopment/PlayerBalancer.git synced 2024-11-23 11:15:30 +01:00
Code Issues Projects Releases Wiki Activity

Close resources on ServerListPing

Browse Source 
Might fix #5
This commit is contained in:
Jaime 2018-12-30 15:02:29 +01:00
parent 7ff5411e76
commit 9873a645f8
1 changed files with 65 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
balancer/src/main/java/com/jaimemartz/playerbalancer/utils/ServerListPing.java
View File

@ -48,27 +48,30 @@ public final class ServerListPing {
}
public StatusResponse ping(InetSocketAddress host, int timeout) throws IOException {
try (Socket socket = new Socket()) {
Socket socket = new Socket();
socket.setSoTimeout(timeout);
socket.connect(host, timeout);
try (DataOutputStream dataOutputStream = new DataOutputStream(socket.getOutputStream());
DataInputStream dataInputStream = new DataInputStream(socket.getInputStream())) {
ByteArrayOutputStream b = new ByteArrayOutputStream();
DataOutputStream handshake = new DataOutputStream(b);
DataOutputStream dataOutputStream = new DataOutputStream(socket.getOutputStream());
DataInputStream dataInputStream = new DataInputStream(socket.getInputStream());
ByteArrayOutputStream byteArrStream = new ByteArrayOutputStream();
DataOutputStream handshake = new DataOutputStream(byteArrStream);
handshake.writeByte(0x00); //packet id for handshake
writeVarInt(handshake, 4); //protocol version
writeVarInt(handshake, host.getHostString().length()); //host length
handshake.writeBytes(host.getHostString()); //host string
handshake.writeShort(host.getPort()); //port
writeVarInt(handshake, 1); //state (1 for handshake)
handshake.close(); // close handshake packet write stream
writeVarInt(dataOutputStream, b.size()); //prepend size
dataOutputStream.write(b.toByteArray()); //write handshake packet
writeVarInt(dataOutputStream, byteArrStream.size()); //prepend size
dataOutputStream.write(byteArrStream.toByteArray()); //write handshake packet
dataOutputStream.writeByte(0x01); //size is only 1
dataOutputStream.writeByte(0x00); //packet id for ping
int size = readVarInt(dataInputStream); //size of packet
int id = readVarInt(dataInputStream); //packet id
@ -93,12 +96,13 @@ public final class ServerListPing {
dataInputStream.readFully(in); //read json string
String json = new String(in);
long now = System.currentTimeMillis();
dataOutputStream.writeByte(0x09); //size of packet
dataOutputStream.writeByte(0x01); //0x01 for ping
dataOutputStream.writeLong(now); //time!?
dataOutputStream.close(); // close request write stream
readVarInt(dataInputStream);
id = readVarInt(dataInputStream);
if (id == -1) {
@ -112,10 +116,12 @@ public final class ServerListPing {
long pingTime = dataInputStream.readLong(); //read response
StatusResponse response = gson.fromJson(json, StatusResponse.class);
response.time = (int) (now - pingTime);
dataInputStream.close(); // close response read stream
socket.close(); // close socket
return response;
}
}
}
public static class StatusResponse {
private BaseComponent description;