...

herodb/examples/tst_index_example.rs

@@ -0,0 +1,93 @@
+use herodb::db::{DB, DBBuilder, Model, IndexKey};
+use herodb::models::biz::Customer;
+use std::path::PathBuf;
+use std::fs;
+
+fn main() -> Result<(), Box<dyn std::error::Error>> {
+    println!("TST Index Example");
+    println!("================");
+
+    // Create a temporary directory for the database
+    let db_path = PathBuf::from("/tmp/tst_index_example");
+    if db_path.exists() {
+        fs::remove_dir_all(&db_path)?;
+    }
+    fs::create_dir_all(&db_path)?;
+    println!("Database path: {:?}", db_path);
+
+    // Create a database instance with the Customer model registered
+    let db = DBBuilder::new(&db_path)
+        .register_model::<Customer>()
+        .build()?;
+
+    // Create some customers
+    let customer1 = Customer::new(
+        1,
+        "John Doe".to_string(),
+        "A regular customer".to_string(),
+        "pk123456".to_string(),
+    );
+
+    let customer2 = Customer::new(
+        2,
+        "Jane Smith".to_string(),
+        "A VIP customer".to_string(),
+        "pk789012".to_string(),
+    );
+
+    let customer3 = Customer::new(
+        3,
+        "John Smith".to_string(),
+        "Another customer".to_string(),
+        "pk345678".to_string(),
+    );
+
+    // Insert the customers
+    db.set(&customer1)?;
+    db.set(&customer2)?;
+    db.set(&customer3)?;
+
+    println!("\nCustomers created:");
+    println!("1. {} ({})", customer1.name, customer1.pubkey);
+    println!("2. {} ({})", customer2.name, customer2.pubkey);
+    println!("3. {} ({})", customer3.name, customer3.pubkey);
+
+    // List all customers
+    println!("\nListing all customers:");
+    let customers = db.list::<Customer>()?;
+    for customer in &customers {
+        println!("- {} (ID: {})", customer.name, customer.id);
+    }
+    println!("Total: {} customers", customers.len());
+
+    // Find customers by name index
+    println!("\nFinding customers by name 'John':");
+    let john_customers = db.find_by_index_prefix::<Customer>("name", "John")?;
+    for customer in &john_customers {
+        println!("- {} (ID: {})", customer.name, customer.id);
+    }
+    println!("Total: {} customers", john_customers.len());
+
+    // Find customers by pubkey index
+    println!("\nFinding customer by pubkey 'pk789012':");
+    let pubkey_customers = db.find_by_index::<Customer>("pubkey", "pk789012")?;
+    for customer in &pubkey_customers {
+        println!("- {} (ID: {})", customer.name, customer.id);
+    }
+    println!("Total: {} customers", pubkey_customers.len());
+
+    // Delete a customer
+    println!("\nDeleting customer with ID 2");
+    db.delete::<Customer>(2)?;
+
+    // List all customers again
+    println!("\nListing all customers after deletion:");
+    let customers = db.list::<Customer>()?;
+    for customer in &customers {
+        println!("- {} (ID: {})", customer.name, customer.id);
+    }
+    println!("Total: {} customers", customers.len());
+
+    println!("\nExample completed successfully!");
+    Ok(())
+}