src: spin shutdown loop while immediates are pending

This allows using `SetImmediate()` and friends at any point during cleanup. PR-URL: #34662 Fixes: #34657 Refs: #34572 Reviewed-By: Gabriel Schulhof <[email protected]> Reviewed-By: James M Snell <[email protected]>
nodejs · Sep 22, 2020 · 01107cc · 01107cc
1 parent 264f003
commit 01107cc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/env.cc b/src/env.cc
@@ -604,7 +604,10 @@ void Environment::RunCleanup() {
                               "RunCleanup", this);
   CleanupHandles();
 
-  while (!cleanup_hooks_.empty()) {
+  while (!cleanup_hooks_.empty() ||
+         native_immediates_.size() > 0 ||
+         native_immediates_threadsafe_.size() > 0 ||
+         native_immediates_interrupts_.size() > 0) {
     // Copy into a vector, since we can't sort an unordered_set in-place.
     std::vector<CleanupHookCallback> callbacks(
         cleanup_hooks_.begin(), cleanup_hooks_.end());