p1-taskqueue 0.1.17__py3-none-any.whl → 0.1.19__py3-none-any.whl

{p1_taskqueue-0.1.17.dist-info → p1_taskqueue-0.1.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: p1-taskqueue
-Version: 0.1.17
+Version: 0.1.19
 Summary: A Task Queue Wrapper for Dekoruma Backend
 Author-email: Chalvin <engineering@dekoruma.com>
 Project-URL: Homepage, https://github.com/Dekoruma/p1-taskqueue

{p1_taskqueue-0.1.17.dist-info → p1_taskqueue-0.1.19.dist-info}/RECORD

@@ -3,8 +3,8 @@ taskqueue/celery_app.py,sha256=BR1wHezSOFanFwSCINnrWvjoiZTNy0o0nUe4o8RYCsI,5261
 taskqueue/cmanager.py,sha256=GSW-RWsNJTFHAoL57-WRf9k1eKD58gJsiU6dlCS7Zq4,16261
 taskqueue/slack_notifier.py,sha256=ZvTbWa1XXHUiciYF14_SH5af0BvGoPCBQohPtxx4FgU,1419
 taskqueue/libs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-taskqueue/libs/helper_test.py,sha256=3f7RoYEDBIpLHG1YY3eQ_RqZvkm4Er3FDEvFJeoAc8o,9572
-p1_taskqueue-0.1.17.dist-info/METADATA,sha256=FB4orsVrxnv00lTH4Mt1OESJCKQhFOZMv-_Ngy1Wvso,1597
-p1_taskqueue-0.1.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-p1_taskqueue-0.1.17.dist-info/top_level.txt,sha256=hA3SM1ik2K8iPqtlt_-_nJ4TAePwHPN3vsOc4EiynqU,10
-p1_taskqueue-0.1.17.dist-info/RECORD,,
+taskqueue/libs/helper_test.py,sha256=NGqvOZth4yO_m5Rm0AX0U4AOKjbcfICELhj5oIkxDe8,9759
+p1_taskqueue-0.1.19.dist-info/METADATA,sha256=LXxoGKzxrZ4Lq92cpadGx3Z8wgZZmi93gu8_aSwAiaU,1597
+p1_taskqueue-0.1.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+p1_taskqueue-0.1.19.dist-info/top_level.txt,sha256=hA3SM1ik2K8iPqtlt_-_nJ4TAePwHPN3vsOc4EiynqU,10
+p1_taskqueue-0.1.19.dist-info/RECORD,,

taskqueue/libs/helper_test.py

@@ -112,6 +112,7 @@ def celery_worker_burst(include_func_names: List[str], channel: str = "default")
 def get_queued_tasks(channel: str = "default"):
     app = current_app
     queued_tasks = []
+    messages = []
 
     try:
         with app.connection_for_read() as conn:
@@ -119,10 +120,11 @@ def get_queued_tasks(channel: str = "default"):
                 queue = app.amqp.queues[channel](chan)
 
                 while True:
-                    message = queue.get(no_ack=True)
+                    message = queue.get(no_ack=False)
                     if not message:
                         break
 
+                    messages.append(message)
                     task_name = message.headers.get("task")
 
                     try:
@@ -173,7 +175,10 @@ def get_queued_tasks(channel: str = "default"):
 
                     except Exception as e:
                         logger.warning(f"Failed to decode message: {e}")
-                        continue
+
+                for message in messages:
+                    if message and not message.acknowledged:
+                        message.reject(requeue=True)
 
     except Exception as e:
         logger.error(