108 lines
3.1 KiB
Rust
108 lines
3.1 KiB
Rust
use runner_rust::{spawn_sync_runner, script_mode::execute_script_mode};
|
|
use clap::Parser;
|
|
use log::{error, info};
|
|
use tokio::sync::mpsc;
|
|
|
|
mod engine;
|
|
use engine::create_osis_engine;
|
|
|
|
#[derive(Parser, Debug)]
|
|
#[command(author, version, about, long_about = None)]
|
|
struct Args {
|
|
/// Runner ID
|
|
runner_id: String,
|
|
|
|
/// Database path
|
|
#[arg(short, long, default_value = "/tmp/osis.db")]
|
|
db_path: String,
|
|
|
|
/// Redis URL
|
|
#[arg(short = 'r', long, default_value = "redis://localhost:6379")]
|
|
redis_url: String,
|
|
|
|
/// Preserve tasks after completion
|
|
#[arg(short, long, default_value_t = false)]
|
|
preserve_tasks: bool,
|
|
|
|
/// Script to execute in single-job mode (optional)
|
|
#[arg(short, long)]
|
|
script: Option<String>,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
|
|
// Initialize logging
|
|
env_logger::init();
|
|
|
|
let args = Args::parse();
|
|
|
|
// Check if we're in script mode
|
|
if let Some(script_content) = args.script {
|
|
info!("Running in script mode with runner ID: {}", args.runner_id);
|
|
|
|
let result = execute_script_mode(
|
|
&script_content,
|
|
&args.runner_id,
|
|
args.redis_url,
|
|
std::time::Duration::from_secs(300), // Default timeout for OSIS
|
|
create_osis_engine,
|
|
).await;
|
|
|
|
match result {
|
|
Ok(output) => {
|
|
println!("Script execution result:\n{}", output);
|
|
return Ok(());
|
|
}
|
|
Err(e) => {
|
|
error!("Script execution failed: {}", e);
|
|
return Err(e);
|
|
}
|
|
}
|
|
}
|
|
|
|
info!("Starting OSIS Sync Runner with ID: {}", args.runner_id);
|
|
info!("Database path: {}", args.db_path);
|
|
info!("Redis URL: {}", args.redis_url);
|
|
info!("Preserve tasks: {}", args.preserve_tasks);
|
|
|
|
// Create shutdown channel
|
|
let (shutdown_tx, shutdown_rx) = mpsc::channel::<()>(1);
|
|
|
|
// Setup signal handling for graceful shutdown
|
|
let shutdown_tx_clone = shutdown_tx.clone();
|
|
tokio::spawn(async move {
|
|
tokio::signal::ctrl_c().await.expect("Failed to listen for ctrl+c");
|
|
info!("Received Ctrl+C, initiating shutdown...");
|
|
let _ = shutdown_tx_clone.send(()).await;
|
|
});
|
|
|
|
// Spawn the sync runner with engine factory
|
|
let runner_handle = spawn_sync_runner(
|
|
args.runner_id.clone(),
|
|
args.db_path,
|
|
args.redis_url,
|
|
shutdown_rx,
|
|
args.preserve_tasks,
|
|
create_osis_engine,
|
|
);
|
|
|
|
info!("OSIS Sync Runner '{}' started successfully", args.runner_id);
|
|
|
|
// Wait for the runner to complete
|
|
match runner_handle.await {
|
|
Ok(Ok(())) => {
|
|
info!("OSIS Sync Runner '{}' shut down successfully", args.runner_id);
|
|
}
|
|
Ok(Err(e)) => {
|
|
error!("OSIS Sync Runner '{}' encountered an error: {}", args.runner_id, e);
|
|
return Err(e);
|
|
}
|
|
Err(e) => {
|
|
error!("Failed to join OSIS Sync Runner '{}' task: {}", args.runner_id, e);
|
|
return Err(Box::new(e));
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|