.
package com.javarush.task.task30.task3008;
import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.SocketAddress;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import static com.javarush.task.task30.task3008.MessageType.*;
public class Server {
private static Map<String, Connection> connectionMap = new ConcurrentHashMap<>();
public static void sendBroadcastMessage(Message message) {
for(String client : connectionMap.keySet()){
try {
connectionMap.get(client).send(message);
} catch (IOException e) {
System.out.println("Сообщение не отправлено!");
}
}
}
private static class Handler extends Thread {
Socket socket;
public Handler(Socket socket) {
this.socket = socket;
}
private String serverHandshake(Connection connection) throws IOException, ClassNotFoundException {
connection.send(new Message(NAME_REQUEST));
Message answer = connection.receive();
if (answer.getType().equals(USER_NAME) && !answer.getData().isEmpty()
&& !connectionMap.containsKey(answer.getData())) {
connectionMap.put(answer.getData(), connection);
connection.send(new Message(NAME_ACCEPTED));
return answer.getData();
} else {
return serverHandshake(connection);
}
}
private void notifyUsers(Connection connection, String userName) throws IOException{
for(String name : connectionMap.keySet()){
Message message = new Message(USER_ADDED, name);
if(!name.equals(userName))
connection.send(message);
}
}
private void serverMainLoop(Connection connection, String userName) throws IOException, ClassNotFoundException{
while (true) {
Message answer = connection.receive();
String messag = answer.getData();
if (answer.getType() == TEXT) {
Message message = new Message(TEXT, userName + ": " + messag);
sendBroadcastMessage(message);
} else {
ConsoleHelper.writeMessage("принятое сообщение не является текстом");
}
}
}
public void run(){
SocketAddress socketAddress = socket.getRemoteSocketAddress();
System.out.println("Было установлено соединение с удаленным адресом " + 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);
socket.connect(socketAddress);
connectionMap.remove(userName);
sendBroadcastMessage(new Message(USER_REMOVED, userName));
} catch (IOException | ClassNotFoundException e) {
System.out.println("Произошла ошибка при обмене данными с удаленным адресом");
try {
socket.close();
} catch (IOException ex) {
ex.printStackTrace();
}
System.out.println("Соединение с удаленным адресом закрыто");
}
}
}
public static void main(String[] args) throws IOException {
int port = ConsoleHelper.readInt();
ServerSocket serverSocket = new ServerSocket(port);
System.out.println("Сервер запущен!");
try {
while (true){
Socket socket = serverSocket.accept();
Handler handler = new Handler(socket);
handler.start();
continue;
}
}catch (Exception e){
System.out.println(e);
serverSocket.close();
}
}
}