fix: bug observed in #916 (comment)

bytebeamio · Sep 30, 2024 · f43ea21 · f43ea21
1 parent 6b0c3ec
commit f43ea21
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/rumqttc/src/v5/state.rs b/rumqttc/src/v5/state.rs
@@ -576,6 +576,10 @@ impl MqttState {
 
         let event = Event::Outgoing(Outgoing::Publish(pkid));
         self.events.push_back(event);
+        match (publish.qos, tx) {
+            (QoS::AtMostOnce, Some(tx)) => tx.resolve(),
+            (_, tx) => self.ack_waiter[publish.pkid as usize] = tx,
+        }
 
         Ok(Some(Packet::Publish(publish)))
     }