in step3/src/server/server.py [0:0]
def read_files_multi():
"""read_files_multi fetchse Shakespeare works from GCS in multi threads.
TODO: This part should be multiprocess.
"""
client = storage.Client()
bucket = client.get_bucket(BUCKET_NAME)
itr = client.list_blobs(bucket, prefix=BUCKET_PREFIX)
blobs = list(itr)
executor = futures.ThreadPoolExecutor(max_workers=8)
results = []
for blob in blobs:
ret = executor.submit(blob.download_as_bytes)
results.append(ret)
executor.shutdown()
logger.info(f"number of files: {len(results)}")
return [r.result().decode("utf-8") for r in results]