使用Java WebSocket获取客户端IP地址的示例代码

@ServerEndpoint("/websocket")
public class WebSocketServer {
    @OnOpen
    public void onOpen(Session session, EndpointConfig config) {
        String clientIp = getClientIpAddress(session);
        System.out.println("New connection from IP: " + clientIp);
    }

    @OnMessage
    public void onMessage(String message, Session session) {
        // 处理收到的消息
    }

    @OnClose
    public void onClose(Session session, CloseReason closeReason) {
        String clientIp = getClientIpAddress(session);
        System.out.println("Connection closed from IP: " + clientIp);
    }

    private String getClientIpAddress(Session session) {
        String ipAddress = null;
        MessageContext messageContext = session.getUserProperties().get("javax.websocket.server.ServerEndpointConfig.MESSAGE_CONTEXT")；
        if (messageContext != null) {
            HttpServletRequest request = (HttpServletRequest) messageContext.get(ServletRequest.class.getName());
            if (request != null) {
                ipAddress = request.getRemoteAddr();
            }
        }
        return ipAddress;
    }
}

@Controller
public class WebSocketController extends TextWebSocketHandler {
    @Override
    public void afterConnectionEstablished(WebSocketSession session) throws Exception {
        String clientIp = getClientIpAddress(session);
        System.out.println("New connection from IP: " + clientIp);
    }

    @Override
    protected void handleTextMessage(WebSocketSession session, TextMessage message) throws Exception {
        // 处理收到的消息
    }

    @Override
    public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
        String clientIp = getClientIpAddress(session);
        System.out.println("Connection closed from IP: " + clientIp);
    }

    private String getClientIpAddress(WebSocketSession session) {
        String ipAddress = null;
        HttpSession httpSession = (HttpSession) session.getAttributes().get("HTTP_SESSION");
        if (httpSession != null) {
            ipAddress = httpSession.getServletContext().getRealPath();
        }
        return ipAddress;
    }
}

@ChannelHandler.Sharable
public class WebSocketHandler extends SimpleChannelInboundHandler<WebSocketFrame> {
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        SocketAddress socketAddress = ctx.channel().remoteAddress();
        if (socketAddress instanceof InetSocketAddress) {
            InetSocketAddress inetSocketAddress = (InetSocketAddress) socketAddress;
            String clientIp = inetSocketAddress.getAddress().getHostAddress();
            System.out.println("New connection from IP: " + clientIp);
        }
    }

    @Override
    protected void channelRead0(ChannelHandlerContext ctx, WebSocketFrame frame) throws Exception {
        // 处理收到的消息
    }

    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        SocketAddress socketAddress = ctx.channel().remoteAddress();
        if (socketAddress instanceof InetSocketAddress) {
            InetSocketAddress inetSocketAddress = (InetSocketAddress) socketAddress;
            String clientIp = inetSocketAddress.getAddress().getHostAddress();
            System.out.println("Connection closed from IP: " + clientIp);
        }
    }
}