2020-04-14 23:20:00 +00:00
|
|
|
package client
|
|
|
|
|
|
|
|
import (
|
2020-10-04 21:01:10 +00:00
|
|
|
"crypto/tls"
|
2020-10-04 23:49:02 +00:00
|
|
|
"net"
|
2020-04-14 23:20:00 +00:00
|
|
|
"net/http"
|
|
|
|
"time"
|
2020-12-25 05:07:18 +00:00
|
|
|
|
|
|
|
"github.com/go-ping/ping"
|
2020-04-14 23:20:00 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
2020-10-23 20:29:20 +00:00
|
|
|
secureHTTPClient *http.Client
|
|
|
|
insecureHTTPClient *http.Client
|
2020-04-14 23:20:00 +00:00
|
|
|
)
|
|
|
|
|
2020-10-23 20:29:20 +00:00
|
|
|
// GetHTTPClient returns the shared HTTP client
|
|
|
|
func GetHTTPClient(insecure bool) *http.Client {
|
2020-10-04 21:01:10 +00:00
|
|
|
if insecure {
|
2020-10-23 20:29:20 +00:00
|
|
|
if insecureHTTPClient == nil {
|
|
|
|
insecureHTTPClient = &http.Client{
|
2021-01-05 04:38:47 +00:00
|
|
|
Timeout: 10 * time.Second,
|
2020-10-04 21:01:10 +00:00
|
|
|
Transport: &http.Transport{
|
2020-10-30 13:50:40 +00:00
|
|
|
MaxIdleConns: 100,
|
|
|
|
MaxIdleConnsPerHost: 20,
|
2020-10-04 21:01:10 +00:00
|
|
|
TLSClientConfig: &tls.Config{
|
|
|
|
InsecureSkipVerify: true,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
2020-04-14 23:20:00 +00:00
|
|
|
}
|
2020-10-23 20:29:20 +00:00
|
|
|
return insecureHTTPClient
|
|
|
|
}
|
|
|
|
if secureHTTPClient == nil {
|
|
|
|
secureHTTPClient = &http.Client{
|
2020-12-25 05:07:18 +00:00
|
|
|
Timeout: 10 * time.Second,
|
2020-10-30 13:51:42 +00:00
|
|
|
Transport: &http.Transport{
|
|
|
|
MaxIdleConns: 100,
|
|
|
|
MaxIdleConnsPerHost: 20,
|
|
|
|
},
|
2020-10-04 21:01:10 +00:00
|
|
|
}
|
2020-04-14 23:20:00 +00:00
|
|
|
}
|
2020-10-23 20:29:20 +00:00
|
|
|
return secureHTTPClient
|
2020-04-14 23:20:00 +00:00
|
|
|
}
|
2020-10-04 23:49:02 +00:00
|
|
|
|
2020-12-25 05:07:18 +00:00
|
|
|
// CanCreateTCPConnection checks whether a connection can be established with a TCP service
|
|
|
|
func CanCreateTCPConnection(address string) bool {
|
2020-10-04 23:49:02 +00:00
|
|
|
conn, err := net.DialTimeout("tcp", address, 5*time.Second)
|
|
|
|
if err != nil {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
_ = conn.Close()
|
|
|
|
return true
|
|
|
|
}
|
2020-12-25 05:07:18 +00:00
|
|
|
|
2020-12-27 22:07:50 +00:00
|
|
|
// Ping checks if an address can be pinged and returns the round-trip time if the address can be pinged
|
|
|
|
//
|
2020-12-25 05:07:18 +00:00
|
|
|
// Note that this function takes at least 100ms, even if the address is 127.0.0.1
|
2020-12-27 22:07:50 +00:00
|
|
|
func Ping(address string) (bool, time.Duration) {
|
2020-12-25 05:07:18 +00:00
|
|
|
pinger, err := ping.NewPinger(address)
|
|
|
|
if err != nil {
|
2020-12-27 22:07:50 +00:00
|
|
|
return false, 0
|
2020-12-25 05:07:18 +00:00
|
|
|
}
|
|
|
|
pinger.Count = 1
|
|
|
|
pinger.Timeout = 5 * time.Second
|
|
|
|
pinger.SetNetwork("ip4")
|
|
|
|
pinger.SetPrivileged(true)
|
|
|
|
err = pinger.Run()
|
|
|
|
if err != nil {
|
2020-12-27 22:07:50 +00:00
|
|
|
return false, 0
|
2020-12-25 05:07:18 +00:00
|
|
|
}
|
2020-12-27 22:07:50 +00:00
|
|
|
if pinger.Statistics() != nil {
|
2021-01-13 02:08:18 +00:00
|
|
|
// If the packet loss is 100, it means that the packet didn't reach the host
|
|
|
|
if pinger.Statistics().PacketLoss == 100 {
|
|
|
|
return false, pinger.Timeout
|
|
|
|
}
|
2020-12-27 22:07:50 +00:00
|
|
|
return true, pinger.Statistics().MaxRtt
|
|
|
|
}
|
|
|
|
return true, 0
|
2020-12-25 05:07:18 +00:00
|
|
|
}
|