fix: hz uses regular expressions to check if there are duplicate regi…

…stration functions in router/register.go
cloudwego · Dec 15, 2023 · a007edb · a007edb
1 parent 82f9bfa
commit a007edb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cmd/hz/generator/router.go b/cmd/hz/generator/router.go
@@ -321,7 +321,10 @@ func (pkgGen *HttpPackageGenerator) updateRegister(pkg, rDir, pkgName string) er
 		}
 
 		insertReg := register.DepPkgAlias + ".Register(r)\n"
-		if bytes.Contains(file, []byte(insertReg)) {
+
+		registerPattern := `(?m)^\s*` + register.DepPkgAlias + `\s*\.\s*Register\(r\)\s*$`
+		re := regexp.MustCompile(registerPattern)
+		if re.Match(file) {
 			return fmt.Errorf("the router(%s) has been registered", insertReg)
 		}