Improve recording

opentripplanner · Jan 5, 2025 · 118eef0 · 118eef0
1 parent 25b00b1
commit 118eef0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/application/src/test/java/org/opentripplanner/transit/speed_test/SpeedTest.java b/application/src/test/java/org/opentripplanner/transit/speed_test/SpeedTest.java
@@ -346,8 +346,9 @@ private void timeTransferCacheComputation() {
       .forEach(reluctance -> {
         RouteRequest routeRequest = new RouteRequest();
         routeRequest.withPreferences(b -> b.withWalk(c -> c.withReluctance(reluctance)));
-        timer.transferCacheTimer.record(() ->
-          timetableRepository.getTransitLayer().initTransferCacheForRequest(routeRequest)
+        timer.recordTimer(
+          "transfer_cache_computation",
+          () -> timetableRepository.getTransitLayer().initTransferCacheForRequest(routeRequest)
         );
       });
   }

diff --git a/...tion/src/test/java/org/opentripplanner/transit/speed_test/model/timer/SpeedTestTimer.java b/...tion/src/test/java/org/opentripplanner/transit/speed_test/model/timer/SpeedTestTimer.java
@@ -139,6 +139,15 @@ public void globalCount(String meterName, long count) {
     }
   }
 
+  public void recordTimer(String meterName, Runnable runnable) {
+    if (uploadRegistry != null) {
+      registry.add(uploadRegistry);
+      var timer = registry.timer(meterName);
+      timer.record(runnable);
+      registry.remove(uploadRegistry);
+    }
+  }
+
   /**
    * Calculate the total time mean for the given timer. If the timer is not
    * found {@link #NOT_AVAILABLE} is returned. This can be the case in unit tests,