rename worker to actor
This commit is contained in:
		
							
								
								
									
										113
									
								
								_archive/core/actor/cmd/README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										113
									
								
								_archive/core/actor/cmd/README.md
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,113 @@
 | 
			
		||||
# Rhai Actor Binary
 | 
			
		||||
 | 
			
		||||
A command-line actor for executing Rhai scripts from Redis task queues.
 | 
			
		||||
 | 
			
		||||
## Binary: `actor`
 | 
			
		||||
 | 
			
		||||
### Installation
 | 
			
		||||
 | 
			
		||||
Build the binary:
 | 
			
		||||
```bash
 | 
			
		||||
cargo build --bin actor --release
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Usage
 | 
			
		||||
 | 
			
		||||
```bash
 | 
			
		||||
# Basic usage - requires circle public key
 | 
			
		||||
actor --circle-public-key <CIRCLE_PUBLIC_KEY>
 | 
			
		||||
 | 
			
		||||
# Custom Redis URL
 | 
			
		||||
actor -c <CIRCLE_PUBLIC_KEY> --redis-url redis://localhost:6379/1
 | 
			
		||||
 | 
			
		||||
# Custom actor ID and database path
 | 
			
		||||
actor -c <CIRCLE_PUBLIC_KEY> --actor-id my_actor --db-path /tmp/actor_db
 | 
			
		||||
 | 
			
		||||
# Preserve tasks for debugging/benchmarking
 | 
			
		||||
actor -c <CIRCLE_PUBLIC_KEY> --preserve-tasks
 | 
			
		||||
 | 
			
		||||
# Remove timestamps from logs
 | 
			
		||||
actor -c <CIRCLE_PUBLIC_KEY> --no-timestamp
 | 
			
		||||
 | 
			
		||||
# Increase verbosity
 | 
			
		||||
actor -c <CIRCLE_PUBLIC_KEY> -v    # Debug logging
 | 
			
		||||
actor -c <CIRCLE_PUBLIC_KEY> -vv   # Full debug
 | 
			
		||||
actor -c <CIRCLE_PUBLIC_KEY> -vvv  # Trace logging
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Command-Line Options
 | 
			
		||||
 | 
			
		||||
| Option | Short | Default | Description |
 | 
			
		||||
|--------|-------|---------|-------------|
 | 
			
		||||
| `--circle-public-key` | `-c` | **Required** | Circle public key to listen for tasks |
 | 
			
		||||
| `--redis-url` | `-r` | `redis://localhost:6379` | Redis connection URL |
 | 
			
		||||
| `--actor-id` | `-w` | `actor_1` | Unique actor identifier |
 | 
			
		||||
| `--preserve-tasks` | | `false` | Preserve task details after completion |
 | 
			
		||||
| `--db-path` | | `actor_rhai_temp_db` | Database path for Rhai engine |
 | 
			
		||||
| `--no-timestamp` | | `false` | Remove timestamps from log output |
 | 
			
		||||
| `--verbose` | `-v` | | Increase verbosity (stackable) |
 | 
			
		||||
 | 
			
		||||
### Features
 | 
			
		||||
 | 
			
		||||
- **Task Queue Processing**: Listens to Redis queues for Rhai script execution tasks
 | 
			
		||||
- **Performance Optimized**: Configured for maximum Rhai engine performance
 | 
			
		||||
- **Graceful Shutdown**: Supports shutdown signals for clean termination
 | 
			
		||||
- **Flexible Logging**: Configurable verbosity and timestamp control
 | 
			
		||||
- **Database Integration**: Uses heromodels for data persistence
 | 
			
		||||
- **Task Cleanup**: Optional task preservation for debugging/benchmarking
 | 
			
		||||
 | 
			
		||||
### How It Works
 | 
			
		||||
 | 
			
		||||
1. **Queue Listening**: Actor listens on Redis queue `baobab:{circle_public_key}`
 | 
			
		||||
2. **Task Processing**: Receives task IDs, fetches task details from Redis
 | 
			
		||||
3. **Script Execution**: Executes Rhai scripts with configured engine
 | 
			
		||||
4. **Result Handling**: Updates task status and sends results to reply queues
 | 
			
		||||
5. **Cleanup**: Optionally cleans up task details after completion
 | 
			
		||||
 | 
			
		||||
### Configuration Examples
 | 
			
		||||
 | 
			
		||||
#### Development Actor
 | 
			
		||||
```bash
 | 
			
		||||
# Simple development actor
 | 
			
		||||
actor -c dev_circle_123
 | 
			
		||||
 | 
			
		||||
# Development with verbose logging (no timestamps)
 | 
			
		||||
actor -c dev_circle_123 -v --no-timestamp
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
#### Production Actor
 | 
			
		||||
```bash
 | 
			
		||||
# Production actor with custom configuration
 | 
			
		||||
actor \
 | 
			
		||||
  --circle-public-key prod_circle_456 \
 | 
			
		||||
  --redis-url redis://redis-server:6379/0 \
 | 
			
		||||
  --actor-id prod_actor_1 \
 | 
			
		||||
  --db-path /var/lib/actor/db \
 | 
			
		||||
  --preserve-tasks
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
#### Benchmarking Actor
 | 
			
		||||
