WebSocket实现实时推送数据到前端

本文详细介绍了如何在Spring框架中配置WebSocket,实现客户端与服务器之间的实时通信。通过自定义WebSocketConfig类,注册了WebSocket处理器,并配置了握手拦截器以处理连接建立前后的逻辑。同时,goodsWebSocketHandler类实现了消息接收、处理及向客户端推送消息的功能。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

@Component
@EnableWebSocket
public class WebSocketConfig implements WebSocketConfigurer{

    @Resource
    goodsWebSocketHandler handler;
 
    public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
     
        //配置对指定的url进行拦截
        registry.addHandler(handler, "/mi/*").addInterceptors(new HandShakeInterceptor());
 
        //允许客户端使用SokcetJs
        registry.addHandler(handler, "/mi/*/*").addInterceptors(new HandShakeInterceptor()).withSockJS();
    }

}

/**
 * 每次建立连接都会进行握手,这个拦截器是用于处理握手前后的预处理工作

 *
 */
public class HandShakeInterceptor implements HandshakeInterceptor {

    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
        System.out.println("将要进入WebSocket预处理过程");
        return true;
    }
 
    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception exception) {
     System.out.println("WebSocket预处理中....");
     System.out.println("webSocket预处理完毕");
    }

}

@Component
public class goodsWebSocketHandler implements WebSocketHandler {
     private volatile static List<WebSocketSession> users = new ArrayList<>();
     
     private static final SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
     @Value("${push.time}")
     private Long pushTime;

    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
          users.add(session);
     /*new Thread(new Runnable() {
               @Override
               public void run() {
                    while(true) {
                         for(WebSocketSession user:users) {//向每个在线的客户端推送消息,4秒推送一次
                              try {
                                   user.sendMessage(new TextMessage("服务器推送消息:" + sdf.format(new Date())));
                              } catch (IOException e) {
                                   e.printStackTrace();
                              }
                         }
                         try {//服务器每5秒向每个在线的客户端推送消息
                              Thread.currentThread().sleep(30000);
                         } catch (InterruptedException e) {
                              e.printStackTrace();
                         }
                    }
               }
          }).start();*/
    }
 
    public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
          String gainMessage = message.getPayload().toString();
//        System.err.println("接收到客户端:" + session.getId() + "发送的消息:" + gainMessage);
          Map params = null;
          if (.isNotEmpty(gainMessage)) {
               params = JSON.parseObject(gainMessage);
          }
          if(.isNotEmpty(params.get("HeartBeat")) || MonkeyUtils.isEmpty(gainMessage)){
               Map map = new HashedMap();
               map.put("HeartBeat",true);
               map.put("HeartBeatDate",sdf.format(new Date()));
               session.sendMessage(new TextMessage(JSON.toJSONString(map)));
          }else{
               for (WebSocketSession mySession : users) {
                    if (mySession.equals(session)) {
                         synchronized (this) {
                              if (session.isOpen()) {
                                   Map finalParams = params;
                                   new Thread(new Runnable() {
                                        @Override
                                        public void run() {
                                             while (true) {
                                                  try {
                                                       if(session.isOpen()){
                                                             //业务操作   查询数据
                                                            List<Map> collages = new HashMap<>();
                                                            session.sendMessage(new TextMessage(JSON.toJSONString(collages)));
                                                       }else{
                                                            break;
                                                       }
                                                  } catch (IOException e) {
                                                       e.printStackTrace();
                                                  }
                                                  try {
                                                       //5秒刷新一次
                                                       Thread.sleep(collagePushTime);
                                                  } catch (InterruptedException e) {
                                                       e.printStackTrace();
                                                  }
                                             }
                                        }
                                   }).start();
                              }
                         }
                    }
               }
          }
    }
 
    public void handleTransportError(WebSocketSession session,Throwable exception) throws Exception {
        users.remove(session);
    }
 
    public void afterConnectionClosed(WebSocketSession session,CloseStatus closeStatus) throws Exception {
        users.remove(session);
        session.close();
    }
 
    public boolean supportsPartialMessages() {
        return false;
    }

    private List<Map> getData(WebSocketSession session,Map param){

     return null;
     }

}






<dependency>
            <groupId>javax.websocket</groupId>
            <artifactId>javax.websocket-api</artifactId>
            <version>1.0</version>
            <scope>provided</scope>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-websocket</artifactId>
            <version>4.0.1.RELEASE</version>
        </dependency>
        <dependency>
            <groupId>org.springframework</groupId>
            <artifactId>spring-messaging</artifactId>
            <version>4.0.1.RELEASE</version>
        </dependency>
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值