beck-8 / s3-tools Goto Github PK

type randomRoundTripper struct {
	resolver *net.Resolver
}

func (r *randomRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
	// 解析域名，获取所有 IP 地址
	ips, err := r.resolver.LookupIP(context.Background(), "ip4", req.URL.Hostname())
	if err != nil {
		return nil, err
	}

	// 随机选择一个 IP 地址
	selectedIP := ips[rand.Intn(len(ips))]
	fmt.Println(req.Host, req.URL.Host)

	// 替换请求的 Host 字段为选定的 IP 地址
	req.URL.Host = selectedIP.String()
	fmt.Println(req.Header, req.Host, req.URL.Host, req.Method)

	// 使用默认的 Transport 进行实际的请求
	return http.DefaultTransport.RoundTrip(req)
}
...
func main() {
	// 创建自定义的 RoundTripper
	transport := &randomRoundTripper{
		resolver: net.DefaultResolver,
	}

	// 创建自定义的 Client
	client := &http.Client{
		Transport: transport,
	}