From 2aca2576f30895dae93333344f21cda2ce0c8640 Mon Sep 17 00:00:00 2001 From: Ayush Chauhan Date: Tue, 11 Jan 2022 22:18:59 +0530 Subject: [PATCH] fix: cloudwatch logs fetch logic --- airflow/providers/amazon/aws/hooks/logs.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/airflow/providers/amazon/aws/hooks/logs.py b/airflow/providers/amazon/aws/hooks/logs.py index bb22ae13d7337..fb12943f65fa8 100644 --- a/airflow/providers/amazon/aws/hooks/logs.py +++ b/airflow/providers/amazon/aws/hooks/logs.py @@ -71,9 +71,7 @@ def get_log_events( | 'ingestionTime' (int): The time in milliseconds the event was ingested. """ next_token = None - - event_count = 1 - while event_count > 0: + while True: if next_token is not None: token_arg: Optional[Dict[str, str]] = {'nextToken': next_token} else: @@ -99,7 +97,7 @@ def get_log_events( yield from events - if 'nextForwardToken' in response: + if next_token != response['nextForwardToken']: next_token = response['nextForwardToken'] else: return