Spring Stomp连接缓慢

时间:2019-11-15 11:22:23

标签: spring websocket stomp

两个Spring微服务之间的Stomp连接速度很慢,客户端大约需要5分钟才能发送CONNECT Stomp消息。两种微服务都在同一节点上运行。

Spring Boot版本:2.1.3.RELEASE

通过以下方式配置Websocket服务器:

@Configuration
@EnableWebSocketMessageBroker
public class WebSocketConfig implements WebSocketMessageBrokerConfigurer {

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

    @Override
    public void registerStompEndpoints(StompEndpointRegistry registry) {
        registry.addEndpoint("/ws-subscribe").setAllowedOrigins("*").withSockJS();
    }
}

以这种方式配置客户端:

@Component
public class WebsocketClient {

    private GlobalConfiguration globalConfiguration;
    private WebsocketListener websocketListener;

    private WebSocketStompClient stompClient;

    @Autowired
    private JwtTokenGenerator jwtTokenGenerator;

    private ApplicationContext appContext;

    private Logger LOGGER = LoggerFactory.getLogger(WebsocketClient.class);

    @Autowired
    public WebsocketClient(ApplicationContext context,GlobalConfiguration globalConfiguration) {
        Transport webSocketTransport = new WebSocketTransport(new StandardWebSocketClient());
        List<Transport> transports = Collections.singletonList(webSocketTransport);

        SockJsClient sockJsClient = new SockJsClient(transports);
        sockJsClient.setMessageCodec(new Jackson2SockJsMessageCodec());

        this.stompClient = new WebSocketStompClient(sockJsClient);

        this.globalConfiguration = globalConfiguration;
        this.websocketListener = new WebsocketListener(context);
    }

    public ListenableFuture<StompSession> connectToWebsocket(){
       String url = globalConfiguration.getOrchestratorWs();

        StompHeaders connectHeaders = new StompHeaders();

        WebSocketHttpHeaders webSocketHttpHeaders = new WebSocketHttpHeaders();
        webSocketHttpHeaders.add("Authorization",jwtTokenGenerator.token);

        LOGGER.info("webSocketHttpHeaders: " + webSocketHttpHeaders.toString());
        LOGGER.info("url: " + url);

        this.stompClient.setMessageConverter(new MappingJackson2MessageConverter());
        return stompClient.connect(url,webSocketHttpHeaders,connectHeaders, websocketListener);
    }
}

在客户端,我们有这个Websocket侦听器:

public class WebsocketListener extends StompSessionHandlerAdapter {

    private static final Logger LOGGER = LoggerFactory.getLogger(WebsocketListener.class);


    private GlobalConfiguration globalConfiguration;
    private ApplicationContext appContext;


    public WebsocketListener(ApplicationContext appContext) {
        this.appContext = appContext;
        this.globalConfiguration = appContext.getBean(GlobalConfiguration.class);
    }

    @Override
    public void afterConnected(StompSession session, StompHeaders connectedHeaders) {
        LOGGER.info("New session established : " + globalConfiguration.getId());
        session.subscribe("/topic/injections/"+globalConfiguration.getId(), this);
        LOGGER.info("Subscribed to /topic/injections/"+globalConfiguration.getId());
      //  session.send("/app/chat", getSampleMessage());
      //  logger.info("Message sent to websocket server");
    }

    @Override
    public void handleException(StompSession session, StompCommand command, StompHeaders headers, byte[] payload, Throwable exception) {
        LOGGER.error("Got an exception", exception);
    }

    @Override
    public Type getPayloadType(StompHeaders headers) {
        return Map.class;
    }

    @Override
    public void handleFrame(StompHeaders headers, Object payload) {
        LOGGER.info("Injection Received : " + payload);
        InjectionHandler injectionHandler = new InjectionHandler(appContext,(HashMap) payload);
        injectionHandler.start();
    }
}

ws://10.2.0.43:7071 / ws-subscribe / 513 / 8306a7ac357847678795b049450fb6c5 / websocket大约5分钟后调用afterConnected

检查Stomp库的源代码,我发现微服务在这5分钟内停留的代码如下:

public ListenableFuture<StompSession> connect(URI url, @Nullable WebSocketHttpHeaders handshakeHeaders, @Nullable StompHeaders connectHeaders, StompSessionHandler sessionHandler) {
    Assert.notNull(url, "'url' must not be null");
    ConnectionHandlingStompSession session = this.createSession(connectHeaders, sessionHandler);
    WebSocketStompClient.WebSocketTcpConnectionHandlerAdapter adapter = new WebSocketStompClient.WebSocketTcpConnectionHandlerAdapter(session);
    this.getWebSocketClient().doHandshake(adapter, handshakeHeaders, url).addCallback(adapter);
    return session.getSessionFuture();
}

尤其是session.getSessionFuture()调用。 关于什么原因可能导致客户端发送的CONNECT消息延迟的任何想法?

0 个答案:

没有答案