brk/crates/brk_computer/src/indexes/quarterindex.rs

use brk_traversable::Traversable;
use brk_types::{Date, MonthIndex, QuarterIndex, StoredU64, Version};
use vecdb::{Database, EagerVec, ImportableVec, PcoVec};

use brk_error::Result;

#[derive(Clone, Traversable)]
pub struct Vecs {
    pub identity: EagerVec<PcoVec<QuarterIndex, QuarterIndex>>,
    pub date: EagerVec<PcoVec<QuarterIndex, Date>>,
    pub first_monthindex: EagerVec<PcoVec<QuarterIndex, MonthIndex>>,
    pub monthindex_count: EagerVec<PcoVec<QuarterIndex, StoredU64>>,
}

impl Vecs {
    pub fn forced_import(db: &Database, version: Version) -> Result<Self> {
        Ok(Self {
            identity: EagerVec::forced_import(db, "quarterindex", version)?,
            date: EagerVec::forced_import(db, "date", version)?,
            first_monthindex: EagerVec::forced_import(db, "first_monthindex", version)?,
            monthindex_count: EagerVec::forced_import(db, "monthindex_count", version)?,
        })
    }
}