Not sure what more they want
package com.codegym.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 (Map.Entry<String, Connection> entry : connectionMap.entrySet()){
try {
entry.getValue().send(message);
} catch (IOException e) {
System.out.println("message could not be sent");
}
}
}
public static void main(String[] args) throws IOException {
ConsoleHelper.writeMessage("please enter port number");
ServerSocket serverSocket = new ServerSocket(ConsoleHelper.readInt());
System.out.println("server is running");
while (true){
try {
Socket socket = serverSocket.accept();
Handler handler = new Handler(socket);
handler.start();
} catch (Exception e){
System.out.println(e.getMessage());
serverSocket.close();
break;
}
}
}
private static class Handler extends Thread{
Socket socket;
Handler(Socket socket){
this.socket = socket;
}
public void run(){
ConsoleHelper.writeMessage("Connection has been established with remote address " + socket.getRemoteSocketAddress() );
Connection connection = null;
String userName = null;
try {
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("Error occurred whilst communicating with the remote address.");
try {
connection.close();
} catch (IOException ioException) {
ConsoleHelper.writeMessage("Error occurred whilst communicating with the remote address.");
}
} finally {
connectionMap.remove(userName);
sendBroadcastMessage(new Message(MessageType.USER_REMOVED, userName));
}
}
private String serverHandshake(Connection connection) throws IOException, ClassNotFoundException{
connection.send(new Message(MessageType.NAME_REQUEST));
Message message = connection.receive();
while (!message.getType().equals(MessageType.USER_NAME) ||
connectionMap.containsKey(message.getData()) ||
message.getData() == null ||
message.getData().equals("")) {
connection.send(new Message(MessageType.NAME_REQUEST));
message = connection.receive();
}
connection.send(new Message(MessageType.NAME_ACCEPTED));
connectionMap.put(message.getData(), connection);
return message.getData();
}
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) {
try {
Message received = connection.receive();
if (received.getType().equals(MessageType.TEXT)) {
String m = userName + ": " + received.getData();
sendBroadcastMessage(new Message(MessageType.TEXT, m));
} else {
String error = "not text error";
ConsoleHelper.writeMessage(error);
}
} catch (Exception e) {
System.out.println(e.getMessage());
}
}
}
}
}