fix: fix io buffer pool alignment (foyer-rs#724)

Signed-off-by: MrCroxx <mrcroxx@outlook.com>
foyer-rs · Sep 20, 2024 · 7ebc971 · 7ebc971
1 parent 474fd2f
commit 7ebc971
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/foyer-storage/src/io_buffer_pool.rs b/foyer-storage/src/io_buffer_pool.rs
@@ -14,7 +14,9 @@
 
 use std::collections::VecDeque;
 
-use crate::{IoBuffer, IoBytes};
+use foyer_common::bits;
+
+use crate::{device::ALIGN, IoBuffer, IoBytes};
 
 pub enum Buffer {
     IoBuffer(IoBuffer),
@@ -41,6 +43,7 @@ pub struct IoBufferPool {
 
 impl IoBufferPool {
     pub fn new(buffer_size: usize, capacity: usize) -> Self {
+        bits::assert_aligned(ALIGN, buffer_size);
         Self {
             capacity,
             buffer_size,

diff --git a/foyer-storage/src/large/batch.rs b/foyer-storage/src/large/batch.rs
@@ -117,6 +117,7 @@ where
     S: HashBuilder + Debug,
 {
     pub fn new(capacity: usize, region_manager: RegionManager, device: MonitoredDevice, indexer: Indexer) -> Self {
+        let capacity = bits::align_up(device.align(), capacity);
         let mut batch = Self {
             buffer: IoBuffer::new(capacity),
             len: 0,