webSocket实现

最新推荐文章于 2023-01-17 21:26:02 发布

原创最新推荐文章于 2023-01-17 21:26:02 发布 · 置顶 · 537 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#websocket

java 专栏收录该内容

2 篇文章

订阅专栏

该博客介绍了如何在Java中实现WebSocket连接，并提供了一个SocketHandler类的详细代码示例，该类处理WebSocket会话的建立、关闭、消息传递以及错误处理。WebSocketConfig配置类用于注册WebSocket处理器，并设置拦截URL。此外，还包含一个简单的测试用例，展示了如何向所有在线用户发送消息。

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

package cn.clgate.atlas.webSocket;

import org.apache.log4j.Logger;
import org.springframework.stereotype.Service;
import org.springframework.web.socket.*;

import java.io.IOException;
import java.util.ArrayList;


@Service
public class SocketHandler implements WebSocketHandler{

   private static final Logger logger;

   private static final ArrayList<WebSocketSession> webSocke;
   
   static{
      webSocke = new ArrayList<WebSocketSession>();
      logger = Logger.getLogger(SocketHandler.class);
   }
   
   @Override
   public void afterConnectionEstablished(WebSocketSession session)
         throws Exception {
      logger.info("成功建立socket连接");
      webSocke.add(session);
      session.sendMessage(new TextMessage("成功建立socket通信了"));
   }

   @Override
   public void handleMessage(WebSocketSession session, WebSocketMessage<?> message)
         throws Exception {
      // TODO Auto-generated method stub
   }

   @Override
   public void handleTransportError(WebSocketSession session, Throwable error)
         throws Exception {
      if(session.isOpen()){
         session.close();
      }
      logger.error("连接出现错误:"+error.toString());
      webSocke.remove(session);
   }
   
   @Override
   public void afterConnectionClosed(WebSocketSession session, CloseStatus arg1)
         throws Exception {
      logger.debug("连接已关闭");
      logger.info("连接已关闭");
      webSocke.remove(session);
   }

   @Override
   public boolean supportsPartialMessages() {
      return false;
   }
   
   /**
     * 给所有在线用户发送消息
     *
     * @param message
     */
    public void sendMessageToUsers(TextMessage message) {
        for (WebSocketSession user : webSocke) {
            try {
                if (user.isOpen()) {
                    user.sendMessage(message);
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
 


}

package cn.clgate.atlas.webSocket;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.EnableWebMvc;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;


@Configuration
@EnableWebMvc
@EnableWebSocket
public class WebSocketConfig extends WebMvcConfigurerAdapter implements WebSocketConfigurer{
   
   @Autowired
   private SocketHandler socketHandler;

   @Override
   public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
      //注册处理拦截器,拦截url为socketServer的请求
      registry.addHandler(socketHandler, "/socketServer");
      
      //注册SockJs的处理拦截器,拦截url为/sockjs/socketServer的请求
      registry.addHandler(socketHandler, "/sockjs/socketServer");
   }

}

测试类:

@org.junit.Test
public void test(){
    SocketHandler socketHandler = new SocketHandler();
    socketHandler.sendMessageToUsers(new TextMessage("webSocket向前端发了一条消息"));
}