add remote cache writes to tail tasks

[ci skip-build-wheels]
pantsbuild · Sep 21, 2022 · 18d490b · 18d490b
1 parent 35757b6
commit 18d490b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/rust/engine/process_execution/src/remote_cache.rs b/src/rust/engine/process_execution/src/remote_cache.rs
@@ -471,7 +471,7 @@ impl crate::CommandRunner for CommandRunner {
     {
       let command_runner = self.clone();
       let result = result.clone();
-      let _write_join = self.executor.spawn(in_workunit!(
+      let write_fut = self.executor.spawn(in_workunit!(
         "remote_cache_write",
         Level::Trace,
         |workunit| async move {
@@ -496,6 +496,8 @@ impl crate::CommandRunner for CommandRunner {
         // NB: We must box the future to avoid a stack overflow.
         .boxed()
       ));
+      let mut tail_tasks = context.tail_tasks.lock();
+      tail_tasks.push(write_fut.boxed());
     }
 
     Ok(result)