springboot 如何设置websocket白名单

在Spring Boot中，你可以通过配置WebSocket的拦截器来实现WebSocket的白名单设置。下面是具体的步骤：

1. 创建一个实现WebSocketHandlerInterceptor接口的拦截器类，例如：

   import org.springframework.web.socket.WebSocketHandler;
   import org.springframework.web.socket.handler.WebSocketHandlerDecorator;

   public class WebSocketInterceptor implements WebSocketHandlerInterceptor {

       @Override
       public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Map<String, Object> attributes) throws Exception {
           // 在此处进行白名单判断，如果不在白名单内，则返回false拒绝连接
           String ip = request.getRemoteAddress().getAddress().getHostAddress();
           if (!isInWhiteList(ip)) {
               return false;
           }
           return true;
       }

       @Override
       public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler, Exception exception) {
           // 无需实现
       }

       private boolean isInWhiteList(String ip) {
           // 在此处编写判断逻辑，比如将白名单IP存储在一个Set中，然后判断ip是否在Set中
           Set<String> whiteList = new HashSet<>();
           whiteList.add("127.0.0.1");
           return whiteList.contains(ip);
       }
   }

2. 配置WebSocket拦截器，将拦截器添加到WebSocket注册器中，例如：

   import org.springframework.context.annotation.Configuration;
   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
   @EnableWebSocket
   public class WebSocketConfig implements WebSocketConfigurer {

       @Override
       public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
           registry.addHandler(new WebSocketHandlerDecorator(new YourWebSocketHandler()), "/your-websocket-endpoint")
                   .addInterceptors(new WebSocketInterceptor());
       }
   }

3. 根据你的具体需求，修改WebSocketInterceptor中的isInWhiteList方法，实现自己的白名单判断逻辑。