基于若依框架开发WebSocket接口

添加依赖

在项目的pom.xml文件中添加WebSocket依赖。若依框架默认可能未包含WebSocket支持,需手动引入:

XML 复制代码
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-websocket</artifactId>
</dependency>
配置WebSocket

创建配置类启用WebSocket支持,并注册WebSocket处理器:

java 复制代码
@Configuration
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer {
    @Override
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
        registry.addHandler(myWebSocketHandler(), "/ws")
                .setAllowedOrigins("*");
    }

    @Bean
    public WebSocketHandler myWebSocketHandler() {
        return new MyWebSocketHandler();
    }
}
实现WebSocket处理器

自定义处理器继承TextWebSocketHandler,处理连接、消息和关闭事件:

java 复制代码
public class MyWebSocketHandler extends TextWebSocketHandler {
    private static final Set<WebSocketSession> sessions = new CopyOnWriteArraySet<>();

    @Override
    public void afterConnectionEstablished(WebSocketSession session) {
        sessions.add(session);
        System.out.println("New connection: " + session.getId());
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) {
        System.out.println("Received message: " + message.getPayload());
        for (WebSocketSession s : sessions) {
            try {
                s.sendMessage(new TextMessage("Echo: " + message.getPayload()));
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) {
        sessions.remove(session);
        System.out.println("Connection closed: " + session.getId());
    }
}
前端连接WebSocket

在Vue或HTML页面中建立WebSocket连接:

javascript 复制代码
const socket = new WebSocket('ws://localhost:8080/ws');
socket.onopen = function() {
    console.log('WebSocket connected');
    socket.send('Hello Server');
};
socket.onmessage = function(event) {
    console.log('Message from server: ', event.data);
};
若依集成注意事项
  1. 权限控制 :若需结合若依的权限系统,可在WebSocket握手时验证Token,例如通过HandshakeInterceptor
  2. 消息广播:结合Redis发布订阅功能实现集群环境下的消息广播。
  3. 心跳检测:客户端需定期发送心跳包,避免连接被意外关闭。
示例:握手拦截器

添加权限验证拦截器:

java 复制代码
public class AuthHandshakeInterceptor implements HandshakeInterceptor {
    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, 
                                   WebSocketHandler wsHandler, Map<String, Object> attributes) {
        String token = ((ServletServerHttpRequest) request).getServletRequest().getParameter("token");
        // 验证Token逻辑
        return true; // 验证通过返回true
    }

    @Override
    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, 
                              WebSocketHandler wsHandler, Exception exception) {}
}

在配置类中注册拦截器:

java 复制代码
registry.addHandler(myWebSocketHandler(), "/ws")
       .addInterceptors(new AuthHandshakeInterceptor())
       .setAllowedOrigins("*");

通过以上步骤,即可在若依框架中实现基础的WebSocket通信功能。

相关推荐
瓯雅爱分享4 小时前
Java搭建高效后端,Vue打造友好前端,联合构建电子采购管理系统,实现采购流程电子化、自动化,涵盖采购全周期管理,功能完备,附详细可运行源码
java·mysql·vue·软件工程·源代码管理
太白IT记4 小时前
form表达和实体类通常有什么不同
java·状态模式
架构师沉默4 小时前
同事查日志太慢,我现场教他一套 grep 组合拳
java·后端·架构
华仔啊4 小时前
前端发版总被用户说“没更新”?一文搞懂浏览器缓存,彻底解决!
前端·javascript
Seven974 小时前
剑指offer-27、字符串的排列
java
SimonKing4 小时前
弃用MinIO,拥抱全新一代分布式文件系统RustFS
java·后端·程序员
AAA修煤气灶刘哥5 小时前
后端人必看!配置中心这玩意儿,用 Nacos 玩明白能少熬 3 个夜
java·后端·spring cloud
一蓑烟雨,一任平生5 小时前
h5实现内嵌微信小程序&支付宝 --截图保存海报分享功能
开发语言·前端·javascript
苏纪云5 小时前
Ajax笔记(下)
前端·javascript·笔记·ajax