```bash
 | 
			
		||||
# Actor optimized for benchmarking
 | 
			
		||||
actor \
 | 
			
		||||
  --circle-public-key bench_circle_789 \
 | 
			
		||||
  --preserve-tasks \
 | 
			
		||||
  --no-timestamp \
 | 
			
		||||
  -vv
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Error Handling
 | 
			
		||||
 | 
			
		||||
The actor provides clear error messages for:
 | 
			
		||||
- Missing or invalid circle public key
 | 
			
		||||
- Redis connection failures
 | 
			
		||||
- Script execution errors
 | 
			
		||||
- Database access issues
 | 
			
		||||
 | 
			
		||||
### Dependencies
 | 
			
		||||
 | 
			
		||||
- `baobab_engine`: Rhai engine with heromodels integration
 | 
			
		||||
- `redis`: Redis client for task queue management
 | 
			
		||||
- `rhai`: Script execution engine
 | 
			
		||||
- `clap`: Command-line argument parsing
 | 
			
		||||
- `env_logger`: Logging infrastructure
 | 
			
		||||
							
								
								
									
										233
									
								
								_archive/core/actor/cmd/osis.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										233
									
								
								_archive/core/actor/cmd/osis.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,233 @@
 | 
			
		||||
//! OSIS Actor Binary - Synchronous actor for system-level operations
 | 
			
		||||
 | 
			
		||||
use clap::Parser;
 | 
			
		||||
use log::{error, info};
 | 
			
		||||
use baobab_actor::config::{ConfigError, ActorConfig};
 | 
			
		||||
use baobab_actor::engine::create_heromodels_engine;
 | 
			
		||||
use baobab_actor::sync_actor::SyncActor;
 | 
			
		||||
use baobab_actor::actor_trait::{spawn_actor, ActorConfig as TraitActorConfig};
 | 
			
		||||
use std::path::PathBuf;
 | 
			
		||||
use std::sync::Arc;
 | 
			
		||||
use tokio::signal;
 | 
			
		||||
use tokio::sync::mpsc;
 | 
			
		||||
 | 
			
		||||
#[derive(Parser, Debug)]
 | 
			
		||||
#[command(
 | 
			
		||||
    name = "osis",
 | 
			
		||||
    version = "0.1.0",
 | 
			
		||||
    about = "OSIS (Operating System Integration Service) - Synchronous Actor",
 | 
			
		||||
    long_about = "A synchronous actor for Hero framework that processes jobs sequentially. \
 | 
			
		||||
                  Ideal for system-level operations that require careful resource management."
 | 
			
		||||
)]
 | 
			
		||||
struct Args {
 | 
			
		||||
    /// Path to TOML configuration file
 | 
			
		||||
    #[arg(short, long, help = "Path to TOML configuration file")]
 | 
			
		||||
    config: PathBuf,
 | 
			
		||||
 | 
			
		||||
    /// Override actor ID from config
 | 
			
		||||
    #[arg(long, help = "Override actor ID from configuration file")]
 | 
			
		||||
    actor_id: Option<String>,
 | 
			
		||||
 | 
			
		||||
    /// Override Redis URL from config
 | 
			
		||||
    #[arg(long, help = "Override Redis URL from configuration file")]
 | 
			
		||||
    redis_url: Option<String>,
 | 
			
		||||
 | 
			
		||||
    /// Override database path from config
 | 
			
		||||
    #[arg(long, help = "Override database path from configuration file")]
 | 
			
		||||
    db_path: Option<String>,
 | 
			
		||||
 | 
			
		||||
    /// Enable verbose logging (debug level)
 | 
			
		||||
    #[arg(short, long, help = "Enable verbose logging")]
 | 
			
		||||
    verbose: bool,
 | 
			
		||||
 | 
			
		||||
    /// Disable timestamps in log output
 | 
			
		||||
    #[arg(long, help = "Remove timestamps from log output")]
 | 
			
		||||
    no_timestamp: bool,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[tokio::main]
 | 
			
		||||
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
 | 
			
		||||
    let args = Args::parse();
 | 
			
		||||
 | 
			
		||||
    // Load configuration from TOML file
 | 
			
		||||
    let mut config = match ActorConfig::from_file(&args.config) {
 | 
			
		||||
        Ok(config) => config,
 | 
			
		||||
        Err(e) => {
 | 
			
		||||
            eprintln!("Failed to load configuration from {:?}: {}", args.config, e);
 | 
			
		||||
            std::process::exit(1);
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    // Validate that this is a sync actor configuration
 | 
			
		||||
    if !config.is_sync() {
 | 
			
		||||
        eprintln!("Error: OSIS actor requires a sync actor configuration");
 | 
			
		||||
        eprintln!("Expected: [actor_type] type = \"sync\"");
 | 
			
		||||
        eprintln!("Found: {:?}", config.actor_type);
 | 
			
		||||
        std::process::exit(1);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Apply command line overrides
 | 
			
		||||
    if let Some(actor_id) = args.actor_id {
 | 
			
		||||
        config.actor_id = actor_id;
 | 
			
		||||
    }
 | 
			
		||||
    if let Some(redis_url) = args.redis_url {
 | 
			
		||||
        config.redis_url = redis_url;
 | 
			
		||||
    }
 | 
			
		||||
    if let Some(db_path) = args.db_path {
 | 
			
		||||
        config.db_path = db_path;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Configure logging
 | 
			
		||||
    setup_logging(&config, args.verbose, args.no_timestamp)?;
 | 
			
		||||
 | 
			
		||||
    info!("🚀 OSIS Actor starting...");
 | 
			
		||||
    info!("Actor ID: {}", config.actor_id);
 | 
			
		||||
    info!("Redis URL: {}", config.redis_url);
 | 
			
		||||
    info!("Database Path: {}", config.db_path);
 | 
			
		||||
    info!("Preserve Tasks: {}", config.preserve_tasks);
 | 
			
		||||
 | 
			
		||||
    // Create Rhai engine
 | 
			
		||||
    let engine = create_heromodels_engine();
 | 
			
		||||
    info!("✅ Rhai engine initialized");
 | 
			
		||||
 | 
			
		||||
    // Create actor configuration for the trait-based interface
 | 
			
		||||
    let actor_config = TraitActorConfig::new(
 | 
			
		||||
        config.actor_id.clone(),
 | 
			
		||||
        config.db_path.clone(),
 | 
			
		||||
        config.redis_url.clone(),
 | 
			
		||||
        config.preserve_tasks,
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    // Create sync actor instance
 | 
			
		||||
    let actor = Arc::new(SyncActor::default());
 | 
			
		||||
    info!("✅ Sync actor instance created");
 | 
			
		||||
 | 
			
		||||
    // Setup shutdown signal handling
 | 
			
		||||
    let (shutdown_tx, shutdown_rx) = mpsc::channel(1);
 | 
			
		||||
    
 | 
			
		||||
    // Spawn shutdown signal handler
 | 
			
		||||
    let shutdown_tx_clone = shutdown_tx.clone();
 | 
			
		||||
    tokio::spawn(async move {
 | 
			
		||||
        if let Err(e) = signal::ctrl_c().await {
 | 
			
		||||
            error!("Failed to listen for shutdown signal: {}", e);
 | 
			
		||||
            return;
 | 
			
		||||
        }
 | 
			
		||||
        info!("🛑 Shutdown signal received");
 | 
			
		||||
        if let Err(e) = shutdown_tx_clone.send(()).await {
 | 
			
		||||
            error!("Failed to send shutdown signal: {}", e);
 | 
			
		||||
        }
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    // Spawn the actor
 | 
			
		||||
    info!("🔄 Starting actor loop...");
 | 
			
		||||
    let actor_handle = spawn_actor(actor, engine, shutdown_rx);
 | 
			
		||||
 | 
			
		||||
    // Wait for the actor to complete
 | 
			
		||||
    match actor_handle.await {
 | 
			
		||||
        Ok(Ok(())) => {
 | 
			
		||||
            info!("✅ OSIS Actor shut down gracefully");
 | 
			
		||||
        }
 | 
			
		||||
        Ok(Err(e)) => {
 | 
			
		||||
            error!("❌ OSIS Actor encountered an error: {}", e);
 | 
			
		||||
            std::process::exit(1);
 | 
			
		||||
        }
 | 
			
		||||
        Err(e) => {
 | 
			
		||||
            error!("❌ Failed to join actor task: {}", e);
 | 
			
		||||
            std::process::exit(1);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Setup logging based on configuration and command line arguments
 | 
			
		||||
fn setup_logging(
 | 
			
		||||
    config: &ActorConfig,
 | 
			
		||||
    verbose: bool,
 | 
			
		||||
    no_timestamp: bool,
 | 
			
		||||
) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
 | 
			
		||||
    let mut builder = env_logger::Builder::new();
 | 
			
		||||
 | 
			
		||||
    // Determine log level
 | 
			
		||||
    let log_level = if verbose {
 | 
			
		||||
        "debug"
 | 
			
		||||
    } else {
 | 
			
		||||
        &config.logging.level
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    // Set log level
 | 
			
		||||
    builder.filter_level(match log_level.to_lowercase().as_str() {
 | 
			
		||||
        "trace" => log::LevelFilter::Trace,
 | 
			
		||||
        "debug" => log::LevelFilter::Debug,
 | 
			
		||||
        "info" => log::LevelFilter::Info,
 | 
			
		||||
        "warn" => log::LevelFilter::Warn,
 | 
			
		||||
        "error" => log::LevelFilter::Error,
 | 
			
		||||
        _ => {
 | 
			
		||||
            eprintln!("Invalid log level: {}. Using 'info'", log_level);
 | 
			
		||||
            log::LevelFilter::Info
 | 
			
		||||
        }
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    // Configure timestamps
 | 
			
		||||
    let show_timestamps = !no_timestamp && config.logging.timestamps;
 | 
			
		||||
    if !show_timestamps {
 | 
			
		||||
        builder.format_timestamp(None);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    builder.init();
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod tests {
 | 
			
		||||
    use super::*;
 | 
			
		||||
    use std::io::Write;
 | 
			
		||||
    use tempfile::NamedTempFile;
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn test_config_validation() {
 | 
			
		||||
        let config_toml = r#"
 | 
			
		||||
actor_id = "test_osis"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/test_db"
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "sync"
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
level = "info"
 | 
			
		||||
"#;
 | 
			
		||||
 | 
			
		||||
        let mut temp_file = NamedTempFile::new().unwrap();
 | 
			
		||||
        temp_file.write_all(config_toml.as_bytes()).unwrap();
 | 
			
		||||
 | 
			
		||||
        let config = ActorConfig::from_file(temp_file.path()).unwrap();
 | 
			
		||||
        assert!(config.is_sync());
 | 
			
		||||
        assert!(!config.is_async());
 | 
			
		||||
        assert_eq!(config.actor_id, "test_osis");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn test_async_config_rejection() {
 | 
			
		||||
        let config_toml = r#"
 | 
			
		||||
actor_id = "test_osis"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/test_db"
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "async"
 | 
			
		||||
default_timeout_seconds = 300
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
level = "info"
 | 
			
		||||
"#;
 | 
			
		||||
 | 
			
		||||
        let mut temp_file = NamedTempFile::new().unwrap();
 | 
			
		||||
        temp_file.write_all(config_toml.as_bytes()).unwrap();
 | 
			
		||||
 | 
			
		||||
        let config = ActorConfig::from_file(temp_file.path()).unwrap();
 | 
			
		||||
        assert!(!config.is_sync());
 | 
			
		||||
        assert!(config.is_async());
 | 
			
		||||
        // This would be rejected in main() function
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										302
									
								
								_archive/core/actor/cmd/system.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										302
									
								
								_archive/core/actor/cmd/system.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,302 @@
 | 
			
		||||
//! System Actor Binary - Asynchronous actor for high-throughput concurrent processing
 | 
			
		||||
 | 
			
		||||
use clap::Parser;
 | 
			
		||||
use log::{error, info, warn};
 | 
			
		||||
use baobab_actor::async_actor_impl::AsyncActor;
 | 
			
		||||
use baobab_actor::config::{ConfigError, ActorConfig};
 | 
			
		||||
use baobab_actor::engine::create_heromodels_engine;
 | 
			
		||||
use baobab_actor::actor_trait::{spawn_actor, ActorConfig as TraitActorConfig};
 | 
			
		||||
use std::path::PathBuf;
 | 
			
		||||
use std::sync::Arc;
 | 
			
		||||
use std::time::Duration;
 | 
			
		||||
use tokio::signal;
 | 
			
		||||
use tokio::sync::mpsc;
 | 
			
		||||
 | 
			
		||||
#[derive(Parser, Debug)]
 | 
			
		||||
#[command(
 | 
			
		||||
    name = "system",
 | 
			
		||||
    version = "0.1.0",
 | 
			
		||||
    about = "System Actor - Asynchronous Actor with Concurrent Job Processing",
 | 
			
		||||
    long_about = "An asynchronous actor for Hero framework that processes multiple jobs \
 | 
			
		||||
                  concurrently with timeout support. Ideal for high-throughput scenarios \
 | 
			
		||||
                  where jobs can be executed in parallel."
 | 
			
		||||
)]
 | 
			
		||||
struct Args {
 | 
			
		||||
    /// Path to TOML configuration file
 | 
			
		||||
    #[arg(short, long, help = "Path to TOML configuration file")]
 | 
			
		||||
    config: PathBuf,
 | 
			
		||||
 | 
			
		||||
    /// Override actor ID from config
 | 
			
		||||
    #[arg(long, help = "Override actor ID from configuration file")]
 | 
			
		||||
    actor_id: Option<String>,
 | 
			
		||||
 | 
			
		||||
    /// Override Redis URL from config
 | 
			
		||||
    #[arg(long, help = "Override Redis URL from configuration file")]
 | 
			
		||||
    redis_url: Option<String>,
 | 
			
		||||
 | 
			
		||||
    /// Override database path from config
 | 
			
		||||
    #[arg(long, help = "Override database path from configuration file")]
 | 
			
		||||
    db_path: Option<String>,
 | 
			
		||||
 | 
			
		||||
    /// Override default timeout in seconds
 | 
			
		||||
    #[arg(long, help = "Override default job timeout in seconds")]
 | 
			
		||||
    timeout: Option<u64>,
 | 
			
		||||
 | 
			
		||||
    /// Enable verbose logging (debug level)
 | 
			
		||||
    #[arg(short, long, help = "Enable verbose logging")]
 | 
			
		||||
    verbose: bool,
 | 
			
		||||
 | 
			
		||||
    /// Disable timestamps in log output
 | 
			
		||||
    #[arg(long, help = "Remove timestamps from log output")]
 | 
			
		||||
    no_timestamp: bool,
 | 
			
		||||
 | 
			
		||||
    /// Show actor statistics periodically
 | 
			
		||||
    #[arg(long, help = "Show periodic actor statistics")]
 | 
			
		||||
    show_stats: bool,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[tokio::main]
 | 
			
		||||
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
 | 
			
		||||
    let args = Args::parse();
 | 
			
		||||
 | 
			
		||||
    // Load configuration from TOML file
 | 
			
		||||
    let mut config = match ActorConfig::from_file(&args.config) {
 | 
			
		||||
        Ok(config) => config,
 | 
			
		||||
        Err(e) => {
 | 
			
		||||
            eprintln!("Failed to load configuration from {:?}: {}", args.config, e);
 | 
			
		||||
            std::process::exit(1);
 | 
			
		||||
        }
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    // Validate that this is an async actor configuration
 | 
			
		||||
    if !config.is_async() {
 | 
			
		||||
        eprintln!("Error: System actor requires an async actor configuration");
 | 
			
		||||
        eprintln!("Expected: [actor_type] type = \"async\"");
 | 
			
		||||
        eprintln!("Found: {:?}", config.actor_type);
 | 
			
		||||
        std::process::exit(1);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Apply command line overrides
 | 
			
		||||
    if let Some(actor_id) = args.actor_id {
 | 
			
		||||
        config.actor_id = actor_id;
 | 
			
		||||
    }
 | 
			
		||||
    if let Some(redis_url) = args.redis_url {
 | 
			
		||||
        config.redis_url = redis_url;
 | 
			
		||||
    }
 | 
			
		||||
    if let Some(db_path) = args.db_path {
 | 
			
		||||
        config.db_path = db_path;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Override timeout if specified
 | 
			
		||||
    if let Some(timeout_secs) = args.timeout {
 | 
			
		||||
        if let baobab_actor::config::ActorType::Async { ref mut default_timeout_seconds } = config.actor_type {
 | 
			
		||||
            *default_timeout_seconds = timeout_secs;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Configure logging
 | 
			
		||||
    setup_logging(&config, args.verbose, args.no_timestamp)?;
 | 
			
		||||
 | 
			
		||||
    info!("🚀 System Actor starting...");
 | 
			
		||||
    info!("Actor ID: {}", config.actor_id);
 | 
			
		||||
    info!("Redis URL: {}", config.redis_url);
 | 
			
		||||
    info!("Database Path: {}", config.db_path);
 | 
			
		||||
    info!("Preserve Tasks: {}", config.preserve_tasks);
 | 
			
		||||
    
 | 
			
		||||
    if let Some(timeout) = config.get_default_timeout() {
 | 
			
		||||
        info!("Default Timeout: {:?}", timeout);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Create Rhai engine
 | 
			
		||||
    let engine = create_heromodels_engine();
 | 
			
		||||
    info!("✅ Rhai engine initialized");
 | 
			
		||||
 | 
			
		||||
    // Create actor configuration for the trait-based interface
 | 
			
		||||
    let mut actor_config = TraitActorConfig::new(
 | 
			
		||||
        config.actor_id.clone(),
 | 
			
		||||
        config.db_path.clone(),
 | 
			
		||||
        config.redis_url.clone(),
 | 
			
		||||
        config.preserve_tasks,
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    // Add timeout configuration for async actor
 | 
			
		||||
    if let Some(timeout) = config.get_default_timeout() {
 | 
			
		||||
        actor_config = actor_config.with_default_timeout(timeout);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Create async actor instance
 | 
			
		||||
    let actor = Arc::new(AsyncActor::default());
 | 
			
		||||
    info!("✅ Async actor instance created");
 | 
			
		||||
 | 
			
		||||
    // Setup shutdown signal handling
 | 
			
		||||
    let (shutdown_tx, shutdown_rx) = mpsc::channel(1);
 | 
			
		||||
    
 | 
			
		||||
    // Spawn shutdown signal handler
 | 
			
		||||
    let shutdown_tx_clone = shutdown_tx.clone();
 | 
			
		||||
    tokio::spawn(async move {
 | 
			
		||||
        if let Err(e) = signal::ctrl_c().await {
 | 
			
		||||
            error!("Failed to listen for shutdown signal: {}", e);
 | 
			
		||||
            return;
 | 
			
		||||
        }
 | 
			
		||||
        info!("🛑 Shutdown signal received");
 | 
			
		||||
        if let Err(e) = shutdown_tx_clone.send(()).await {
 | 
			
		||||
            error!("Failed to send shutdown signal: {}", e);
 | 
			
		||||
        }
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    // Spawn statistics reporter if requested
 | 
			
		||||
    if args.show_stats {
 | 
			
		||||
        let actor_stats = Arc::clone(&actor);
 | 
			
		||||
        tokio::spawn(async move {
 | 
			
		||||
            let mut interval = tokio::time::interval(Duration::from_secs(30));
 | 
			
		||||
            loop {
 | 
			
		||||
                interval.tick().await;
 | 
			
		||||
                let running_count = actor_stats.running_job_count().await;
 | 
			
		||||
                if running_count > 0 {
 | 
			
		||||
                    info!("📊 Actor Stats: {} jobs currently running", running_count);
 | 
			
		||||
                } else {
 | 
			
		||||
                    info!("📊 Actor Stats: No jobs currently running");
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        });
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Spawn the actor
 | 
			
		||||
    info!("🔄 Starting actor loop...");
 | 
			
		||||
    let actor_handle = spawn_actor(actor, engine, shutdown_rx);
 | 
			
		||||
 | 
			
		||||
    // Wait for the actor to complete
 | 
			
		||||
    match actor_handle.await {
 | 
			
		||||
        Ok(Ok(())) => {
 | 
			
		||||
            info!("✅ System Actor shut down gracefully");
 | 
			
		||||
        }
 | 
			
		||||
        Ok(Err(e)) => {
 | 
			
		||||
            error!("❌ System Actor encountered an error: {}", e);
 | 
			
		||||
            std::process::exit(1);
 | 
			
		||||
        }
 | 
			
		||||
        Err(e) => {
 | 
			
		||||
            error!("❌ Failed to join actor task: {}", e);
 | 
			
		||||
            std::process::exit(1);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Setup logging based on configuration and command line arguments
 | 
			
		||||
fn setup_logging(
 | 
			
		||||
    config: &ActorConfig,
 | 
			
		||||
    verbose: bool,
 | 
			
		||||
    no_timestamp: bool,
 | 
			
		||||
) -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
 | 
			
		||||
    let mut builder = env_logger::Builder::new();
 | 
			
		||||
 | 
			
		||||
    // Determine log level
 | 
			
		||||
    let log_level = if verbose {
 | 
			
		||||
        "debug"
 | 
			
		||||
    } else {
 | 
			
		||||
        &config.logging.level
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    // Set log level
 | 
			
		||||
    builder.filter_level(match log_level.to_lowercase().as_str() {
 | 
			
		||||
        "trace" => log::LevelFilter::Trace,
 | 
			
		||||
        "debug" => log::LevelFilter::Debug,
 | 
			
		||||
        "info" => log::LevelFilter::Info,
 | 
			
		||||
        "warn" => log::LevelFilter::Warn,
 | 
			
		||||
        "error" => log::LevelFilter::Error,
 | 
			
		||||
        _ => {
 | 
			
		||||
            warn!("Invalid log level: {}. Using 'info'", log_level);
 | 
			
		||||
            log::LevelFilter::Info
 | 
			
		||||
        }
 | 
			
		||||
    });
 | 
			
		||||
 | 
			
		||||
    // Configure timestamps
 | 
			
		||||
    let show_timestamps = !no_timestamp && config.logging.timestamps;
 | 
			
		||||
    if !show_timestamps {
 | 
			
		||||
        builder.format_timestamp(None);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    builder.init();
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod tests {
 | 
			
		||||
    use super::*;
 | 
			
		||||
    use std::io::Write;
 | 
			
		||||
    use tempfile::NamedTempFile;
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn test_config_validation() {
 | 
			
		||||
        let config_toml = r#"
 | 
			
		||||
actor_id = "test_system"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/test_db"
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "async"
 | 
			
		||||
default_timeout_seconds = 600
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
level = "info"
 | 
			
		||||
"#;
 | 
			
		||||
 | 
			
		||||
        let mut temp_file = NamedTempFile::new().unwrap();
 | 
			
		||||
        temp_file.write_all(config_toml.as_bytes()).unwrap();
 | 
			
		||||
 | 
			
		||||
        let config = ActorConfig::from_file(temp_file.path()).unwrap();
 | 
			
		||||
        assert!(!config.is_sync());
 | 
			
		||||
        assert!(config.is_async());
 | 
			
		||||
        assert_eq!(config.actor_id, "test_system");
 | 
			
		||||
        assert_eq!(config.get_default_timeout(), Some(Duration::from_secs(600)));
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn test_sync_config_rejection() {
 | 
			
		||||
        let config_toml = r#"
 | 
			
		||||
actor_id = "test_system"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/test_db"
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "sync"
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
level = "info"
 | 
			
		||||
"#;
 | 
			
		||||
 | 
			
		||||
        let mut temp_file = NamedTempFile::new().unwrap();
 | 
			
		||||
        temp_file.write_all(config_toml.as_bytes()).unwrap();
 | 
			
		||||
 | 
			
		||||
        let config = ActorConfig::from_file(temp_file.path()).unwrap();
 | 
			
		||||
        assert!(config.is_sync());
 | 
			
		||||
        assert!(!config.is_async());
 | 
			
		||||
        // This would be rejected in main() function
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn test_timeout_override() {
 | 
			
		||||
        let config_toml = r#"
 | 
			
		||||
actor_id = "test_system"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/test_db"
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "async"
 | 
			
		||||
default_timeout_seconds = 300
 | 
			
		||||
"#;
 | 
			
		||||
 | 
			
		||||
        let mut temp_file = NamedTempFile::new().unwrap();
 | 
			
		||||
        temp_file.write_all(config_toml.as_bytes()).unwrap();
 | 
			
		||||
 | 
			
		||||
        let mut config = ActorConfig::from_file(temp_file.path()).unwrap();
 | 
			
		||||
        assert_eq!(config.get_default_timeout(), Some(Duration::from_secs(300)));
 | 
			
		||||
 | 
			
		||||
        // Test timeout override
 | 
			
		||||
        if let baobab_actor::config::ActorType::Async { ref mut default_timeout_seconds } = config.actor_type {
 | 
			
		||||
            *default_timeout_seconds = 600;
 | 
			
		||||
        }
 | 
			
		||||
        assert_eq!(config.get_default_timeout(), Some(Duration::from_secs(600)));
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										95
									
								
								_archive/core/actor/cmd/worker.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										95
									
								
								_archive/core/actor/cmd/worker.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,95 @@
 | 
			
		||||
use clap::Parser;
 | 
			
		||||
use baobab_actor::engine::create_heromodels_engine;
 | 
			
		||||
use baobab_actor::spawn_rhai_actor;
 | 
			
		||||
use tokio::sync::mpsc;
 | 
			
		||||
 | 
			
		||||
#[derive(Parser, Debug)]
 | 
			
		||||
#[command(author, version, about, long_about = None)]
 | 
			
		||||
struct Args {
 | 
			
		||||
    /// Actor ID for identification
 | 
			
		||||
    #[arg(short, long)]
 | 
			
		||||
    actor_id: String,
 | 
			
		||||
 | 
			
		||||
    /// Redis URL
 | 
			
		||||
    #[arg(short, long, default_value = "redis://localhost:6379")]
 | 
			
		||||
    redis_url: String,
 | 
			
		||||
 | 
			
		||||
    /// Preserve task details after completion (for benchmarking)
 | 
			
		||||
    #[arg(long, default_value = "false")]
 | 
			
		||||
    preserve_tasks: bool,
 | 
			
		||||
 | 
			
		||||
    /// Root directory for engine database
 | 
			
		||||
    #[arg(long, default_value = "actor_rhai_temp_db")]
 | 
			
		||||
    db_path: String,
 | 
			
		||||
 | 
			
		||||
    /// Disable timestamps in log output
 | 
			
		||||
    #[arg(long, help = "Remove timestamps from log output")]
 | 
			
		||||
    no_timestamp: bool,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[tokio::main]
 | 
			
		||||
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
 | 
			
		||||
    let args = Args::parse();
 | 
			
		||||
 | 
			
		||||
    // Configure env_logger with or without timestamps
 | 
			
		||||
    if args.no_timestamp {
 | 
			
		||||
        env_logger::Builder::from_default_env()
 | 
			
		||||
            .format_timestamp(None)
 | 
			
		||||
            .init();
 | 
			
		||||
    } else {
 | 
			
		||||
        env_logger::init();
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    log::info!("Rhai Actor (binary) starting with performance-optimized engine.");
 | 
			
		||||
    log::info!(
 | 
			
		||||
        "Actor ID: {}, Redis: {}",
 | 
			
		||||
        args.actor_id,
 | 
			
		||||
        args.redis_url
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    let mut engine = create_heromodels_engine();
 | 
			
		||||
 | 
			
		||||
    // Performance optimizations for benchmarking
 | 
			
		||||
    engine.set_max_operations(0); // Unlimited operations for performance testing
 | 
			
		||||
    engine.set_max_expr_depths(0, 0); // Unlimited expression depth
 | 
			
		||||
    engine.set_max_string_size(0); // Unlimited string size
 | 
			
		||||
    engine.set_max_array_size(0); // Unlimited array size
 | 
			
		||||
    engine.set_max_map_size(0); // Unlimited map size
 | 
			
		||||
 | 
			
		||||
    // Enable full optimization for maximum performance
 | 
			
		||||
    engine.set_optimization_level(rhai::OptimizationLevel::Full);
 | 
			
		||||
 | 
			
		||||
    log::info!("Engine configured for maximum performance");
 | 
			
		||||
 | 
			
		||||
    // Create shutdown channel (for graceful shutdown, though not used in benchmarks)
 | 
			
		||||
    let (_shutdown_tx, shutdown_rx) = mpsc::channel::<()>(1);
 | 
			
		||||
 | 
			
		||||
    // Spawn the actor
 | 
			
		||||
    let actor_handle = spawn_rhai_actor(
 | 
			
		||||
        args.actor_id,
 | 
			
		||||
        args.db_path,
 | 
			
		||||
        engine,
 | 
			
		||||
        args.redis_url,
 | 
			
		||||
        shutdown_rx,
 | 
			
		||||
        args.preserve_tasks,
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    // Wait for the actor to complete
 | 
			
		||||
    match actor_handle.await {
 | 
			
		||||
        Ok(result) => match result {
 | 
			
		||||
            Ok(_) => {
 | 
			
		||||
                log::info!("Actor completed successfully");
 | 
			
		||||
                Ok(())
 | 
			
		||||
            }
 | 
			
		||||
            Err(e) => {
 | 
			
		||||
                log::error!("Actor failed: {}", e);
 | 
			
		||||
                Err(e)
 | 
			
		||||
            }
 | 
			
		||||
        },
 | 
			
		||||
        Err(e) => {
 | 
			
		||||
            log::error!("Actor task panicked: {}", e);
 | 
			
		||||
            Err(Box::new(e) as Box<dyn std::error::Error + Send + Sync>)
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										197
									
								
								_archive/core/actor/examples/README.md
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										197
									
								
								_archive/core/actor/examples/README.md
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,197 @@
 | 
			
		||||
# Actor Examples
 | 
			
		||||
 | 
			
		||||
This directory contains example configurations and test scripts for both OSIS and System actor binaries.
 | 
			
		||||
 | 
			
		||||
## Overview
 | 
			
		||||
 | 
			
		||||
Both examples demonstrate the ping/pong functionality built into the Hero actors:
 | 
			
		||||
- Actors automatically detect jobs with script content "ping"
 | 
			
		||||
- They respond immediately with "pong" without executing the Rhai engine
 | 
			
		||||
- This provides a fast health check and connectivity test mechanism
 | 
			
		||||
 | 
			
		||||
## Prerequisites
 | 
			
		||||
 | 
			
		||||
1. **Redis Server**: Both examples require a running Redis server
 | 
			
		||||
   ```bash
 | 
			
		||||
   # Install Redis (macOS)
 | 
			
		||||
   brew install redis
 | 
			
		||||
   
 | 
			
		||||
   # Start Redis server
 | 
			
		||||
   redis-server
 | 
			
		||||
   ```
 | 
			
		||||
 | 
			
		||||
2. **Rust Environment**: Make sure you can build the actor binaries
 | 
			
		||||
   ```bash
 | 
			
		||||
   cd /path/to/herocode/hero/core/actor
 | 
			
		||||
   cargo build --bin osis --bin system
 | 
			
		||||
   ```
 | 
			
		||||
 | 
			
		||||
## OSIS Actor Example
 | 
			
		||||
 | 
			
		||||
**Location**: `examples/osis/`
 | 
			
		||||
 | 
			
		||||
The OSIS (Operating System Integration Service) actor processes jobs synchronously, one at a time.
 | 
			
		||||
 | 
			
		||||
### Files
 | 
			
		||||
- `config.toml` - Configuration for the OSIS actor
 | 
			
		||||
- `example.sh` - Test script that demonstrates ping/pong functionality
 | 
			
		||||
 | 
			
		||||
### Usage
 | 
			
		||||
```bash
 | 
			
		||||
