springboot websocket session共享 代码编写

你好，关于springboot websocket session共享的代码编写，可以使用Spring的内置WebSocket支持来实现。下面是参考示例代码：

1、首先，在Spring boot项目的依赖中加入WebSocket支持：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>

2、编写WebSocket配置类：

@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(myHandler(), "/myHandler").setAllowedOrigins("*");
    }
    
    @Bean
    public WebSocketHandler myHandler() {
        return new MyWebSocketHandler();
    }
    
}

3、实现WebSocketHandler接口：

public class MyWebSocketHandler implements WebSocketHandler {
    
    private List<WebSocketSession> sessions = new ArrayList<>();
    
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        sessions.add(session);
    }
    
    @Override
    public void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        for (WebSocketSession s : sessions) {
            s.sendMessage(message);
        }
    }
    
    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
        sessions.remove(session);
    }
    
    @Override
    public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
        sessions.remove(session);
    }
    
}

4、在需要共享session的Controller中注入WebSocketSession对象：

@Controller
public class MyController {
    
    @Autowired
    private WebSocketSession webSocketSession;
    
    @MessageMapping("/sendMessage")
    @SendTo("/topic/messages")
    public String sendMessage(String message) throws Exception {
        webSocketSession.sendMessage(new TextMessage(message));
        return message;
    }
    
}

以上代码仅供参考，具体实现需根据业务需求进行改造。希望能对你有所帮助！