optimize(hz): middleware search method (#762)

cloudwego · May 6, 2023 · 0d8b98b · 0d8b98b
1 parent c8ffa58
commit 0d8b98b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cmd/hz/generator/router.go b/cmd/hz/generator/router.go
@@ -405,7 +405,11 @@ func (pkgGen *HttpPackageGenerator) updateMiddlewareReg(router interface{}, midd
 	}
 
 	for _, mw := range middlewareList {
-		if bytes.Contains(file, []byte(mw)) {
+		mwName := mw + "Mw"
+		if pkgGen.SnakeStyleMiddleware {
+			mwName = mw + "_mw"
+		}
+		if bytes.Contains(file, []byte(mwName)) {
 			continue
 		}
 		middlewareSingleTpl := pkgGen.tpls[middlewareSingleTplName]