lighthouse/beacon_node/beacon_chain/src/blob_cache.rs

use lru::LruCache;
use parking_lot::Mutex;
use types::{BlobSidecarList, EthSpec, Hash256};

pub const DEFAULT_BLOB_CACHE_SIZE: usize = 10;

/// A cache blobs by beacon block root.
pub struct BlobCache<T: EthSpec> {
    blobs: Mutex<LruCache<BlobCacheId, BlobSidecarList<T>>>,
}

#[derive(Hash, PartialEq, Eq)]
struct BlobCacheId(Hash256);

impl<T: EthSpec> Default for BlobCache<T> {
    fn default() -> Self {
        BlobCache {
            blobs: Mutex::new(LruCache::new(DEFAULT_BLOB_CACHE_SIZE)),
        }
    }
}

impl<T: EthSpec> BlobCache<T> {
    pub fn put(
        &self,
        beacon_block: Hash256,
        blobs: BlobSidecarList<T>,
    ) -> Option<BlobSidecarList<T>> {
        self.blobs.lock().put(BlobCacheId(beacon_block), blobs)
    }

    pub fn pop(&self, root: &Hash256) -> Option<BlobSidecarList<T>> {
        self.blobs.lock().pop(&BlobCacheId(*root))
    }
}