Add profiles to the call sites of updateRunfiles

coeuvre · copybara-github · commit 18c8839c09d7 · 2024-01-08T23:25:17.000-08:00
To make it easier to debug #20748. Closes #20783. PiperOrigin-RevId: 596824058 Change-Id: I1dd6b940c30bd6c4b99904d19667f48a389b3a41
diff --git a/src/main/java/com/google/devtools/build/lib/exec/local/LocalSpawnRunner.java b/src/main/java/com/google/devtools/build/lib/exec/local/LocalSpawnRunner.java
@@ -131,8 +131,10 @@ public SpawnResult exec(Spawn spawn, SpawnExecutionContext context)
     SpawnMetrics.Builder spawnMetrics = SpawnMetrics.Builder.forLocalExec();
     Stopwatch totalTimeStopwatch = Stopwatch.createStarted();
     Stopwatch setupTimeStopwatch = Stopwatch.createStarted();
-    runfilesTreeUpdater.updateRunfiles(
-        spawn.getRunfilesSupplier(), spawn.getEnvironment(), context.getFileOutErr());
+    try (var s = Profiler.instance().profile("updateRunfiles")) {
+      runfilesTreeUpdater.updateRunfiles(
+          spawn.getRunfilesSupplier(), spawn.getEnvironment(), context.getFileOutErr());
+    }
     if (Spawns.shouldPrefetchInputsForLocalExecution(spawn)) {
       context.prefetchInputsAndWait();
     }
diff --git a/src/main/java/com/google/devtools/build/lib/worker/WorkerSpawnRunner.java b/src/main/java/com/google/devtools/build/lib/worker/WorkerSpawnRunner.java
@@ -177,8 +177,10 @@ public SpawnResult exec(Spawn spawn, SpawnExecutionContext context)
                 String.format(
                     "%s worker %s", spawn.getMnemonic(), spawn.getResourceOwner().describe()))) {
 
-      runfilesTreeUpdater.updateRunfiles(
-          spawn.getRunfilesSupplier(), spawn.getEnvironment(), context.getFileOutErr());
+      try (var s = Profiler.instance().profile("updateRunfiles")) {
+        runfilesTreeUpdater.updateRunfiles(
+            spawn.getRunfilesSupplier(), spawn.getEnvironment(), context.getFileOutErr());
+      }
 
       InputMetadataProvider inputFileCache = context.getInputMetadataProvider();
 
