starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-04-23 16:58:31 +02:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

refactor: ping if-flow and remove timeout

Browse Source
This commit is contained in:
yusing
2025-04-17 14:21:47 +08:00
parent cfd1d8fff0
commit c7e0dcbfd8
1 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
internal/net/ping.go
View File

@@ -94,24 +94,20 @@ func Ping(ctx context.Context, ip net.IP) (bool, error) {
} }
} }
var pingDialer = &net.Dialer{
Timeout: 2 * time.Second,
}
// PingWithTCPFallback pings the IP address using ICMP and TCP fallback. // PingWithTCPFallback pings the IP address using ICMP and TCP fallback.
// //
// If the ICMP ping fails due to permission error, it will try to connect to the specified port. // If the ICMP ping fails due to permission error, it will try to connect to the specified port.
func PingWithTCPFallback(ctx context.Context, ip net.IP, port int) (bool, error) { func PingWithTCPFallback(ctx context.Context, ip net.IP, port int) (bool, error) {
ok, err := Ping(ctx, ip) ok, err := Ping(ctx, ip)
if err != nil { if err == nil {
if !errors.Is(err, os.ErrPermission) {
return false, err
}
} else {
return ok, nil return ok, nil
} }
if !errors.Is(err, os.ErrPermission) {
return false, err
}
conn, err := pingDialer.DialContext(ctx, "tcp", fmt.Sprintf("%s:%d", ip, port)) var dialer net.Dialer
conn, err := dialer.DialContext(ctx, "tcp", fmt.Sprintf("%s:%d", ip, port))
if err != nil { if err != nil {
return false, err return false, err
} }