Replace multiproccessing with concurrent.futures.

Author: kurtisvg

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 tasks
+    pool = ThreadPoolExecutor()
     results = []
     start = time.time()
 

spanner/cloud-client/requirements.txt

@@ -1 +1,2 @@
 google-cloud-spanner==1.1.0
+futures==3.2.0