Go中将http.Client绑定到IP地址

在分布式计算领域，经常需要控制源IP HTTP 请求源自的地址。对于客户端计算机上有多个 NIC，这种粒度至关重要。

考虑以下基本 HTTP 客户端代码：

package main

import "net/http"

func main() {
    webclient := &http.Client{}
    req, _ := http.NewRequest("GET", "http://www.google.com", nil)
    httpResponse, _ := webclient.Do(req)
    defer httpResponse.Body.Close()
}

要将此客户端绑定到特定的 NIC 或 IP 地址，我们需要修改其 Transport 字段。我们将使用使用自定义 net.Dialer 的自定义 net.Transport。 net.Dialer 反过来又允许我们指定传出连接的本地地址。

import (
    "net"
    "net/http"
)

func main() {
    localAddr, err := net.ResolveIPAddr("ip", "")
    if err != nil {
        panic(err)
    }
    localTCPAddr := net.TCPAddr{
        IP: localAddr.IP,
    }

    webclient := &http.Client{
        Transport: &http.Transport{
            Proxy:                http.ProxyFromEnvironment,
            DialContext:          (&net.Dialer{LocalAddr: &localTCPAddr}).DialContext,
            MaxIdleConns:          100,
            IdleConnTimeout:       90 * time.Second,
            TLSHandshakeTimeout:   10 * time.Second,
            ExpectContinueTimeout: 1 * time.Second,
        },
    }
}

通过此修改，我们的 HTTP 客户端现在绑定到指定的 IP 地址，确保所有传出请求均源自所需的 NIC。