WebSocket

import org.springframework.web.socket.*;
import org.springframework.messaging.handler.annotation.*;
import org.springframework.messaging.simp.SimpMessagingTemplate;

/* ── WebSocket 설정 ──────────────────────────────────────── */
@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

    @Override
    public void configureMessageBroker(MessageBrokerRegistry registry) {
        // 인메모리 메시지 브로커 — /topic(브로드캐스트), /queue(1:1)
        registry.enableSimpleBroker("/topic", "/queue");
        // 클라이언트 → 서버 메시지 prefix
        registry.setApplicationDestinationPrefixes("/app");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws")
                .setAllowedOriginPatterns("*")
                .withSockJS(); // WebSocket 미지원 환경 폴백
    }
}

/* ── 채팅 컨트롤러 ──────────────────────────────────────── */
@Controller
public class ChatController {

    private final SimpMessagingTemplate messagingTemplate;

    public ChatController(SimpMessagingTemplate messagingTemplate) {
        this.messagingTemplate = messagingTemplate;
    }

    // 클라이언트 → /app/chat.send → 브로드캐스트
    @MessageMapping("/chat.send")
    @SendTo("/topic/chat")
    public ChatMessage handleMessage(ChatMessage message) {
        message.setTimestamp(Instant.now());
        return message; // /topic/chat 구독자 전체에 전달
    }

    // 서버에서 특정 사용자에게 1:1 메시지 전송
    public void sendToUser(String userId, String message) {
        messagingTemplate.convertAndSendToUser(
            userId, "/queue/notification",
            new Notification(message)
        );
    }

    // 연결/해제 이벤트 처리
    @EventListener
    public void handleConnect(SessionConnectedEvent event) {
        log.info("WebSocket 연결: {}", event.getUser());
    }

    @EventListener
    public void handleDisconnect(SessionDisconnectEvent event) {
        log.info("WebSocket 해제: {}", event.getSessionId());
    }
}

// ── 기본 WebSocket API ───────────────────────────────────
const ws = new WebSocket('wss://example.com/ws');

ws.onopen = () => {
    console.log('연결 수립', ws.readyState); // OPEN = 1
    ws.send('안녕하세요!');               // 텍스트 전송
    ws.send(new Uint8Array([1,2,3]));    // 바이너리 전송
};

ws.onmessage = (event) => {
    if (typeof event.data === 'string') {
        const msg = JSON.parse(event.data);
        console.log('수신:', msg);
    }
};

ws.onclose = (event) => {
    // event.code: 1000(정상), 1006(비정상), 1011(서버오류)
    console.log(`종료: ${event.code} ${event.reason}`);
};

ws.onerror = (error) => console.error('오류:', error);

// ── 자동 재연결 래퍼 ─────────────────────────────────────
class ReconnectingWebSocket {
    constructor(url, options = {}) {
        this.url        = url;
        this.maxRetries = options.maxRetries || 5;
        this.retryDelay = options.retryDelay || 1000;
        this.retries    = 0;
        this.connect();
    }

    connect() {
        this.ws = new WebSocket(this.url);

        this.ws.onopen = () => {
            this.retries = 0; // 성공 시 카운터 리셋
            this.onopen?.();
        };

        this.ws.onmessage = (e) => this.onmessage?.(e);

        this.ws.onclose = (e) => {
            if (e.code !== 1000 && this.retries < this.maxRetries) {
                const delay = this.retryDelay * Math.pow(2, this.retries);
                setTimeout(() => this.connect(), delay); // 지수 백오프
                this.retries++;
            }
        };
    }

    send(data) {
        if (this.ws.readyState === WebSocket.OPEN)
            this.ws.send(JSON.stringify(data));
    }
}