├── .gitignore
├── README.md
├── pom.xml
└── src
└── main
├── java
└── hello
│ ├── Application.java
│ ├── CustomSubProtocolWebSocketHandler.java
│ ├── SessionHandler.java
│ └── WebSocketConfig.java
└── resources
└── static
├── index.html
├── sockjs-0.3.4.js
└── stomp.js
/.gitignore:
--------------------------------------------------------------------------------
1 | target
2 |
3 | .idea
4 | *.iml
5 |
--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
1 | # spring-websocket-disconnect
2 | 1. Run the application using: mvn spring-boot:run
3 | 2. Point your browser at http://localhost:8080 and click "Connect" button
4 |
There is job on background, which closes websocket session each 10 seconds
5 |
--------------------------------------------------------------------------------
/pom.xml:
--------------------------------------------------------------------------------
1 |
2 |
4 | 4.0.0
5 |
6 | spring-websocket-disconnect
7 | spring-websocket-disconnect
8 | 0.1.0
9 |
10 |
11 | org.springframework.boot
12 | spring-boot-starter-parent
13 | 1.2.5.RELEASE
14 |
15 |
16 |
17 |
18 | org.springframework.boot
19 | spring-boot-starter-websocket
20 |
21 |
22 | org.springframework
23 | spring-messaging
24 |
25 |
26 |
27 |
28 | 1.8
29 |
30 |
31 |
32 |
33 |
34 | org.springframework.boot
35 | spring-boot-maven-plugin
36 |
37 |
38 |
39 |
40 |
41 |
--------------------------------------------------------------------------------
/src/main/java/hello/Application.java:
--------------------------------------------------------------------------------
1 | package hello;
2 |
3 | import org.springframework.boot.SpringApplication;
4 | import org.springframework.boot.autoconfigure.SpringBootApplication;
5 |
6 | @SpringBootApplication
7 | public class Application {
8 |
9 | public static void main(String[] args) {
10 | SpringApplication.run(Application.class, args);
11 | }
12 | }
13 |
--------------------------------------------------------------------------------
/src/main/java/hello/CustomSubProtocolWebSocketHandler.java:
--------------------------------------------------------------------------------
1 | package hello;
2 |
3 | import org.slf4j.Logger;
4 | import org.slf4j.LoggerFactory;
5 | import org.springframework.beans.factory.annotation.Autowired;
6 | import org.springframework.messaging.MessageChannel;
7 | import org.springframework.messaging.SubscribableChannel;
8 | import org.springframework.stereotype.Component;
9 | import org.springframework.web.socket.WebSocketSession;
10 | import org.springframework.web.socket.messaging.SubProtocolWebSocketHandler;
11 |
12 | import java.io.IOException;
13 | import java.util.Map;
14 | import java.util.concurrent.ConcurrentHashMap;
15 | import java.util.concurrent.Executors;
16 | import java.util.concurrent.ScheduledExecutorService;
17 | import java.util.concurrent.TimeUnit;
18 |
19 | public class CustomSubProtocolWebSocketHandler extends SubProtocolWebSocketHandler {
20 | private static final Logger LOGGER = LoggerFactory.getLogger(CustomSubProtocolWebSocketHandler.class);
21 |
22 | @Autowired
23 | private SessionHandler sessionHandler;
24 |
25 | public CustomSubProtocolWebSocketHandler(MessageChannel clientInboundChannel, SubscribableChannel clientOutboundChannel) {
26 | super(clientInboundChannel, clientOutboundChannel);
27 | }
28 |
29 |
30 | @Override
31 | public void afterConnectionEstablished(WebSocketSession session) throws Exception {
32 | LOGGER.info("New websocket connection was established");
33 | sessionHandler.register(session);
34 | super.afterConnectionEstablished(session);
35 | }
36 | }
37 |
--------------------------------------------------------------------------------
/src/main/java/hello/SessionHandler.java:
--------------------------------------------------------------------------------
1 | package hello;
2 |
3 | import org.slf4j.Logger;
4 | import org.slf4j.LoggerFactory;
5 | import org.springframework.stereotype.Service;
6 | import org.springframework.web.socket.WebSocketSession;
7 |
8 | import java.io.IOException;
9 | import java.util.Map;
10 | import java.util.concurrent.ConcurrentHashMap;
11 | import java.util.concurrent.Executors;
12 | import java.util.concurrent.ScheduledExecutorService;
13 | import java.util.concurrent.TimeUnit;
14 |
15 | @Service
16 | public class SessionHandler {
17 | private static final Logger LOGGER = LoggerFactory.getLogger(SessionHandler.class);
18 | private final ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor();
19 |
20 | private final Map sessionMap = new ConcurrentHashMap<>();
21 |
22 | public SessionHandler() {
23 | scheduler.scheduleAtFixedRate(new Runnable() {
24 | @Override
25 | public void run() {
26 | sessionMap.keySet().forEach(k -> {
27 | try {
28 | sessionMap.get(k).close();
29 | sessionMap.remove(k);
30 | } catch (IOException e) {
31 | LOGGER.error("Error while closing websocket session: {}", e);
32 | }
33 | });
34 | }
35 | }, 10, 10, TimeUnit.SECONDS);
36 | }
37 |
38 | public void register(WebSocketSession session) {
39 | sessionMap.put(session.getId(), session);
40 | }
41 |
42 | }
43 |
--------------------------------------------------------------------------------
/src/main/java/hello/WebSocketConfig.java:
--------------------------------------------------------------------------------
1 | package hello;
2 |
3 | import org.springframework.context.annotation.Bean;
4 | import org.springframework.context.annotation.Configuration;
5 | import org.springframework.messaging.converter.MessageConverter;
6 | import org.springframework.messaging.handler.invocation.HandlerMethodArgumentResolver;
7 | import org.springframework.messaging.handler.invocation.HandlerMethodReturnValueHandler;
8 | import org.springframework.messaging.simp.config.ChannelRegistration;
9 | import org.springframework.messaging.simp.config.MessageBrokerRegistry;
10 | import org.springframework.web.socket.WebSocketHandler;
11 | import org.springframework.web.socket.config.annotation.*;
12 |
13 | import java.util.List;
14 |
15 | @Configuration
16 | @EnableWebSocketMessageBroker
17 | public class WebSocketConfig extends WebSocketMessageBrokerConfigurationSupport implements WebSocketMessageBrokerConfigurer {
18 |
19 | @Override
20 | public void configureWebSocketTransport(WebSocketTransportRegistration registry) {
21 | super.configureWebSocketTransport(registry);
22 | }
23 |
24 | @Override
25 | public boolean configureMessageConverters(List messageConverters) {
26 | return super.configureMessageConverters(messageConverters);
27 | }
28 |
29 | @Override
30 | public void configureClientInboundChannel(ChannelRegistration registration) {
31 | super.configureClientInboundChannel(registration);
32 | }
33 |
34 | @Override
35 | public void configureClientOutboundChannel(ChannelRegistration registration) {
36 | super.configureClientOutboundChannel(registration);
37 | }
38 |
39 |
40 | @Override
41 | public void addArgumentResolvers(List argumentResolvers) {
42 | super.addArgumentResolvers(argumentResolvers);
43 | }
44 |
45 | @Override
46 | public void addReturnValueHandlers(List returnValueHandlers) {
47 | super.addReturnValueHandlers(returnValueHandlers);
48 | }
49 |
50 | @Override
51 | public void configureMessageBroker(MessageBrokerRegistry config) {
52 | config.enableSimpleBroker("/topic");
53 | config.setApplicationDestinationPrefixes("/app");
54 | }
55 |
56 | @Override
57 | public void registerStompEndpoints(StompEndpointRegistry registry) {
58 | registry.addEndpoint("/hello").withSockJS();
59 | }
60 |
61 | @Bean
62 | public WebSocketHandler subProtocolWebSocketHandler() {
63 | return new CustomSubProtocolWebSocketHandler(clientInboundChannel(), clientOutboundChannel());
64 | }
65 | }
66 |
--------------------------------------------------------------------------------
/src/main/resources/static/index.html:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 | Hello WebSocket
5 |
6 |
7 |
18 |
19 |
20 |
21 |
22 |
23 |
24 |