Allow SimpleClient to yield in case of garbage responses

FactbirdHQ · Feb 4, 2025 · 8c159ec · 8c159ec
1 parent aeb67d5
commit 8c159ec
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/atat/Cargo.toml b/atat/Cargo.toml
@@ -23,6 +23,7 @@ embedded-io-async = "0.6"
 futures = { version = "0.3", default-features = false }
 embassy-sync = "0.6"
 embassy-time = "0.4"
+embassy-futures = "0.1"
 heapless = { version = "^0.8", features = ["serde"] }
 serde_at = { path = "../serde_at", version = "^0.23.0", optional = true }
 atat_derive = { path = "../atat_derive", version = "^0.23.0", optional = true }

diff --git a/atat/src/asynch/simple_client.rs b/atat/src/asynch/simple_client.rs
@@ -57,7 +57,11 @@ impl<'a, RW: Read + Write, D: Digester> SimpleClient<'a, RW, D> {
                 _ => return Err(Error::Read),
             };
 
-            trace!("Buffer contents: '{:?}'", LossyStr(&self.buf[..self.pos]));
+            trace!(
+                "Buffer contents: ({:?} bytes) '{:?}'",
+                self.pos,
+                LossyStr(&self.buf[..self.pos])
+            );
 
             while self.pos > 0 {
                 let (res, swallowed) = match self.digester.digest(&self.buf[..self.pos]) {
@@ -108,6 +112,7 @@ impl<'a, RW: Read + Write, D: Digester> SimpleClient<'a, RW, D> {
                 };
 
                 if swallowed == 0 {
+                    embassy_futures::yield_now().await;
                     break;
                 }