handler malformed json

cuenca-mx · Apr 3, 2023 · fab314d · fab314d
1 parent 995d376
commit fab314d
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/fast_agave/tasks/sqs_tasks.py b/fast_agave/tasks/sqs_tasks.py
@@ -2,6 +2,7 @@
 import json
 from functools import wraps
 from itertools import count
+from json import JSONDecodeError
 from typing import AsyncGenerator, Callable, Coroutine
 
 from aiobotocore.httpsession import HTTPClientError
@@ -99,7 +100,11 @@ async def concurrency_controller(coro: Coroutine) -> None:
                     can_read,
                     sqs,
                 ):
-                    body = json.loads(message['Body'])
+                    try:
+                        body = json.loads(message['Body'])
+                    except JSONDecodeError:
+                        continue
+
                     message_receive_count = int(
                         message['Attributes']['ApproximateReceiveCount']
                     )

diff --git a/fast_agave/version.py b/fast_agave/version.py
@@ -1 +1 @@
-__version__ = '0.9.0.dev0'
+__version__ = '0.9.0.dev1'
diff --git a/tests/tasks/test_sqs_celery_client.py b/tests/tasks/test_sqs_celery_client.py
@@ -24,7 +24,6 @@ async def test_send_task(sqs_client) -> None:
         base64.b64decode(message['body'].encode('utf-8')).decode()
     )
 
-    # breakpoint()
     assert body_json[0] == args
     assert body_json[1] == kwargs
     assert message['headers']['lang'] == 'py'