Spring boot获取websocket客户端ip

在Spring Boot中，你可以通过实现HandshakeInterceptor接口来获取WebSocket客户端的IP地址。以下是一个简单的例子：

import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.server.HandshakeInterceptor;
 
import java.util.Map;
 
public class WebSocketHandshakeInterceptor implements HandshakeInterceptor {
 
    @Override
    public boolean beforeHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
                                   Map<String, Object> attributes) throws Exception {
        // 获取客户端IP地址
        String clientIP = request.getRemoteAddress().getAddress().getHostAddress();
        attributes.put("clientIP", clientIP);
        return true; // 继续握手
    }
 
    @Override
    public void afterHandshake(ServerHttpRequest request, ServerHttpResponse response, WebSocketHandler wsHandler,
                               Exception ex) {
        // 握手后的操作
    }
}

然后，你需要将这个拦截器注册到Spring的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) {
        // 注册WebSocket拦截器
        registry.addHandler(myWebSocketHandler(), "/websocket-endpoint")
                .addInterceptors(new WebSocketHandshakeInterceptor());
    }
 
    // 你的WebSocketHandler定义
    private MyWebSocketHandler myWebSocketHandler() {
        return new MyWebSocketHandler();
    }
}

在你的WebSocketHandler实现中，你可以通过attributes.get("clientIP")来获取客户端IP地址：

import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.TextWebSocketHandler;
 
import java.util.Map;
 
public class MyWebSocketHandler extends Te