인프런 커뮤니티 질문&답변

08_chatServerSelector 소스코드의 마지막 부분인 브로드캐스팅하는 함수 sendMessageAll에서 오류를 발견했습니다. 제공하여주신 소스 코드를 작성하고, 여러 클라이언트를 실행시켜보았습니다. 연결은 문제가 없었지만, 쓰기 부분에서 문제가 발생했습니다.

 private static void sendMessageAll(Selector selector,
                  SocketChannel sender, String msg) throws IOException {
        ByteBuffer msgBuffer = ByteBuffer.wrap((msg + "\n").getBytes());

        for (SelectionKey key : selector.keys()) {
            Channel channel = key.channel();
            if (channel instanceof SocketChannel) {
                SocketChannel target = (SocketChannel) channel;
                target.write(msgBuffer); // <- 문제 지점
            }
        }
    }

msgBuffer를 반복문을 통하여 타겟 클라이언트 채널에 쓰기를 할 때에, 첫 클라이언트에게만 메시지가 보내지는 것을 발견하였습니다. 이유는 write호출시 msgBuffer의 position이 마지막으로 이동이 되어서, 다른 클라이언트 소켓 채널에게는 빈 msgBuffer가 쓰기가 되어서 메시지가 보내지지가 않았습니다.

해결방법)

msgBuffer의 값을 할당받는 부분을 반복문 안쪽으로 이동시켜 매 이터레이션 마다 값을 받게 코드를 수정하였습니다.

  // 브로드캐스팅 함수
    private static void sendMessageAll(Selector selector, SocketChannel sender, String msg) throws IOException {

        for (SelectionKey key : selector.keys()) {
            Channel channel = key.channel();

            if (channel instanceof SocketChannel) {
                SocketChannel target = (SocketChannel) channel;

                ByteBuffer msgBuffer = ByteBuffer.wrap((msg + "\n").getBytes());

                target.write(msgBuffer);
            }
        }
    }