brk_computer 0.2.5

use brk_traversable::Traversable;
use brk_types::{Date, Day1, Height, StoredU64, Version};
use vecdb::{Database, EagerVec, ImportableVec, PcoVec, Rw, StorageMode};

use brk_error::Result;

#[derive(Traversable)]
pub struct Vecs<M: StorageMode = Rw> {
    pub identity: M::Stored<EagerVec<PcoVec<Day1, Day1>>>,
    pub date: M::Stored<EagerVec<PcoVec<Day1, Date>>>,
    pub first_height: M::Stored<EagerVec<PcoVec<Day1, Height>>>,
    pub height_count: M::Stored<EagerVec<PcoVec<Day1, StoredU64>>>,
}

impl Vecs {
    pub(crate) fn forced_import(db: &Database, version: Version) -> Result<Self> {
        Ok(Self {
            identity: EagerVec::forced_import(db, "day1_index", version)?,
            date: EagerVec::forced_import(db, "date", version + Version::ONE)?,
            first_height: EagerVec::forced_import(db, "first_height", version)?,
            height_count: EagerVec::forced_import(db, "height_count", version)?,
        })
    }
}