cd examples/osis
 | 
			
		||||
./example.sh
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### What the script does:
 | 
			
		||||
1. Checks Redis connectivity
 | 
			
		||||
2. Cleans up any existing jobs
 | 
			
		||||
3. Starts the OSIS actor in the background
 | 
			
		||||
4. Sends 3 ping jobs sequentially
 | 
			
		||||
5. Verifies each job receives a "pong" response
 | 
			
		||||
6. Reports success/failure statistics
 | 
			
		||||
7. Cleans up actor and Redis data
 | 
			
		||||
 | 
			
		||||
### Expected Output
 | 
			
		||||
```
 | 
			
		||||
=== OSIS Actor Example ===
 | 
			
		||||
✅ Redis is running
 | 
			
		||||
✅ OSIS actor started (PID: 12345)
 | 
			
		||||
📤 Sending ping job: ping_job_1_1234567890
 | 
			
		||||
✅ Job ping_job_1_1234567890 completed successfully with result: pong
 | 
			
		||||
...
 | 
			
		||||
🎉 All tests passed! OSIS actor is working correctly.
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## System Actor Example
 | 
			
		||||
 | 
			
		||||
**Location**: `examples/system/`
 | 
			
		||||
 | 
			
		||||
The System actor processes jobs asynchronously, handling multiple jobs concurrently.
 | 
			
		||||
 | 
			
		||||
### Files
 | 
			
		||||
- `config.toml` - Configuration for the System actor (includes async settings)
 | 
			
		||||
- `example.sh` - Test script that demonstrates concurrent ping/pong functionality
 | 
			
		||||
 | 
			
		||||
### Usage
 | 
			
		||||
```bash
 | 
			
		||||
cd examples/system
 | 
			
		||||
./example.sh
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### What the script does:
 | 
			
		||||
1. Checks Redis connectivity
 | 
			
		||||
2. Cleans up any existing jobs
 | 
			
		||||
3. Starts the System actor with stats reporting
 | 
			
		||||
4. Sends 5 concurrent ping jobs
 | 
			
		||||
5. Sends 10 rapid-fire ping jobs to test async capabilities
 | 
			
		||||
6. Verifies all jobs receive "pong" responses
 | 
			
		||||
7. Reports comprehensive success/failure statistics
 | 
			
		||||
8. Cleans up actor and Redis data
 | 
			
		||||
 | 
			
		||||
### Expected Output
 | 
			
		||||
```
 | 
			
		||||
=== System Actor Example ===
 | 
			
		||||
✅ Redis is running
 | 
			
		||||
✅ System actor started (PID: 12345)
 | 
			
		||||
📤 Sending ping job: ping_job_1_1234567890123
 | 
			
		||||
✅ Job ping_job_1_1234567890123 completed successfully with result: pong
 | 
			
		||||
...
 | 
			
		||||
🎉 All tests passed! System actor is handling concurrent jobs correctly.
 | 
			
		||||
Overall success rate: 15/15
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## Configuration Details
 | 
			
		||||
 | 
			
		||||
### OSIS Configuration (`examples/osis/config.toml`)
 | 
			
		||||
```toml
 | 
			
		||||
actor_id = "osis_example_actor"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/osis_example_db"
 | 
			
		||||
preserve_tasks = false
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "sync"
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
timestamps = true
 | 
			
		||||
level = "info"
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### System Configuration (`examples/system/config.toml`)
 | 
			
		||||
```toml
 | 
			
		||||
actor_id = "system_example_actor"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/system_example_db"
 | 
			
		||||
preserve_tasks = false
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "async"
 | 
			
		||||
default_timeout_seconds = 30
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
timestamps = true
 | 
			
		||||
level = "info"
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
## Key Differences
 | 
			
		||||
 | 
			
		||||
| Feature | OSIS Actor | System Actor |
 | 
			
		||||
|---------|-------------|---------------|
 | 
			
		||||
| **Processing** | Sequential (one job at a time) | Concurrent (multiple jobs simultaneously) |
 | 
			
		||||
| **Use Case** | System-level operations requiring resource management | High-throughput job processing |
 | 
			
		||||
| **Timeout** | No timeout configuration | Configurable job timeouts |
 | 
			
		||||
| **Stats** | Basic logging | Optional statistics reporting (`--show-stats`) |
 | 
			
		||||
| **Job Handling** | Blocking job execution | Non-blocking async job execution |
 | 
			
		||||
 | 
			
		||||
## Troubleshooting
 | 
			
		||||
 | 
			
		||||
### Redis Connection Issues
 | 
			
		||||
```bash
 | 
			
		||||
# Check if Redis is running
 | 
			
		||||
redis-cli ping
 | 
			
		||||
 | 
			
		||||
# Check Redis logs
 | 
			
		||||
redis-server --loglevel verbose
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Actor Compilation Issues
 | 
			
		||||
```bash
 | 
			
		||||
# Clean and rebuild
 | 
			
		||||
cargo clean
 | 
			
		||||
cargo build --bin osis --bin system
 | 
			
		||||
```
 | 
			
		||||
 | 
			
		||||
### Job Processing Issues
 | 
			
		||||
- Check Redis for stuck jobs: `redis-cli keys "hero:*"`
 | 
			
		||||
- Clear all Hero jobs: `redis-cli eval "return redis.call('del', unpack(redis.call('keys', 'hero:*')))" 0`
 | 
			
		||||
- Check actor logs for detailed error messages
 | 
			
		||||
 | 
			
		||||
## Extending the Examples
 | 
			
		||||
 | 
			
		||||
### Adding Custom Jobs
 | 
			
		||||
To test with custom Rhai scripts instead of ping jobs:
 | 
			
		||||
 | 
			
		||||
1. Modify the job creation in the shell scripts:
 | 
			
		||||
   ```bash
 | 
			
		||||
   # Replace "ping" with your Rhai script
 | 
			
		||||
   redis-cli -u "$REDIS_URL" hset "hero:job:$job_id" \
 | 
			
		||||
       script "your_rhai_script_here"
 | 
			
		||||
   ```
 | 
			
		||||
 | 
			
		||||
2. Update result verification to expect your script's output instead of "pong"
 | 
			
		||||
 | 
			
		||||
### Testing Different Configurations
 | 
			
		||||
- Modify `config.toml` files to test different Redis URLs, database paths, or logging levels
 | 
			
		||||
- Test with `preserve_tasks = true` to inspect job details after completion
 | 
			
		||||
- Adjust timeout values in the System actor configuration
 | 
			
		||||
 | 
			
		||||
## Architecture Notes
 | 
			
		||||
 | 
			
		||||
Both examples demonstrate the unified Actor trait architecture:
 | 
			
		||||
- **Common Interface**: Both actors implement the same `Actor` trait
 | 
			
		||||
- **Ping/Pong Handling**: Built into the trait's `spawn` method before job delegation
 | 
			
		||||
- **Redis Integration**: Uses the shared Job struct from `hero_job` crate
 | 
			
		||||
- **Configuration**: TOML-based configuration with CLI overrides
 | 
			
		||||
- **Graceful Shutdown**: Both actors handle SIGTERM/SIGINT properly
 | 
			
		||||
 | 
			
		||||
This architecture allows for easy extension with new actor types while maintaining consistent behavior and configuration patterns.
 | 
			
		||||
							
								
								
									
										11
									
								
								_archive/core/actor/examples/osis/config.toml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								_archive/core/actor/examples/osis/config.toml
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,11 @@
 | 
			
		||||
actor_id = "osis_example_actor"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/osis_example_db"
 | 
			
		||||
preserve_tasks = false
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "sync"
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
timestamps = true
 | 
			
		||||
level = "info"
 | 
			
		||||
							
								
								
									
										138
									
								
								_archive/core/actor/examples/osis/example.sh
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										138
									
								
								_archive/core/actor/examples/osis/example.sh
									
									
									
									
									
										Executable file
									
								
							@@ -0,0 +1,138 @@
 | 
			
		||||
#!/bin/bash
 | 
			
		||||
 | 
			
		||||
# OSIS Actor Example Script
 | 
			
		||||
# This script demonstrates the OSIS actor by:
 | 
			
		||||
# 1. Starting the actor with the config.toml
 | 
			
		||||
# 2. Sending ping jobs to Redis
 | 
			
		||||
# 3. Verifying pong responses
 | 
			
		||||
 | 
			
		||||
set -e
 | 
			
		||||
 | 
			
		||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 | 
			
		||||
CONFIG_FILE="$SCRIPT_DIR/config.toml"
 | 
			
		||||
ACTOR_ID="osis_example_actor"
 | 
			
		||||
REDIS_URL="redis://localhost:6379"
 | 
			
		||||
 | 
			
		||||
echo "=== OSIS Actor Example ==="
 | 
			
		||||
echo "Script directory: $SCRIPT_DIR"
 | 
			
		||||
echo "Config file: $CONFIG_FILE"
 | 
			
		||||
echo "Actor ID: $ACTOR_ID"
 | 
			
		||||
echo "Redis URL: $REDIS_URL"
 | 
			
		||||
echo
 | 
			
		||||
 | 
			
		||||
# Check if Redis is running
 | 
			
		||||
echo "Checking Redis connection..."
 | 
			
		||||
if ! redis-cli -u "$REDIS_URL" ping > /dev/null 2>&1; then
 | 
			
		||||
    echo "❌ Error: Redis is not running or not accessible at $REDIS_URL"
 | 
			
		||||
    echo "Please start Redis server first: redis-server"
 | 
			
		||||
    exit 1
 | 
			
		||||
fi
 | 
			
		||||
echo "✅ Redis is running"
 | 
			
		||||
echo
 | 
			
		||||
 | 
			
		||||
# Clean up any existing jobs in the queue
 | 
			
		||||
echo "Cleaning up existing jobs in Redis..."
 | 
			
		||||
redis-cli -u "$REDIS_URL" del "hero:jobs:$ACTOR_ID" > /dev/null 2>&1 || true
 | 
			
		||||
redis-cli -u "$REDIS_URL" eval "return redis.call('del', unpack(redis.call('keys', 'hero:job:*')))" 0 > /dev/null 2>&1 || true
 | 
			
		||||
echo "✅ Redis queues cleaned"
 | 
			
		||||
echo
 | 
			
		||||
 | 
			
		||||
# Start the OSIS actor in the background
 | 
			
		||||
echo "Starting OSIS actor..."
 | 
			
		||||
cd "$SCRIPT_DIR/../.."
 | 
			
		||||
cargo run --bin osis -- --config "$CONFIG_FILE" &
 | 
			
		||||
ACTOR_PID=$!
 | 
			
		||||
echo "✅ OSIS actor started (PID: $ACTOR_PID)"
 | 
			
		||||
echo
 | 
			
		||||
 | 
			
		||||
# Wait a moment for the actor to initialize
 | 
			
		||||
echo "Waiting for actor to initialize..."
 | 
			
		||||
sleep 3
 | 
			
		||||
 | 
			
		||||
# Function to send a ping job and check for pong response
 | 
			
		||||
