From b77106b526930990f51a306fd70cd00856f481e8 Mon Sep 17 00:00:00 2001
From: Igor Tolmachev <me@igorek.dev>
Date: Fri, 12 Jul 2024 21:40:08 +0900
Subject: Add zstd compression and fix bugs

---
 src/utils/read.rs | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 src/utils/read.rs

(limited to 'src/utils/read.rs')

diff --git a/src/utils/read.rs b/src/utils/read.rs
new file mode 100644
index 0000000..185758a
--- /dev/null
+++ b/src/utils/read.rs
@@ -0,0 +1,23 @@
+use std::io::{Read, Result as IOResult};
+
+pub trait ReadUtils {
+    fn read_arr<const S: usize>(&mut self) -> IOResult<[u8; S]>;
+
+    fn read_vec(&mut self, size: usize) -> IOResult<Vec<u8>>;
+}
+
+impl<R: Read> ReadUtils for R {
+    #[inline]
+    fn read_arr<const S: usize>(&mut self) -> Result<[u8; S], std::io::Error> {
+        let mut arr = [0; S];
+        self.read(&mut arr)?;
+        Ok(arr)
+    }
+
+    #[inline]
+    fn read_vec(&mut self, size: usize) -> Result<Vec<u8>, std::io::Error> {
+        let mut vec = vec![0; size];
+        self.read(&mut vec)?;
+        Ok(vec)
+    }
+}
-- 
cgit v1.2.3