Форматирование проверял - работает. Вводом некорректных сообщений тоже. Не знаю, в чём причина.
Должен сказать, что чат у меня не работает. Сервер запускается до стадии "Server started" после ввода адреса, а вот клиент и бот останавливаются после пары вводов, ничего путного не выдав.
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 class Handler extends Thread {
Socket socket;
Handler(Socket socket){
this.socket = socket;
}
private String serverHandshake(Connection connection)
throws IOException, ClassNotFoundException {
connection.send(new Message(MessageType.NAME_REQUEST, "Enter your name"));
Message message = connection.receive();
String name;
if (message.getType().equals(MessageType.USER_NAME)){
if (!message.getData().isEmpty()) name = message.getData();
else return serverHandshake(connection);
if (!connectionMap.containsKey(name)) connectionMap.put(name, connection);
else return serverHandshake(connection);
connection.send(new Message(MessageType.NAME_ACCEPTED, "Welcome"));
return name;
}
else return serverHandshake(connection);
}
private void notifyUsers(Connection connection, String userName) throws IOException {
for (Map.Entry<String, Connection> entry: connectionMap.entrySet()){
if (!entry.getKey().equals(userName))
connection.send(new Message(MessageType.USER_ADDED, entry.getKey()));
}
}
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("Error");
}
}
@Override
public void run() {
ConsoleHelper.writeMessage("Адрес"+socket.getRemoteSocketAddress());
try {
Connection connection = new Connection(socket);
String name = serverHandshake(connection);
sendBroadcastMessage(new Message(MessageType.USER_ADDED, name));
notifyUsers(connection, name);
serverMainLoop(connection, name);
connectionMap.remove(name);
sendBroadcastMessage(new Message(MessageType.USER_REMOVED, name));
ConsoleHelper.writeMessage("Соединение закрыто");
} catch (IOException | ClassNotFoundException e) {
ConsoleHelper.writeMessage("Произошла ошибка при обмене данными с удаленным адресом");
}
}
}
private static Map<String, Connection> connectionMap = new ConcurrentHashMap<>();
public static void sendBroadcastMessage(Message message) {
for (String name : connectionMap.keySet()) {
try {
connectionMap.get(name).send(message);
} catch (IOException e) {
ConsoleHelper.writeMessage(String.format("Can't send the message to %s", name));
}
}
}
public static void main(String[] args) {
try (ServerSocket socket = new ServerSocket(ConsoleHelper.readInt())) {
ConsoleHelper.writeMessage("Server started");
while (true) {
new Handler(socket.accept()).start();
}
} catch (Exception e) {
e.printStackTrace();
}
}
}