feat: add load_by_id function to catalog
This commit is contained in:
@@ -1,2 +1,5 @@
|
||||
pub mod sqlite_init_db;
|
||||
pub mod sqlite_load_by_id;
|
||||
pub mod sqlite_map_row_to_attribute;
|
||||
pub mod sqlite_map_row_to_entity;
|
||||
pub mod sqlite_persist_catalog;
|
||||
|
||||
36
01.workspace/heave/src/fun/sqlite_load_by_id.rs
Normal file
36
01.workspace/heave/src/fun/sqlite_load_by_id.rs
Normal file
@@ -0,0 +1,36 @@
|
||||
use crate::*;
|
||||
use rusqlite::*;
|
||||
|
||||
const SELECT_ENTITY_BY_ID: &str = r#"
|
||||
SELECT * FROM entity
|
||||
WHERE id = ?1;
|
||||
"#;
|
||||
|
||||
const SELECT_ATTRIBUTE_BY_FK: &str = r#"
|
||||
SELECT * FROM attribute
|
||||
WHERE entity_id = ?1;
|
||||
"#;
|
||||
|
||||
pub fn run(path: &path::Path, entity_id: &str) -> Option<Entity> {
|
||||
let connection = Connection::open(path).unwrap();
|
||||
let result = connection
|
||||
.query_one(SELECT_ENTITY_BY_ID, [entity_id], sqlite::map::row_to_entity)
|
||||
.optional();
|
||||
let mut entity = result.unwrap();
|
||||
if let Some(ref mut entity) = entity {
|
||||
let mut select_attributes_statement = connection.prepare(SELECT_ATTRIBUTE_BY_FK).unwrap();
|
||||
let attributes = select_attributes_statement
|
||||
.query_map([&entity.id], sqlite::map::row_to_attribute)
|
||||
.unwrap();
|
||||
for attribute in attributes {
|
||||
let attribute = attribute.unwrap();
|
||||
entity.attributes.insert(attribute.id.clone(), attribute);
|
||||
}
|
||||
}
|
||||
entity
|
||||
}
|
||||
|
||||
// #[cfg(test)]
|
||||
// mod unit_tests {
|
||||
// use super::*;
|
||||
// }
|
||||
25
01.workspace/heave/src/fun/sqlite_map_row_to_attribute.rs
Normal file
25
01.workspace/heave/src/fun/sqlite_map_row_to_attribute.rs
Normal file
@@ -0,0 +1,25 @@
|
||||
use crate::*;
|
||||
|
||||
pub fn run(row: &rusqlite::Row) -> rusqlite::Result<Attribute> {
|
||||
let id: String = row.get(0)?;
|
||||
let _entity_id: String = row.get(1)?;
|
||||
let signed_int: Option<i64> = row.get(2)?;
|
||||
let unsigned_int: Option<u64> = row.get(3)?;
|
||||
let real: Option<f64> = row.get(4)?;
|
||||
let text: Option<String> = row.get(5)?;
|
||||
let bool: Option<bool> = row.get(6)?;
|
||||
let value: Value = match (signed_int, unsigned_int, real, text, bool) {
|
||||
(Some(value), None, None, None, None) => Value::SignedInt(value),
|
||||
(None, Some(value), None, None, None) => Value::UnsignedInt(value),
|
||||
(None, None, Some(value), None, None) => Value::Real(value),
|
||||
(None, None, None, Some(value), None) => Value::Text(value),
|
||||
(None, None, None, None, Some(value)) => Value::Bool(value),
|
||||
_ => panic!(),
|
||||
};
|
||||
Ok(Attribute { id, value })
|
||||
}
|
||||
|
||||
// #[cfg(test)]
|
||||
// mod unit_tests {
|
||||
// use super::*;
|
||||
// }
|
||||
12
01.workspace/heave/src/fun/sqlite_map_row_to_entity.rs
Normal file
12
01.workspace/heave/src/fun/sqlite_map_row_to_entity.rs
Normal file
@@ -0,0 +1,12 @@
|
||||
use crate::*;
|
||||
|
||||
pub fn run(row: &rusqlite::Row) -> rusqlite::Result<Entity> {
|
||||
let id: String = row.get(0)?;
|
||||
let class: String = row.get(1)?;
|
||||
Ok(Entity::default().with_id(&id).with_class(&class))
|
||||
}
|
||||
|
||||
// #[cfg(test)]
|
||||
// mod unit_tests {
|
||||
// use super::*;
|
||||
// }
|
||||
@@ -17,6 +17,13 @@ mod sqlite {
|
||||
pub mod init {
|
||||
pub use crate::fun::sqlite_init_db::run as db;
|
||||
}
|
||||
pub mod load {
|
||||
pub use crate::fun::sqlite_load_by_id::run as by_id;
|
||||
}
|
||||
pub mod map {
|
||||
pub use crate::fun::sqlite_map_row_to_attribute::run as row_to_attribute;
|
||||
pub use crate::fun::sqlite_map_row_to_entity::run as row_to_entity;
|
||||
}
|
||||
pub mod persist {
|
||||
pub use crate::fun::sqlite_persist_catalog::run as catalog;
|
||||
}
|
||||
|
||||
@@ -38,6 +38,14 @@ impl O {
|
||||
let path = path::Path::new(&self.path);
|
||||
sqlite::persist::catalog(path, self);
|
||||
}
|
||||
pub fn load_by_id(&mut self, id: &str) {
|
||||
let path = path::Path::new(&self.path);
|
||||
let entity = sqlite::load::by_id(path, id);
|
||||
match entity {
|
||||
None => (),
|
||||
Some(entity) => self.insert(entity),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// impl std::fmt::Display for O {
|
||||
|
||||
@@ -161,4 +161,33 @@ mod tests {
|
||||
let read_product: Product = catalog.get(&expected_product.id).unwrap();
|
||||
assert_eq!(read_product, expected_product);
|
||||
}
|
||||
#[test]
|
||||
fn check_010() {
|
||||
// demonstrates load entity by id
|
||||
let tempfile = tempfile::NamedTempFile::new().unwrap();
|
||||
let path = tempfile.path().to_str().unwrap();
|
||||
// insert a new entity
|
||||
let mut catalog = Catalog::new(path);
|
||||
catalog.init();
|
||||
let entity = Entity::new("test")
|
||||
.with_attribute("int", Value::SignedInt(50))
|
||||
.with_attribute("string", Value::Text("text".to_string()));
|
||||
let id = entity.id.clone();
|
||||
catalog.insert(entity);
|
||||
catalog.persist();
|
||||
// read the entity in a new catalog
|
||||
let mut catalog = Catalog::new(path);
|
||||
catalog.load_by_id(&id);
|
||||
let entity: Option<Entity> = catalog.get(&id);
|
||||
assert_eq!(
|
||||
entity,
|
||||
Some(
|
||||
Entity::default()
|
||||
.with_id(&id)
|
||||
.with_class("test")
|
||||
.with_attribute("int", Value::SignedInt(50))
|
||||
.with_attribute("string", Value::Text("text".to_string()))
|
||||
)
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user