used buffered download
This commit is contained in:
@@ -1,7 +1,9 @@
|
|||||||
"""Definition of get function."""
|
"""Definition of get function."""
|
||||||
|
|
||||||
|
import logging
|
||||||
import os
|
import os
|
||||||
from io import BytesIO
|
from io import BytesIO
|
||||||
|
from traceback import print_exc
|
||||||
|
|
||||||
from minio import Minio
|
from minio import Minio
|
||||||
|
|
||||||
@@ -16,10 +18,22 @@ def get(
|
|||||||
bucket_name = os.getenv('MINIO_BUCKET_NAME')
|
bucket_name = os.getenv('MINIO_BUCKET_NAME')
|
||||||
# get buffer
|
# get buffer
|
||||||
try:
|
try:
|
||||||
response = client.get_object(bucket_name, object_name)
|
response = client.get_object(
|
||||||
buffer = BytesIO(response.data)
|
bucket_name=bucket_name,
|
||||||
|
object_name=object_name,
|
||||||
|
)
|
||||||
|
assert response.status == 200
|
||||||
|
buffer = BytesIO()
|
||||||
|
chunk_size = 2**14
|
||||||
|
while chunk := response.read(chunk_size):
|
||||||
|
buffer.write(chunk)
|
||||||
|
buffer.seek(0)
|
||||||
|
logging.debug('got %s', object_name)
|
||||||
|
return buffer
|
||||||
|
except Exception as exc:
|
||||||
|
logging.error('failed getting data from MinIO')
|
||||||
|
print_exc()
|
||||||
|
raise exc
|
||||||
finally:
|
finally:
|
||||||
response.close()
|
response.close()
|
||||||
response.release_conn()
|
response.release_conn()
|
||||||
buffer.seek(0)
|
|
||||||
return buffer
|
|
||||||
|
|||||||
Reference in New Issue
Block a user