fabaccess-bffh/src/db/resources.rs

use rkyv::{Archive, Serialize, Deserialize};

use super::{
    AllocAdapter,
    DB,
};
use crate::db::AlignedAdapter;
use crate::db::raw::RawDB;
use std::sync::Arc;
use crate::db::{Environment, DatabaseFlags};
use crate::db::Result;

#[derive(Clone, Debug, PartialEq, Eq)]
#[derive(Archive, Serialize, Deserialize)]
#[derive(serde::Serialize, serde::Deserialize)]
pub struct Resource {
    uuid: u128,
    id: String,
    name_idx: u64,
    description_idx: u64,
}

#[derive(Clone)]
pub struct ResourceDB {
    env: Arc<Environment>,
    db: DB<AllocAdapter<Resource>>,
    id_index: DB<AlignedAdapter<u64>>,
}

impl ResourceDB {
    pub unsafe fn new(env: Arc<Environment>, db: RawDB, id_index: RawDB) -> Self {
        let db = DB::new_unchecked(db);
        let id_index = DB::new_unchecked(id_index);

        Self { env, db, id_index }
    }

    pub unsafe fn open(env: Arc<Environment>) -> Result<Self> {
        let db = RawDB::open(&env, Some("resources"))?;
        let idx = RawDB::open(&env, Some("resources-idx"))?;
        Ok(Self::new(env, db, idx))
    }

    pub unsafe fn create(env: Arc<Environment>) -> Result<Self> {
        let flags = DatabaseFlags::empty();
        let db = RawDB::create(&env, Some("resources"), flags)?;
        let idx = RawDB::create(&env, Some("resources-idx"), flags)?;
        Ok(Self::new(env, db, idx))
    }

    pub fn lookup_id<S: AsRef<str>>(&self, id: S) -> Result<Option<u64>> {
        let txn = self.env.begin_ro_txn()?;
        let id = self.id_index.get(&txn, &id.as_ref().as_bytes()).map(|ok| {
            ok.map(|num| *num)
        })?;
        Ok(id)
    }
}