Fix thread pool memory leak

Fixes #108.
mtth · Dec 11, 2018 · 25778f8 · 25778f8
1 parent 2e86523
commit 25778f8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/hdfs/client.py b/hdfs/client.py
@@ -1125,6 +1125,10 @@ def _map_async(pool_size, func, args):
   """
   pool = ThreadPool(pool_size)
   if sys.version_info <= (2, 6):
-    return pool.map(func, args)
+    results = pool.map(func, args)
   else:
-    return pool.map_async(func, args).get(1 << 24) # 6+ months.
+    results = pool.map_async(func, args).get(1 << 24) # 6+ months.
+
+  pool.close()
+  pool.join()
+  return results