Package org.springframework.web.socket.messaging

Class SubProtocolWebSocketHandler

java.lang.Object
org.springframework.web.socket.messaging.SubProtocolWebSocketHandler
All Implemented Interfaces:
org.springframework.context.Lifecycle, org.springframework.context.Phased, org.springframework.context.SmartLifecycle, org.springframework.messaging.MessageHandler, SubProtocolCapable, WebSocketHandler
public class SubProtocolWebSocketHandler extends Object implements WebSocketHandler, SubProtocolCapable, org.springframework.messaging.MessageHandler, org.springframework.context.SmartLifecycle
An implementation of WebSocketHandler that delegates incoming WebSocket messages to a SubProtocolHandler along with a MessageChannel to which the sub-protocol handler can send messages from WebSocket clients to the application.

Also an implementation of MessageHandler that finds the WebSocket session associated with the Message and passes it, along with the message, to the sub-protocol handler to send messages from the application back to the client.

Since:
4.0
Author:
Rossen Stoyanchev, Juergen Hoeller, Andy Wilkinson, Artem Bilan

  • Constructor Details

    • SubProtocolWebSocketHandler

      public SubProtocolWebSocketHandler(org.springframework.messaging.MessageChannel clientInboundChannel, org.springframework.messaging.SubscribableChannel clientOutboundChannel)
      Create a new SubProtocolWebSocketHandler for the given inbound and outbound channels.
      Parameters:
      clientInboundChannel - the inbound MessageChannel
      clientOutboundChannel - the outbound MessageChannel

  • Method Details

    • setProtocolHandlers

      public void setProtocolHandlers(List<SubProtocolHandler> protocolHandlers)
      Configure one or more handlers to use depending on the sub-protocol requested by the client in the WebSocket handshake request.
      Parameters:
      protocolHandlers - the sub-protocol handlers to use

    • getProtocolHandlers

      public List<SubProtocolHandler> getProtocolHandlers()

    • addProtocolHandler

      public void addProtocolHandler(SubProtocolHandler handler)
      Register a sub-protocol handler.

    • getProtocolHandlerMap

      public Map<String,SubProtocolHandler> getProtocolHandlerMap()
      Return the sub-protocols keyed by protocol name.

    • setDefaultProtocolHandler

      public void setDefaultProtocolHandler(@Nullable SubProtocolHandler defaultProtocolHandler)
      Set the SubProtocolHandler to use when the client did not request a sub-protocol.
      Parameters:
      defaultProtocolHandler - the default handler

    • getDefaultProtocolHandler

      @Nullable public SubProtocolHandler getDefaultProtocolHandler()
      Return the default sub-protocol handler to use.

    • getSubProtocols

      public List<String> getSubProtocols()
      Return all supported protocols.
      Specified by:
      getSubProtocols in interface SubProtocolCapable

    • setSendTimeLimit

      public void setSendTimeLimit(int sendTimeLimit)
      Specify the send-time limit (milliseconds).
      See Also:

    • getSendTimeLimit

      public int getSendTimeLimit()
      Return the send-time limit (milliseconds).

    • setSendBufferSizeLimit

      public void setSendBufferSizeLimit(int sendBufferSizeLimit)
      Specify the buffer-size limit (number of bytes).
      See Also:

    • getSendBufferSizeLimit

      public int getSendBufferSizeLimit()
      Return the buffer-size limit (number of bytes).

    • setTimeToFirstMessage

      public void setTimeToFirstMessage(int timeToFirstMessage)
      Set the maximum time allowed in milliseconds after the WebSocket connection is established and before the first sub-protocol message is received.

      This handler is for WebSocket connections that use a sub-protocol. Therefore, we expect the client to send at least one sub-protocol message in the beginning, or else we assume the connection isn't doing well, e.g. proxy issue, slow network, and can be closed.

      By default this is set to 60,000 (1 minute).

      Parameters:
      timeToFirstMessage - the maximum time allowed in milliseconds
      Since:
      5.1
      See Also:
      • checkSessions()

    • getTimeToFirstMessage

      public int getTimeToFirstMessage()
      Return the maximum time allowed after the WebSocket connection is established and before the first sub-protocol message.
      Since:
      5.1

    • setPhase

      public void setPhase(int phase)
      Set the phase that this handler should run in.

      By default, this is SmartLifecycle.DEFAULT_PHASE.

      Since:
      6.1.4

    • getPhase

      public int getPhase()
      Specified by:
      getPhase in interface org.springframework.context.Phased
      Specified by:
      getPhase in interface org.springframework.context.SmartLifecycle

    • getStatsInfo

      public String getStatsInfo()
      Return a String describing internal state and counters. Effectively toString() on getStats().

    • getStats

      public SubProtocolWebSocketHandler.Stats getStats()
      Return a structured object with various session counters.
      Since:
      5.2

    • start

      public final void start()
      Specified by:
      start in interface org.springframework.context.Lifecycle

    • stop

      public final void stop()
      Specified by:
      stop in interface org.springframework.context.Lifecycle

    • stop

      public final void stop(Runnable callback)
      Specified by:
      stop in interface org.springframework.context.SmartLifecycle

    • isRunning

      public final boolean isRunning()
      Specified by:
      isRunning in interface org.springframework.context.Lifecycle

    • afterConnectionEstablished

      public void afterConnectionEstablished(WebSocketSession session) throws Exception
      Description copied from interface: WebSocketHandler
      Invoked after WebSocket negotiation has succeeded and the WebSocket connection is opened and ready for use.
      Specified by:
      afterConnectionEstablished in interface WebSocketHandler
      Throws:
      Exception - this method can handle or propagate exceptions; see class-level Javadoc for details.

    • handleMessage

      public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception
      Handle an inbound message from a WebSocket client.
      Specified by:
      handleMessage in interface WebSocketHandler
      Throws:
      Exception - this method can handle or propagate exceptions; see class-level Javadoc for details.

    • handleMessage

      public void handleMessage(org.springframework.messaging.Message<?> message) throws org.springframework.messaging.MessagingException
      Handle an outbound Spring Message to a WebSocket client.
      Specified by:
      handleMessage in interface org.springframework.messaging.MessageHandler
      Throws:
      org.springframework.messaging.MessagingException

    • handleTransportError

      public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception
      Description copied from interface: WebSocketHandler
      Handle an error from the underlying WebSocket message transport.
      Specified by:
      handleTransportError in interface WebSocketHandler
      Throws:
      Exception - this method can handle or propagate exceptions; see class-level Javadoc for details.

    • afterConnectionClosed

      public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception
      Description copied from interface: WebSocketHandler
      Invoked after the WebSocket connection has been closed by either side, or after a transport error has occurred. Although the session may technically still be open, depending on the underlying implementation, sending messages at this point is discouraged and most likely will not succeed.
      Specified by:
      afterConnectionClosed in interface WebSocketHandler
      Throws:
      Exception - this method can handle or propagate exceptions; see class-level Javadoc for details.

    • supportsPartialMessages

      public boolean supportsPartialMessages()
      Description copied from interface: WebSocketHandler
      Whether the WebSocketHandler handles partial messages. If this flag is set to true and the underlying WebSocket server supports partial messages, then a large WebSocket message, or one of an unknown size may be split and maybe received over multiple calls to WebSocketHandler.handleMessage(WebSocketSession, WebSocketMessage). The flag WebSocketMessage.isLast() indicates if the message is partial and whether it is the last part.
      Specified by:
      supportsPartialMessages in interface WebSocketHandler

    • decorateSession

      protected WebSocketSession decorateSession(WebSocketSession session)
      Decorate the given WebSocketSession, if desired.

      The default implementation builds a ConcurrentWebSocketSessionDecorator with the configured send-time limit and buffer-size limit.

      Parameters:
      session - the original WebSocketSession
      Returns:
      the decorated WebSocketSession, or potentially the given session as-is
      Since:
      4.3.13

    • findProtocolHandler

      protected final SubProtocolHandler findProtocolHandler(WebSocketSession session)
      Find a SubProtocolHandler for the given session.
      Parameters:
      session - the WebSocketSession to find a handler for

    • toString

      public String toString()
      Overrides:
      toString in class Object