transport.go

package main

import (
	"github.com/sdvcrx/cuttlefish/config"
	"net"
	"net/http"
	"net/url"
	"time"
)

var DefaultProxyTransport = &http.Transport{
	Proxy: SelectProxy,
	// Copy from https://golang.org/pkg/net/http/#RoundTripper
	// http.DefaultTransport
	DialContext: (&net.Dialer{
		Timeout:   30 * time.Second,
		KeepAlive: 30 * time.Second,
		DualStack: true,
	}).DialContext,
	MaxIdleConns:          100,
	IdleConnTimeout:       90 * time.Second,
	TLSHandshakeTimeout:   10 * time.Second,
	ExpectContinueTimeout: 1 * time.Second,
}

func SelectProxy(r *http.Request) (*url.URL, error) {
	proxy := config.GetInstance().ParentProxies.Next()
	if proxy != "" {
		logger.Info().Msgf("select proxy: %s", proxy)
		return url.Parse(proxy)
	}
	return nil, nil
}