59 lines
1.8 KiB
Rust
59 lines
1.8 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,
|
|
pub login: String,
|
|
pub full_name: String,
|
|
}
|
|
|
|
fn main() {
|
|
println!("Hero Models - Simple Model Example");
|
|
println!("==================================");
|
|
|
|
// Create a new DB instance, reset before every run
|
|
let db_path = "/tmp/ourdb_simple_model_example";
|
|
let db = heromodels::db::hero::OurDB::new(db_path, true).expect("Can create DB");
|
|
|
|
// Example usage of the generated implementation
|
|
println!("SimpleUser DB Prefix: {}", SimpleUser::db_prefix());
|
|
|
|
// Create a new user with ID 0 (will be auto-generated when saved)
|
|
let user = SimpleUser {
|
|
base_data: BaseModelData::new(),
|
|
login: "johndoe".to_string(),
|
|
full_name: "John Doe".to_string(),
|
|
};
|
|
|
|
println!("\nBefore saving - SimpleUser ID: {}", user.get_id());
|
|
println!("Before saving - SimpleUser DB Keys: {:?}", user.db_keys());
|
|
|
|
// Save the user to the database
|
|
let collection = db
|
|
.collection::<SimpleUser>()
|
|
.expect("can open user collection");
|
|
let (user_id, saved_user) = collection.set(&user).expect("can save user");
|
|
|
|
println!("\nAfter saving - SimpleUser ID: {}", saved_user.get_id());
|
|
println!(
|
|
"After saving - SimpleUser DB Keys: {:?}",
|
|
saved_user.db_keys()
|
|
);
|
|
println!("Returned ID: {}", user_id);
|
|
|
|
// Verify that the ID was auto-generated
|
|
assert_eq!(saved_user.get_id(), user_id);
|
|
assert_ne!(saved_user.get_id(), 0);
|
|
|
|
println!("\nExample finished. DB stored at {}", db_path);
|
|
println!(
|
|
"To clean up, you can manually delete the directory: {}",
|
|
db_path
|
|
);
|
|
}
|