https://www.baeldung.com/spring-websockets-send-message-to-user

https://kmdigit.github.io/2020/03/12/send-message-direct-user-in-spring-websocket/

https://withseungryu.tistory.com/136

https://stackoverflow.com/questions/34929578/spring-websocket-sendtosession-send-message-to-specific-session

https://stackoverflow.com/questions/37853727/where-user-comes-from-in-convertandsendtouser-works-in-sockjsspring-websocket

웹소켓 접속시마다 클라이언트를 구분할 수 있도록 Principal을 상속한 클래스를 만든다.

import java.security.Principal;

public class StompPrincipal implements Principal {
    private String name;

    public StompPrincipal(String name) {
        this.name = name;
    }

    @Override
    public String getName() {
        return name;
    }
}

웹소켓 접속시에 handshake 부분에서 클라이언트에게 principal을 하나씩 주어준다.

import com.fm.webinar.config.handler.StompHandler;
import com.fm.webinar.security.StompPrincipal;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.messaging.simp.config.ChannelRegistration;
import org.springframework.messaging.simp.config.MessageBrokerRegistry;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker;
import org.springframework.web.socket.config.annotation.StompEndpointRegistry;
import org.springframework.web.socket.config.annotation.WebSocketMessageBrokerConfigurer;
import org.springframework.web.socket.server.support.DefaultHandshakeHandler;

import java.security.Principal;
import java.util.Map;
import java.util.UUID;

@Slf4j
@RequiredArgsConstructor
@Configuration
@EnableWebSocketMessageBroker
public class WebSockConfig implements WebSocketMessageBrokerConfigurer {

    private final StompHandler stompHandler;

    @Override
    public void configureMessageBroker(MessageBrokerRegistry config) {
        config.enableSimpleBroker("/sub");
        config.setApplicationDestinationPrefixes("/pub");
    }

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/webinar/stomp")
                .setAllowedOriginPatterns("*") // 스프링 5.3에서부터는 패턴으로 해주지 않으면 안된다.
                .setHandshakeHandler(
                        new DefaultHandshakeHandler() {
                            @Override
                            protected Principal determineUser(ServerHttpRequest request, WebSocketHandler wsHandler, Map<String, Object> attributes) {
                                return new StompPrincipal(UUID.randomUUID().toString());
                            }
                        }
                )
                .withSockJS()
        ; // sock.js를 통하여 낮은 버전의 브라우저에서도 websocket이 동작할수 있게 합니다.
    }

    @Override
    public void configureClientInboundChannel(ChannelRegistration registration) {
        registration.interceptors(stompHandler);
    }

}

handshake 에서 할당한 principal.name을 socketPrincipalName 이름으로 받아와서 redis에 사용자 정보와 함께 저장한다.

import com.fm.webinar.domain.session.Session;
import com.fm.webinar.exception.SessionNullException;
import com.fm.webinar.exception.SocketAccessDeniedException;
import com.fm.webinar.service.SessionService;
import com.fm.webinar.service.WhoamiService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessageChannel;
import org.springframework.messaging.simp.stomp.StompCommand;
import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
import org.springframework.messaging.support.ChannelInterceptor;
import org.springframework.messaging.support.MessageHeaderAccessor;
import org.springframework.stereotype.Component;

import java.sql.Timestamp;
import java.util.Optional;

@Slf4j
@RequiredArgsConstructor
@Component
public class StompHandler implements ChannelInterceptor {

    private final WhoamiService whoamiService;
    private final SessionService sessionService;

    private String getSessionId(String destination) {
        String[] route = destination.split("/");
        if (route.length < 4) {
            return null;
        } else {
            return route[3];
        }
    }


    // websocket을 통해 들어온 요청이 처리 되기전 실행된다.
    @Override
    public Message<?> preSend(Message<?> message, MessageChannel channel) {

        StompHeaderAccessor accessor = MessageHeaderAccessor.getAccessor(message, StompHeaderAccessor.class);
        String jwtToken = accessor.getFirstNativeHeader("Authorization");
        String eventSessionId = accessor.getFirstNativeHeader("EventSessionId");
        String socketPrincipalName = accessor.getUser().getName();

        if (StompCommand.CONNECT == accessor.getCommand()) { // websocket 연결요청
            log.info("CONNECT {}", jwtToken);
            try {
                whoamiService.setSessionUser(socketPrincipalName, eventSessionId, jwtToken);
            } catch (Exception e) {
                log.error("CONNECT {}", e.getMessage(), e);
                throw new SocketAccessDeniedException("권한이 없습니다.");
            }

        } else if (StompCommand.SUBSCRIBE == accessor.getCommand()) { // 채팅룸 구독요청
            // header정보에서 구독 destination정보를 얻고, sessionId를 추출한다.
            String sessionId = getSessionId(Optional.ofNullable((String) message.getHeaders().get("simpDestination")).orElse("InvalidSessionId"));

            // 개인 메시지는 /user/로 시작한다.
            if (!message.getHeaders().get("simpDestination").toString().startsWith("/user/")) {
                Session session = sessionService.findById(sessionId);
                if (session == null) {
                    throw new SessionNullException("해당 세션이 없습니다.");
                } else {
                    Timestamp now = new Timestamp(System.currentTimeMillis());
                    if (session.getStartDate().after(now) || now.after(session.getEndDate())) {
                        throw new SessionNullException("해당 세션이 없습니다.");
                    }
                }
            }

        } else if (StompCommand.DISCONNECT == accessor.getCommand()) { // Websocket 연결 종료
            whoamiService.removeSessionUser(socketPrincipalName);
        }

        return message;
    }
}

채팅 메시지 수신시에 principal을 함께 받아서 누가 보낸 메시지인지 확인할 수 있다.

@Slf4j
@RequiredArgsConstructor
@Controller
public class ChatController {

    private final ChatService chatService;
    private final WhoamiService whoamiService;

    /**
     * 채팅에 관한 메시지만 처리한다.
     * websocket "/pub/chat"로 들어오는 메시징을 처리한다.
     */
    @MessageMapping("/chat")
    public void chatMessage(ChatMessage message, Principal principal) {
        // 토큰 체크 및 사용자 셋팅
        User user = whoamiService.getUserByPrincipalName(principal.getName());
        ChatMessage addMessage = message.toBuilder()
                .senderName(user.getName())
                .senderId(user.getUserId())
                .userCount(chatService.getUserCount(message.getSessionId()))
                .build();
        chatService.sendChatMessage(addMessage);
    }

}

message.targetId는 principal.name 이다.

이렇게 하면 특정 사용자에게만 메시지를 보낼 수 있으며, 해당 사용자가 누구인지도 확인이 가능하다.

@Slf4j
@RequiredArgsConstructor
@Service
public class CommandSubscriber {
    private final ObjectMapper objectMapper;
    private final SimpMessagingTemplate template;

    public void receive(String commandMessage) throws JsonProcessingException {
        log.info("@@@@@@@@@@ command 구독함 @@@@@@@@@ {}", commandMessage);
        CommandMessage message = objectMapper.readValue(commandMessage, CommandMessage.class);

        if (message.getType() == CommandMessageType.FORCED_EXIT_COMMAND) {
            // 중복로그인으로 인한 강제 퇴장인 경우 개인에게만 보낸다.
            this.template.convertAndSendToUser(message.getTargetId(), "/sub/command/" + message.getSessionId(), message);
        } else {
            this.template.convertAndSend("/sub/command/" + message.getSessionId(), message);
        }
    }
}

Spring WebSockets: 특정 사용자에게 메시지 보내기