feat(nats): flush after sending invocation

Signed-off-by: Roman Volosatovs <[email protected]>
bytecodealliance · Nov 22, 2024 · 0f7cdee · 0f7cdee
1 parent b8185f6
commit 0f7cdee
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/crates/transport-nats/src/lib.rs b/crates/transport-nats/src/lib.rs
@@ -1116,6 +1116,12 @@ impl wrpc_transport::Invoke for Client {
                 .await
         }
         .context("failed to publish handshake")?;
+        let nats = Arc::clone(&self.nats);
+        tokio::spawn(async move {
+            if let Err(err) = nats.flush().await {
+                error!(?err, "failed to flush");
+            }
+        });
         Ok((
             ParamWriter::Root(RootParamWriter::new(
                 (*self.nats).clone(),