style: improve code of http client init

This commit is contained in:
Mmx
2024-10-12 17:07:00 +08:00
parent 0110a553da
commit 5b673af27b
4 changed files with 40 additions and 51 deletions

View File

@@ -104,7 +104,7 @@ func doLogin(conf LoginSingleConf) error {
logger := conf.Logger logger := conf.Logger
// 登录配置初始化 // 登录配置初始化
httpClient := http_client.HttpPackSelect(conf.Eth).Client httpClient := http_client.ClientSelect(conf.Eth)
srunClient := srun.New(&srun.Conf{ srunClient := srun.New(&srun.Conf{
Logger: logger, Logger: logger,
Https: config.Settings.Basic.Https, Https: config.Settings.Basic.Https,

View File

@@ -9,13 +9,9 @@ import (
"net/http" "net/http"
) )
type Http struct { var _DefaultClient *http.Client
Client *http.Client
}
var HttpPack *Http var _EthClientMap map[string]*http.Client
var httpTools map[string]*Http
func init() { func init() {
logger := config.Logger.WithField(keys.LogComponent, "init http") logger := config.Logger.WithField(keys.LogComponent, "init http")
@@ -37,21 +33,21 @@ func init() {
} }
} }
HttpPack = genHttpPack(eth) _DefaultClient = CreateClientFromEth(eth)
} else { } else {
httpTools = make(map[string]*Http) _EthClientMap = make(map[string]*http.Client)
} }
} }
func HttpPackSelect(eth *tools.Eth) *Http { func ClientSelect(eth *tools.Eth) *http.Client {
if HttpPack != nil { if _DefaultClient != nil {
return HttpPack return _DefaultClient
} }
if addrHttp, ok := httpTools[eth.Name]; ok { if client, ok := _EthClientMap[eth.Name]; ok {
return addrHttp return client
} else { } else {
addrHttp = genHttpPack(eth) client = CreateClientFromEth(eth)
httpTools[eth.Name] = addrHttp _EthClientMap[eth.Name] = client
return addrHttp return client
} }
} }

View File

@@ -6,30 +6,27 @@ import (
"crypto/tls" "crypto/tls"
"github.com/Mmx233/BitSrunLoginGo/internal/config" "github.com/Mmx233/BitSrunLoginGo/internal/config"
"github.com/Mmx233/BitSrunLoginGo/tools" "github.com/Mmx233/BitSrunLoginGo/tools"
"github.com/Mmx233/tool"
"net" "net"
"net/http" "net/http"
) )
func genHttpPack(eth *tools.Eth) *Http { func CreateClientFromEth(eth *tools.Eth) *http.Client {
var addr net.Addr var addr net.Addr
if eth != nil { if eth != nil {
addr = eth.Addr addr = eth.Addr
} }
tr := &http.Transport{
DialContext: (&net.Dialer{ return &http.Client{
Timeout: config.Timeout, Transport: &http.Transport{
LocalAddr: addr, DialContext: (&net.Dialer{
}).DialContext, Timeout: config.Timeout,
TLSClientConfig: &tls.Config{ LocalAddr: addr,
InsecureSkipVerify: config.Settings.Basic.SkipCertVerify, }).DialContext,
TLSClientConfig: &tls.Config{
InsecureSkipVerify: config.Settings.Basic.SkipCertVerify,
},
Proxy: http.ProxyFromEnvironment,
}, },
Proxy: http.ProxyFromEnvironment, Timeout: config.Timeout,
}
return &Http{
Client: tool.GenHttpClient(&tool.HttpClientOptions{
Transport: tr,
Timeout: config.Timeout,
}),
} }
} }

View File

@@ -4,13 +4,12 @@ import (
"crypto/tls" "crypto/tls"
"github.com/Mmx233/BitSrunLoginGo/internal/config" "github.com/Mmx233/BitSrunLoginGo/internal/config"
"github.com/Mmx233/BitSrunLoginGo/tools" "github.com/Mmx233/BitSrunLoginGo/tools"
"github.com/Mmx233/tool"
"net" "net"
"net/http" "net/http"
"syscall" "syscall"
) )
func genHttpPack(eth *tools.Eth) *Http { func CreateClientFromEth(eth *tools.Eth) *http.Client {
dialer := net.Dialer{ dialer := net.Dialer{
Timeout: config.Timeout, Timeout: config.Timeout,
} }
@@ -18,31 +17,28 @@ func genHttpPack(eth *tools.Eth) *Http {
dialer.LocalAddr = eth.Addr dialer.LocalAddr = eth.Addr
ethName := eth.Name ethName := eth.Name
dialer.Control = func(network string, address string, c syscall.RawConn) error { dialer.Control = func(network string, address string, c syscall.RawConn) error {
var operr error var opErr error
fn := func(fd uintptr) { fn := func(fd uintptr) {
operr = syscall.SetsockoptString(int(fd), syscall.SOL_SOCKET, syscall.SO_BINDTODEVICE, ethName) opErr = syscall.SetsockoptString(int(fd), syscall.SOL_SOCKET, syscall.SO_BINDTODEVICE, ethName)
} }
if err := c.Control(fn); err != nil { if err := c.Control(fn); err != nil {
return err return err
} }
if operr != nil { if opErr != nil {
return operr return opErr
} }
return nil return nil
} }
} }
tr := &http.Transport{ return &http.Client{
DialContext: dialer.DialContext, Transport: &http.Transport{
TLSHandshakeTimeout: config.Timeout, DialContext: dialer.DialContext,
IdleConnTimeout: config.Timeout, TLSHandshakeTimeout: config.Timeout,
TLSClientConfig: &tls.Config{InsecureSkipVerify: config.Settings.Basic.SkipCertVerify}, IdleConnTimeout: config.Timeout,
} TLSClientConfig: &tls.Config{InsecureSkipVerify: config.Settings.Basic.SkipCertVerify},
tr.Proxy = http.ProxyFromEnvironment Proxy: http.ProxyFromEnvironment,
return &Http{ },
Client: tool.GenHttpClient(&tool.HttpClientOptions{ Timeout: config.Timeout,
Transport: tr,
Timeout: config.Timeout,
}),
} }
} }