109 lines
3.2 KiB
Rust
109 lines
3.2 KiB
Rust
use heromodels::db::{Collection, Db};
|
|
use heromodels_core::{BaseModelData, Model};
|
|
use heromodels_derive::model;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
// Basic usage
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
#[model]
|
|
pub struct SimpleUser {
|
|
pub base_data: BaseModelData,
|
|
|
|
#[index]
|
|
pub login: String,
|
|
|
|
pub full_name: String,
|
|
}
|
|
|
|
// With customization options
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
#[model]
|
|
pub struct CustomUser {
|
|
pub base_data: BaseModelData,
|
|
|
|
#[index(name = "user_name")]
|
|
pub login_name: String,
|
|
|
|
#[index]
|
|
pub is_active: bool,
|
|
|
|
pub full_name: String,
|
|
}
|
|
|
|
fn main() {
|
|
println!("Hero Models - Model Macro Example");
|
|
println!("=================================");
|
|
|
|
// Create a new DB instance, reset before every run
|
|
let db_path = "/tmp/ourdb_model_macro_example";
|
|
let db = heromodels::db::hero::OurDB::new(db_path, true).expect("Can create DB");
|
|
|
|
// Example usage of the generated implementations
|
|
println!("SimpleUser DB Prefix: {}", SimpleUser::db_prefix());
|
|
println!("CustomUser DB Prefix: {}", CustomUser::db_prefix());
|
|
|
|
let user = SimpleUser {
|
|
base_data: BaseModelData::new(), // ID will be auto-generated by OurDB
|
|
login: "johndoe".to_string(),
|
|
full_name: "John Doe".to_string(),
|
|
};
|
|
|
|
let custom_user = CustomUser {
|
|
base_data: BaseModelData::new(), // ID will be auto-generated by OurDB
|
|
login_name: "janesmith".to_string(),
|
|
is_active: true,
|
|
full_name: "Jane Smith".to_string(),
|
|
};
|
|
|
|
println!("\nBefore saving - SimpleUser ID: {}", user.get_id());
|
|
println!("Before saving - SimpleUser DB Keys: {:?}", user.db_keys());
|
|
|
|
println!("\nBefore saving - CustomUser ID: {}", custom_user.get_id());
|
|
println!(
|
|
"Before saving - CustomUser DB Keys: {:?}",
|
|
custom_user.db_keys()
|
|
);
|
|
|
|
// Save the models to the database
|
|
let simple_collection = db
|
|
.collection::<SimpleUser>()
|
|
.expect("can open simple user collection");
|
|
let custom_collection = db
|
|
.collection::<CustomUser>()
|
|
.expect("can open custom user collection");
|
|
|
|
let (user_id, saved_user) = simple_collection.set(&user).expect("can save simple user");
|
|
let (custom_user_id, saved_custom_user) = custom_collection
|
|
.set(&custom_user)
|
|
.expect("can save custom user");
|
|
|
|
println!("\nAfter saving - SimpleUser ID: {}", saved_user.get_id());
|
|
println!(
|
|
"After saving - SimpleUser DB Keys: {:?}",
|
|
saved_user.db_keys()
|
|
);
|
|
println!("Returned SimpleUser ID: {}", user_id);
|
|
|
|
println!(
|
|
"\nAfter saving - CustomUser ID: {}",
|
|
saved_custom_user.get_id()
|
|
);
|
|
println!(
|
|
"After saving - CustomUser DB Keys: {:?}",
|
|
saved_custom_user.db_keys()
|
|
);
|
|
println!("Returned CustomUser ID: {}", custom_user_id);
|
|
|
|
// Verify that the IDs were auto-generated
|
|
assert_eq!(saved_user.get_id(), user_id);
|
|
assert_ne!(saved_user.get_id(), 0);
|
|
assert_eq!(saved_custom_user.get_id(), custom_user_id);
|
|
assert_ne!(saved_custom_user.get_id(), 0);
|
|
|
|
println!("\nExample finished. DB stored at {}", db_path);
|
|
println!(
|
|
"To clean up, you can manually delete the directory: {}",
|
|
db_path
|
|
);
|
|
}
|