protected void sendTextMessage(String text) {
Message message = new Message(MessageType.TEXT, text);
try {
connection.send(message);
} catch (IOException e) {
ConsoleHelper.writeMessage("Не удалось отправить сообщение");
clientConnected = false;
}
}package com.javarush.task.task30.task3008;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Collection;
import java.util.Collections;
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) {
ConsoleHelper.writeMessage("Введите порт сервера:");
int serverPort = ConsoleHelper.readInt();
try (ServerSocket serverSocket = new ServerSocket(serverPort)){
ConsoleHelper.writeMessage("Сервер запущен");
while (true) {
Socket clientSocket = serverSocket.accept();
Handler session = new Handler(clientSocket);
session.start();
}
} catch (IOException e) {
e.printStackTrace();
}
}
public static void sendBroadcastMessage(Message message) {
Set<Map.Entry<String, Connection>> set = connectionMap.entrySet();
for (Map.Entry<String, Connection> m : set) {
try {
m.getValue().send(message);
} catch (IOException e) {
ConsoleHelper.writeMessage("Невозможно отправить сообщение пользователю " + m.getKey());
}
}
}
private static class Handler extends Thread {
private Socket socket;
public Handler(Socket socket) {
this.socket = socket;
}
@Override
public void run() {
ConsoleHelper.writeMessage("Установлено новое соединение с удалённым адресом: " + socket.getRemoteSocketAddress());
String userName = null;
try (Connection connection = new Connection(socket)) {
userName = serverHandshake(connection);
Message message = new Message(MessageType.USER_ADDED, userName);
sendBroadcastMessage(message);
notifyUsers(connection, userName);
serverMainLoop(connection, userName);
} catch (IOException | ClassNotFoundException e) {
ConsoleHelper.writeMessage("При обмене данными с удалённым адресом произошла ошибка!");
}
if(userName != null) {
connectionMap.remove(userName);
Message message = new Message(MessageType.USER_REMOVED, userName);
sendBroadcastMessage(message);
ConsoleHelper.writeMessage("Соединение с удалённым адресом " + socket.getRemoteSocketAddress() + " закрыто.");
}
}
private String serverHandshake(Connection connection) throws IOException, ClassNotFoundException {
String userName;
while (true) {
Message queryName = new Message(MessageType.NAME_REQUEST);
connection.send(queryName);
Message nameResponse = connection.receive();
if(nameResponse.getType() == MessageType.USER_NAME) {
userName = nameResponse.getData();
if(userName != null && !connectionMap.containsKey(userName) && !userName.equals("")) {
connectionMap.put(userName, connection);
connection.send(new Message(MessageType.NAME_ACCEPTED));
break;
}
}
}
return userName;
}
private void notifyUsers(Connection connection, String userName) throws IOException {
Set<String> namesSet = connectionMap.keySet();
for (String s : namesSet) {
Message message = new Message(MessageType.USER_ADDED, s);
if(!s.equals(userName)) {
connection.send(message);
}
}
}
private void serverMainLoop(Connection connection, String userName) throws IOException, ClassNotFoundException {
while (true) {
Message message = connection.receive();
if(message.getType() == MessageType.TEXT) {
String resp = userName + ": " + message.getData();
Message response = new Message(MessageType.TEXT, resp);
Collection<Connection> connectionSet = connectionMap.values();
sendBroadcastMessage(response);
} else {
ConsoleHelper.writeMessage("Полученное не текстовое сообщение!");
}
}
}
}
}