style: improve code of http client init
This commit is contained in:
@@ -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,
|
||||||
|
|||||||
@@ -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
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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,
|
|
||||||
}),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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,
|
|
||||||
}),
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user