perf(validator): check Ipv4、Ipv6 by more graceful method (#220)

duke-git · May 25, 2024 · 4b3a62b · 4b3a62b
1 parent e054680
commit 4b3a62b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/validator/validator.go b/validator/validator.go
@@ -214,7 +214,7 @@ func IsIpV4(ipstr string) bool {
 	if ip == nil {
 		return false
 	}
-	return strings.Contains(ipstr, ".")
+	return ip.To4() != nil
 }
 
 // IsIpV6 check if the string is a ipv6 address.
@@ -224,7 +224,7 @@ func IsIpV6(ipstr string) bool {
 	if ip == nil {
 		return false
 	}
-	return strings.Contains(ipstr, ":")
+	return ip.To4() == nil && len(ip) == net.IPv6len
 }
 
 // IsPort check if the string is a valid net port.