feat: add from_file API

yankun1992 · Dec 4, 2023 · 465fca5 · 465fca5
1 parent 3d57d80
commit 465fca5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 10 deletions.
diff --git a/fastbloom-rs/src/bloom.rs b/fastbloom-rs/src/bloom.rs
@@ -1,5 +1,6 @@
 use std::cmp::min;
 use std::fs::{File, OpenOptions};
+use std::fs;
 use std::io::{Write, Read};
 use std::ptr::slice_from_raw_parts;
 

diff --git a/fastbloom-rs/src/vec.rs b/fastbloom-rs/src/vec.rs
@@ -1,5 +1,5 @@
 use core::slice;
-use std::{fs::File, os::windows::fs::FileExt};
+use std::{fs::File, io::{Read, Seek}};
 
 use crate::builder::SUFFIX;
 
@@ -49,15 +49,8 @@ impl BloomBitVec {
             slice::from_raw_parts_mut(buf, bytes_len.try_into().unwrap())
         };
 
-        let mut cursor = seek;
-        let mut read = 0u64;
-
-        while read < bytes_len {
-            let size = file.seek_read(&mut buf[read as usize ..], cursor).unwrap() as u64;
-            read = read + size;
-            cursor += size;
-        }
-
+        file.seek(std::io::SeekFrom::Start(seek)).unwrap();
+        file.read_exact(buf).unwrap();
 
         BloomBitVec {
             storage,