devkobe24.com

@Component
public class IRCServer {
    private Selector selector;
    private ServerSocketChannel serverSocket;
    
    // Spring이 관리하는 컴포넌트로 등록
}

@Component
public class ServerInitializer implements ApplicationRunner {
    @Autowired
    private IRCServer ircServer;
    
    @Override
    public void run(ApplicationArguments args) throws Exception {
        // Spring Boot가 완전히 뜨고 나서 IRC 서버 시작
        ircServer.start();
    }
}

public class NioIrcServer implements Runnable {
    private Selector selector;
    private ServerSocketChannel serverSocket;

    public void start() throws Exception {
        // 1. Selector 생성 - 이벤트를 감시할 관제탑
        selector = Selector.open();
        
        // 2. 서버 소켓 생성 및 바인딩
        serverSocket = ServerSocketChannel.open();
        serverSocket.bind(new InetSocketAddress(6667)); // IRC 표준 포트
        
        // 3. 🔑 Non-blocking 모드 설정 (이게 NIO의 핵심!)
        serverSocket.configureBlocking(false);
        
        // 4. Accept 이벤트를 Selector에 등록
        serverSocket.register(selector, SelectionKey.OP_ACCEPT);

        // 5. 별도 스레드에서 이벤트 루프 시작
        new Thread(this, "IRC-NIO-Thread").start();
    }

    @Override
    public void run() {
        while (true) { // 무한 루프 - 서버가 살아있는 한 계속
            try {
                // ⏰ 이벤트가 발생할 때까지 대기
                // (blocking이지만, 여러 채널을 동시에 감시하는 효율적인 대기)
                selector.select();

                Set<SelectionKey> selectedKeys = selector.selectedKeys();
                Iterator<SelectionKey> iter = selectedKeys.iterator();

                while (iter.hasNext()) {
                    SelectionKey key = iter.next();

                    if (key.isAcceptable()) {
                        // 🆕 새로운 클라이언트 접속!
                        handleAccept(key);
                    } else if (key.isReadable()) {
                        // 📨 클라이언트가 데이터를 보냈어요
                        handleRead(key);
                    }
                    
                    iter.remove(); // 처리한 키는 반드시 제거!
                }
            } catch (Exception e) {
                // 에러 처리...
            }
        }
    }
}

public class MessageFrameDecoder {
    private ByteBuffer buffer = ByteBuffer.allocate(1024);
    
    public List<String> decode(ByteBuffer incoming) {
        List<String> messages = new ArrayList<>();
        
        // 기존 버퍼에 새 데이터 추가
        buffer.put(incoming);
        buffer.flip();
        
        // \r\n을 찾아 메시지 추출
        while (buffer.hasRemaining()) {
            byte b = buffer.get();
            if (b == '\n') {
                // 완전한 메시지 발견!
                messages.add(extractMessage());
            }
        }
        
        buffer.compact(); // 남은 데이터는 다음 번을 위해 보관
        return messages;
    }
}

// ❌ 흔한 실수
buffer.put(data);
// flip() 없이 바로 읽으면? → 쓰레기 값!
buffer.get(); 

// ✅ 올바른 사용
buffer.put(data);    // Write 모드
buffer.flip();       // Read 모드로 전환
buffer.get();        // 이제 읽을 수 있음
buffer.compact();    // 남은 데이터 정리

try {
    readData(key);
} catch (IOException e) {
    // ❌ 이렇게만 하면?
    e.printStackTrace();
    // 소켓은 열린 채로, SelectionKey는 등록된 채로...
    // 시간이 지나면 "Too many open files" 에러 발생!
}

try {
    readData(key);
} catch (IOException e) {
    // ✅ 리소스 정리
    key.cancel();                    // Selector에서 제거
    key.channel().close();           // 소켓 닫기
    clientMap.remove(key.channel()); // 애플리케이션 상태 정리
    log.info("Client disconnected: {}", e.getMessage());
}