aqs: Only set already_acked_or_nacked after ack succeeded

svix · Aug 13, 2024 · b07b93f · b07b93f
1 parent f498ce1
commit b07b93f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/omniqueue/src/backends/azure_queue_storage.rs b/omniqueue/src/backends/azure_queue_storage.rs
@@ -156,13 +156,14 @@ impl Acker for AqsAcker {
         if self.already_acked_or_nacked {
             return Err(QueueError::CannotAckOrNackTwice);
         }
-        self.already_acked_or_nacked = true;
+
         self.client
             .pop_receipt_client(self.pop_receipt.clone())
             .delete()
             .await
-            .map_err(QueueError::generic)
-            .map(|_| ())
+            .map_err(QueueError::generic)?;
+        self.already_acked_or_nacked = true;
+        Ok(())
     }
 
     async fn nack(&mut self) -> Result<()> {