From a007edb271a374c2017ad412a3216b0002c4e60c Mon Sep 17 00:00:00 2001 From: ozline Date: Sat, 16 Dec 2023 05:03:10 +0800 Subject: [PATCH] fix: hz uses regular expressions to check if there are duplicate registration functions in router/register.go --- cmd/hz/generator/router.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/cmd/hz/generator/router.go b/cmd/hz/generator/router.go index 10f431657..db141b0e7 100644 --- 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) }