////
package com.javarush.task.task30.task3008;
import java.io.IOException;
import java.net.Socket;
import java.net.ServerSocket;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import com.javarush.task.task30.task3008.client.Client;
public class Server {
static private Map<String, Connection> connectionMap = new ConcurrentHashMap<>();
public static void main(String[] args) {
int port = ConsoleHelper.readInt();
try {
ConsoleHelper.writeMessage("Сервер запущен");
try (ServerSocket ss = new ServerSocket(port)) {
while (true) {
Socket socket = ss.accept();
Handler handler = new Handler(socket);
handler.start();
}
}
} catch (Exception e) {
System.out.println(e.getStackTrace());
}
}
public static void sendBroadcastMessage(Message message) {
for (Map.Entry<String, Connection> pair : connectionMap.entrySet()) {
try {
pair.getValue().send(message);
} catch (IOException e) {
System.out.println("На сервере ошибка");
}
}
}
private static class Handler extends Thread {
private Socket socket;
public Handler(Socket socket) {
this.socket = socket;
}
public void run(){
ConsoleHelper.writeMessage("AdressWas"+socket.getRemoteSocketAddress());
Connection cn1 = null;
String userName = null;
try {
cn1 = new Connection(socket);
userName = serverHandshake(cn1);
sendBroadcastMessage(new Message(MessageType.USER_ADDED,userName));
sendListOfUsers(cn1,userName);
serverMainLoop(cn1,userName);
connectionMap.remove(userName);
} catch (IOException e) {
ConsoleHelper.writeMessage("ConnectionWasDeleted");
} catch (ClassNotFoundException e) {
ConsoleHelper.writeMessage("ConnectionWasDeleted");
}
sendBroadcastMessage(new Message(MessageType.USER_REMOVED,userName));
ConsoleHelper.writeMessage("ConnectionWasDeleted");
}
private String serverHandshake(Connection connection) throws IOException, ClassNotFoundException {
while (true) {
Message newMessageBitch = new Message(MessageType.NAME_REQUEST);
connection.send(newMessageBitch);
Message mge = connection.receive();
if ((mge.getType() == MessageType.USER_NAME)&&(!mge.getData().isEmpty())&&(!connectionMap.containsKey(mge.getData()))) {
connectionMap.putIfAbsent(mge.getData(), connection);
Message message = new Message(MessageType.NAME_ACCEPTED);
connection.send(message);
return mge.getData();
}
}
}
private void serverMainLoop(Connection connection, String userName) throws IOException, ClassNotFoundException{
while(true){
Message newMessageForClient = connection.receive();
if(newMessageForClient.getType()==MessageType.TEXT){
sendBroadcastMessage(newMessageForClient);
}
else{
ConsoleHelper.writeMessage("ErorrBlyaaaat");
}
}
}
private void sendListOfUsers(Connection connection, String userName) throws IOException{
for(Map.Entry<String,Connection>pair:connectionMap.entrySet()){
String name = pair.getKey();
if(name==userName){
}
else{
Message command = new Message(MessageType.USER_ADDED,name);
connection.send(command);}
}
}
}
}