private void notifyUsers(Connection connection, String userName) throws IOException {
Message message = new Message(MessageType.USER_ADDED, "User, called " + userName + " has been added.");
for (Map.Entry<String, Connection> connectionEntry : connectionMap.entrySet()) {
if (!connectionEntry.getKey().equals(userName)) {
connectionEntry.getValue().send(message);
}
}
}
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;
import java.util.concurrent.ConcurrentMap;
public class Server {
private static Map<String, Connection> connectionMap = new ConcurrentHashMap<>();
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 receivedMessage;
do {
connection.send(new Message(MessageType.NAME_REQUEST));
} while ((receivedMessage = connection.receive()) == null || receivedMessage.getType() != MessageType.USER_NAME || receivedMessage.getData() == null || receivedMessage.getData().isEmpty() || connectionMap.containsKey(receivedMessage.getData()));
connectionMap.put(receivedMessage.getData(), connection);
connection.send(new Message(MessageType.NAME_ACCEPTED));
return receivedMessage.getData();
}
private void notifyUsers(Connection connection, String userName) throws IOException {
Message message = new Message(MessageType.USER_ADDED, "User, called " + userName + " has been added.");
for (Map.Entry<String, Connection> connectionEntry : connectionMap.entrySet()) {
if (!connectionEntry.getKey().equals(userName)) {
connectionEntry.getValue().send(message);
}
}
}
}
public static void sendBroadcastMessage(Message message) {
for (Map.Entry<String, Connection> connectionEntry : connectionMap.entrySet()) {
try {
connectionEntry.getValue().send(message);
} catch (IOException e) {
System.out.println("Exception while trying to send message to '" + connectionEntry.getKey() + "':");
e.printStackTrace();
}
}
}
public static void main(String[] args) throws IOException {
ConsoleHelper.writeMessage("Please, enter port number:");
int portNumber = ConsoleHelper.readInt();
try (ServerSocket serverSocket = new ServerSocket(portNumber)) {
System.out.println("The server is running, listening on port " + portNumber);
while (true) {
new Handler(serverSocket.accept()).start();
System.out.println("Client connection accepted!");
}
} catch (Exception e) {
System.out.println(e.getMessage());
}
}
}