Программа работает слишком долго. Уже и так, и сяк, и решения все перепробовал, вроде бы. В чем проблема, подскажите, пожалуйста?
P.S. при дебаге зависает на вызове метода wait. В этом ли проблема?
package com.javarush.task.task30.task3008;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
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 (Connection entry : connectionMap.values()) {
try {
entry.send(message);
} catch (IOException e) {
System.out.println("Не удалось отправить сообщение!");
}
}
}
private static class Handler extends Thread {
private Socket socket;
public Handler(Socket socket) {
this.socket = socket;
}
private String serverHandshake(Connection connection) throws IOException, ClassNotFoundException {
Message message = null;
boolean isRecieved = false;
while (!isRecieved) {
connection.send(new Message(MessageType.NAME_REQUEST));
message = connection.receive();
if (message.getData() != null && !message.getData().equals("")) {
if (message.getType() == MessageType.USER_NAME) {
isRecieved = true;
if (connectionMap.containsKey(message.getData())) {
isRecieved = false;
} else {
connectionMap.put(message.getData(), connection);
connection.send(new Message(MessageType.NAME_ACCEPTED));
}
}
}
}
return message.getData();
}
private void serverMainLoop(Connection connection, String userName) throws IOException, ClassNotFoundException {
while (true) {
Message message = connection.receive();
if (message.getType() == MessageType.TEXT) {
sendBroadcastMessage(new Message(MessageType.TEXT, userName + ": " + message.getData()));
} else {
ConsoleHelper.writeMessage("Введите корректное сообщение!");
}
}
}
private void notifyUsers(Connection connection, String userName) throws IOException {
for (String entry : connectionMap.keySet()) {
if (!entry.equals(userName)) {
connection.send(new Message(MessageType.USER_ADDED, entry));
}
}
}
@Override
public void run() {
ConsoleHelper.writeMessage("Установлено новое соединение с удаленным адресом: " + socket.getRemoteSocketAddress());
String userName = null;
try (Connection connection = new Connection(socket)) {
userName = serverHandshake(connection);
sendBroadcastMessage(new Message(MessageType.USER_ADDED, userName));
notifyUsers(connection, userName);
serverMainLoop(connection, userName);
} catch (IOException | ClassNotFoundException e) {
ConsoleHelper.writeMessage("Произошла ошибка при обмене данными с удаленным адресом!");
} finally {
if (userName != null) {
connectionMap.remove(userName);
sendBroadcastMessage(new Message(MessageType.USER_REMOVED, userName));
}
ConsoleHelper.writeMessage("Соединение с удаленным адресом полностью закрыто!");
}
}
}
public static void main(String[] args) throws Exception {
int port = ConsoleHelper.readInt();
try (ServerSocket serverSocket = new ServerSocket(port)) {
System.out.println("Сервер запущен!");
try {
while (true) {
Socket client = serverSocket.accept();
Thread handler = new Handler(client);
handler.start();
}
} catch (Exception e) {
e.printStackTrace();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}