in read_stall_retry/max_latency_per_pod.py [0:0]
def process_files(path):
"""Processes CSV files in a GCS directory and calculates max overall latency."""
try:
client = storage.Client()
bucket_name, blob_prefix = path.replace("gs://", "").split("/", 1)
bucket = client.bucket(bucket_name)
blobs = bucket.list_blobs(prefix=blob_prefix, delimiter='/')
global_max_latency = 0
global_max_latency_file = None
global_max_latency_object = None
futures = []
with ThreadPoolExecutor() as executor:
for blob in blobs:
if blob.name.endswith('.csv') and '/' not in blob.name[len(blob_prefix):]:
logger.info(f"Processing file: {blob.name}")
futures.append(executor.submit(process_csv_blob, blob))
for future in as_completed(futures):
max_latency, max_latency_object, file_name = future.result()
if max_latency is not None:
logger.info(f" Max Overall Latency in {file_name}: {max_latency}, Object Name: {max_latency_object}")
if max_latency > global_max_latency:
global_max_latency = max_latency
global_max_latency_file = file_name
global_max_latency_object = max_latency_object
if global_max_latency > 0:
logger.info(f"\nGlobal Max Overall Latency: {global_max_latency}")
logger.info(f"File Name: {global_max_latency_file}")
logger.info(f"Object Name: {global_max_latency_object}")
else:
logger.info("\nNo valid latency values found.")
except Exception as e:
logger.error(f"Error processing GCS files: {e}")