feat: support hostname target (#1455)

Signed-off-by: ldysdu <ldysdu@163.com>
apache · Mar 9, 2024 · 3171465 · 3171465
1 parent 72fedbb
commit 3171465
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/common/src/main/java/org/dromara/hertzbeat/common/support/valid/HostValid.java b/common/src/main/java/org/dromara/hertzbeat/common/support/valid/HostValid.java
@@ -37,7 +37,7 @@
 @Constraint(validatedBy = HostParamValidator.class)
 public @interface HostValid {
 
-    String message() default "Host need ipv4,ipv6 or domain,<br>EG:127.0.0.1  hertzbeat.com";
+    String message() default "Host need ipv4,ipv6,hostname or domain,<br>EG:127.0.0.1  hertzbeat.com";
 
     Class<?>[] groups() default {};
 

diff --git a/common/src/main/java/org/dromara/hertzbeat/common/util/IpDomainUtil.java b/common/src/main/java/org/dromara/hertzbeat/common/util/IpDomainUtil.java
@@ -36,7 +36,7 @@
 public class IpDomainUtil {
 
     private static final Pattern DOMAIN_PATTERN =
-            Pattern.compile("^(?=^.{3,255}$)[_a-zA-Z0-9][-_a-zA-Z0-9]{0,62}(\\.[a-zA-Z0-9][-_a-zA-Z0-9]{0,62})+$");
+            Pattern.compile("^[-\\w]+(\\.[-\\w]+)*$");
 
     private static final String LOCALHOST = "localhost";
 

diff --git a/common/src/test/java/org/dromara/hertzbeat/common/util/IpDomainUtilTest.java b/common/src/test/java/org/dromara/hertzbeat/common/util/IpDomainUtilTest.java
@@ -40,11 +40,14 @@ void validateIpDomain() {
         assertTrue(IpDomainUtil.validateIpDomain("good.didi"));
         assertTrue(IpDomainUtil.validateIpDomain("spider_rds.sogou"));
         assertTrue(IpDomainUtil.validateIpDomain("_redis.hn.sogou"));
-        assertFalse(IpDomainUtil.validateIpDomain("tmp"));
-        assertFalse(IpDomainUtil.validateIpDomain("good"));
+        assertTrue(IpDomainUtil.validateIpDomain("host"));
+        assertTrue(IpDomainUtil.validateIpDomain("host-1"));
         assertFalse(IpDomainUtil.validateIpDomain("www.baidu.com."));
         assertFalse(IpDomainUtil.validateIpDomain("good."));
         assertFalse(IpDomainUtil.validateIpDomain(".good."));
+        assertFalse(IpDomainUtil.validateIpDomain("www.baidu..com"));
+        assertFalse(IpDomainUtil.validateIpDomain("www.baidu*com"));
+        assertFalse(IpDomainUtil.validateIpDomain("host$-1"));
     }
 
     @Test
@@ -54,4 +57,4 @@ void isHasSchema() {
         assertFalse(IpDomainUtil.isHasSchema("www.baidu.com"));
         assertFalse(IpDomainUtil.isHasSchema("https_www.baidu.com"));
     }
-}
+}