maokaihe/mini_lsm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(code): part 3 iterators

Browse Source 
Signed-off-by: Alex Chi <iskyzh@gmail.com>
This commit is contained in:
Alex Chi
2022-12-24 10:11:06 -05:00
parent b263ea4fac
commit 4eb2177a3e
39 changed files with 1304 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
mini-lsm/src/table/tests.rs
View File

@@ -2,41 +2,30 @@ use std::sync::Arc;
use bytes::Bytes;
use super::*;
use crate::iterators::impls::StorageIterator;
use crate::table::SsTableBuilder;
use super::{SsTable, SsTableIterator};
#[test]
fn test_sst_build_single_key() {
let mut builder = SsTableBuilder::new(16, 16);
assert!(builder.add(b"233", b"233333"));
let mut builder = SsTableBuilder::new(16);
builder.add(b"233", b"233333");
builder.build("").unwrap();
}
#[test]
fn test_sst_build_two_blocks() {
let mut builder = SsTableBuilder::new(1024, 16);
assert!(builder.add(b"11", b"11"));
assert!(builder.add(b"22", b"22"));
assert!(builder.add(b"33", b"11"));
assert!(builder.add(b"44", b"22"));
assert!(builder.add(b"55", b"11"));
assert!(builder.add(b"66", b"22"));
let mut builder = SsTableBuilder::new(16);
builder.add(b"11", b"11");
builder.add(b"22", b"22");
builder.add(b"33", b"11");
builder.add(b"44", b"22");
builder.add(b"55", b"11");
builder.add(b"66", b"22");
assert!(builder.meta.len() >= 2);
builder.build("").unwrap();
}
#[test]
fn test_sst_build_full() {
let mut builder = SsTableBuilder::new(32, 16);
assert!(builder.add(b"11", b"11"));
assert!(builder.add(b"22", b"22"));
assert!(builder.add(b"33", b"11"));
assert!(builder.add(b"44", b"22"));
assert!(!builder.add(b"55", b"11"));
builder.build("").unwrap();
}
fn key_of(idx: usize) -> Vec<u8> {
format!("key_{:03}", idx * 5).into_bytes()
}
@@ -50,11 +39,11 @@ fn num_of_keys() -> usize {
}
fn generate_sst() -> SsTable {
let mut builder = SsTableBuilder::new(65536, 128);
let mut builder = SsTableBuilder::new(128);
for idx in 0..num_of_keys() {
let key = key_of(idx);
let value = value_of(idx);
assert!(builder.add(&key[..], &value[..]));
builder.add(&key[..], &value[..]);
}
builder.build("").unwrap()
}