feat(transport/http): add walk http handler (go-kratos#2943)

* add walk http handler
superryanguo · Aug 3, 2023 · 3109546 · 3109546
1 parent 072b2ee
commit 3109546
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/transport/http/server.go b/transport/http/server.go
@@ -220,6 +220,14 @@ func (s *Server) WalkRoute(fn WalkRouteFunc) error {
 	})
 }
 
+// WalkHandle walks the router and all its sub-routers, calling walkFn for each route in the tree.
+func (s *Server) WalkHandle(handle func(method, path string, handler http.HandlerFunc)) error {
+	return s.WalkRoute(func(r RouteInfo) error {
+		handle(r.Method, r.Path, s.ServeHTTP)
+		return nil
+	})
+}
+
 // Route registers an HTTP router.
 func (s *Server) Route(prefix string, filters ...FilterFunc) *Router {
 	return newRouter(prefix, s, filters...)