java 负载均衡下获取客户端ip

原创

mob649e81643021 2024-11-10 05:09:22 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e81643021的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java 负载均衡下获取客户端 IP 的方法解析

在微服务架构中，负载均衡是提升系统性能与可用性的关键策略之一。负载均衡可以在多台服务器之间分配请求，确保系统的高可用性与良好的用户体验。然而，负载均衡在处理客户端请求时，常常会导致获取客户端 IP 的问题。本文将探讨在 Java 中如何在负载均衡环境下正确地获取客户端 IP。

为什么获取客户端 IP 变得复杂

当采用负载均衡器（如 Nginx、HAProxy 或云负载均衡服务）时，请求首先到达负载均衡器，然后再转发到后端的应用服务器。这一过程中，客户端的真实 IP 地址可能会被替换为负载均衡器的地址。为了从请求中恢复原来的客户端 IP 地址，通常需要使用 X-Forwarded-For 头部。

基本流程

获取客户端 IP 的基本流程如下：

客户端发送请求至负载均衡器。
负载均衡器将请求转发至后端应用，同时在请求头中添加 X-Forwarded-For。
后端应用分析请求头中的 X-Forwarded-For，获取真实的客户端 IP。

下面是用 Mermaid 语法表示的流程图：

flowchart TD
    A[客户端发送请求] --> B[负载均衡器转发请求]
    B --> C[后端应用分析请求]
    C --> D[获取客户端真实 IP]

实现代码

在 Java 中，我们可以通过 Servlet API 获取请求的头信息。以下是一个示例代码，演示如何获取客户端的真实 IP 地址：

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServlet;
import java.io.IOException;

public class ClientIPServlet extends HttpServlet {
    protected void doGet(HttpServletRequest request, HttpServletResponse response) 
            throws ServletException, IOException {
        String clientIp = getClientIP(request);
        response.getWriter().write("Client IP: " + clientIp);
    }

    private String getClientIP(HttpServletRequest request) {
        String clientIp = request.getHeader("X-Forwarded-For");

        if (clientIp == null || clientIp.isEmpty() || "unknown".equalsIgnoreCase(clientIp)) {
            clientIp = request.getRemoteAddr();
        } else {
            // 如果有多个 IP，取第一个（真实 IP）
            int commaIndex = clientIp.indexOf(",");
            if (commaIndex != -1) {
                clientIp = clientIp.substring(0, commaIndex);
            }
        }

        return clientIp;
    }
}

代码解读

获取请求头：我们使用 request.getHeader("X-Forwarded-For") 获取负载均衡器传递的客户端 IP。
Fallback：如果 X-Forwarded-For 为空，则使用 request.getRemoteAddr() 返回的 IP。
处理多重代理：如果有多个 IP 地址，取第一个地址作为真实 IP。

关系图

我们也可以用歌表示系统组件之间的关系。下面是用 Mermaid 语法表示的关系图：

erDiagram
    CLIENT {
        string ipAddress
    }
    LOAD_BALANCER {
        string address
    }
    BACKEND_APP {
        string requestData
    }

    CLIENT ||--o| LOAD_BALANCER : sends_request
    LOAD_BALANCER ||--o| BACKEND_APP : forwards_request

在这个关系图中，客户端与负载均衡器之间有请求关系，负载均衡器与后端应用之间也存在请求转发的关系。