Циклы не прерывал, так как написано
Цикл будет завершен автоматически если произойдет ошибка (будет брошено исключение) или поток, в котором работает цикл, будет прерван.
package com.javarush.task.task30.task3008;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.UnknownHostException;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
public class Server {
private static Map<String, Connection> connectionMap = new ConcurrentHashMap<>();
public static void sendBroadcastMessage(Message message) {
for (Map.Entry<String, Connection> stringConnectionEntry : connectionMap.entrySet()) {
try {
stringConnectionEntry.getValue().send(message);
} catch (IOException e) {
System.out.println("Message for user " + stringConnectionEntry.getKey() + " not send!");
}
}
}
public static void main(String[] args) throws IOException {
ConsoleHelper consoleHelper = new ConsoleHelper();
try (ServerSocket ss = new ServerSocket(consoleHelper.readInt())) {
System.out.println("Server started..");
while (true) {
Socket socket = ss.accept();
Handler handler = new Handler(socket);
handler.start();
}
} catch (Exception e) {
System.out.println("Error");
}
}
private static class Handler extends Thread {
private Socket socket;
public Handler(Socket socket) {
this.socket = socket;
}
private String serverHandshake(Connection connection) throws IOException, ClassNotFoundException {
connection.send(new Message(MessageType.NAME_REQUEST, "Give me your name"));
Message ansver = connection.receive();
String userName = "";
if (ansver.getType().equals(MessageType.USER_NAME)) {
userName = ansver.getData();
if (!userName.isEmpty() && !connectionMap.containsKey(userName)) {
connectionMap.put(userName, connection);
connection.send(new Message(MessageType.NAME_ACCEPTED, "You are approved!"));
} else return serverHandshake(connection);
} else return serverHandshake(connection);
return userName;
}
private void notifyUsers(Connection connection, String userName) throws IOException {
for (String s : connectionMap.keySet()) {
if (!s.equals(userName)) {
connection.send(new Message(MessageType.USER_ADDED, s));
}
}
}
private void serverMainLoop(Connection connection, String userName) throws IOException, ClassNotFoundException {
while (true) {
Message mes = connection.receive();
if (mes.getType() == MessageType.TEXT) {
sendBroadcastMessage(new Message(MessageType.TEXT, userName + ": " + mes.getData()));
} else ConsoleHelper.writeMessage("Error");
}
}
@Override
public void run() {
System.out.println("New connection " + socket.getRemoteSocketAddress());
try (Connection connection = new Connection(socket)){
String userName = serverHandshake(connection);
sendBroadcastMessage(new Message(MessageType.USER_ADDED,userName));
notifyUsers(connection, userName);
serverMainLoop(connection,userName);
connectionMap.remove(userName);
sendBroadcastMessage(new Message(MessageType.USER_REMOVED,userName));
ConsoleHelper.writeMessage("Remote server connection closed");
} catch (ClassNotFoundException | IOException e) {
ConsoleHelper.writeMessage("Remote server connection error");
}
}
}
}