Replace multiproccessing with concurrent.futures.

GoogleCloudPlatform · Mar 14, 2018 · 1578a08 · 1578a08
1 parent cdd7da1
commit 1578a08
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/spanner/cloud-client/batch_sample.py b/spanner/cloud-client/batch_sample.py
@@ -19,7 +19,7 @@
 """
 
 import argparse
-from multiprocessing.pool import ThreadPool
+from concurrent.futures import ThreadPoolExecutor
 import time
 
 from google.cloud import spanner
@@ -48,8 +48,8 @@ def run_batch_query(instance_id, database_id):
         keyset=spanner.KeySet(all_=True)
     )
 
-    # pool = multiprocessing.Pool()
-    pool = ThreadPool()
+    # Create a pool of workers for the task
+    pool = ThreadPoolExecutor()
     results = []
     start = time.time()
 

diff --git a/spanner/cloud-client/requirements.txt b/spanner/cloud-client/requirements.txt
@@ -1 +1,2 @@
 google-cloud-spanner==1.1.0
+futures==3.2.0