Код работал слишком долго.

Вот код, который "не прошёл":

package com.javarush.task.task30.task3008;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.net.SocketAddress;
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> pair : connectionMap.entrySet()) {
            try {
                pair.getValue().send(message);
            }
            catch (Exception e) {
                System.out.println("Ошибка при отправке сообщения!" + e.getMessage());
            }
        }
    }


    public static void main(String[] args) {
        int port = ConsoleHelper.readInt();
        try (ServerSocket serverSocket = new ServerSocket(port)){
            System.out.println("Server started");
            while (true) {
                Handler handler = new Handler(serverSocket.accept());
                handler.start();
            }
        }
        catch (Exception e){
            System.out.println(e.getMessage());
        }

    }


    private static class Handler extends Thread {
        private Socket socket;

        public Handler(Socket socket) {
            this.socket = socket;
        }

        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) {
                    continue;
                }
                String userName = answer.getData();
                if (userName == null || userName.isEmpty()) {
                    continue;
                }
                if (connectionMap.containsKey(userName)) {
                    continue;
                }
                connectionMap.put(userName, connection);
                connection.send(new Message(MessageType.NAME_ACCEPTED));
                return userName;
            }
        }

        private void notifyUsers(Connection connection, String userName) throws IOException {
            for(Map.Entry<String, Connection> pair : connectionMap.entrySet()) {
                String key = pair.getKey();
                if(key != userName) {
                    Message message = new Message(MessageType.USER_ADDED, key);
                    connection.send(message);
                }
            }
        }

        private void serverMainLoop(Connection connection, String userName) throws IOException, ClassNotFoundException {
            while(true) {
                Message message = connection.receive();
                if (message != null) {
                    if (message.getType() == MessageType.TEXT) {
                        String text = userName + ": " + message.getData();
                        Message message1 = new Message(MessageType.TEXT, text);
                        sendBroadcastMessage(message1);
                    }
                    else {
                        ConsoleHelper.writeMessage("Error!");
                    }
                }
            }
        }

        @Override
        public  void run() {
            ConsoleHelper.writeMessage("Соединение установлено с" + socket.getRemoteSocketAddress());
            String userName = null;
            if(socket != null) {
                try (Connection 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("Произошла ошибка при обмене данными с удаленным адресом");
                }
                if (userName != null) {
                    connectionMap.remove(userName);
                    sendBroadcastMessage(new Message(MessageType.USER_REMOVED, userName));
                }
            }
            ConsoleHelper.writeMessage("Соединение закрыто");
        }

    }
}

А вот код, который я взял из "помощи" и модифицировал (этот код прошёл) :

package com.javarush.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 main(String[] args) throws IOException {
        int port = ConsoleHelper.readInt();

        try (ServerSocket serverSocket = new ServerSocket(port)){
            System.out.println("Server started");
            while (true) {
                Handler handler = new Handler(serverSocket.accept());
                handler.start();
            }
        }
        catch (Exception e){
            System.out.println(e.getMessage());
        }
    }

    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("Message not sended. Error: " + e.getMessage());
            }
        }
    }

    private static class Handler extends Thread {
        private Socket socket;

        public Handler (Socket socket) {
            this.socket = socket;
        }

        public void run() {
            ConsoleHelper.writeMessage(String.valueOf(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("Произошла ошибка при обмене данными с удаленным адресом.");
            }
            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 {
            Message nameRequest = new Message(MessageType.NAME_REQUEST, "Введите, пожалуйста, ваше имя!");
            Message query;
            do {
                connection.send(nameRequest);
                query = connection.receive();
            }
            while (query.getType() != MessageType.USER_NAME
                    || query.getData().isEmpty()
                    || connectionMap.containsKey(query.getData()));

            connectionMap.put(query.getData(), connection);
            connection.send(new Message(MessageType.NAME_ACCEPTED, "Name accepted"));

            return query.getData();
        }

        private void notifyUsers(Connection connection, String userName) throws IOException {
            for (Map.Entry<String, Connection> pair : connectionMap.entrySet()) {
                Message request = new Message(MessageType.USER_ADDED, pair.getKey());
                if (!pair.getKey().equals(userName))
                    connection.send(request);
            }
        }

        private void serverMainLoop(Connection connection, String userName) throws IOException, ClassNotFoundException {
            while (true) {
                Message clientMessage = connection.receive();
                if (clientMessage.getType() == MessageType.TEXT)
                    sendBroadcastMessage(new Message(MessageType.TEXT, userName + ": " + clientMessage.getData()));
                else
                    ConsoleHelper.writeMessage("Error!");
            }
        }
    }
}

Проблема была именно в файле Server, поэтому не вижу смысла выкладывать остальные. Что не так в "неправильном" коде?

Комментарии (8)

популярные
новые
старые

Для того, чтобы оставить комментарий Вы должны авторизоваться

Anonymous #2631266

Уровень 41

17 мая 2021, 08:10

Message not sended. Это неправильный глагол. Send - sent.

Roman Pryshchepa Java Developer

Expert

14 августа 2019, 11:57

while(true) {
    Message message = connection.receive();
    if (message != null) {
        ...
    }
}

При тестировании message == null и получается вечный ~~двигатель~~ цикл? :) Вот почему ты получил "Код работал слишком долго." Поправим тесты.

IceBerg

Master

14 августа 2019, 12:35

Здесь же в любом случае вечный цикл, если сообщение будет null, то просто пропустим и будем ждать следующее сообщение, как и в случае с нетекстовым типом, но так мы себя обезопасим от NPE, которое вылетит в следующей строке.

Антон Full Stack Developer

14 августа 2019, 15:07

Согласен с IceBerg. Как другим способом заставить ввести корректные данные? Можно, конечно, написать костыль типа

for(int i = 0; i < 1; i++) {
    if (message == null) {
        i = 0;
    }
}

Но это именно костыль. + Бесконечный цикл в любом случае что в изначальном варианте коде, что в том, который я "модифицировал". Проблема точно не в этом была.

15 августа 2019, 05:44

Твой кейс не учтен в тестировании (все не возможно учесть), поправим тесты.

Даниил

Уровень 36

31 июля 2020, 22:28

Год прошел - так и не поправили. На это же самое сегодня напоролся)

14 августа 2019, 07:42

полезный

Как вариант, пустое сообщение

connection.send(new Message(MessageType.NAME_ACCEPTED));

14 августа 2019, 07:50

Может быть. Я точно помню, что начал химичить с этими message, ибо валидатор всё время выдавал "код работал слишком долго". То есть, изначально я передавал вторым параметром что-то вроде "Назовите своё имя".