Fix stats

eclipse-zenoh · Dec 12, 2024 · ee9f0a5 · ee9f0a5
1 parent c5ba11e
commit ee9f0a5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/io/zenoh-transport/src/common/batch.rs b/io/zenoh-transport/src/common/batch.rs
@@ -363,7 +363,14 @@ impl Encode<&TransportMessage> for &mut WBatch {
 
     fn encode(self, x: &TransportMessage) -> Self::Output {
         let mut writer = self.buffer.writer();
-        self.codec.write(&mut writer, x)
+        let res = self.codec.write(&mut writer, x);
+        #[cfg(feature = "stats")]
+        {
+            if let Ok(()) = res {
+                self.stats.t_msgs += 1;
+            }
+        }
+        res
     }
 }