send_ping_job() {
 | 
			
		||||
    local job_num=$1
 | 
			
		||||
    local job_id="ping_job_${job_num}_$(date +%s)"
 | 
			
		||||
    
 | 
			
		||||
    echo "📤 Sending ping job: $job_id"
 | 
			
		||||
    
 | 
			
		||||
    # Create job in Redis
 | 
			
		||||
    redis-cli -u "$REDIS_URL" hset "hero:job:$job_id" \
 | 
			
		||||
        id "$job_id" \
 | 
			
		||||
        script "ping" \
 | 
			
		||||
        status "Queued" \
 | 
			
		||||
        created_at "$(date -u +%Y-%m-%dT%H:%M:%SZ)" \
 | 
			
		||||
        actor_id "$ACTOR_ID" > /dev/null
 | 
			
		||||
    
 | 
			
		||||
    # Add job to actor queue
 | 
			
		||||
    redis-cli -u "$REDIS_URL" lpush "hero:jobs:$ACTOR_ID" "$job_id" > /dev/null
 | 
			
		||||
    
 | 
			
		||||
    # Wait for job completion and check result
 | 
			
		||||
    local timeout=10
 | 
			
		||||
    local elapsed=0
 | 
			
		||||
    while [ $elapsed -lt $timeout ]; do
 | 
			
		||||
        local status=$(redis-cli -u "$REDIS_URL" hget "hero:job:$job_id" status 2>/dev/null || echo "")
 | 
			
		||||
        if [ "$status" = "Finished" ]; then
 | 
			
		||||
            local result=$(redis-cli -u "$REDIS_URL" hget "hero:job:$job_id" result 2>/dev/null || echo "")
 | 
			
		||||
            if [ "$result" = "pong" ]; then
 | 
			
		||||
                echo "✅ Job $job_id completed successfully with result: $result"
 | 
			
		||||
                return 0
 | 
			
		||||
            else
 | 
			
		||||
                echo "❌ Job $job_id completed but with unexpected result: $result"
 | 
			
		||||
                return 1
 | 
			
		||||
            fi
 | 
			
		||||
        elif [ "$status" = "Error" ]; then
 | 
			
		||||
            local error=$(redis-cli -u "$REDIS_URL" hget "hero:job:$job_id" error 2>/dev/null || echo "")
 | 
			
		||||
            echo "❌ Job $job_id failed with error: $error"
 | 
			
		||||
            return 1
 | 
			
		||||
        fi
 | 
			
		||||
        sleep 1
 | 
			
		||||
        elapsed=$((elapsed + 1))
 | 
			
		||||
    done
 | 
			
		||||
    
 | 
			
		||||
    echo "❌ Job $job_id timed out after ${timeout}s"
 | 
			
		||||
    return 1
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
# Send multiple ping jobs to test the actor
 | 
			
		||||
echo "Testing ping/pong functionality..."
 | 
			
		||||
success_count=0
 | 
			
		||||
total_jobs=3
 | 
			
		||||
 | 
			
		||||
for i in $(seq 1 $total_jobs); do
 | 
			
		||||
    echo
 | 
			
		||||
    echo "--- Test $i/$total_jobs ---"
 | 
			
		||||
    if send_ping_job $i; then
 | 
			
		||||
        success_count=$((success_count + 1))
 | 
			
		||||
    fi
 | 
			
		||||
    sleep 1
 | 
			
		||||
done
 | 
			
		||||
 | 
			
		||||
echo
 | 
			
		||||
echo "=== Test Results ==="
 | 
			
		||||
echo "Successful ping/pong tests: $success_count/$total_jobs"
 | 
			
		||||
 | 
			
		||||
if [ $success_count -eq $total_jobs ]; then
 | 
			
		||||
    echo "🎉 All tests passed! OSIS actor is working correctly."
 | 
			
		||||
    exit_code=0
 | 
			
		||||
else
 | 
			
		||||
    echo "⚠️  Some tests failed. Check the actor logs for details."
 | 
			
		||||
    exit_code=1
 | 
			
		||||
fi
 | 
			
		||||
 | 
			
		||||
# Clean up
 | 
			
		||||
echo
 | 
			
		||||
echo "Cleaning up..."
 | 
			
		||||
echo "Stopping OSIS actor (PID: $ACTOR_PID)..."
 | 
			
		||||
kill $ACTOR_PID 2>/dev/null || true
 | 
			
		||||
wait $ACTOR_PID 2>/dev/null || true
 | 
			
		||||
echo "✅ Actor stopped"
 | 
			
		||||
 | 
			
		||||
echo "Cleaning up Redis jobs..."
 | 
			
		||||
redis-cli -u "$REDIS_URL" del "hero:jobs:$ACTOR_ID" > /dev/null 2>&1 || true
 | 
			
		||||
redis-cli -u "$REDIS_URL" eval "return redis.call('del', unpack(redis.call('keys', 'hero:job:*')))" 0 > /dev/null 2>&1 || true
 | 
			
		||||
echo "✅ Redis cleaned up"
 | 
			
		||||
 | 
			
		||||
echo
 | 
			
		||||
echo "=== OSIS Actor Example Complete ==="
 | 
			
		||||
exit $exit_code
 | 
			
		||||
							
								
								
									
										14
									
								
								_archive/core/actor/examples/osis_config.toml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										14
									
								
								_archive/core/actor/examples/osis_config.toml
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,14 @@
 | 
			
		||||
# OSIS Actor Configuration
 | 
			
		||||
# Synchronous actor for system-level operations
 | 
			
		||||
 | 
			
		||||
actor_id = "osis_actor_1"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/osis_actor_db"
 | 
			
		||||
preserve_tasks = false
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "sync"
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
timestamps = true
 | 
			
		||||
level = "info"
 | 
			
		||||
							
								
								
									
										60
									
								
								_archive/core/actor/examples/osis_worker_demo.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										60
									
								
								_archive/core/actor/examples/osis_worker_demo.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,60 @@
 | 
			
		||||
use std::process::{Command, Stdio};
 | 
			
		||||
use std::path::Path;
 | 
			
		||||
use std::env;
 | 
			
		||||
use std::io::{self, Write};
 | 
			
		||||
 | 
			
		||||
/// OSIS Actor Demo Runner
 | 
			
		||||
/// 
 | 
			
		||||
/// This Rust wrapper executes the OSIS actor bash script example.
 | 
			
		||||
/// It provides a way to run shell-based examples through Cargo.
 | 
			
		||||
fn main() -> Result<(), Box<dyn std::error::Error>> {
 | 
			
		||||
    println!("🚀 OSIS Actor Demo");
 | 
			
		||||
    println!("==================");
 | 
			
		||||
    println!();
 | 
			
		||||
 | 
			
		||||
    // Get the current working directory and construct the path to the shell script
 | 
			
		||||
    let current_dir = env::current_dir()?;
 | 
			
		||||
    let script_path = current_dir.join("examples").join("osis").join("example.sh");
 | 
			
		||||
    
 | 
			
		||||
    // Check if the script exists
 | 
			
		||||
    if !script_path.exists() {
 | 
			
		||||
        eprintln!("❌ Error: Script not found at {:?}", script_path);
 | 
			
		||||
        eprintln!("   Make sure you're running this from the actor crate root directory.");
 | 
			
		||||
        std::process::exit(1);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    println!("📁 Script location: {:?}", script_path);
 | 
			
		||||
    println!("🔧 Executing OSIS actor example...");
 | 
			
		||||
    println!();
 | 
			
		||||
 | 
			
		||||
    // Make sure the script is executable
 | 
			
		||||
    #[cfg(unix)]
 | 
			
		||||
    {
 | 
			
		||||
        use std::os::unix::fs::PermissionsExt;
 | 
			
		||||
        let mut perms = std::fs::metadata(&script_path)?.permissions();
 | 
			
		||||
        perms.set_mode(0o755);
 | 
			
		||||
        std::fs::set_permissions(&script_path, perms)?;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Execute the shell script
 | 
			
		||||
    let mut child = Command::new("bash")
 | 
			
		||||
        .arg(&script_path)
 | 
			
		||||
        .current_dir(¤t_dir)
 | 
			
		||||
        .stdin(Stdio::inherit())
 | 
			
		||||
        .stdout(Stdio::inherit())
 | 
			
		||||
        .stderr(Stdio::inherit())
 | 
			
		||||
        .spawn()?;
 | 
			
		||||
 | 
			
		||||
    // Wait for the script to complete
 | 
			
		||||
    let status = child.wait()?;
 | 
			
		||||
 | 
			
		||||
    println!();
 | 
			
		||||
    if status.success() {
 | 
			
		||||
        println!("✅ OSIS actor demo completed successfully!");
 | 
			
		||||
    } else {
 | 
			
		||||
        println!("❌ OSIS actor demo failed with exit code: {:?}", status.code());
 | 
			
		||||
        std::process::exit(status.code().unwrap_or(1));
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										12
									
								
								_archive/core/actor/examples/system/config.toml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								_archive/core/actor/examples/system/config.toml
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,12 @@
 | 
			
		||||
actor_id = "system_example_actor"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/system_example_db"
 | 
			
		||||
preserve_tasks = false
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "async"
 | 
			
		||||
default_timeout_seconds = 30
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
timestamps = true
 | 
			
		||||
level = "info"
 | 
			
		||||
							
								
								
									
										183
									
								
								_archive/core/actor/examples/system/example.sh
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										183
									
								
								_archive/core/actor/examples/system/example.sh
									
									
									
									
									
										Executable file
									
								
							@@ -0,0 +1,183 @@
 | 
			
		||||
#!/bin/bash
 | 
			
		||||
 | 
			
		||||
# System Actor Example Script
 | 
			
		||||
# This script demonstrates the System actor by:
 | 
			
		||||
# 1. Starting the actor with the config.toml
 | 
			
		||||
# 2. Sending multiple concurrent ping jobs to Redis
 | 
			
		||||
# 3. Verifying pong responses
 | 
			
		||||
 | 
			
		||||
set -e
 | 
			
		||||
 | 
			
		||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 | 
			
		||||
CONFIG_FILE="$SCRIPT_DIR/config.toml"
 | 
			
		||||
ACTOR_ID="system_example_actor"
 | 
			
		||||
REDIS_URL="redis://localhost:6379"
 | 
			
		||||
 | 
			
		||||
echo "=== System Actor Example ==="
 | 
			
		||||
echo "Script directory: $SCRIPT_DIR"
 | 
			
		||||
echo "Config file: $CONFIG_FILE"
 | 
			
		||||
echo "Actor ID: $ACTOR_ID"
 | 
			
		||||
echo "Redis URL: $REDIS_URL"
 | 
			
		||||
echo
 | 
			
		||||
 | 
			
		||||
# Check if Redis is running
 | 
			
		||||
echo "Checking Redis connection..."
 | 
			
		||||
if ! redis-cli -u "$REDIS_URL" ping > /dev/null 2>&1; then
 | 
			
		||||
    echo "❌ Error: Redis is not running or not accessible at $REDIS_URL"
 | 
			
		||||
    echo "Please start Redis server first: redis-server"
 | 
			
		||||
    exit 1
 | 
			
		||||
fi
 | 
			
		||||
echo "✅ Redis is running"
 | 
			
		||||
echo
 | 
			
		||||
 | 
			
		||||
# Clean up any existing jobs in the queue
 | 
			
		||||
echo "Cleaning up existing jobs in Redis..."
 | 
			
		||||
redis-cli -u "$REDIS_URL" del "hero:jobs:$ACTOR_ID" > /dev/null 2>&1 || true
 | 
			
		||||
redis-cli -u "$REDIS_URL" eval "return redis.call('del', unpack(redis.call('keys', 'hero:job:*')))" 0 > /dev/null 2>&1 || true
 | 
			
		||||
echo "✅ Redis queues cleaned"
 | 
			
		||||
echo
 | 
			
		||||
 | 
			
		||||
# Start the System actor in the background
 | 
			
		||||
echo "Starting System actor..."
 | 
			
		||||
cd "$SCRIPT_DIR/../.."
 | 
			
		||||
cargo run --bin system -- --config "$CONFIG_FILE" --show-stats &
 | 
			
		||||
ACTOR_PID=$!
 | 
			
		||||
echo "✅ System actor started (PID: $ACTOR_PID)"
 | 
			
		||||
echo
 | 
			
		||||
 | 
			
		||||
# Wait a moment for the actor to initialize
 | 
			
		||||
echo "Waiting for actor to initialize..."
 | 
			
		||||
sleep 3
 | 
			
		||||
 | 
			
		||||
# Function to send a ping job (non-blocking)
 | 
			
		||||
send_ping_job() {
 | 
			
		||||
    local job_num=$1
 | 
			
		||||
    local job_id="ping_job_${job_num}_$(date +%s%N)"
 | 
			
		||||
    
 | 
			
		||||
    echo "📤 Sending ping job: $job_id"
 | 
			
		||||
    
 | 
			
		||||
    # Create job in Redis
 | 
			
		||||
    redis-cli -u "$REDIS_URL" hset "hero:job:$job_id" \
 | 
			
		||||
        id "$job_id" \
 | 
			
		||||
        script "ping" \
 | 
			
		||||
        status "Queued" \
 | 
			
		||||
        created_at "$(date -u +%Y-%m-%dT%H:%M:%SZ)" \
 | 
			
		||||
        actor_id "$ACTOR_ID" > /dev/null
 | 
			
		||||
    
 | 
			
		||||
    # Add job to actor queue
 | 
			
		||||
    redis-cli -u "$REDIS_URL" lpush "hero:jobs:$ACTOR_ID" "$job_id" > /dev/null
 | 
			
		||||
    
 | 
			
		||||
    echo "$job_id"
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
# Function to check job result
 | 
			
		||||
check_job_result() {
 | 
			
		||||
    local job_id=$1
 | 
			
		||||
    local timeout=15
 | 
			
		||||
    local elapsed=0
 | 
			
		||||
    
 | 
			
		||||
    while [ $elapsed -lt $timeout ]; do
 | 
			
		||||
        local status=$(redis-cli -u "$REDIS_URL" hget "hero:job:$job_id" status 2>/dev/null || echo "")
 | 
			
		||||
        if [ "$status" = "Finished" ]; then
 | 
			
		||||
            local result=$(redis-cli -u "$REDIS_URL" hget "hero:job:$job_id" result 2>/dev/null || echo "")
 | 
			
		||||
            if [ "$result" = "pong" ]; then
 | 
			
		||||
                echo "✅ Job $job_id completed successfully with result: $result"
 | 
			
		||||
                return 0
 | 
			
		||||
            else
 | 
			
		||||
                echo "❌ Job $job_id completed but with unexpected result: $result"
 | 
			
		||||
                return 1
 | 
			
		||||
            fi
 | 
			
		||||
        elif [ "$status" = "Error" ]; then
 | 
			
		||||
            local error=$(redis-cli -u "$REDIS_URL" hget "hero:job:$job_id" error 2>/dev/null || echo "")
 | 
			
		||||
            echo "❌ Job $job_id failed with error: $error"
 | 
			
		||||
            return 1
 | 
			
		||||
        fi
 | 
			
		||||
        sleep 0.5
 | 
			
		||||
        elapsed=$((elapsed + 1))
 | 
			
		||||
    done
 | 
			
		||||
    
 | 
			
		||||
    echo "❌ Job $job_id timed out after ${timeout}s"
 | 
			
		||||
    return 1
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
# Send multiple concurrent ping jobs to test async processing
 | 
			
		||||
echo "Testing concurrent ping/pong functionality..."
 | 
			
		||||
total_jobs=5
 | 
			
		||||
job_ids=()
 | 
			
		||||
 | 
			
		||||
echo
 | 
			
		||||
echo "--- Sending $total_jobs concurrent ping jobs ---"
 | 
			
		||||
for i in $(seq 1 $total_jobs); do
 | 
			
		||||
    job_id=$(send_ping_job $i)
 | 
			
		||||
    job_ids+=("$job_id")
 | 
			
		||||
    sleep 0.1  # Small delay between job submissions
 | 
			
		||||
done
 | 
			
		||||
 | 
			
		||||
echo
 | 
			
		||||
echo "--- Waiting for all jobs to complete ---"
 | 
			
		||||
success_count=0
 | 
			
		||||
 | 
			
		||||
for job_id in "${job_ids[@]}"; do
 | 
			
		||||
    echo "Checking job: $job_id"
 | 
			
		||||
    if check_job_result "$job_id"; then
 | 
			
		||||
        success_count=$((success_count + 1))
 | 
			
		||||
    fi
 | 
			
		||||
done
 | 
			
		||||
 | 
			
		||||
echo
 | 
			
		||||
echo "=== Test Results ==="
 | 
			
		||||
echo "Successful concurrent ping/pong tests: $success_count/$total_jobs"
 | 
			
		||||
 | 
			
		||||
if [ $success_count -eq $total_jobs ]; then
 | 
			
		||||
    echo "🎉 All tests passed! System actor is handling concurrent jobs correctly."
 | 
			
		||||
    exit_code=0
 | 
			
		||||
else
 | 
			
		||||
    echo "⚠️  Some tests failed. Check the actor logs for details."
 | 
			
		||||
    exit_code=1
 | 
			
		||||
fi
 | 
			
		||||
 | 
			
		||||
# Test rapid job submission to showcase async capabilities
 | 
			
		||||
echo
 | 
			
		||||
echo "--- Testing rapid job submission (10 jobs in quick succession) ---"
 | 
			
		||||
rapid_jobs=10
 | 
			
		||||
rapid_job_ids=()
 | 
			
		||||
 | 
			
		||||
for i in $(seq 1 $rapid_jobs); do
 | 
			
		||||
    job_id=$(send_ping_job "rapid_$i")
 | 
			
		||||
    rapid_job_ids+=("$job_id")
 | 
			
		||||
done
 | 
			
		||||
 | 
			
		||||
echo "Waiting for rapid jobs to complete..."
 | 
			
		||||
rapid_success=0
 | 
			
		||||
for job_id in "${rapid_job_ids[@]}"; do
 | 
			
		||||
    if check_job_result "$job_id"; then
 | 
			
		||||
        rapid_success=$((rapid_success + 1))
 | 
			
		||||
    fi
 | 
			
		||||
done
 | 
			
		||||
 | 
			
		||||
echo "Rapid submission test: $rapid_success/$rapid_jobs successful"
 | 
			
		||||
 | 
			
		||||
# Clean up
 | 
			
		||||
echo
 | 
			
		||||
echo "Cleaning up..."
 | 
			
		||||
echo "Stopping System actor (PID: $ACTOR_PID)..."
 | 
			
		||||
kill $ACTOR_PID 2>/dev/null || true
 | 
			
		||||
wait $ACTOR_PID 2>/dev/null || true
 | 
			
		||||
echo "✅ Actor stopped"
 | 
			
		||||
 | 
			
		||||
echo "Cleaning up Redis jobs..."
 | 
			
		||||
redis-cli -u "$REDIS_URL" del "hero:jobs:$ACTOR_ID" > /dev/null 2>&1 || true
 | 
			
		||||
redis-cli -u "$REDIS_URL" eval "return redis.call('del', unpack(redis.call('keys', 'hero:job:*')))" 0 > /dev/null 2>&1 || true
 | 
			
		||||
echo "✅ Redis cleaned up"
 | 
			
		||||
 | 
			
		||||
echo
 | 
			
		||||
echo "=== System Actor Example Complete ==="
 | 
			
		||||
total_success=$((success_count + rapid_success))
 | 
			
		||||
total_tests=$((total_jobs + rapid_jobs))
 | 
			
		||||
echo "Overall success rate: $total_success/$total_tests"
 | 
			
		||||
 | 
			
		||||
if [ $total_success -eq $total_tests ]; then
 | 
			
		||||
    exit 0
 | 
			
		||||
else
 | 
			
		||||
    exit 1
 | 
			
		||||
fi
 | 
			
		||||
							
								
								
									
										15
									
								
								_archive/core/actor/examples/system_config.toml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								_archive/core/actor/examples/system_config.toml
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,15 @@
 | 
			
		||||
# System Actor Configuration
 | 
			
		||||
# Asynchronous actor for high-throughput concurrent processing
 | 
			
		||||
 | 
			
		||||
actor_id = "system_actor_1"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/system_actor_db"
 | 
			
		||||
preserve_tasks = false
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "async"
 | 
			
		||||
default_timeout_seconds = 300  # 5 minutes
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
timestamps = true
 | 
			
		||||
level = "info"
 | 
			
		||||
							
								
								
									
										60
									
								
								_archive/core/actor/examples/system_worker_demo.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										60
									
								
								_archive/core/actor/examples/system_worker_demo.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,60 @@
 | 
			
		||||
use std::process::{Command, Stdio};
 | 
			
		||||
use std::path::Path;
 | 
			
		||||
use std::env;
 | 
			
		||||
use std::io::{self, Write};
 | 
			
		||||
 | 
			
		||||
/// System Actor Demo Runner
 | 
			
		||||
/// 
 | 
			
		||||
/// This Rust wrapper executes the System actor bash script example.
 | 
			
		||||
/// It provides a way to run shell-based examples through Cargo.
 | 
			
		||||
fn main() -> Result<(), Box<dyn std::error::Error>> {
 | 
			
		||||
    println!("🚀 System Actor Demo");
 | 
			
		||||
    println!("====================");
 | 
			
		||||
    println!();
 | 
			
		||||
 | 
			
		||||
    // Get the current working directory and construct the path to the shell script
 | 
			
		||||
    let current_dir = env::current_dir()?;
 | 
			
		||||
    let script_path = current_dir.join("examples").join("system").join("example.sh");
 | 
			
		||||
    
 | 
			
		||||
    // Check if the script exists
 | 
			
		||||
    if !script_path.exists() {
 | 
			
		||||
        eprintln!("❌ Error: Script not found at {:?}", script_path);
 | 
			
		||||
        eprintln!("   Make sure you're running this from the actor crate root directory.");
 | 
			
		||||
        std::process::exit(1);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    println!("📁 Script location: {:?}", script_path);
 | 
			
		||||
    println!("🔧 Executing System actor example...");
 | 
			
		||||
    println!();
 | 
			
		||||
 | 
			
		||||
    // Make sure the script is executable
 | 
			
		||||
    #[cfg(unix)]
 | 
			
		||||
    {
 | 
			
		||||
        use std::os::unix::fs::PermissionsExt;
 | 
			
		||||
        let mut perms = std::fs::metadata(&script_path)?.permissions();
 | 
			
		||||
        perms.set_mode(0o755);
 | 
			
		||||
        std::fs::set_permissions(&script_path, perms)?;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Execute the shell script
 | 
			
		||||
    let mut child = Command::new("bash")
 | 
			
		||||
        .arg(&script_path)
 | 
			
		||||
        .current_dir(¤t_dir)
 | 
			
		||||
        .stdin(Stdio::inherit())
 | 
			
		||||
        .stdout(Stdio::inherit())
 | 
			
		||||
        .stderr(Stdio::inherit())
 | 
			
		||||
        .spawn()?;
 | 
			
		||||
 | 
			
		||||
    // Wait for the script to complete
 | 
			
		||||
    let status = child.wait()?;
 | 
			
		||||
 | 
			
		||||
    println!();
 | 
			
		||||
    if status.success() {
 | 
			
		||||
        println!("✅ System actor demo completed successfully!");
 | 
			
		||||
    } else {
 | 
			
		||||
        println!("❌ System actor demo failed with exit code: {:?}", status.code());
 | 
			
		||||
        std::process::exit(status.code().unwrap_or(1));
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										322
									
								
								_archive/core/actor/examples/trait_based_worker_demo.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										322
									
								
								_archive/core/actor/examples/trait_based_worker_demo.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,322 @@
 | 
			
		||||
//! # Trait-Based Actor Demo
 | 
			
		||||
//!
 | 
			
		||||
//! This example demonstrates the new unified actor interface using the Actor trait.
 | 
			
		||||
//! It shows how both synchronous and asynchronous actors can be used with the same
 | 
			
		||||
//! API, eliminating code duplication and providing a clean, consistent interface.
 | 
			
		||||
//!
 | 
			
		||||
//! ## Features Demonstrated
 | 
			
		||||
//!
 | 
			
		||||
//! - Unified actor interface using the Actor trait
 | 
			
		||||
//! - Both sync and async actor implementations
 | 
			
		||||
//! - Shared configuration and spawn logic
 | 
			
		||||
//! - Clean shutdown handling
 | 
			
		||||
//! - Job processing with different strategies
 | 
			
		||||
//!
 | 
			
		||||
//! ## Usage
 | 
			
		||||
//!
 | 
			
		||||
//! Make sure Redis is running on localhost:6379, then run:
 | 
			
		||||
//! ```bash
 | 
			
		||||
//! cargo run --example trait_based_actor_demo
 | 
			
		||||
//! ```
 | 
			
		||||
 | 
			
		||||
use hero_job::{Job, JobStatus, ScriptType};
 | 
			
		||||
use log::{info, warn, error};
 | 
			
		||||
use baobab_actor::{
 | 
			
		||||
    SyncActor, AsyncActor,
 | 
			
		||||
    spawn_sync_actor, spawn_async_actor,
 | 
			
		||||
    engine::create_heromodels_engine,
 | 
			
		||||
    actor_trait::{spawn_actor, Actor}
 | 
			
		||||
};
 | 
			
		||||
use redis::AsyncCommands;
 | 
			
		||||
use std::sync::Arc;
 | 
			
		||||
use std::time::Duration;
 | 
			
		||||
use tokio::sync::mpsc;
 | 
			
		||||
use tokio::time::sleep;
 | 
			
		||||
 | 
			
		||||
const REDIS_URL: &str = "redis://127.0.0.1:6379";
 | 
			
		||||
 | 
			
		||||
#[tokio::main]
 | 
			
		||||
async fn main() -> Result<(), Box<dyn std::error::Error>> {
 | 
			
		||||
    // Initialize logging
 | 
			
		||||
    env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("info")).init();
 | 
			
		||||
 | 
			
		||||
    info!("Starting Trait-Based Actor Demo");
 | 
			
		||||
 | 
			
		||||
    // Create Redis connection for job creation
 | 
			
		||||
    let redis_client = redis::Client::open(REDIS_URL)?;
 | 
			
		||||
    let mut redis_conn = redis_client.get_multiplexed_async_connection().await?;
 | 
			
		||||
 | 
			
		||||
    // Demo 1: Using the unified trait-based interface
 | 
			
		||||
    info!("=== Demo 1: Unified Trait-Based Interface ===");
 | 
			
		||||
    
 | 
			
		||||
    // Create shutdown channels for both actors
 | 
			
		||||
    let (sync_shutdown_tx, sync_shutdown_rx) = mpsc::channel::<()>(1);
 | 
			
		||||
    let (async_shutdown_tx, async_shutdown_rx) = mpsc::channel::<()>(1);
 | 
			
		||||
 | 
			
		||||
    // Actors are now configured using builder pattern directly
 | 
			
		||||
 | 
			
		||||
    // Create actor instances using builder pattern
 | 
			
		||||
    let sync_actor = Arc::new(
 | 
			
		||||
        SyncActor::builder()
 | 
			
		||||
            .actor_id("demo_sync_actor")
 | 
			
		||||
            .db_path("/tmp")
 | 
			
		||||
            .redis_url("redis://localhost:6379")
 | 
			
		||||
            .preserve_tasks(false)
 | 
			
		||||
            .build()
 | 
			
		||||
            .expect("Failed to build SyncActor")
 | 
			
		||||
    );
 | 
			
		||||
    
 | 
			
		||||
    let async_actor = Arc::new(
 | 
			
		||||
        AsyncActor::builder()
 | 
			
		||||
            .actor_id("demo_async_actor")
 | 
			
		||||
            .db_path("/tmp")
 | 
			
		||||
            .redis_url("redis://localhost:6379")
 | 
			
		||||
            .default_timeout(Duration::from_secs(300))
 | 
			
		||||
            .build()
 | 
			
		||||
            .expect("Failed to build AsyncActor")
 | 
			
		||||
    );
 | 
			
		||||
    
 | 
			
		||||
    let sync_engine = create_heromodels_engine();
 | 
			
		||||
    let async_engine = create_heromodels_engine();
 | 
			
		||||
 | 
			
		||||
    info!("Spawning {} actor: {}", sync_actor.actor_type(), sync_actor.actor_id());
 | 
			
		||||
    let sync_handle = spawn_actor(sync_actor.clone(), sync_engine, sync_shutdown_rx);
 | 
			
		||||
 | 
			
		||||
    info!("Spawning {} actor: {}", async_actor.actor_type(), async_actor.actor_id());
 | 
			
		||||
    let async_handle = spawn_actor(async_actor.clone(), async_engine, async_shutdown_rx);
 | 
			
		||||
 | 
			
		||||
    // Give actors time to start
 | 
			
		||||
    sleep(Duration::from_secs(1)).await;
 | 
			
		||||
 | 
			
		||||
    // Create and dispatch jobs to both actors
 | 
			
		||||
    info!("Creating demo jobs for both actors...");
 | 
			
		||||
 | 
			
		||||
    // Job for sync actor - simple calculation
 | 
			
		||||
    let sync_job = create_demo_job(
 | 
			
		||||
        "sync_calculation",
 | 
			
		||||
        r#"
 | 
			
		||||
        print("Sync actor: Starting calculation...");
 | 
			
		||||
        let result = 0;
 | 
			
		||||
        for i in 1..=100 {
 | 
			
		||||
            result += i;
 | 
			
		||||
        }
 | 
			
		||||
        print("Sync actor: Sum of 1-100 = " + result);
 | 
			
		||||
        result
 | 
			
		||||
        "#,
 | 
			
		||||
        None,
 | 
			
		||||
    ).await?;
 | 
			
		||||
 | 
			
		||||
    dispatch_job(&mut redis_conn, &sync_job, sync_actor.actor_id()).await?;
 | 
			
		||||
    info!("Dispatched job to sync actor: {}", sync_job.id);
 | 
			
		||||
 | 
			
		||||
    // Job for async actor - with timeout demonstration
 | 
			
		||||
    let async_job = create_demo_job(
 | 
			
		||||
        "async_calculation",
 | 
			
		||||
        r#"
 | 
			
		||||
        print("Async actor: Starting calculation...");
 | 
			
		||||
        let result = 1;
 | 
			
		||||
        for i in 1..=10 {
 | 
			
		||||
            result *= i;
 | 
			
		||||
        }
 | 
			
		||||
        print("Async actor: 10! = " + result);
 | 
			
		||||
        result
 | 
			
		||||
        "#,
 | 
			
		||||
        Some(15), // 15 second timeout
 | 
			
		||||
    ).await?;
 | 
			
		||||
 | 
			
		||||
    dispatch_job(&mut redis_conn, &async_job, async_actor.actor_id()).await?;
 | 
			
		||||
    info!("Dispatched job to async actor: {}", async_job.id);
 | 
			
		||||
 | 
			
		||||
    // Monitor job execution
 | 
			
		||||
    info!("Monitoring job execution for 10 seconds...");
 | 
			
		||||
    let monitor_start = std::time::Instant::now();
 | 
			
		||||
    let monitor_duration = Duration::from_secs(10);
 | 
			
		||||
 | 
			
		||||
    while monitor_start.elapsed() < monitor_duration {
 | 
			
		||||
        // Check sync job status
 | 
			
		||||
        if let Ok(status) = Job::get_status(&mut redis_conn, &sync_job.id).await {
 | 
			
		||||
            match status {
 | 
			
		||||
                JobStatus::Finished => {
 | 
			
		||||
                    let job_key = format!("hero:job:{}", sync_job.id);
 | 
			
		||||
                    if let Ok(result) = redis_conn.hget::<_, _, String>(&job_key, "output").await {
 | 
			
		||||
                        info!("✅ Sync Job {} COMPLETED with result: {}", sync_job.id, result);
 | 
			
		||||
                    } else {
 | 
			
		||||
                        info!("✅ Sync Job {} COMPLETED", sync_job.id);
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                JobStatus::Error => {
 | 
			
		||||
                    let job_key = format!("hero:job:{}", sync_job.id);
 | 
			
		||||
                    if let Ok(error) = redis_conn.hget::<_, _, String>(&job_key, "error").await {
 | 
			
		||||
                        warn!("❌ Sync Job {} FAILED with error: {}", sync_job.id, error);
 | 
			
		||||
                    } else {
 | 
			
		||||
                        warn!("❌ Sync Job {} FAILED", sync_job.id);
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                _ => info!("🔄 Sync Job {} status: {:?}", sync_job.id, status),
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // Check async job status
 | 
			
		||||
        if let Ok(status) = Job::get_status(&mut redis_conn, &async_job.id).await {
 | 
			
		||||
            match status {
 | 
			
		||||
                JobStatus::Finished => {
 | 
			
		||||
                    let job_key = format!("hero:job:{}", async_job.id);
 | 
			
		||||
                    if let Ok(result) = redis_conn.hget::<_, _, String>(&job_key, "output").await {
 | 
			
		||||
                        info!("✅ Async Job {} COMPLETED with result: {}", async_job.id, result);
 | 
			
		||||
                    } else {
 | 
			
		||||
                        info!("✅ Async Job {} COMPLETED", async_job.id);
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                JobStatus::Error => {
 | 
			
		||||
                    let job_key = format!("hero:job:{}", async_job.id);
 | 
			
		||||
                    if let Ok(error) = redis_conn.hget::<_, _, String>(&job_key, "error").await {
 | 
			
		||||
                        warn!("❌ Async Job {} FAILED with error: {}", async_job.id, error);
 | 
			
		||||
                    } else {
 | 
			
		||||
                        warn!("❌ Async Job {} FAILED", async_job.id);
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                _ => info!("🔄 Async Job {} status: {:?}", async_job.id, status),
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        sleep(Duration::from_secs(2)).await;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // Demo 2: Using convenience functions (backward compatibility)
 | 
			
		||||
    info!("\n=== Demo 2: Convenience Functions (Backward Compatibility) ===");
 | 
			
		||||
 | 
			
		||||
    let (conv_sync_shutdown_tx, conv_sync_shutdown_rx) = mpsc::channel::<()>(1);
 | 
			
		||||
    let (conv_async_shutdown_tx, conv_async_shutdown_rx) = mpsc::channel::<()>(1);
 | 
			
		||||
 | 
			
		||||
    // Spawn actors using convenience functions
 | 
			
		||||
    let conv_sync_engine = create_heromodels_engine();
 | 
			
		||||
    let conv_async_engine = create_heromodels_engine();
 | 
			
		||||
 | 
			
		||||
    info!("Spawning sync actor using convenience function...");
 | 
			
		||||
    let conv_sync_handle = spawn_sync_actor(
 | 
			
		||||
        "convenience_sync_actor".to_string(),
 | 
			
		||||
        "/tmp".to_string(),
 | 
			
		||||
        conv_sync_engine,
 | 
			
		||||
        REDIS_URL.to_string(),
 | 
			
		||||
        conv_sync_shutdown_rx,
 | 
			
		||||
        false,
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    info!("Spawning async actor using convenience function...");
 | 
			
		||||
    let conv_async_handle = spawn_async_actor(
 | 
			
		||||
        "convenience_async_actor".to_string(),
 | 
			
		||||
        "/tmp".to_string(),
 | 
			
		||||
        conv_async_engine,
 | 
			
		||||
        REDIS_URL.to_string(),
 | 
			
		||||
        conv_async_shutdown_rx,
 | 
			
		||||
        Duration::from_secs(20), // 20 second timeout
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    // Give convenience actors time to start
 | 
			
		||||
    sleep(Duration::from_secs(1)).await;
 | 
			
		||||
 | 
			
		||||
    // Create jobs for convenience actors
 | 
			
		||||
    let conv_sync_job = create_demo_job(
 | 
			
		||||
        "convenience_sync",
 | 
			
		||||
        r#"
 | 
			
		||||
        print("Convenience sync actor: Hello World!");
 | 
			
		||||
        "Hello from convenience sync actor"
 | 
			
		||||
        "#,
 | 
			
		||||
        None,
 | 
			
		||||
    ).await?;
 | 
			
		||||
 | 
			
		||||
    let conv_async_job = create_demo_job(
 | 
			
		||||
        "convenience_async",
 | 
			
		||||
        r#"
 | 
			
		||||
        print("Convenience async actor: Hello World!");
 | 
			
		||||
        "Hello from convenience async actor"
 | 
			
		||||
        "#,
 | 
			
		||||
        Some(10),
 | 
			
		||||
    ).await?;
 | 
			
		||||
 | 
			
		||||
    dispatch_job(&mut redis_conn, &conv_sync_job, "convenience_sync_actor").await?;
 | 
			
		||||
    dispatch_job(&mut redis_conn, &conv_async_job, "convenience_async_actor").await?;
 | 
			
		||||
 | 
			
		||||
    info!("Dispatched jobs to convenience actors");
 | 
			
		||||
 | 
			
		||||
    // Wait a bit for jobs to complete
 | 
			
		||||
    sleep(Duration::from_secs(5)).await;
 | 
			
		||||
 | 
			
		||||
    // Shutdown all actors gracefully
 | 
			
		||||
    info!("\n=== Shutting Down All Actors ===");
 | 
			
		||||
 | 
			
		||||
    info!("Sending shutdown signals...");
 | 
			
		||||
    let _ = sync_shutdown_tx.send(()).await;
 | 
			
		||||
    let _ = async_shutdown_tx.send(()).await;
 | 
			
		||||
    let _ = conv_sync_shutdown_tx.send(()).await;
 | 
			
		||||
    let _ = conv_async_shutdown_tx.send(()).await;
 | 
			
		||||
 | 
			
		||||
    info!("Waiting for actors to shutdown...");
 | 
			
		||||
    
 | 
			
		||||
    // Wait for all actors to shutdown
 | 
			
		||||
    let results = tokio::join!(
 | 
			
		||||
        sync_handle,
 | 
			
		||||
        async_handle,
 | 
			
		||||
        conv_sync_handle,
 | 
			
		||||
        conv_async_handle
 | 
			
		||||
    );
 | 
			
		||||
 | 
			
		||||
    match results {
 | 
			
		||||
        (Ok(Ok(())), Ok(Ok(())), Ok(Ok(())), Ok(Ok(()))) => {
 | 
			
		||||
            info!("All actors shut down successfully!");
 | 
			
		||||
        }
 | 
			
		||||
        _ => {
 | 
			
		||||
            error!("Some actors encountered errors during shutdown");
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    info!("Trait-Based Actor Demo completed successfully!");
 | 
			
		||||
    
 | 
			
		||||
    // Summary
 | 
			
		||||
    info!("\n=== Summary ===");
 | 
			
		||||
    info!("✅ Demonstrated unified Actor trait interface");
 | 
			
		||||
    info!("✅ Showed both sync and async actor implementations");
 | 
			
		||||
    info!("✅ Used shared configuration and spawn logic");
 | 
			
		||||
    info!("✅ Maintained backward compatibility with convenience functions");
 | 
			
		||||
    info!("✅ Eliminated code duplication between actor types");
 | 
			
		||||
    info!("✅ Provided clean, consistent API for all actor operations");
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Create a demo job with the specified script and timeout
 | 
			
		||||
async fn create_demo_job(
 | 
			
		||||
    name: &str,
 | 
			
		||||
    script: &str,
 | 
			
		||||
    timeout_seconds: Option<i32>,
 | 
			
		||||
) -> Result<Job, Box<dyn std::error::Error>> {
 | 
			
		||||
    let mut job = Job::new(
 | 
			
		||||
        format!("demo_{}", name), // caller_id
 | 
			
		||||
        "demo_context".to_string(), // context_id
 | 
			
		||||
        script.to_string(),
 | 
			
		||||
        ScriptType::OSIS,
 | 
			
		||||
    );
 | 
			
		||||
    
 | 
			
		||||
    // Set timeout if provided
 | 
			
		||||
    if let Some(timeout) = timeout_seconds {
 | 
			
		||||
        job.timeout = Duration::from_secs(timeout as u64);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    Ok(job)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Dispatch a job to the actor queue
 | 
			
		||||
async fn dispatch_job(
 | 
			
		||||
    redis_conn: &mut redis::aio::MultiplexedConnection,
 | 
			
		||||
    job: &Job,
 | 
			
		||||
    actor_queue: &str,
 | 
			
		||||
) -> Result<(), Box<dyn std::error::Error>> {
 | 
			
		||||
    // Store job in Redis
 | 
			
		||||
    job.store_in_redis(redis_conn).await?;
 | 
			
		||||
 | 
			
		||||
    // Add job to actor queue
 | 
			
		||||
    let queue_key = format!("hero:job:{}", actor_queue);
 | 
			
		||||
    let _: () = redis_conn.rpush(&queue_key, &job.id).await?;
 | 
			
		||||
 | 
			
		||||
    Ok(())
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										420
									
								
								_archive/core/actor/src/async_worker_impl.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										420
									
								
								_archive/core/actor/src/async_worker_impl.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,420 @@
 | 
			
		||||
//! # Asynchronous Actor Implementation
 | 
			
		||||
//!
 | 
			
		||||
//! This module provides an asynchronous actor implementation that can process
 | 
			
		||||
//! multiple jobs concurrently with timeout support. Each job is spawned as a
 | 
			
		||||
//! separate Tokio task, allowing for parallel execution and proper timeout handling.
 | 
			
		||||
//!
 | 
			
		||||
//! ## Features
 | 
			
		||||
//!
 | 
			
		||||
//! - **Concurrent Processing**: Multiple jobs can run simultaneously
 | 
			
		||||
//! - **Timeout Support**: Jobs that exceed their timeout are automatically cancelled
 | 
			
		||||
//! - **Resource Cleanup**: Proper cleanup of aborted/cancelled jobs
 | 
			
		||||
//! - **Non-blocking**: Actor continues processing new jobs while others are running
 | 
			
		||||
//! - **Scalable**: Can handle high job throughput with parallel execution
 | 
			
		||||
//!
 | 
			
		||||
//! ## Usage
 | 
			
		||||
//!
 | 
			
		||||
//! ```rust
 | 
			
		||||
//! use std::sync::Arc;
 | 
			
		||||
//! use std::time::Duration;
 | 
			
		||||
//! use baobab_actor::async_actor_impl::AsyncActor;
 | 
			
		||||
//! use baobab_actor::actor_trait::{spawn_actor, ActorConfig};
 | 
			
		||||
//! use baobab_actor::engine::create_heromodels_engine;
 | 
			
		||||
//! use tokio::sync::mpsc;
 | 
			
		||||
//!
 | 
			
		||||
//! let config = ActorConfig::new(
 | 
			
		||||
//!     "async_actor_1".to_string(),
 | 
			
		||||
//!     "/path/to/db".to_string(),
 | 
			
		||||
//!     "redis://localhost:6379".to_string(),
 | 
			
		||||
//!     false, // preserve_tasks
 | 
			
		||||
//! ).with_default_timeout(Duration::from_secs(300));
 | 
			
		||||
//!
 | 
			
		||||
//! let actor = Arc::new(AsyncActor::new());
 | 
			
		||||
//! let engine = create_heromodels_engine();
 | 
			
		||||
//! let (shutdown_tx, shutdown_rx) = mpsc::channel(1);
 | 
			
		||||
//!
 | 
			
		||||
//! let handle = spawn_actor(actor, config, engine, shutdown_rx);
 | 
			
		||||
//!
 | 
			
		||||
//! // Later, shutdown the actor
 | 
			
		||||
//! shutdown_tx.send(()).await.unwrap();
 | 
			
		||||
//! handle.await.unwrap().unwrap();
 | 
			
		||||
//! ```
 | 
			
		||||
 | 
			
		||||
use async_trait::async_trait;
 | 
			
		||||
use hero_job::{Job, JobStatus};
 | 
			
		||||
use log::{debug, error, info, warn};
 | 
			
		||||
use rhai::Engine;
 | 
			
		||||
use std::collections::HashMap;
 | 
			
		||||
use std::sync::Arc;
 | 
			
		||||
use std::time::Duration;
 | 
			
		||||
use tokio::sync::Mutex;
 | 
			
		||||
use tokio::task::JoinHandle;
 | 
			
		||||
use tokio::time::timeout;
 | 
			
		||||
 | 
			
		||||
use crate::engine::eval_script;
 | 
			
		||||
use crate::actor_trait::{Actor, ActorConfig};
 | 
			
		||||
use crate::initialize_redis_connection;
 | 
			
		||||
 | 
			
		||||
/// Represents a running job with its handle and metadata
 | 
			
		||||
#[derive(Debug)]
 | 
			
		||||
struct RunningJob {
 | 
			
		||||
    job_id: String,
 | 
			
		||||
    handle: JoinHandle<()>,
 | 
			
		||||
    started_at: std::time::Instant,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Builder for AsyncActor
 | 
			
		||||
#[derive(Debug, Default)]
 | 
			
		||||
pub struct AsyncActorBuilder {
 | 
			
		||||
    actor_id: Option<String>,
 | 
			
		||||
    db_path: Option<String>,
 | 
			
		||||
    redis_url: Option<String>,
 | 
			
		||||
    default_timeout: Option<Duration>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl AsyncActorBuilder {
 | 
			
		||||
    pub fn new() -> Self {
 | 
			
		||||
        Self::default()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn actor_id<S: Into<String>>(mut self, actor_id: S) -> Self {
 | 
			
		||||
        self.actor_id = Some(actor_id.into());
 | 
			
		||||
        self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn db_path<S: Into<String>>(mut self, db_path: S) -> Self {
 | 
			
		||||
        self.db_path = Some(db_path.into());
 | 
			
		||||
        self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn redis_url<S: Into<String>>(mut self, redis_url: S) -> Self {
 | 
			
		||||
        self.redis_url = Some(redis_url.into());
 | 
			
		||||
        self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn default_timeout(mut self, timeout: Duration) -> Self {
 | 
			
		||||
        self.default_timeout = Some(timeout);
 | 
			
		||||
        self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn build(self) -> Result<AsyncActor, String> {
 | 
			
		||||
        Ok(AsyncActor {
 | 
			
		||||
            actor_id: self.actor_id.ok_or("actor_id is required")?,
 | 
			
		||||
            db_path: self.db_path.ok_or("db_path is required")?,
 | 
			
		||||
            redis_url: self.redis_url.ok_or("redis_url is required")?,
 | 
			
		||||
            default_timeout: self.default_timeout.unwrap_or(Duration::from_secs(300)),
 | 
			
		||||
            running_jobs: Arc::new(Mutex::new(HashMap::new())),
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Asynchronous actor that processes jobs concurrently
 | 
			
		||||
#[derive(Debug, Clone)]
 | 
			
		||||
pub struct AsyncActor {
 | 
			
		||||
    pub actor_id: String,
 | 
			
		||||
    pub db_path: String,
 | 
			
		||||
    pub redis_url: String,
 | 
			
		||||
    pub default_timeout: Duration,
 | 
			
		||||
    running_jobs: Arc<Mutex<HashMap<String, RunningJob>>>,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl AsyncActor {
 | 
			
		||||
    /// Create a new AsyncActorBuilder
 | 
			
		||||
    pub fn builder() -> AsyncActorBuilder {
 | 
			
		||||
        AsyncActorBuilder::new()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /// Add a running job to the tracking map
 | 
			
		||||
    async fn add_running_job(&self, job_id: String, handle: JoinHandle<()>) {
 | 
			
		||||
        let running_job = RunningJob {
 | 
			
		||||
            job_id: job_id.clone(),
 | 
			
		||||
            handle,
 | 
			
		||||
            started_at: std::time::Instant::now(),
 | 
			
		||||
        };
 | 
			
		||||
        
 | 
			
		||||
        let mut jobs = self.running_jobs.lock().await;
 | 
			
		||||
        jobs.insert(job_id.clone(), running_job);
 | 
			
		||||
        debug!("Async Actor: Added running job '{}'. Total running: {}", 
 | 
			
		||||
               job_id, jobs.len());
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /// Remove a completed job from the tracking map
 | 
			
		||||
    async fn remove_running_job(&self, job_id: &str) {
 | 
			
		||||
        let mut jobs = self.running_jobs.lock().await;
 | 
			
		||||
        if let Some(job) = jobs.remove(job_id) {
 | 
			
		||||
            let duration = job.started_at.elapsed();
 | 
			
		||||
            debug!("Async Actor: Removed completed job '{}' after {:?}. Remaining: {}", 
 | 
			
		||||
                   job_id, duration, jobs.len());
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /// Get the count of currently running jobs
 | 
			
		||||
    pub async fn running_job_count(&self) -> usize {
 | 
			
		||||
        let jobs = self.running_jobs.lock().await;
 | 
			
		||||
        jobs.len()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /// Cleanup any finished jobs from the running jobs map
 | 
			
		||||
    async fn cleanup_finished_jobs(&self) {
 | 
			
		||||
        let mut jobs = self.running_jobs.lock().await;
 | 
			
		||||
        let mut to_remove = Vec::new();
 | 
			
		||||
        
 | 
			
		||||
        for (job_id, running_job) in jobs.iter() {
 | 
			
		||||
            if running_job.handle.is_finished() {
 | 
			
		||||
                to_remove.push(job_id.clone());
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        for job_id in to_remove {
 | 
			
		||||
            if let Some(job) = jobs.remove(&job_id) {
 | 
			
		||||
                let duration = job.started_at.elapsed();
 | 
			
		||||
                debug!("Async Actor: Cleaned up finished job '{}' after {:?}", 
 | 
			
		||||
                       job_id, duration);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /// Execute a single job asynchronously with timeout support
 | 
			
		||||
    async fn execute_job_with_timeout(
 | 
			
		||||
        job: Job,
 | 
			
		||||
        engine: Engine,
 | 
			
		||||
        actor_id: String,
 | 
			
		||||
        redis_url: String,
 | 
			
		||||
        job_timeout: Duration,
 | 
			
		||||
    ) {
 | 
			
		||||
        let job_id = job.id.clone();
 | 
			
		||||
        info!("Async Actor '{}', Job {}: Starting execution with timeout {:?}", 
 | 
			
		||||
              actor_id, job_id, job_timeout);
 | 
			
		||||
 | 
			
		||||
        // Create a new Redis connection for this job
 | 
			
		||||
        let mut redis_conn = match initialize_redis_connection(&actor_id, &redis_url).await {
 | 
			
		||||
            Ok(conn) => conn,
 | 
			
		||||
            Err(e) => {
 | 
			
		||||
                error!("Async Actor '{}', Job {}: Failed to initialize Redis connection: {}", 
 | 
			
		||||
                       actor_id, job_id, e);
 | 
			
		||||
                return;
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        // Update job status to Started
 | 
			
		||||
        if let Err(e) = Job::update_status(&mut redis_conn, &job_id, JobStatus::Started).await {
 | 
			
		||||
            error!("Async Actor '{}', Job {}: Failed to update status to Started: {}", 
 | 
			
		||||
                   actor_id, job_id, e);
 | 
			
		||||
            return;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // Create the script execution task
 | 
			
		||||
        let script_task = async {
 | 
			
		||||
            // Execute the Rhai script
 | 
			
		||||
            match eval_script(&engine, &job.script) {
 | 
			
		||||
                Ok(result) => {
 | 
			
		||||
                    let result_str = format!("{:?}", result);
 | 
			
		||||
                    info!("Async Actor '{}', Job {}: Script executed successfully. Result: {}", 
 | 
			
		||||
                          actor_id, job_id, result_str);
 | 
			
		||||
                    
 | 
			
		||||
                    // Update job with success result
 | 
			
		||||
                    if let Err(e) = Job::set_result(&mut redis_conn, &job_id, &result_str).await {
 | 
			
		||||
                        error!("Async Actor '{}', Job {}: Failed to set result: {}", 
 | 
			
		||||
                               actor_id, job_id, e);
 | 
			
		||||
                        return;
 | 
			
		||||
                    }
 | 
			
		||||
                    
 | 
			
		||||
                    if let Err(e) = Job::update_status(&mut redis_conn, &job_id, JobStatus::Finished).await {
 | 
			
		||||
                        error!("Async Actor '{}', Job {}: Failed to update status to Finished: {}", 
 | 
			
		||||
                               actor_id, job_id, e);
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
                Err(e) => {
 | 
			
		||||
                    let error_msg = format!("Script execution error: {}", e);
 | 
			
		||||
                    error!("Async Actor '{}', Job {}: {}", actor_id, job_id, error_msg);
 | 
			
		||||
                    
 | 
			
		||||
                    // Update job with error
 | 
			
		||||
                    if let Err(e) = Job::set_error(&mut redis_conn, &job_id, &error_msg).await {
 | 
			
		||||
                        error!("Async Actor '{}', Job {}: Failed to set error: {}", 
 | 
			
		||||
                               actor_id, job_id, e);
 | 
			
		||||
                        return;
 | 
			
		||||
                    }
 | 
			
		||||
                    
 | 
			
		||||
                    if let Err(e) = Job::update_status(&mut redis_conn, &job_id, JobStatus::Error).await {
 | 
			
		||||
                        error!("Async Actor '{}', Job {}: Failed to update status to Error: {}", 
 | 
			
		||||
                               actor_id, job_id, e);
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        // Execute the script with timeout
 | 
			
		||||
        match timeout(job_timeout, script_task).await {
 | 
			
		||||
            Ok(()) => {
 | 
			
		||||
                info!("Async Actor '{}', Job {}: Completed within timeout", actor_id, job_id);
 | 
			
		||||
            }
 | 
			
		||||
            Err(_) => {
 | 
			
		||||
                warn!("Async Actor '{}', Job {}: Timed out after {:?}, marking as error", 
 | 
			
		||||
                      actor_id, job_id, job_timeout);
 | 
			
		||||
                
 | 
			
		||||
                let timeout_msg = format!("Job timed out after {:?}", job_timeout);
 | 
			
		||||
                if let Err(e) = Job::set_error(&mut redis_conn, &job_id, &timeout_msg).await {
 | 
			
		||||
                    error!("Async Actor '{}', Job {}: Failed to set timeout error: {}", 
 | 
			
		||||
                           actor_id, job_id, e);
 | 
			
		||||
                }
 | 
			
		||||
                
 | 
			
		||||
                if let Err(e) = Job::update_status(&mut redis_conn, &job_id, JobStatus::Error).await {
 | 
			
		||||
                    error!("Async Actor '{}', Job {}: Failed to update status to Error after timeout: {}", 
 | 
			
		||||
                           actor_id, job_id, e);
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        info!("Async Actor '{}', Job {}: Job processing completed", actor_id, job_id);
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Default for AsyncActor {
 | 
			
		||||
    fn default() -> Self {
 | 
			
		||||
        // Default AsyncActor with placeholder values
 | 
			
		||||
        // In practice, use the builder pattern instead
 | 
			
		||||
        Self {
 | 
			
		||||
            actor_id: "default_async_actor".to_string(),
 | 
			
		||||
            db_path: "/tmp".to_string(),
 | 
			
		||||
            redis_url: "redis://localhost:6379".to_string(),
 | 
			
		||||
            default_timeout: Duration::from_secs(300),
 | 
			
		||||
            running_jobs: Arc::new(Mutex::new(HashMap::new())),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[async_trait]
 | 
			
		||||
impl Actor for AsyncActor {
 | 
			
		||||
    async fn process_job(
 | 
			
		||||
        &self,
 | 
			
		||||
        job: Job,
 | 
			
		||||
        engine: Engine, // Reuse the stateless engine
 | 
			
		||||
        _redis_conn: &mut redis::aio::MultiplexedConnection,
 | 
			
		||||
    ) {
 | 
			
		||||
        let job_id = job.id.clone();
 | 
			
		||||
        let actor_id = &self.actor_id.clone();
 | 
			
		||||
 | 
			
		||||
        // Determine timeout (use job-specific timeout if available, otherwise default)
 | 
			
		||||
        let job_timeout = if job.timeout.as_secs() > 0 {
 | 
			
		||||
            job.timeout
 | 
			
		||||
        } else {
 | 
			
		||||
            self.default_timeout // Use actor's default timeout
 | 
			
		||||
        };
 | 
			
		||||
 | 
			
		||||
        info!("Async Actor '{}', Job {}: Spawning job execution task with timeout {:?}", 
 | 
			
		||||
              actor_id, job_id, job_timeout);
 | 
			
		||||
 | 
			
		||||
        // Clone necessary data for the spawned task
 | 
			
		||||
        let job_id_clone = job_id.clone();
 | 
			
		||||
        let actor_id_clone = actor_id.clone();
 | 
			
		||||
        let actor_id_debug = actor_id.clone(); // Additional clone for debug statement
 | 
			
		||||
        let job_id_debug = job_id.clone(); // Additional clone for debug statement
 | 
			
		||||
        let redis_url_clone = self.redis_url.clone();
 | 
			
		||||
        let running_jobs_clone = Arc::clone(&self.running_jobs);
 | 
			
		||||
 | 
			
		||||
        // Spawn the job execution task
 | 
			
		||||
        let job_handle = tokio::spawn(async move {
 | 
			
		||||
            Self::execute_job_with_timeout(
 | 
			
		||||
                job,
 | 
			
		||||
                engine,
 | 
			
		||||
                actor_id_clone,
 | 
			
		||||
                redis_url_clone,
 | 
			
		||||
                job_timeout,
 | 
			
		||||
            ).await;
 | 
			
		||||
            
 | 
			
		||||
            // Remove this job from the running jobs map when it completes
 | 
			
		||||
            let mut jobs = running_jobs_clone.lock().await;
 | 
			
		||||
            if let Some(running_job) = jobs.remove(&job_id_clone) {
 | 
			
		||||
                let duration = running_job.started_at.elapsed();
 | 
			
		||||
                debug!("Async Actor '{}': Removed completed job '{}' after {:?}", 
 | 
			
		||||
                       actor_id_debug, job_id_debug, duration);
 | 
			
		||||
            }
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        // Add the job to the running jobs map
 | 
			
		||||
        self.add_running_job(job_id, job_handle).await;
 | 
			
		||||
 | 
			
		||||
        // Cleanup finished jobs periodically
 | 
			
		||||
        self.cleanup_finished_jobs().await;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn actor_type(&self) -> &'static str {
 | 
			
		||||
        "Async"
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fn actor_id(&self) -> &str {
 | 
			
		||||
        &self.actor_id
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fn redis_url(&self) -> &str {
 | 
			
		||||
        &self.redis_url
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod tests {
 | 
			
		||||
    use super::*;
 | 
			
		||||
    use crate::engine::create_heromodels_engine;
 | 
			
		||||
    use hero_job::ScriptType;
 | 
			
		||||
 | 
			
		||||
    #[tokio::test]
 | 
			
		||||
    async fn test_async_actor_creation() {
 | 
			
		||||
        let actor = AsyncActor::new();
 | 
			
		||||
        assert_eq!(actor.actor_type(), "Async");
 | 
			
		||||
        assert_eq!(actor.running_job_count().await, 0);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[tokio::test]
 | 
			
		||||
    async fn test_async_actor_default() {
 | 
			
		||||
        let actor = AsyncActor::default();
 | 
			
		||||
        assert_eq!(actor.actor_type(), "Async");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[tokio::test]
 | 
			
		||||
    async fn test_async_actor_job_tracking() {
 | 
			
		||||
        let actor = AsyncActor::new();
 | 
			
		||||
        
 | 
			
		||||
        // Simulate adding a job
 | 
			
		||||
        let handle = tokio::spawn(async {
 | 
			
		||||
            tokio::time::sleep(Duration::from_millis(100)).await;
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
        actor.add_running_job("job_1".to_string(), handle).await;
 | 
			
		||||
        assert_eq!(actor.running_job_count().await, 1);
 | 
			
		||||
 | 
			
		||||
        // Wait for job to complete
 | 
			
		||||
        tokio::time::sleep(Duration::from_millis(200)).await;
 | 
			
		||||
        actor.cleanup_finished_jobs().await;
 | 
			
		||||
        assert_eq!(actor.running_job_count().await, 0);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[tokio::test]
 | 
			
		||||
    async fn test_async_actor_process_job_interface() {
 | 
			
		||||
        let actor = AsyncActor::new();
 | 
			
		||||
        let engine = create_heromodels_engine();
 | 
			
		||||
        
 | 
			
		||||
        // Create a simple test job
 | 
			
		||||
        let job = Job::new(
 | 
			
		||||
            "test_caller".to_string(),
 | 
			
		||||
            "test_context".to_string(),
 | 
			
		||||
            r#"print("Hello from async actor test!"); 42"#.to_string(),
 | 
			
		||||
            ScriptType::OSIS,
 | 
			
		||||
        );
 | 
			
		||||
 | 
			
		||||
        let config = ActorConfig::new(
 | 
			
		||||
            "test_async_actor".to_string(),
 | 
			
		||||
            "/tmp".to_string(),
 | 
			
		||||
            "redis://localhost:6379".to_string(),
 | 
			
		||||
            false,
 | 
			
		||||
        ).with_default_timeout(Duration::from_secs(60));
 | 
			
		||||
 | 
			
		||||
        // Note: This test doesn't actually connect to Redis, it just tests the interface
 | 
			
		||||
        // In a real test environment, you'd need a Redis instance or mock
 | 
			
		||||
        
 | 
			
		||||
        // The process_job method should be callable (interface test)
 | 
			
		||||
        // actor.process_job(job, engine, &mut redis_conn, &config).await;
 | 
			
		||||
        
 | 
			
		||||
        // For now, just verify the actor was created successfully
 | 
			
		||||
        assert_eq!(actor.actor_type(), "Async");
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										250
									
								
								_archive/core/actor/src/config.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										250
									
								
								_archive/core/actor/src/config.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,250 @@
 | 
			
		||||
//! Actor Configuration Module - TOML-based configuration for Hero actors
 | 
			
		||||
 | 
			
		||||
use serde::{Deserialize, Serialize};
 | 
			
		||||
use std::fs;
 | 
			
		||||
use std::path::Path;
 | 
			
		||||
use std::time::Duration;
 | 
			
		||||
 | 
			
		||||
/// Actor configuration loaded from TOML file
 | 
			
		||||
#[derive(Debug, Clone, Serialize, Deserialize)]
 | 
			
		||||
pub struct ActorConfig {
 | 
			
		||||
    /// Actor identification
 | 
			
		||||
    pub actor_id: String,
 | 
			
		||||
    
 | 
			
		||||
    /// Redis connection URL
 | 
			
		||||
    pub redis_url: String,
 | 
			
		||||
    
 | 
			
		||||
    /// Database path for Rhai engine
 | 
			
		||||
    pub db_path: String,
 | 
			
		||||
    
 | 
			
		||||
    /// Whether to preserve task details after completion
 | 
			
		||||
    #[serde(default = "default_preserve_tasks")]
 | 
			
		||||
    pub preserve_tasks: bool,
 | 
			
		||||
    
 | 
			
		||||
    /// Actor type configuration
 | 
			
		||||
    pub actor_type: ActorType,
 | 
			
		||||
    
 | 
			
		||||
    /// Logging configuration
 | 
			
		||||
    #[serde(default)]
 | 
			
		||||
    pub logging: LoggingConfig,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Actor type configuration
 | 
			
		||||
#[derive(Debug, Clone, Serialize, Deserialize)]
 | 
			
		||||
#[serde(tag = "type")]
 | 
			
		||||
pub enum ActorType {
 | 
			
		||||
    /// Synchronous actor configuration
 | 
			
		||||
    #[serde(rename = "sync")]
 | 
			
		||||
    Sync,
 | 
			
		||||
    
 | 
			
		||||
    /// Asynchronous actor configuration
 | 
			
		||||
    #[serde(rename = "async")]
 | 
			
		||||
    Async {
 | 
			
		||||
        /// Default timeout for jobs in seconds
 | 
			
		||||
        #[serde(default = "default_timeout_seconds")]
 | 
			
		||||
        default_timeout_seconds: u64,
 | 
			
		||||
    },
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Logging configuration
 | 
			
		||||
#[derive(Debug, Clone, Serialize, Deserialize)]
 | 
			
		||||
pub struct LoggingConfig {
 | 
			
		||||
    /// Whether to include timestamps in log output
 | 
			
		||||
    #[serde(default = "default_timestamps")]
 | 
			
		||||
    pub timestamps: bool,
 | 
			
		||||
    
 | 
			
		||||
    /// Log level (trace, debug, info, warn, error)
 | 
			
		||||
    #[serde(default = "default_log_level")]
 | 
			
		||||
    pub level: String,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Default for LoggingConfig {
 | 
			
		||||
    fn default() -> Self {
 | 
			
		||||
        Self {
 | 
			
		||||
            timestamps: default_timestamps(),
 | 
			
		||||
            level: default_log_level(),
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl ActorConfig {
 | 
			
		||||
    /// Load configuration from TOML file
 | 
			
		||||
    pub fn from_file<P: AsRef<Path>>(path: P) -> Result<Self, ConfigError> {
 | 
			
		||||
        let content = fs::read_to_string(&path)
 | 
			
		||||
            .map_err(|e| ConfigError::IoError(format!("Failed to read config file: {}", e)))?;
 | 
			
		||||
        
 | 
			
		||||
        let config: ActorConfig = toml::from_str(&content)
 | 
			
		||||
            .map_err(|e| ConfigError::ParseError(format!("Failed to parse TOML: {}", e)))?;
 | 
			
		||||
        
 | 
			
		||||
        config.validate()?;
 | 
			
		||||
        Ok(config)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    /// Validate the configuration
 | 
			
		||||
    fn validate(&self) -> Result<(), ConfigError> {
 | 
			
		||||
        if self.actor_id.is_empty() {
 | 
			
		||||
            return Err(ConfigError::ValidationError("actor_id cannot be empty".to_string()));
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        if self.redis_url.is_empty() {
 | 
			
		||||
            return Err(ConfigError::ValidationError("redis_url cannot be empty".to_string()));
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        if self.db_path.is_empty() {
 | 
			
		||||
            return Err(ConfigError::ValidationError("db_path cannot be empty".to_string()));
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        // Validate log level
 | 
			
		||||
        match self.logging.level.to_lowercase().as_str() {
 | 
			
		||||
            "trace" | "debug" | "info" | "warn" | "error" => {},
 | 
			
		||||
            _ => return Err(ConfigError::ValidationError(
 | 
			
		||||
                format!("Invalid log level: {}. Must be one of: trace, debug, info, warn, error", self.logging.level)
 | 
			
		||||
            )),
 | 
			
		||||
        }
 | 
			
		||||
        
 | 
			
		||||
        Ok(())
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    /// Get the default timeout duration for async actors
 | 
			
		||||
    pub fn get_default_timeout(&self) -> Option<Duration> {
 | 
			
		||||
        match &self.actor_type {
 | 
			
		||||
            ActorType::Sync => None,
 | 
			
		||||
            ActorType::Async { default_timeout_seconds } => {
 | 
			
		||||
                Some(Duration::from_secs(*default_timeout_seconds))
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    /// Check if this is a sync actor configuration
 | 
			
		||||
    pub fn is_sync(&self) -> bool {
 | 
			
		||||
        matches!(self.actor_type, ActorType::Sync)
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    /// Check if this is an async actor configuration
 | 
			
		||||
    pub fn is_async(&self) -> bool {
 | 
			
		||||
        matches!(self.actor_type, ActorType::Async { .. })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Configuration error types
 | 
			
		||||
#[derive(Debug, thiserror::Error)]
 | 
			
		||||
pub enum ConfigError {
 | 
			
		||||
    #[error("IO error: {0}")]
 | 
			
		||||
    IoError(String),
 | 
			
		||||
    
 | 
			
		||||
    #[error("Parse error: {0}")]
 | 
			
		||||
    ParseError(String),
 | 
			
		||||
    
 | 
			
		||||
    #[error("Validation error: {0}")]
 | 
			
		||||
    ValidationError(String),
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Default value functions for serde
 | 
			
		||||
fn default_preserve_tasks() -> bool {
 | 
			
		||||
    false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn default_timeout_seconds() -> u64 {
 | 
			
		||||
    300 // 5 minutes
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn default_timestamps() -> bool {
 | 
			
		||||
    true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
fn default_log_level() -> String {
 | 
			
		||||
    "info".to_string()
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod tests {
 | 
			
		||||
    use super::*;
 | 
			
		||||
    use std::io::Write;
 | 
			
		||||
    use tempfile::NamedTempFile;
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn test_sync_actor_config() {
 | 
			
		||||
        let config_toml = r#"
 | 
			
		||||
actor_id = "sync_actor_1"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/actor_db"
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "sync"
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
timestamps = false
 | 
			
		||||
level = "debug"
 | 
			
		||||
"#;
 | 
			
		||||
        
 | 
			
		||||
        let config: ActorConfig = toml::from_str(config_toml).unwrap();
 | 
			
		||||
        assert_eq!(config.actor_id, "sync_actor_1");
 | 
			
		||||
        assert!(config.is_sync());
 | 
			
		||||
        assert!(!config.is_async());
 | 
			
		||||
        assert_eq!(config.get_default_timeout(), None);
 | 
			
		||||
        assert!(!config.logging.timestamps);
 | 
			
		||||
        assert_eq!(config.logging.level, "debug");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn test_async_actor_config() {
 | 
			
		||||
        let config_toml = r#"
 | 
			
		||||
actor_id = "async_actor_1"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/actor_db"
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "async"
 | 
			
		||||
default_timeout_seconds = 600
 | 
			
		||||
 | 
			
		||||
[logging]
 | 
			
		||||
timestamps = true
 | 
			
		||||
level = "info"
 | 
			
		||||
"#;
 | 
			
		||||
        
 | 
			
		||||
        let config: ActorConfig = toml::from_str(config_toml).unwrap();
 | 
			
		||||
        assert_eq!(config.actor_id, "async_actor_1");
 | 
			
		||||
        assert!(!config.is_sync());
 | 
			
		||||
        assert!(config.is_async());
 | 
			
		||||
        assert_eq!(config.get_default_timeout(), Some(Duration::from_secs(600)));
 | 
			
		||||
        assert!(config.logging.timestamps);
 | 
			
		||||
        assert_eq!(config.logging.level, "info");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn test_config_from_file() {
 | 
			
		||||
        let config_toml = r#"
 | 
			
		||||
actor_id = "test_actor"
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/test_db"
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "sync"
 | 
			
		||||
"#;
 | 
			
		||||
        
 | 
			
		||||
        let mut temp_file = NamedTempFile::new().unwrap();
 | 
			
		||||
        temp_file.write_all(config_toml.as_bytes()).unwrap();
 | 
			
		||||
        
 | 
			
		||||
        let config = ActorConfig::from_file(temp_file.path()).unwrap();
 | 
			
		||||
        assert_eq!(config.actor_id, "test_actor");
 | 
			
		||||
        assert!(config.is_sync());
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[test]
 | 
			
		||||
    fn test_config_validation() {
 | 
			
		||||
        let config_toml = r#"
 | 
			
		||||
actor_id = ""
 | 
			
		||||
redis_url = "redis://localhost:6379"
 | 
			
		||||
db_path = "/tmp/test_db"
 | 
			
		||||
 | 
			
		||||
[actor_type]
 | 
			
		||||
type = "sync"
 | 
			
		||||
"#;
 | 
			
		||||
        
 | 
			
		||||
        let result: Result<ActorConfig, _> = toml::from_str(config_toml);
 | 
			
		||||
        assert!(result.is_ok());
 | 
			
		||||
        
 | 
			
		||||
        let config = result.unwrap();
 | 
			
		||||
        assert!(config.validate().is_err());
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										260
									
								
								_archive/core/actor/src/engine.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										260
									
								
								_archive/core/actor/src/engine.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,260 @@
 | 
			
		||||
//! # Rhai Engine Module
 | 
			
		||||
//!
 | 
			
		||||
//! The central Rhai scripting engine for the heromodels ecosystem. This module provides
 | 
			
		||||
//! a unified interface for creating, configuring, and executing Rhai scripts with access
 | 
			
		||||
//! to all business domain modules.
 | 
			
		||||
//!
 | 
			
		||||
//! ## Features
 | 
			
		||||
//!
 | 
			
		||||
//! - **Unified Engine Creation**: Pre-configured Rhai engine with all DSL modules
 | 
			
		||||
//! - **Script Execution Utilities**: Direct evaluation, file-based execution, and AST compilation
 | 
			
		||||
//! - **Mock Database System**: Complete testing environment with seeded data
 | 
			
		||||
//! - **Feature-Based Architecture**: Modular compilation based on required domains
 | 
			
		||||
//!
 | 
			
		||||
//! ## Quick Start
 | 
			
		||||
//!
 | 
			
		||||
//! ```rust
 | 
			
		||||
//! use baobab_actor::engine::{create_heromodels_engine, eval_script};
 | 
			
		||||
//!
 | 
			
		||||
//! // Create a fully configured engine
 | 
			
		||||
//! let engine = create_heromodels_engine();
 | 
			
		||||
//!
 | 
			
		||||
//! // Execute a business logic script
 | 
			
		||||
//! let result = eval_script(&engine, r#"
 | 
			
		||||
//!     let company = new_company()
 | 
			
		||||
//!         .name("Acme Corp")
 | 
			
		||||
//!         .business_type("global");
 | 
			
		||||
//!     company.name
 | 
			
		||||
//! "#)?;
 | 
			
		||||
//!
 | 
			
		||||
//! println!("Company name: {}", result.as_string().unwrap());
 | 
			
		||||
//! ```
 | 
			
		||||
//!
 | 
			
		||||
//! ## Available Features
 | 
			
		||||
//!
 | 
			
		||||
//! - `calendar` (default): Calendar and event management
 | 
			
		||||
//! - `finance` (default): Financial accounts, assets, and marketplace
 | 
			
		||||
//! - `flow`: Workflow and approval processes
 | 
			
		||||
//! - `legal`: Contract and legal document management
 | 
			
		||||
//! - `projects`: Project and task management
 | 
			
		||||
//! - `biz`: Business operations and entities
 | 
			
		||||
 | 
			
		||||
use rhai::{Engine, EvalAltResult, Scope, AST};
 | 
			
		||||
use std::fs;
 | 
			
		||||
use std::path::Path;
 | 
			
		||||
 | 
			
		||||
/// Creates a fully configured Rhai engine with all available DSL modules.
 | 
			
		||||
///
 | 
			
		||||
/// This function creates a new Rhai engine and registers all available heromodels
 | 
			
		||||
/// DSL modules based on the enabled features. The engine comes pre-configured
 | 
			
		||||
/// with all necessary functions and types for business logic scripting.
 | 
			
		||||
///
 | 
			
		||||
/// # Returns
 | 
			
		||||
///
 | 
			
		||||
/// A fully configured `Engine` instance ready for script execution.
 | 
			
		||||
///
 | 
			
		||||
/// # Features
 | 
			
		||||
///
 | 
			
		||||
/// The engine includes modules based on enabled Cargo features:
 | 
			
		||||
/// - `calendar`: Calendar and event management functions
 | 
			
		||||
/// - `finance`: Financial accounts, assets, and marketplace operations
 | 
			
		||||
/// - `flow`: Workflow and approval process management
 | 
			
		||||
/// - `legal`: Contract and legal document handling
 | 
			
		||||
/// - `projects`: Project and task management
 | 
			
		||||
/// - `biz`: General business operations and entities
 | 
			
		||||
///
 | 
			
		||||
/// # Example
 | 
			
		||||
///
 | 
			
		||||
/// ```rust
 | 
			
		||||
/// use baobab_actor::engine::create_heromodels_engine;
 | 
			
		||||
///
 | 
			
		||||
/// let engine = create_heromodels_engine();
 | 
			
		||||
/// 
 | 
			
		||||
/// // The engine is now ready to execute business logic scripts
 | 
			
		||||
/// let result = engine.eval::<String>(r#"
 | 
			
		||||
///     "Hello from heromodels engine!"
 | 
			
		||||
/// "#)?;
 | 
			
		||||
/// ```
 | 
			
		||||
///
 | 
			
		||||
/// # Performance Notes
 | 
			
		||||
///
 | 
			
		||||
/// The engine is optimized for production use with reasonable defaults for
 | 
			
		||||
/// operation limits, expression depth, and memory usage. For benchmarking
 | 
			
		||||
/// or special use cases, you may want to adjust these limits after creation.
 | 
			
		||||
// pub fn create_heromodels_engine() -> Engine {
 | 
			
		||||
//     let mut engine = Engine::new();
 | 
			
		||||
 | 
			
		||||
//     // Register all heromodels Rhai modules
 | 
			
		||||
//     baobab_dsl::register_dsl_modules(&mut engine);
 | 
			
		||||
 | 
			
		||||
//     engine
 | 
			
		||||
// }
 | 
			
		||||
 | 
			
		||||
/// Evaluates a Rhai script string and returns the result.
 | 
			
		||||
///
 | 
			
		||||
/// This function provides a convenient way to execute Rhai script strings directly
 | 
			
		||||
/// using the provided engine. It's suitable for one-off script execution or when
 | 
			
		||||
/// the script content is dynamically generated.
 | 
			
		||||
///
 | 
			
		||||
/// # Arguments
 | 
			
		||||
///
 | 
			
		||||
/// * `engine` - The Rhai engine to use for script execution
 | 
			
		||||
/// * `script` - The Rhai script content as a string
 | 
			
		||||
///
 | 
			
		||||
/// # Returns
 | 
			
		||||
///
 | 
			
		||||
/// * `Ok(Dynamic)` - The result of script execution
 | 
			
		||||
/// * `Err(Box<EvalAltResult>)` - Script compilation or execution error
 | 
			
		||||
///
 | 
			
		||||
/// # Example
 | 
			
		||||
///
 | 
			
		||||
/// ```rust
 | 
			
		||||
/// use baobab_actor::engine::{create_heromodels_engine, eval_script};
 | 
			
		||||
///
 | 
			
		||||
/// let engine = create_heromodels_engine();
 | 
			
		||||
/// let result = eval_script(&engine, r#"
 | 
			
		||||
///     let x = 42;
 | 
			
		||||
///     let y = 8;
 | 
			
		||||
///     x + y
 | 
			
		||||
/// "#)?;
 | 
			
		||||
/// assert_eq!(result.as_int().unwrap(), 50);
 | 
			
		||||
/// ```
 | 
			
		||||
pub fn eval_script(
 | 
			
		||||
    engine: &Engine,
 | 
			
		||||
    script: &str,
 | 
			
		||||
) -> Result<rhai::Dynamic, Box<rhai::EvalAltResult>> {
 | 
			
		||||
    engine.eval(script)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Evaluates a Rhai script from a file and returns the result.
 | 
			
		||||
///
 | 
			
		||||
/// This function reads a Rhai script from the filesystem and executes it using
 | 
			
		||||
/// the provided engine. It handles file reading errors gracefully and provides
 | 
			
		||||
/// meaningful error messages.
 | 
			
		||||
///
 | 
			
		||||
/// # Arguments
 | 
			
		||||
///
 | 
			
		||||
/// * `engine` - The Rhai engine to use for script execution
 | 
			
		||||
/// * `file_path` - Path to the Rhai script file
 | 
			
		||||
///
 | 
			
		||||
/// # Returns
 | 
			
		||||
///
 | 
			
		||||
/// * `Ok(Dynamic)` - The result of script execution
 | 
			
		||||
/// * `Err(Box<EvalAltResult>)` - File reading, compilation, or execution error
 | 
			
		||||
///
 | 
			
		||||
/// # Example
 | 
			
		||||
///
 | 
			
		||||
/// ```rust
 | 
			
		||||
/// use baobab_actor::engine::{create_heromodels_engine, eval_file};
 | 
			
		||||
/// use std::path::Path;
 | 
			
		||||
///
 | 
			
		||||
/// let engine = create_heromodels_engine();
 | 
			
		||||
/// let result = eval_file(&engine, Path::new("scripts/business_logic.rhai"))?;
 | 
			
		||||
/// println!("Script result: {:?}", result);
 | 
			
		||||
/// ```
 | 
			
		||||
///
 | 
			
		||||
/// # Error Handling
 | 
			
		||||
///
 | 
			
		||||
/// File reading errors are converted to Rhai `ErrorSystem` variants with
 | 
			
		||||
/// descriptive messages including the file path that failed to load.
 | 
			
		||||
pub fn eval_file(
 | 
			
		||||
    engine: &Engine,
 | 
			
		||||
    file_path: &Path,
 | 
			
		||||
) -> Result<rhai::Dynamic, Box<rhai::EvalAltResult>> {
 | 
			
		||||
    let script_content = fs::read_to_string(file_path).map_err(|e| {
 | 
			
		||||
        Box::new(EvalAltResult::ErrorSystem(
 | 
			
		||||
            format!("Failed to read script file '{}': {}", file_path.display(), e),
 | 
			
		||||
            e.into(),
 | 
			
		||||
        ))
 | 
			
		||||
    })?;
 | 
			
		||||
    
 | 
			
		||||
    engine.eval(&script_content)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Compiles a Rhai script string into an Abstract Syntax Tree (AST).
 | 
			
		||||
///
 | 
			
		||||
/// This function compiles a Rhai script into an AST that can be executed multiple
 | 
			
		||||
/// times with different scopes. This is more efficient than re-parsing the script
 | 
			
		||||
/// for each execution when the same script needs to be run repeatedly.
 | 
			
		||||
///
 | 
			
		||||
/// # Arguments
 | 
			
		||||
///
 | 
			
		||||
/// * `engine` - The Rhai engine to use for compilation
 | 
			
		||||
/// * `script` - The Rhai script content as a string
 | 
			
		||||
///
 | 
			
		||||
/// # Returns
 | 
			
		||||
///
 | 
			
		||||
/// * `Ok(AST)` - The compiled Abstract Syntax Tree
 | 
			
		||||
/// * `Err(Box<EvalAltResult>)` - Script compilation error
 | 
			
		||||
///
 | 
			
		||||
/// # Example
 | 
			
		||||
///
 | 
			
		||||
/// ```rust
 | 
			
		||||
/// use baobab_actor::engine::{create_heromodels_engine, compile_script, run_ast};
 | 
			
		||||
/// use rhai::Scope;
 | 
			
		||||
///
 | 
			
		||||
/// let engine = create_heromodels_engine();
 | 
			
		||||
/// let ast = compile_script(&engine, r#"
 | 
			
		||||
///     let company = new_company().name(company_name);
 | 
			
		||||
///     save_company(company)
 | 
			
		||||
/// "#)?;
 | 
			
		||||
///
 | 
			
		||||
/// // Execute the compiled script multiple times with different variables
 | 
			
		||||
/// let mut scope1 = Scope::new();
 | 
			
		||||
/// scope1.push("company_name", "Acme Corp");
 | 
			
		||||
/// let result1 = run_ast(&engine, &ast, &mut scope1)?;
 | 
			
		||||
///
 | 
			
		||||
/// let mut scope2 = Scope::new();
 | 
			
		||||
/// scope2.push("company_name", "Tech Startup");
 | 
			
		||||
/// let result2 = run_ast(&engine, &ast, &mut scope2)?;
 | 
			
		||||
/// ```
 | 
			
		||||
pub fn compile_script(engine: &Engine, script: &str) -> Result<AST, Box<rhai::EvalAltResult>> {
 | 
			
		||||
    Ok(engine.compile(script)?)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Executes a compiled Rhai script AST with the provided scope.
 | 
			
		||||
///
 | 
			
		||||
/// This function runs a pre-compiled AST using the provided engine and scope.
 | 
			
		||||
/// The scope can contain variables and functions that will be available to
 | 
			
		||||
/// the script during execution.
 | 
			
		||||
///
 | 
			
		||||
/// # Arguments
 | 
			
		||||
///
 | 
			
		||||
/// * `engine` - The Rhai engine to use for execution
 | 
			
		||||
/// * `ast` - The compiled Abstract Syntax Tree to execute
 | 
			
		||||
/// * `scope` - Mutable scope containing variables and functions for the script
 | 
			
		||||
///
 | 
			
		||||
/// # Returns
 | 
			
		||||
///
 | 
			
		||||
/// * `Ok(Dynamic)` - The result of script execution
 | 
			
		||||
/// * `Err(Box<EvalAltResult>)` - Script execution error
 | 
			
		||||
///
 | 
			
		||||
/// # Example
 | 
			
		||||
///
 | 
			
		||||
/// ```rust
 | 
			
		||||
/// use baobab_actor::engine::{create_heromodels_engine, compile_script, run_ast};
 | 
			
		||||
/// use rhai::Scope;
 | 
			
		||||
///
 | 
			
		||||
/// let engine = create_heromodels_engine();
 | 
			
		||||
/// let ast = compile_script(&engine, "x + y")?;
 | 
			
		||||
///
 | 
			
		||||
/// let mut scope = Scope::new();
 | 
			
		||||
/// scope.push("x", 10_i64);
 | 
			
		||||
/// scope.push("y", 32_i64);
 | 
			
		||||
///
 | 
			
		||||
/// let result = run_ast(&engine, &ast, &mut scope)?;
 | 
			
		||||
/// assert_eq!(result.as_int().unwrap(), 42);
 | 
			
		||||
/// ```
 | 
			
		||||
///
 | 
			
		||||
/// # Performance Notes
 | 
			
		||||
///
 | 
			
		||||
/// Using compiled ASTs is significantly more efficient than re-parsing scripts
 | 
			
		||||
/// for repeated execution, especially for complex scripts or when executing
 | 
			
		||||
/// the same logic with different input parameters.
 | 
			
		||||
pub fn run_ast(
 | 
			
		||||
    engine: &Engine,
 | 
			
		||||
    ast: &AST,
 | 
			
		||||
    scope: &mut Scope,
 | 
			
		||||
) -> Result<rhai::Dynamic, Box<rhai::EvalAltResult>> {
 | 
			
		||||
    engine.eval_ast_with_scope(scope, ast)
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										255
									
								
								_archive/core/actor/src/sync_worker.rs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										255
									
								
								_archive/core/actor/src/sync_worker.rs
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,255 @@
 | 
			
		||||
//! # Synchronous Actor Implementation
 | 
			
		||||
//!
 | 
			
		||||
//! This module provides a synchronous actor implementation that processes jobs
 | 
			
		||||
//! one at a time in sequence. This is the original actor behavior that's suitable
 | 
			
		||||
//! for scenarios where job execution should not overlap or when resource constraints
 | 
			
		||||
//! require sequential processing.
 | 
			
		||||
//!
 | 
			
		||||
//! ## Features
 | 
			
		||||
//!
 | 
			
		||||
//! - **Sequential Processing**: Jobs are processed one at a time
 | 
			
		||||
//! - **Simple Resource Management**: No concurrent job tracking needed
 | 
			
		||||
//! - **Predictable Behavior**: Jobs complete in the order they're received
 | 
			
		||||
//! - **Lower Memory Usage**: Only one job active at a time
 | 
			
		||||
//!
 | 
			
		||||
//! ## Usage
 | 
			
		||||
//!
 | 
			
		||||
//! ```rust
 | 
			
		||||
//! use std::sync::Arc;
 | 
			
		||||
//! use baobab_actor::sync_actor::SyncActor;
 | 
			
		||||
//! use baobab_actor::actor_trait::{spawn_actor, ActorConfig};
 | 
			
		||||
//! use baobab_actor::engine::create_heromodels_engine;
 | 
			
		||||
//! use tokio::sync::mpsc;
 | 
			
		||||
//!
 | 
			
		||||
//! let config = ActorConfig::new(
 | 
			
		||||
//!     "sync_actor_1".to_string(),
 | 
			
		||||
//!     "/path/to/db".to_string(),
 | 
			
		||||
//!     "redis://localhost:6379".to_string(),
 | 
			
		||||
//!     false, // preserve_tasks
 | 
			
		||||
//! );
 | 
			
		||||
//!
 | 
			
		||||
//! let actor = Arc::new(SyncActor::new());
 | 
			
		||||
//! let engine = create_heromodels_engine();
 | 
			
		||||
//! let (shutdown_tx, shutdown_rx) = mpsc::channel(1);
 | 
			
		||||
//!
 | 
			
		||||
//! let handle = spawn_actor(actor, config, engine, shutdown_rx);
 | 
			
		||||
//!
 | 
			
		||||
//! // Later, shutdown the actor
 | 
			
		||||
//! shutdown_tx.send(()).await.unwrap();
 | 
			
		||||
//! handle.await.unwrap().unwrap();
 | 
			
		||||
//! ```
 | 
			
		||||
 | 
			
		||||
use async_trait::async_trait;
 | 
			
		||||
use hero_job::{Job, JobStatus};
 | 
			
		||||
use log::{debug, error, info};
 | 
			
		||||
use rhai::Engine;
 | 
			
		||||
 | 
			
		||||
use crate::engine::eval_script;
 | 
			
		||||
use crate::actor_trait::{Actor, ActorConfig};
 | 
			
		||||
 | 
			
		||||
/// Builder for SyncActor
 | 
			
		||||
#[derive(Debug, Default)]
 | 
			
		||||
pub struct SyncActorBuilder {
 | 
			
		||||
    actor_id: Option<String>,
 | 
			
		||||
    db_path: Option<String>,
 | 
			
		||||
    redis_url: Option<String>,
 | 
			
		||||
    preserve_tasks: bool,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl SyncActorBuilder {
 | 
			
		||||
    pub fn new() -> Self {
 | 
			
		||||
        Self::default()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn actor_id<S: Into<String>>(mut self, actor_id: S) -> Self {
 | 
			
		||||
        self.actor_id = Some(actor_id.into());
 | 
			
		||||
        self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn db_path<S: Into<String>>(mut self, db_path: S) -> Self {
 | 
			
		||||
        self.db_path = Some(db_path.into());
 | 
			
		||||
        self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn redis_url<S: Into<String>>(mut self, redis_url: S) -> Self {
 | 
			
		||||
        self.redis_url = Some(redis_url.into());
 | 
			
		||||
        self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn preserve_tasks(mut self, preserve: bool) -> Self {
 | 
			
		||||
        self.preserve_tasks = preserve;
 | 
			
		||||
        self
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    pub fn build(self) -> Result<SyncActor, String> {
 | 
			
		||||
        Ok(SyncActor {
 | 
			
		||||
            actor_id: self.actor_id.ok_or("actor_id is required")?,
 | 
			
		||||
            db_path: self.db_path.ok_or("db_path is required")?,
 | 
			
		||||
            redis_url: self.redis_url.ok_or("redis_url is required")?,
 | 
			
		||||
            preserve_tasks: self.preserve_tasks,
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/// Synchronous actor that processes jobs sequentially
 | 
			
		||||
#[derive(Debug, Clone)]
 | 
			
		||||
pub struct SyncActor {
 | 
			
		||||
    pub actor_id: String,
 | 
			
		||||
    pub db_path: String,
 | 
			
		||||
    pub redis_url: String,
 | 
			
		||||
    pub preserve_tasks: bool,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl SyncActor {
 | 
			
		||||
    /// Create a new SyncActorBuilder
 | 
			
		||||
    pub fn builder() -> SyncActorBuilder {
 | 
			
		||||
        SyncActorBuilder::new()
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
impl Default for SyncActor {
 | 
			
		||||
    fn default() -> Self {
 | 
			
		||||
        // Default SyncActor with placeholder values
 | 
			
		||||
        // In practice, use the builder pattern instead
 | 
			
		||||
        Self {
 | 
			
		||||
            actor_id: "default_sync_actor".to_string(),
 | 
			
		||||
            db_path: "/tmp".to_string(),
 | 
			
		||||
            redis_url: "redis://localhost:6379".to_string(),
 | 
			
		||||
            preserve_tasks: false,
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[async_trait]
 | 
			
		||||
impl Actor for SyncActor {
 | 
			
		||||
    async fn process_job(
 | 
			
		||||
        &self,
 | 
			
		||||
        job: Job,
 | 
			
		||||
        engine: Engine,
 | 
			
		||||
        redis_conn: &mut redis::aio::MultiplexedConnection,
 | 
			
		||||
    ) {
 | 
			
		||||
        let job_id = &job.id;
 | 
			
		||||
        let actor_id = &self.actor_id;
 | 
			
		||||
        let db_path = &self.db_path;
 | 
			
		||||
        
 | 
			
		||||
        info!("Sync Actor '{}', Job {}: Starting sequential processing", actor_id, job_id);
 | 
			
		||||
 | 
			
		||||
        // Update job status to Started
 | 
			
		||||
        if let Err(e) = Job::update_status(redis_conn, job_id, JobStatus::Started).await {
 | 
			
		||||
            error!("Sync Actor '{}', Job {}: Failed to update status to Started: {}", 
 | 
			
		||||
                   actor_id, job_id, e);
 | 
			
		||||
            return;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // Execute the Rhai script
 | 
			
		||||
        match eval_script(&engine, &job.script) {
 | 
			
		||||
            Ok(result) => {
 | 
			
		||||
                let result_str = format!("{:?}", result);
 | 
			
		||||
                info!("Sync Actor '{}', Job {}: Script executed successfully. Result: {}", 
 | 
			
		||||
                      actor_id, job_id, result_str);
 | 
			
		||||
                
 | 
			
		||||
                // Update job with success result
 | 
			
		||||
                if let Err(e) = Job::set_result(redis_conn, job_id, &result_str).await {
 | 
			
		||||
                    error!("Sync Actor '{}', Job {}: Failed to set result: {}", 
 | 
			
		||||
                           actor_id, job_id, e);
 | 
			
		||||
                    return;
 | 
			
		||||
                }
 | 
			
		||||
                
 | 
			
		||||
                if let Err(e) = Job::update_status(redis_conn, job_id, JobStatus::Finished).await {
 | 
			
		||||
                    error!("Sync Actor '{}', Job {}: Failed to update status to Finished: {}", 
 | 
			
		||||
                           actor_id, job_id, e);
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
            Err(e) => {
 | 
			
		||||
                let error_msg = format!("Script execution error: {}", e);
 | 
			
		||||
                error!("Sync Actor '{}', Job {}: {}", actor_id, job_id, error_msg);
 | 
			
		||||
                
 | 
			
		||||
                // Update job with error
 | 
			
		||||
                if let Err(e) = Job::set_error(redis_conn, job_id, &error_msg).await {
 | 
			
		||||
                    error!("Sync Actor '{}', Job {}: Failed to set error: {}", 
 | 
			
		||||
                           actor_id, job_id, e);
 | 
			
		||||
                    return;
 | 
			
		||||
                }
 | 
			
		||||
                
 | 
			
		||||
                if let Err(e) = Job::update_status(redis_conn, job_id, JobStatus::Error).await {
 | 
			
		||||
                    error!("Sync Actor '{}', Job {}: Failed to update status to Error: {}", 
 | 
			
		||||
                           actor_id, job_id, e);
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // Cleanup job if preserve_tasks is false
 | 
			
		||||
        if !self.preserve_tasks {
 | 
			
		||||
            if let Err(e) = Job::delete_from_redis(redis_conn, job_id).await {
 | 
			
		||||
                error!("Sync Actor '{}', Job {}: Failed to cleanup job: {}", 
 | 
			
		||||
                       actor_id, job_id, e);
 | 
			
		||||
            } else {
 | 
			
		||||
                debug!("Sync Actor '{}', Job {}: Job cleaned up from Redis", actor_id, job_id);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        info!("Sync Actor '{}', Job {}: Sequential processing completed", actor_id, job_id);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fn actor_type(&self) -> &'static str {
 | 
			
		||||
        "Sync"
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fn actor_id(&self) -> &str {
 | 
			
		||||
        &self.actor_id
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    fn redis_url(&self) -> &str {
 | 
			
		||||
        &self.redis_url
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
#[cfg(test)]
 | 
			
		||||
mod tests {
 | 
			
		||||
    use super::*;
 | 
			
		||||
    use crate::engine::create_heromodels_engine;
 | 
			
		||||
    use hero_job::ScriptType;
 | 
			
		||||
    use std::time::Duration;
 | 
			
		||||
 | 
			
		||||
    #[tokio::test]
 | 
			
		||||
    async fn test_sync_actor_creation() {
 | 
			
		||||
        let actor = SyncActor::new();
 | 
			
		||||
        assert_eq!(actor.actor_type(), "Sync");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[tokio::test]
 | 
			
		||||
    async fn test_sync_actor_default() {
 | 
			
		||||
        let actor = SyncActor::default();
 | 
			
		||||
        assert_eq!(actor.actor_type(), "Sync");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    #[tokio::test]
 | 
			
		||||
    async fn test_sync_actor_process_job_interface() {
 | 
			
		||||
        let actor = SyncActor::new();
 | 
			
		||||
        let engine = create_heromodels_engine();
 | 
			
		||||
        
 | 
			
		||||
        // Create a simple test job
 | 
			
		||||
        let job = Job::new(
 | 
			
		||||
            "test_caller".to_string(),
 | 
			
		||||
            "test_context".to_string(),
 | 
			
		||||
            r#"print("Hello from sync actor test!"); 42"#.to_string(),
 | 
			
		||||
            ScriptType::OSIS,
 | 
			
		||||
        );
 | 
			
		||||
 | 
			
		||||
        let config = ActorConfig::new(
 | 
			
		||||
            "test_sync_actor".to_string(),
 | 
			
		||||
            "/tmp".to_string(),
 | 
			
		||||
            "redis://localhost:6379".to_string(),
 | 
			
		||||
            false,
 | 
			
		||||
        );
 | 
			
		||||
 | 
			
		||||
        // Note: This test doesn't actually connect to Redis, it just tests the interface
 | 
			
		||||
        // In a real test environment, you'd need a Redis instance or mock
 | 
			
		||||
        
 | 
			
		||||
        // The process_job method should be callable (interface test)
 | 
			
		||||
        // actor.process_job(job, engine, &mut redis_conn, &config).await;
 | 
			
		||||
        
 | 
			
		||||
        // For now, just verify the actor was created successfully
 | 
			
		||||
        assert_eq!(actor.actor_type(), "Sync");
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
		Reference in New Issue
	
	Block a user