chinhwajie/BitSrunLoginGo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

style: improve code of http client init

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

32
internal/pkg/http_client/http_linux.go
View File

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