don't bench system calls

umputun · Jun 1, 2022 · f29706c · f29706c
1 parent b44ef71
commit f29706c
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/app/server/rlb_server.go b/app/server/rlb_server.go
@@ -110,25 +110,31 @@ func (s *RLBServer) routes() chi.Router {
 	router := chi.NewRouter()
 	s.bench = rest.NewBenchmarks()
 
-	router.Use(middleware.RequestID, middleware.RealIP, rest.Recoverer(log.Default()), s.bench.Handler)
+	router.Use(middleware.RequestID, middleware.RealIP, rest.Recoverer(log.Default()))
 	router.Use(middleware.Throttle(10000), middleware.Timeout(60*time.Second))
 	router.Use(rest.AppInfo("RLB", "Umputun", s.version), rest.Ping)
 	router.Use(tollbooth_chi.LimitHandler(tollbooth.NewLimiter(50, nil)), middleware.NoCache)
 
 	router.Use(logger.New(logger.Log(log.Default()), logger.WithBody, logger.Prefix("[DEBUG]"),
 		logger.IPfn(logger.AnonymizeIP)).Handler)
 	router.Use()
+
 	// current routes
 	router.Route("/api/v1/jump", func(r chi.Router) {
+		r.Use(s.bench.Handler)
+		r.Get("/{svc}", s.DoJump)
+		r.Head("/{svc}", s.DoJump)
+	})
+
+	// legacy routes
+	router.Group(func(r chi.Router) {
+		r.Use(s.bench.Handler)
 		r.Get("/{svc}", s.DoJump)
 		r.Head("/{svc}", s.DoJump)
 	})
 
 	router.Get("/api/v1/status", s.statusCtrl)
 	router.Get("/api/v1/bench", s.benchCtrl)
-	// legacy routes
-	router.Get("/{svc}", s.DoJump)
-	router.Head("/{svc}", s.DoJump)
 
 	return router
 }