refactor server room operations

This commit is contained in:
2025-03-01 13:06:55 +01:00
parent 63ec7b3aaa
commit 0533c16cf2
2 changed files with 28 additions and 42 deletions

View File

@@ -34,50 +34,52 @@ public class Server implements PacketHandler {
public String getRoomName(ServerConnexion connexion) {
for (Map.Entry<String, ArrayList<ServerConnexion>> entry : rooms.entrySet()) {
if(entry.getValue().contains(connexion)) {
if (entry.getValue().contains(connexion)) {
return entry.getKey();
}
}
return null;
}
public void createRoom(String roomName, ServerConnexion connexion) throws SocketException {
if(rooms.containsKey(roomName)) {
throw new SocketException("Room already exists");
public boolean createRoom(String roomName, ServerConnexion connexion) {
if (rooms.containsKey(roomName)) {
return false;
}
rooms.put(roomName, new ArrayList<>());
rooms.get(roomName).add(connexion);
return true;
}
public void leaveRoom(ServerConnexion connexion) throws SocketException {
public boolean leaveRoom(ServerConnexion connexion) {
String roomName = getRoomName(connexion);
if(roomName != null) {
if (roomName != null) {
rooms.get(roomName).remove(connexion);
// Remove the room if it is empty
if(rooms.get(roomName).isEmpty()) {
if (rooms.get(roomName).isEmpty()) {
rooms.remove(roomName);
}
return;
return true;
}
throw new SocketException("Room does not exist");
return false;
}
public void joinRoom(String roomName, ServerConnexion connexion) throws SocketException {
if(rooms.containsKey(roomName) && !rooms.get(roomName).contains(connexion)) {
public boolean joinRoom(String roomName, ServerConnexion connexion) {
if (rooms.containsKey(roomName) && !rooms.get(roomName).contains(connexion)) {
rooms.get(roomName).add(connexion);
return;
return true;
}
throw new SocketException("Room does not exist");
return false;
}
public void sendToRoom(ServerConnexion connexion, SendChatMessagePacket packet) throws SocketException {
public boolean sendToRoom(ServerConnexion connexion, SendChatMessagePacket packet) {
String roomName = getRoomName(connexion);
ChatMessagePacket chatPacket = new ChatMessagePacket(Instant.now(), connexion.getChatterName(), packet.getContent());
if(roomName != null && rooms.containsKey(roomName)) {
ChatMessagePacket chatPacket = new ChatMessagePacket(Instant.now(), connexion.getChatterName(),
packet.getContent());
if (roomName != null && rooms.containsKey(roomName)) {
rooms.get(roomName).forEach(con -> con.sendPacket(chatPacket));
return;
return true;
}
throw new SocketException("You are not in a room or the room does not exist");
return false;
}
public void close() {

View File

@@ -3,7 +3,6 @@ package server;
import java.io.IOException;
import java.net.DatagramSocket;
import java.net.InetSocketAddress;
import java.net.SocketException;
import network.SocketWriter;
import network.protocol.Packet;
@@ -48,32 +47,20 @@ public class ServerConnexion implements PacketVisitor {
@Override
public void visitPacket(CreateRoomPacket packet) {
try {
server.createRoom(packet.getRoomName(), this);
sendPacket(new ServerResponsePacket(Response.RoomCreated));
} catch (SocketException e) {
sendPacket(new ServerResponsePacket(Response.RoomNotCreated));
}
boolean created = server.createRoom(packet.getRoomName(), this);
sendPacket(new ServerResponsePacket(created ? Response.RoomCreated : Response.RoomNotCreated));
}
@Override
public void visitPacket(JoinRoomPacket packet) {
try {
server.joinRoom(packet.getRoomName(), this);
sendPacket(new ServerResponsePacket(Response.RoomJoined));
} catch (SocketException e) {
sendPacket(new ServerResponsePacket(Response.RoomNotJoined));
}
boolean joined = server.joinRoom(packet.getRoomName(), this);
sendPacket(new ServerResponsePacket(joined ? Response.RoomJoined : Response.RoomNotJoined));
}
@Override
public void visitPacket(LeaveRoomPacket packet) {
try {
server.leaveRoom(this);
sendPacket(new ServerResponsePacket(Response.RoomLeft));
} catch (SocketException e) {
sendPacket(new ServerResponsePacket(Response.RoomNotLeft));
}
boolean left = server.leaveRoom(this);
sendPacket(new ServerResponsePacket(left ? Response.RoomLeft : Response.RoomNotLeft));
}
@Override
@@ -95,11 +82,8 @@ public class ServerConnexion implements PacketVisitor {
@Override
public void visitPacket(SendChatMessagePacket packet) {
try {
server.sendToRoom(this, packet);
} catch (SocketException e) {
sendPacket(new ServerResponsePacket(Response.MessageNotSent));
}
boolean messageSent = server.sendToRoom(this, packet);
sendPacket(new ServerResponsePacket(messageSent ? Response.MessageSent : Response.MessageNotSent));
}
@Override