Persist aborted_or_skipped on Message.options

richrliu · Sep 25, 2024 · 28e42b5 · 28e42b5
1 parent 7b108ca
commit 28e42b5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/dramatiq_abort/middleware.py b/src/dramatiq_abort/middleware.py
@@ -99,9 +99,11 @@ def before_process_message(
 
         event = self.backend.poll(self.id_to_key(message.message_id, AbortMode.CANCEL))
         if event:
+            message.options["aborted_or_skipped"] = True
             raise SkipMessage()
 
         self.manager.add_abortable(message.message_id)
+        message.options["aborted_or_skipped"] = True
 
     def after_process_message(
         self,