58 lines
1.8 KiB
Rust
58 lines
1.8 KiB
Rust
use crate::kvs::kv::Key;
|
|
use crate::sql::statements::DefineDatabaseStatement;
|
|
use crate::sql::statements::DefineEventStatement;
|
|
use crate::sql::statements::DefineFieldStatement;
|
|
use crate::sql::statements::DefineIndexStatement;
|
|
use crate::sql::statements::DefineLoginStatement;
|
|
use crate::sql::statements::DefineNamespaceStatement;
|
|
use crate::sql::statements::DefineParamStatement;
|
|
use crate::sql::statements::DefineScopeStatement;
|
|
use crate::sql::statements::DefineTableStatement;
|
|
use crate::sql::statements::DefineTokenStatement;
|
|
use crate::sql::statements::LiveStatement;
|
|
use std::collections::HashMap;
|
|
use std::sync::Arc;
|
|
|
|
#[derive(Clone)]
|
|
pub enum Entry {
|
|
Ns(Arc<DefineNamespaceStatement>),
|
|
Db(Arc<DefineDatabaseStatement>),
|
|
Tb(Arc<DefineTableStatement>),
|
|
Nss(Arc<[DefineNamespaceStatement]>),
|
|
Nls(Arc<[DefineLoginStatement]>),
|
|
Nts(Arc<[DefineTokenStatement]>),
|
|
Dbs(Arc<[DefineDatabaseStatement]>),
|
|
Dls(Arc<[DefineLoginStatement]>),
|
|
Dts(Arc<[DefineTokenStatement]>),
|
|
Scs(Arc<[DefineScopeStatement]>),
|
|
Sts(Arc<[DefineTokenStatement]>),
|
|
Pas(Arc<[DefineParamStatement]>),
|
|
Tbs(Arc<[DefineTableStatement]>),
|
|
Evs(Arc<[DefineEventStatement]>),
|
|
Fds(Arc<[DefineFieldStatement]>),
|
|
Ixs(Arc<[DefineIndexStatement]>),
|
|
Fts(Arc<[DefineTableStatement]>),
|
|
Lvs(Arc<[LiveStatement]>),
|
|
}
|
|
|
|
#[derive(Default)]
|
|
pub struct Cache(pub HashMap<Key, Entry>);
|
|
|
|
impl Cache {
|
|
// Check if key exists
|
|
pub fn exi(&mut self, key: &Key) -> bool {
|
|
self.0.contains_key(key)
|
|
}
|
|
// Set a key in the cache
|
|
pub fn set(&mut self, key: Key, val: Entry) {
|
|
self.0.insert(key, val);
|
|
}
|
|
// Get a key from the cache
|
|
pub fn get(&mut self, key: &Key) -> Option<Entry> {
|
|
self.0.get(key).cloned()
|
|
}
|
|
// Delete a key from the cache
|
|
pub fn del(&mut self, key: &Key) -> Option<Entry> {
|
|
self.0.remove(key)
|
|
}
|
|
}
|