use reifydb_core::event::EventBus;
use reifydb_runtime::{actor::system::ActorSpawner, context::clock::Clock};
#[cfg(all(feature = "sqlite", not(target_arch = "wasm32")))]
use reifydb_sqlite::{SqliteConfig, SqliteTempPathGuard};
use reifydb_value::value::duration::Duration;
use crate::tier::{commit::buffer::MultiCommitBufferTier, persistent::MultiPersistentTier, read::ReadBufferConfig};
#[derive(Clone)]
pub struct MultiStoreConfig {
pub commit: CommitBufferConfig,
pub persistent: Option<PersistentConfig>,
pub read: Option<ReadBufferConfig>,
pub retention: RetentionConfig,
pub merge_config: MergeConfig,
pub event_bus: EventBus,
pub spawner: ActorSpawner,
pub clock: Clock,
}
impl MultiStoreConfig {
pub fn memory(spawner: ActorSpawner, clock: Clock, event_bus: EventBus) -> Self {
Self {
commit: CommitBufferConfig {
storage: MultiCommitBufferTier::memory(),
},
persistent: None,
read: None,
retention: Default::default(),
merge_config: Default::default(),
event_bus,
spawner,
clock,
}
}
#[cfg(all(feature = "sqlite", not(target_arch = "wasm32")))]
pub fn sqlite(persistent: PersistentConfig, spawner: ActorSpawner, clock: Clock, event_bus: EventBus) -> Self {
Self {
commit: CommitBufferConfig {
storage: MultiCommitBufferTier::memory(),
},
persistent: Some(persistent),
read: Some(ReadBufferConfig::default()),
retention: Default::default(),
merge_config: Default::default(),
event_bus,
spawner,
clock,
}
}
}
#[derive(Clone)]
pub struct CommitBufferConfig {
pub storage: MultiCommitBufferTier,
}
#[derive(Clone)]
pub struct PersistentConfig {
pub storage: MultiPersistentTier,
}
impl PersistentConfig {
pub fn opened(storage: MultiPersistentTier) -> Self {
Self {
storage,
}
}
#[cfg(all(feature = "sqlite", not(target_arch = "wasm32")))]
pub fn sqlite(sqlite_config: SqliteConfig) -> Self {
Self {
storage: MultiPersistentTier::sqlite(sqlite_config),
}
}
#[cfg(all(feature = "sqlite", not(target_arch = "wasm32")))]
pub fn sqlite_in_memory() -> (Self, SqliteTempPathGuard) {
let (storage, guard) = MultiPersistentTier::sqlite_in_memory();
(
Self {
storage,
},
guard,
)
}
}
#[derive(Clone, Debug)]
pub struct RetentionConfig {
pub buffer: Duration,
pub persistent: Duration,
}
#[derive(Clone, Debug)]
pub struct MergeConfig {
pub merge_threshold_rows: usize,
pub merge_batch_size: usize,
pub enable_auto_eviction: bool,
}
impl Default for RetentionConfig {
fn default() -> Self {
Self {
buffer: Duration::from_seconds(300).unwrap(),
persistent: Duration::from_seconds(3600).unwrap(),
}
}
}
impl Default for MergeConfig {
fn default() -> Self {
Self {
merge_threshold_rows: 100_000,
merge_batch_size: 10_000,
enable_auto_eviction: true,
}
}
}