diff --git a/benches/common.rs b/benches/common.rs index 3abb7c5..537da11 100644 --- a/benches/common.rs +++ b/benches/common.rs @@ -16,6 +16,7 @@ use tokio::fs::create_dir_all; use tonbo::{ executor::tokio::TokioExecutor, stream, transaction::TransactionEntry, DbOption, Projection, }; +use tonbo_ext_reader::foyer_reader::FoyerReader; use tonbo_macros::Record; const RNG_SEED: u64 = 3; @@ -185,12 +186,12 @@ pub trait BenchReader { } pub struct TonboBenchDataBase { - db: tonbo::DB, + db: tonbo::DB, } impl TonboBenchDataBase { #[allow(dead_code)] - pub fn new(db: tonbo::DB) -> Self { + pub fn new(db: tonbo::DB) -> Self { TonboBenchDataBase { db } } } @@ -234,7 +235,7 @@ impl BenchDatabase for TonboBenchDataBase { } pub struct TonboBenchReadTransaction<'a> { - txn: tonbo::transaction::Transaction<'a, Customer>, + txn: tonbo::transaction::Transaction<'a, Customer, FoyerReader>, } impl<'db> BenchReadTransaction for TonboBenchReadTransaction<'db> { @@ -249,7 +250,7 @@ impl<'db> BenchReadTransaction for TonboBenchReadTransaction<'db> { } pub struct TonboBenchReader<'db, 'txn> { - txn: &'txn tonbo::transaction::Transaction<'db, Customer>, + txn: &'txn tonbo::transaction::Transaction<'db, Customer, FoyerReader>, } impl BenchReader for TonboBenchReader<'_, '_> { @@ -289,7 +290,7 @@ impl BenchReader for TonboBenchReader<'_, '_> { } pub struct TonboBenchWriteTransaction<'a> { - txn: tonbo::transaction::Transaction<'a, Customer>, + txn: tonbo::transaction::Transaction<'a, Customer, FoyerReader>, } impl<'db> BenchWriteTransaction for TonboBenchWriteTransaction<'db> { @@ -309,7 +310,7 @@ impl<'db> BenchWriteTransaction for TonboBenchWriteTransaction<'db> { } pub struct TonboBenchInserter<'db, 'txn> { - txn: &'txn mut tonbo::transaction::Transaction<'db, Customer>, + txn: &'txn mut tonbo::transaction::Transaction<'db, Customer, FoyerReader>, } impl BenchInserter for TonboBenchInserter<'_, '_> { diff --git a/benches/criterion/writes.rs b/benches/criterion/writes.rs index 41f3af2..d46342b 100644 --- a/benches/criterion/writes.rs +++ b/benches/criterion/writes.rs @@ -3,6 +3,7 @@ use std::{iter::repeat_with, sync::Arc}; use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion}; use mimalloc::MiMalloc; use tonbo::{executor::tokio::TokioExecutor, DbOption, Record, DB}; +use tonbo_ext_reader::foyer_reader::FoyerReader; #[global_allocator] static GLOBAL: MiMalloc = MiMalloc; @@ -15,7 +16,7 @@ pub struct KV { } #[inline(never)] -async fn tonbo_write(db: &DB, batch_size: usize) { +async fn tonbo_write(db: &DB, batch_size: usize) { let mut kvs = Vec::with_capacity(128); for _ in 0..batch_size { let key = repeat_with(fastrand::alphanumeric).take(256).collect();