如何在Vapor 3.0中获取客户端IP地址？

嘿，我来帮你搞定这个问题！在Vapor 3里获取客户端IP其实并不复杂，主要分两种场景：用户直接访问你的应用，或者应用部署在反向代理（比如Nginx、Cloudflare）后面的情况，下面分别说明：

1. 直接访问场景：获取原始远程地址 #

当用户直接请求你的Vapor应用时，可以通过Request对象的remoteAddress属性来获取客户端IP。这个属性返回的是SocketAddress?类型，我们需要解包后提取其中的ipAddress字符串：

func getClientIP(_ req: Request) throws -> String {
    guard let ip = req.remoteAddress?.ipAddress else {
        throw Abort(.internalServerError, reason: "无法获取客户端IP地址")
    }
    return "你的IP地址是：\(ip)"
}

2. 反向代理场景：处理转发头 #

如果你的Vapor应用部署在反向代理之后，remoteAddress拿到的会是代理服务器的IP，这时候需要从请求头里读取真实的客户端IP。常用的请求头是X-Forwarded-For和X-Real-IP：

func getRealClientIP(_ req: Request) throws -> String {
    // 优先读取X-Forwarded-For，这个头可能包含多个IP（用逗号分隔），第一个就是客户端真实IP
    if let forwardedFor = req.http.headers.first(name: .xForwardedFor) {
        let ipList = forwardedFor.split(separator: ",").map { $0.trimmingCharacters(in: .whitespaces) }
        if let realIP = ipList.first {
            return "你的真实IP地址是：\(realIP)"
        }
    }
    
    // 如果X-Forwarded-For不存在，尝试读取X-Real-IP
    if let realIP = req.http.headers.first(name: .xRealIP) {
        return "你的真实IP地址是：\(realIP)"
    }
    
    // 最后 fallback 到原始remoteAddress
    guard let ip = req.remoteAddress?.ipAddress else {
        throw Abort(.internalServerError, reason: "无法获取客户端IP地址")
    }
    return "你的IP地址是：\(ip)"
}

小提醒 #

如果你使用反向代理，一定要确保代理服务器正确设置了X-Forwarded-For或X-Real-IP头，并且你的Vapor应用只信任这些代理的请求，避免被恶意伪造IP头哦。

内容的提问来源于stack exchange，提问作者5t111111