62 lines
1.4 KiB
Python
62 lines
1.4 KiB
Python
import json
|
|
import logging
|
|
|
|
import boto3
|
|
|
|
from mev_inspect.text_io import BytesIteratorIO
|
|
|
|
MEV_SUMMARY_EXPORT_QUERY = """
|
|
SELECT to_json(mev_summary)
|
|
FROM mev_summary
|
|
WHERE
|
|
block_number >= :after_block_number AND
|
|
block_number < :before_block_number
|
|
"""
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def export_block_range(
|
|
inspect_db_session, after_block_number: int, before_block_number
|
|
) -> None:
|
|
client = get_s3_client()
|
|
bucket_name = get_export_bucket_name()
|
|
|
|
mev_summary_json_results = inspect_db_session.execute(
|
|
statement=MEV_SUMMARY_EXPORT_QUERY,
|
|
params={
|
|
"after_block_number": after_block_number,
|
|
"before_block_number": before_block_number,
|
|
},
|
|
)
|
|
|
|
mev_summary_json_fileobj = BytesIteratorIO(
|
|
(f"{json.dumps(row)}\n".encode("utf-8") for (row,) in mev_summary_json_results)
|
|
)
|
|
|
|
key = f"mev_summary/{after_block_number}-{before_block_number}.json"
|
|
|
|
client.upload_fileobj(
|
|
mev_summary_json_fileobj,
|
|
Bucket=bucket_name,
|
|
Key=key,
|
|
)
|
|
|
|
logger.info(f"Exported to {key}")
|
|
|
|
|
|
# TODO - replaced by ConfigMap
|
|
def get_export_bucket_name() -> str:
|
|
return "local-export"
|
|
|
|
|
|
# TODO - handle for production
|
|
def get_s3_client():
|
|
return boto3.client(
|
|
"s3",
|
|
region_name="us-east-1",
|
|
endpoint_url="http://localstack:4566",
|
|
aws_access_key_id="test",
|
|
aws_secret_access_key="test",
|
|
)
|