0g-storage-node/node/chunk_pool/src/lib.rs

#[macro_use]
extern crate tracing;

mod handler;
mod mem_pool;

pub use handler::{ChunkPoolHandler, ChunkPoolMessage};
pub use mem_pool::{FileID, MemoryChunkPool, SegmentInfo};

use std::sync::Arc;
use std::time::Duration;
use storage_async::ShardConfig;

#[derive(Clone, Copy, Debug)]
pub struct Config {
    pub write_window_size: usize,
    pub max_cached_chunks_all: usize,
    pub max_writings: usize,
    pub expiration_time_secs: u64,
    pub shard_config: ShardConfig,
}

impl Config {
    pub fn expiration_time(&self) -> Duration {
        Duration::from_secs(self.expiration_time_secs)
    }
}

pub fn unbounded(
    config: Config,
    log_store: Arc<storage_async::Store>,
    network_send: tokio::sync::mpsc::UnboundedSender<network::NetworkMessage>,
) -> (Arc<MemoryChunkPool>, ChunkPoolHandler) {
    let (sender, receiver) = tokio::sync::mpsc::unbounded_channel();

    let mem_pool = Arc::new(MemoryChunkPool::new(config, log_store.clone(), sender));
    let handler = ChunkPoolHandler::new(receiver, mem_pool.clone(), log_store, network_send);

    (mem_pool, handler)
}