From c3386c5fcf193bc3a13621af5f10162962f4308f Mon Sep 17 00:00:00 2001 From: thinhnd2104 <90733094+thinhnd2104@users.noreply.github.com> Date: Mon, 9 May 2022 05:32:09 +0700 Subject: [PATCH] Fix conn close error on retrieving log events (#23470) related: [#23469] (https://github.com/apache/airflow/issues/23469). --- airflow/providers/amazon/aws/operators/ecs.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/airflow/providers/amazon/aws/operators/ecs.py b/airflow/providers/amazon/aws/operators/ecs.py index b1f710895d721..66aaa19389055 100644 --- a/airflow/providers/amazon/aws/operators/ecs.py +++ b/airflow/providers/amazon/aws/operators/ecs.py @@ -25,7 +25,7 @@ from threading import Event, Thread from typing import Dict, Generator, Optional, Sequence -from botocore.exceptions import ClientError +from botocore.exceptions import ClientError, ConnectionClosedError from botocore.waiter import Waiter from airflow.exceptions import AirflowException @@ -139,6 +139,9 @@ def _get_log_events(self, skip: int = 0) -> Generator: self.logger.warning('Error on retrieving Cloudwatch log events', error) yield from () + except ConnectionClosedError as error: + self.logger.warning('ConnectionClosedError on retrieving Cloudwatch log events', error) + yield from () def _event_to_str(self, event: dict) -> str: event_dt = datetime.utcfromtimestamp(event['timestamp'] / 1000.0)