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。