Spring启动Web套接字中的实时通知

Question

在我的应用程序中，我需要向特定用户发送实时通知。我的WebSocketConfig类如下，

    @Configuration
    @EnableWebSocketMessageBroker
    public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer {
        @Override
        public void registerStompEndpoints(StompEndpointRegistry stompEndpointRegistry) {
            stompEndpointRegistry.addEndpoint("/websocket-example")
                    .withSockJS();
        }

        @Override
        public void configureMessageBroker(MessageBrokerRegistry registry) {
            registry.enableSimpleBroker("/topic");
    }
}

大多数情况下，信息将由服务器端发送。因此，我没有设置应用程序目标。

在客户端，我订阅目标“/主题/用户”，

function connect() {
    var socket = new SockJS('/websocket-example');
    stompClient = Stomp.over(socket);
    stompClient.connect({}, function (frame) {
        setConnected(true);
        console.log('Connected: ' + frame);
        stompClient.subscribe('/topic/user', function (greeting) {
//            showGreeting(JSON.parse(greeting.body).content);
console.log("Received message through WS");
        });
    });
}

在我的一个RestController中，我有一个方法可以将消息广播给所有连接的客户端。

    @GetMapping("/test")
    public void test()
    {
        template.convertAndSend("/topic/user", "Hurray");
    }

在这部分之前，一切正常。我收到消息并正在登录控制台。

现在，如果我想只向特定用户发送通知，我必须使用template.convertAndSendToUser（String user， String目标，String消息）。但是我不明白我应该传递给user参数的内容。我将在哪里以及何时获得user？

我问了几个与此相关的问题，但我不清楚这些概念。

Answer 1

这个用户名是java的一部分。安全主要接口。每个StompHeaderAccessor或WebSocket会话对象都有一个此原则的实例，您可以从中获取用户名。它不是自动生成的<代码>它必须由服务器为每个会话手动生成

您可以在此处查看有关为每个会话生成唯一id的更多信息。

然后像这样使用：

@MessageMapping('/test')
public void test(SimpMessageHeaderAccessor sha)
{
  String userName = sha.session.principal.name;
  template.convertAndSend(userName, '/topic/user', "Hurray");
}

Answer 2

在向用户发送任何消息之前，您需要首先通过服务器对其进行身份验证。有不同的方法可以做到这一点。Spring安全性是这里的一个关键短语

https://docs.spring.io/spring-security/site/docs/current/guides/html5/helloworld-boot.html

身份验证完成后，只需调用以下命令即可获取用户名：

Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
String currentPrincipalName = authentication.getName();

https://www.baeldung.com/get-user-in-spring-security