在Spring websockets中发送错误消息

Question

@MessageExceptionHandler
@SendToUser(value = "/queue/error",broadcast = false)
public ApplicationError handleException(Exception message) throws ApplicationError {
        return  new ApplicationError("test");
}

我订阅了

stompClient.subscribe('/user/queue/error', stompErrorCallback, {token: accessToken});

在我的例子中，用户没有通过身份验证，但从这里

虽然用户目标通常意味着一个经过身份验证的用户，但它并不是严格要求的。不与经过身份验证的用户关联的WebSocket会话可以订阅用户目的地。在这种情况下，@sendtouser注释的行为与broadcast=false完全相同，即只针对发送正在处理的消息的会话。

public class ClientInboundChannelInterceptor extends ChannelInterceptorAdapter {
    @Autowired
    @Lazy(value = true)
    @Qualifier("brokerMessagingTemplate")
    private SimpMessagingTemplate simpMessagingTemplate;

    @Override
    public Message<?> preSend(Message message, MessageChannel channel) throws IllegalArgumentException{
         if(some thing goes wrong)
           throw new RuntimeException();
    }

    @MessageExceptionHandler
    @SendToUser(value = "/queue/error",broadcast = false)
    public ApplicationError handleException(RuntimeException message) throws    ApplicationError {
        return  new ApplicationError("test");
    }
}

@MessageExceptionHandler未捕获此异常。所以我尝试使用SimpMessagingTemplate直接将其发送给用户。

我基本上想做的是：

simpMessagingTemplate.convertAndSendToUser(SOMETHING,"/queue/error",e);

某些内容应该是正确的用户名，但在我的情况下，用户没有经过身份验证，因此我不能使用HeaderAccessor.getUser().getName()

我甚至试过

simpMessagingTemplate.convertAndSendToUser(headerAccessor.getHeader("","/queue/error",e, Collections.singletonMap(SimpMessageHeaderAccessor.SESSION_ID_HEADER, headerAccessor.getSessionId()));

但这不起作用。

我甚至尝试过headeraccessor.getSessionID()来代替username，但这似乎不起作用。

Answer 1

我最初的直觉是正确的，在未经身份验证的用户情况下使用sessionId作为用户名，但问题出在头上。

通过@Sendtouser和SimpMessagingTemplate.ConvertandSendtouser()调试了几个小时后，我意识到如果我们使用@Sendtouser标头将自动设置，如果我们使用的是SimpMessagingTemplate.ConvertandSendtouser()则必须显式定义标头。

@sendtouser正在设置两个标头,

SimpMessageType：SimpMessageType.Message,SimpSessionID：SessionID

所以我尝试添加标题,

String sessionId = headerAccessor.getSessionId();
Map<String,Object> headerMap = new HashMap<>();
headerMap.put("simpMessageType", SimpMessageType.MESSAGE);
headerMap.put("simpSessionId",sessionId);   
simpMessagingTemplate.convertAndSendToUser(headerAccessor.getSessionId(),"/queue/error",e,headerMap);

String sessionId = headerAccessor.getSessionId();
Map<String,Object> headerMap = new HashMap<>();
headerMap.put("simpMessageType", SimpMessageType.MESSAGE);
headerMap.put("simpSessionId",sessionId); 
MessageHeaders headers = new MessageHeaders(headerMap);

simpMessagingTemplate.convertAndSendToUser(headerAccessor.getSessionId(),"/queue/error",e,headers);

经过更多的调试，我找到了设置头的正确方法，这可能是创建这些头的唯一方法（从SendtoMethodReturnValueHandler.java)。

private MessageHeaders createHeaders(String sessionId) {
    SimpMessageHeaderAccessor headerAccessor = SimpMessageHeaderAccessor.create(SimpMessageType.MESSAGE);
    headerAccessor.setSessionId(sessionId);
    headerAccessor.setLeaveMutable(true);
    return headerAccessor.getMessageHeaders();
}

所以最后,

String sessionId = headerAccessor.getSessionId();
template.convertAndSendToUser(sessionId,"/queue/error","tesssssts",createHeaders(sessionId));

成功了。