Хелп
package com.javarush.task.task30.task3008;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.net.Inet4Address;
import java.net.InetAddress;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
public class Server {
private static Map<String,Connection> connectionMap = new ConcurrentHashMap<>();
public static void main(String[] args) throws IOException{
ConsoleHelper.writeMessage("Введите порт");
ServerSocket serverSocket = new ServerSocket(ConsoleHelper.readInt());
System.out.println("Socket is accept");
try {
while (true) {
Socket socket = serverSocket.accept();
Handler handler = new Handler(socket);
handler.start();
}
}catch(Exception e) {
System.out.println("Ошибка");
serverSocket.close();
}
}
private static class Handler extends Thread {
private Socket socket;
public Handler(Socket socket) {
this.socket = socket;
}
@Override
public void run() {
ConsoleHelper.writeMessage("Connect this "+socket.getRemoteSocketAddress());
String name = null;
try(Connection connection = new Connection(socket)) {
name = serverHandshake(connection);
sendBroadcastMessage(new Message(MessageType.USER_ADDED,name));
notifyUsers(connection,name);
serverMainLoop(connection,name);
} catch (IOException | ClassNotFoundException e) {
ConsoleHelper.writeMessage("Error with connection");
}
if(name != null) {
connectionMap.remove(name);
sendBroadcastMessage(new Message(MessageType.USER_REMOVED, name));
}
ConsoleHelper.writeMessage("Connection is cosed!");
}
private String serverHandshake(Connection connection) throws IOException, ClassNotFoundException {
while (true) {
connection.send(new Message(MessageType.NAME_REQUEST));
Message answer = connection.receive();
if (answer.getType() == MessageType.USER_NAME) {
if (!answer.getData().isEmpty()) {
if (!connectionMap.containsKey(answer.getData())) {
connectionMap.put(answer.getData(), connection);
connection.send(new Message(MessageType.NAME_ACCEPTED));
return answer.getData();
}
}
}
}
}
private void notifyUsers(Connection connection, String userName) throws IOException {
for (String client : connectionMap.keySet()) {
if (!client.equals(userName))
connection.send(new Message(MessageType.USER_ADDED,client));
}
}
private void serverMainLoop(Connection connection, String userName) throws IOException, ClassNotFoundException {
while(true) {
Message message = connection.receive();
if (message.getType() == MessageType.TEXT) {
Message newMessage = new Message(MessageType.TEXT,userName+": " +message.getData());
sendBroadcastMessage(newMessage);
} else ConsoleHelper.writeMessage("Error");
}
}
}
public static void sendBroadcastMessage(Message message) {
for (String client : connectionMap.keySet()) {
try {
connectionMap.get(client).send(message);
} catch (IOException e) {
ConsoleHelper.writeMessage("Exception this client #"+client);
}
}
}
}