87 lines
2.8 KiB
Java
87 lines
2.8 KiB
Java
package nl.andrewl.aos2_server;
|
|
|
|
import nl.andrewl.aos_core.model.Player;
|
|
import nl.andrewl.aos_core.net.UdpReceiver;
|
|
import nl.andrewl.aos_core.net.udp.DatagramInit;
|
|
import nl.andrewl.record_net.Message;
|
|
|
|
import java.io.IOException;
|
|
import java.net.*;
|
|
import java.util.HashMap;
|
|
import java.util.HashSet;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
|
|
public class Server implements Runnable {
|
|
private final ServerSocket serverSocket;
|
|
private final DatagramSocket datagramSocket;
|
|
private volatile boolean running;
|
|
|
|
private int nextClientId = 1;
|
|
private final Set<ClientCommunicationHandler> clientHandlers;
|
|
private final Map<Integer, Player> players;
|
|
private final Map<Integer, ClientCommunicationHandler> playerClientHandlers;
|
|
|
|
public Server() throws IOException {
|
|
this.serverSocket = new ServerSocket(24464, 5);
|
|
this.serverSocket.setReuseAddress(true);
|
|
this.datagramSocket = new DatagramSocket(24464);
|
|
this.datagramSocket.setReuseAddress(true);
|
|
this.clientHandlers = new HashSet<>();
|
|
this.players = new HashMap<>();
|
|
this.playerClientHandlers = new HashMap<>();
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
running = true;
|
|
new Thread(new UdpReceiver(datagramSocket, this::handleUdpMessage)).start();
|
|
System.out.println("Started AOS2-Server on TCP/UDP port " + serverSocket.getLocalPort() + "; now accepting connections.");
|
|
while (running) {
|
|
acceptClientConnection();
|
|
}
|
|
datagramSocket.close();
|
|
for (var handler : clientHandlers) handler.shutdown();
|
|
}
|
|
|
|
public void handleUdpMessage(Message msg, DatagramPacket packet) {
|
|
// Echo any init message from known clients.
|
|
if (msg instanceof DatagramInit init) {
|
|
var handler = getHandler(init.clientId());
|
|
if (handler != null) {
|
|
handler.setClientUdpPort(packet.getPort());
|
|
handler.sendDatagramPacket(msg);
|
|
}
|
|
}
|
|
}
|
|
|
|
public synchronized Player registerPlayer(ClientCommunicationHandler handler, String username) {
|
|
Player player = new Player(nextClientId++, username);
|
|
players.put(player.getId(), player);
|
|
playerClientHandlers.put(player.getId(), handler);
|
|
System.out.println("Registered player " + username + " with id " + player.getId());
|
|
return player;
|
|
}
|
|
|
|
public ClientCommunicationHandler getHandler(int id) {
|
|
return playerClientHandlers.get(id);
|
|
}
|
|
|
|
private void acceptClientConnection() {
|
|
try {
|
|
Socket clientSocket = serverSocket.accept();
|
|
ClientCommunicationHandler handler = new ClientCommunicationHandler(this, clientSocket, datagramSocket);
|
|
clientHandlers.add(handler);
|
|
} catch (IOException e) {
|
|
if (e instanceof SocketException && !this.running && e.getMessage().equalsIgnoreCase("Socket closed")) {
|
|
return; // Ignore this exception, since it is expected on shutdown.
|
|
}
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static void main(String[] args) throws IOException {
|
|
new Server().run();
|
|
}
|
|
}
|