coordinode-lsm-tree 4.2.0

// Copyright (c) 2024-present, fjall-rs
// This source code is licensed under both the Apache 2.0 and MIT License
// (found in the LICENSE-* files in the repository)

use crate::TableId;

/// The hidden set keeps track of which tables are currently being compacted
///
/// When a table is hidden (being compacted), no other compaction task can include that
/// table, or it will be declined to be run.
///
/// If a compaction task fails, the tables are shown again (removed from the hidden set).
#[derive(Clone, Default)]
pub struct HiddenSet {
    pub(crate) set: crate::HashSet<TableId>,
}

impl HiddenSet {
    pub(crate) fn hide<T: IntoIterator<Item = TableId>>(&mut self, keys: T) {
        self.set.extend(keys);
    }

    pub(crate) fn show<T: IntoIterator<Item = TableId>>(&mut self, keys: T) {
        for key in keys {
            self.set.remove(&key);
        }
    }

    pub(crate) fn is_blocked<T: IntoIterator<Item = TableId>>(&self, ids: T) -> bool {
        ids.into_iter().any(|id| self.is_hidden(id))
    }

    pub(crate) fn is_hidden(&self, key: TableId) -> bool {
        self.set.contains(&key)
    }

    pub(crate) fn is_empty(&self) -> bool {
        self.set.is_empty()
    }

    pub(crate) fn should_decline_compaction<T: IntoIterator<Item = TableId>>(
        &self,
        ids: T,
    ) -> bool {
        self.is_blocked(ids)
    }
}