wait for thread pools to be idle on unload

apache · Aug 2, 2022 · d38781f · d38781f
1 parent 2a02757
commit d38781f
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 0 deletions.
diff --git a/r/R/arrow-package.R b/r/R/arrow-package.R
@@ -81,6 +81,10 @@
   invisible()
 }
 
+.onUnload <- function(...) {
+  WaitForIdleThreadPool()
+}
+
 configure_tzdb <- function() {
   # This is needed on Windows to support timezone-aware calculations
   if (requireNamespace("tzdb", quietly = TRUE)) {

diff --git a/r/R/arrowExports.R b/r/R/arrowExports.R
diff --git a/r/src/arrowExports.cpp b/r/src/arrowExports.cpp
diff --git a/r/src/threadpool.cpp b/r/src/threadpool.cpp
@@ -54,3 +54,17 @@ int GetIOThreadPoolCapacity() { return arrow::io::GetIOThreadPoolCapacity(); }
 void SetIOThreadPoolCapacity(int threads) {
   StopIfNotOk(arrow::io::SetIOThreadPoolCapacity(threads));
 }
+
+namespace arrow {
+namespace io {
+namespace internal {
+arrow::internal::ThreadPool* GetIOThreadPool();
+}
+}  // namespace io
+}  // namespace arrow
+
+// [[arrow::export]]
+void WaitForIdleThreadPool() {
+  arrow::internal::GetCpuThreadPool()->WaitForIdle();
+  arrow::io::internal::GetIOThreadPool()->WaitForIdle();
+}