Fix handling Vec of varint types (#5)

ntex-rs · May 2, 2023 · ff590dd · ff590dd
1 parent 0397aa5
commit ff590dd
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 4 deletions.
diff --git a/ntex-grpc/CHANGES.md b/ntex-grpc/CHANGES.md
@@ -1,5 +1,9 @@
 # Changes
 
+## [0.3.6] - 2023-05-02
+
+* Fix handling Vec of varint types
+
 ## [0.3.5] - 2023-04-06
 
 * Fix panic on error after stream eof

diff --git a/ntex-grpc/Cargo.toml b/ntex-grpc/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "ntex-grpc"
-version = "0.3.5"
+version = "0.3.6"
 license = "MIT"
 authors = ["Nikolay Kim <[email protected]>"]
 description = "GRPC Client/Server framework"

diff --git a/ntex-grpc/src/types.rs b/ntex-grpc/src/types.rs
@@ -359,9 +359,18 @@ impl<T: NativeType> NativeType for Vec<T> {
         wtype: WireType,
         src: &mut Bytes,
     ) -> Result<(), DecodeError> {
-        let mut value: T = Default::default();
-        value.deserialize(tag, wtype, src)?;
-        self.push(value);
+        if T::TYPE == WireType::Varint {
+            let len = encoding::decode_varint(src)? as usize;
+            for _ in 0..len {
+                let mut value: T = Default::default();
+                value.merge(src)?;
+                self.push(value);
+            }
+        } else {
+            let mut value: T = Default::default();
+            value.deserialize(tag, wtype, src)?;
+            self.push(value);
+        }
         Ok(())
     }