Вроде все по условию
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 main(String[] args) throws IOException {
int port = ConsoleHelper.readInt();
try (ServerSocket serverSocket = new ServerSocket(port);) {
ConsoleHelper.writeMessage("Сервер запущен");
while (true) {
new Handler(serverSocket.accept()).start();
}
}
catch (Exception e){
e.printStackTrace();
}
}
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;
String name;
while (true){
connection.send(new Message(MessageType.NAME_REQUEST, "Enter name:"));
message = connection.receive();
name = message.getData();
if(message.getType() == MessageType.USER_NAME & name != null & name != ""){
if (!connectionMap.containsKey(name)){
connectionMap.put(name, connection);
break;
}
}
}
connection.send(new Message(MessageType.NAME_ACCEPTED));
ConsoleHelper.writeMessage("Name " + name + " is accepted");
return name;
}
private void notifyUsers(Connection connection, String userName) throws IOException, ClassNotFoundException {
for (Map.Entry<String, Connection> map:connectionMap.entrySet()) {
if(map.getKey() != userName) {
connection.send(new Message(MessageType.USER_ADDED, map.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("Type of message must be Text");
}
}
}
public void run(){
ConsoleHelper.writeMessage("Установлено новое соединение с сервером " + socket.getRemoteSocketAddress());
String userName = "Nick";
try (Connection connection = new Connection(socket)) {
userName = serverHandshake(connection);
sendBroadcastMessage(new Message(MessageType.USER_ADDED, userName));
notifyUsers(connection, userName);
serverMainLoop(connection, userName);
connection.close();
if (userName != null){
connectionMap.remove(userName);
sendBroadcastMessage(new Message(MessageType.USER_REMOVED, userName));
}
ConsoleHelper.writeMessage("Соединение с сервером закрыто");
} catch (IOException | ClassNotFoundException e) {
ConsoleHelper.writeMessage("Произошла ошибка при обмене данными с удаленным адресом");
}
}
}
public static void sendBroadcastMessage(Message message){
try {
for (Map.Entry<String, Connection> map: connectionMap.entrySet()) {
map.getValue().send(message);
}
}
catch (IOException e){
ConsoleHelper.writeMessage("Сообщение не отправлено");
}
}
}