openzeppelin_relayer/queues/redis/

worker.rs

//! Redis/Apalis worker initialization.
//!
//! This module contains all Apalis-specific worker creation logic for the Redis
//! queue backend, including WorkerBuilder configurations, Monitor setup,
//! backoff strategies, and token swap cron workers.

use actix_web::web::ThinData;

use crate::{
    config::ServerConfig,
    constants::{
        SYSTEM_CLEANUP_CRON_SCHEDULE, TRANSACTION_CLEANUP_CRON_SCHEDULE,
        WORKER_SYSTEM_CLEANUP_RETRIES, WORKER_TOKEN_SWAP_REQUEST_RETRIES,
        WORKER_TRANSACTION_CLEANUP_RETRIES,
    },
    jobs::{
        notification_handler, relayer_health_check_handler, system_cleanup_handler,
        token_swap_cron_handler, token_swap_request_handler, transaction_cleanup_handler,
        transaction_request_handler, transaction_status_handler, transaction_submission_handler,
        Job, JobProducerTrait, NotificationSend, RelayerHealthCheck, SystemCleanupCronReminder,
        TokenSwapCronReminder, TokenSwapRequest, TransactionCleanupCronReminder,
        TransactionRequest, TransactionSend, TransactionStatusCheck,
    },
    models::{
        DefaultAppState, NetworkRepoModel, NotificationRepoModel, RelayerNetworkPolicy,
        RelayerRepoModel, SignerRepoModel, ThinDataAppState, TransactionRepoModel,
    },
    repositories::{
        ApiKeyRepositoryTrait, NetworkRepository, PluginRepositoryTrait, RelayerRepository,
        Repository, TransactionCounterTrait, TransactionRepository,
    },
};
use apalis::prelude::*;

use apalis::layers::retry::backoff::MakeBackoff;
use apalis::layers::retry::{backoff::ExponentialBackoffMaker, RetryPolicy};
use apalis::layers::ErrorHandlingLayer;

/// Re-exports from [`tower::util`]
pub use tower::util::rng::HasherRng;

use apalis_cron::CronStream;
use eyre::Result;
use std::{str::FromStr, time::Duration};
use tokio::signal::unix::SignalKind;
use tracing::{debug, error, info};

use super::{filter_relayers_for_swap, QueueType, WorkerContext};
use crate::queues::retry_config::{
    RetryBackoffConfig, NOTIFICATION_BACKOFF, RELAYER_HEALTH_BACKOFF, STATUS_EVM_BACKOFF,
    STATUS_GENERIC_BACKOFF, STATUS_STELLAR_BACKOFF, SYSTEM_CLEANUP_BACKOFF,
    TOKEN_SWAP_CRON_BACKOFF, TOKEN_SWAP_REQUEST_BACKOFF, TX_CLEANUP_BACKOFF, TX_REQUEST_BACKOFF,
    TX_SUBMISSION_BACKOFF,
};

// ---------------------------------------------------------------------------
// Apalis adapter functions
//
// These thin adapters are the ONLY place where Apalis-specific handler types
// (Data, Attempt, Worker<Context>, TaskId, RedisContext) appear. They convert
// Apalis types → WorkerContext and HandlerError → apalis::prelude::Error,
// keeping all handler business logic backend-neutral.
// ---------------------------------------------------------------------------

async fn apalis_transaction_request_handler(
    job: Job<TransactionRequest>,
    state: Data<ThinData<DefaultAppState>>,
    attempt: Attempt,
    task_id: TaskId,
) -> Result<(), apalis::prelude::Error> {
    let ctx = WorkerContext::new(attempt.current(), task_id.to_string());
    transaction_request_handler(job, (*state).clone(), ctx)
        .await
        .map_err(Into::into)
}

async fn apalis_transaction_submission_handler(
    job: Job<TransactionSend>,
    state: Data<ThinData<DefaultAppState>>,
    attempt: Attempt,
    task_id: TaskId,
) -> Result<(), apalis::prelude::Error> {
    let ctx = WorkerContext::new(attempt.current(), task_id.to_string());
    transaction_submission_handler(job, (*state).clone(), ctx)
        .await
        .map_err(Into::into)
}

async fn apalis_transaction_status_handler(
    job: Job<TransactionStatusCheck>,
    state: Data<ThinData<DefaultAppState>>,
    attempt: Attempt,
    task_id: TaskId,
) -> Result<(), apalis::prelude::Error> {
    let ctx = WorkerContext::new(attempt.current(), task_id.to_string());
    transaction_status_handler(job, (*state).clone(), ctx)
        .await
        .map_err(Into::into)
}

async fn apalis_notification_handler(
    job: Job<NotificationSend>,
    state: Data<ThinData<DefaultAppState>>,
    attempt: Attempt,
    task_id: TaskId,
) -> Result<(), apalis::prelude::Error> {
    let ctx = WorkerContext::new(attempt.current(), task_id.to_string());
    notification_handler(job, (*state).clone(), ctx)
        .await
        .map_err(Into::into)
}

async fn apalis_token_swap_request_handler(
    job: Job<TokenSwapRequest>,
    state: Data<ThinData<DefaultAppState>>,
    attempt: Attempt,
    task_id: TaskId,
) -> Result<(), apalis::prelude::Error> {
    let ctx = WorkerContext::new(attempt.current(), task_id.to_string());
    token_swap_request_handler(job, (*state).clone(), ctx)
        .await
        .map_err(Into::into)
}

async fn apalis_relayer_health_check_handler(
    job: Job<RelayerHealthCheck>,
    state: Data<ThinData<DefaultAppState>>,
    attempt: Attempt,
    task_id: TaskId,
) -> Result<(), apalis::prelude::Error> {
    let ctx = WorkerContext::new(attempt.current(), task_id.to_string());
    relayer_health_check_handler(job, (*state).clone(), ctx)
        .await
        .map_err(Into::into)
}

async fn apalis_transaction_cleanup_handler(
    _job: TransactionCleanupCronReminder,
    state: Data<ThinData<DefaultAppState>>,
    attempt: Attempt,
    task_id: TaskId,
) -> Result<(), apalis::prelude::Error> {
    let ctx = WorkerContext::new(attempt.current(), task_id.to_string());
    transaction_cleanup_handler(TransactionCleanupCronReminder(), (*state).clone(), ctx)
        .await
        .map_err(Into::into)
}

async fn apalis_system_cleanup_handler(
    _job: SystemCleanupCronReminder,
    state: Data<ThinData<DefaultAppState>>,
    attempt: Attempt,
    task_id: TaskId,
) -> Result<(), apalis::prelude::Error> {
    let ctx = WorkerContext::new(attempt.current(), task_id.to_string());
    system_cleanup_handler(SystemCleanupCronReminder(), (*state).clone(), ctx)
        .await
        .map_err(Into::into)
}

async fn apalis_token_swap_cron_handler(
    _job: TokenSwapCronReminder,
    relayer_id: Data<String>,
    state: Data<ThinData<DefaultAppState>>,
    attempt: Attempt,
    task_id: TaskId,
) -> Result<(), apalis::prelude::Error> {
    let ctx = WorkerContext::new(attempt.current(), task_id.to_string());
    token_swap_cron_handler(
        TokenSwapCronReminder(),
        (*relayer_id).clone(),
        (*state).clone(),
        ctx,
    )
    .await
    .map_err(Into::into)
}

const TRANSACTION_REQUEST: &str = "transaction_request";
const TRANSACTION_SENDER: &str = "transaction_sender";
// Generic transaction status checker
const TRANSACTION_STATUS_CHECKER: &str = "transaction_status_checker";
// Network specific status checkers
const TRANSACTION_STATUS_CHECKER_EVM: &str = "transaction_status_checker_evm";
const TRANSACTION_STATUS_CHECKER_STELLAR: &str = "transaction_status_checker_stellar";
const NOTIFICATION_SENDER: &str = "notification_sender";
const TOKEN_SWAP_REQUEST: &str = "token_swap_request";
const TRANSACTION_CLEANUP: &str = "transaction_cleanup";
const RELAYER_HEALTH_CHECK: &str = "relayer_health_check";
const SYSTEM_CLEANUP: &str = "system_cleanup";

/// Creates an exponential backoff with configurable parameters
///
/// # Arguments
/// * `initial_ms` - Initial delay in milliseconds (e.g., 200)
/// * `max_ms` - Maximum delay in milliseconds (e.g., 5000)
/// * `jitter` - Jitter factor 0.0-1.0 (e.g., 0.99 for high jitter)
///
/// # Returns
/// A configured backoff instance ready for use with RetryPolicy
fn create_backoff(initial_ms: u64, max_ms: u64, jitter: f64) -> Result<ExponentialBackoffMaker> {
    let maker = ExponentialBackoffMaker::new(
        Duration::from_millis(initial_ms),
        Duration::from_millis(max_ms),
        jitter,
        HasherRng::default(),
    )?;

    Ok(maker)
}

fn create_backoff_from_config(cfg: RetryBackoffConfig) -> Result<ExponentialBackoffMaker> {
    create_backoff(cfg.initial_ms, cfg.max_ms, cfg.jitter)
}

/// Initializes Redis/Apalis workers and starts the lifecycle monitor.
///
/// # Arguments
/// * `app_state` - Application state containing the job producer and configuration
pub async fn initialize_redis_workers<J, RR, TR, NR, NFR, SR, TCR, PR, AKR>(
    app_state: ThinDataAppState<J, RR, TR, NR, NFR, SR, TCR, PR, AKR>,
) -> Result<()>
where
    J: JobProducerTrait + Send + Sync + 'static,
    RR: RelayerRepository + Repository<RelayerRepoModel, String> + Send + Sync + 'static,
    TR: TransactionRepository + Repository<TransactionRepoModel, String> + Send + Sync + 'static,
    NR: NetworkRepository + Repository<NetworkRepoModel, String> + Send + Sync + 'static,
    NFR: Repository<NotificationRepoModel, String> + Send + Sync + 'static,
    SR: Repository<SignerRepoModel, String> + Send + Sync + 'static,
    TCR: TransactionCounterTrait + Send + Sync + 'static,
    PR: PluginRepositoryTrait + Send + Sync + 'static,
    AKR: ApiKeyRepositoryTrait + Send + Sync + 'static,
{
    let queue_backend = app_state
        .job_producer
        .get_queue_backend()
        .ok_or_else(|| eyre::eyre!("Queue backend is not available"))?;
    let queue = queue_backend
        .queue()
        .cloned()
        .ok_or_else(|| eyre::eyre!("Redis queue is not available for active backend"))?;

    let transaction_request_queue_worker = WorkerBuilder::new(TRANSACTION_REQUEST)
        .layer(ErrorHandlingLayer::new())
        .retry(
            RetryPolicy::retries(QueueType::TransactionRequest.max_retries())
                .with_backoff(create_backoff_from_config(TX_REQUEST_BACKOFF)?.make_backoff()),
        )
        .enable_tracing()
        .catch_panic()
        .concurrency(ServerConfig::get_worker_concurrency(
            QueueType::TransactionRequest.concurrency_env_key(),
            QueueType::TransactionRequest.default_concurrency(),
        ))
        .data(app_state.clone())
        .backend(queue.transaction_request_queue.clone())
        .build_fn(apalis_transaction_request_handler);

    let transaction_submission_queue_worker = WorkerBuilder::new(TRANSACTION_SENDER)
        .layer(ErrorHandlingLayer::new())
        .enable_tracing()
        .catch_panic()
        .retry(
            RetryPolicy::retries(QueueType::TransactionSubmission.max_retries())
                .with_backoff(create_backoff_from_config(TX_SUBMISSION_BACKOFF)?.make_backoff()),
        )
        .concurrency(ServerConfig::get_worker_concurrency(
            QueueType::TransactionSubmission.concurrency_env_key(),
            QueueType::TransactionSubmission.default_concurrency(),
        ))
        .data(app_state.clone())
        .backend(queue.transaction_submission_queue.clone())
        .build_fn(apalis_transaction_submission_handler);

    // Generic status checker
    // Uses medium settings that work reasonably for most chains
    let transaction_status_queue_worker = WorkerBuilder::new(TRANSACTION_STATUS_CHECKER)
        .layer(ErrorHandlingLayer::new())
        .enable_tracing()
        .catch_panic()
        .retry(
            RetryPolicy::retries(QueueType::StatusCheck.max_retries())
                .with_backoff(create_backoff_from_config(STATUS_GENERIC_BACKOFF)?.make_backoff()),
        )
        .concurrency(ServerConfig::get_worker_concurrency(
            QueueType::StatusCheck.concurrency_env_key(),
            QueueType::StatusCheck.default_concurrency(),
        ))
        .data(app_state.clone())
        .backend(queue.transaction_status_queue.clone())
        .build_fn(apalis_transaction_status_handler);

    // EVM status checker - slower retries to avoid premature resubmission
    // EVM has longer block times (~12s) and needs time for resubmission logic
    let transaction_status_queue_worker_evm = WorkerBuilder::new(TRANSACTION_STATUS_CHECKER_EVM)
        .layer(ErrorHandlingLayer::new())
        .enable_tracing()
        .catch_panic()
        .retry(
            RetryPolicy::retries(QueueType::StatusCheck.max_retries())
                .with_backoff(create_backoff_from_config(STATUS_EVM_BACKOFF)?.make_backoff()),
        )
        .concurrency(ServerConfig::get_worker_concurrency(
            QueueType::StatusCheckEvm.concurrency_env_key(),
            QueueType::StatusCheckEvm.default_concurrency(),
        ))
        .data(app_state.clone())
        .backend(queue.transaction_status_queue_evm.clone())
        .build_fn(apalis_transaction_status_handler);

    // Stellar status checker - fast retries for fast finality
    // Stellar has sub-second finality, needs more frequent status checks
    let transaction_status_queue_worker_stellar =
        WorkerBuilder::new(TRANSACTION_STATUS_CHECKER_STELLAR)
            .layer(ErrorHandlingLayer::new())
            .enable_tracing()
            .catch_panic()
            .retry(
                RetryPolicy::retries(QueueType::StatusCheckStellar.max_retries()).with_backoff(
                    create_backoff_from_config(STATUS_STELLAR_BACKOFF)?.make_backoff(),
                ),
            )
            .concurrency(ServerConfig::get_worker_concurrency(
                QueueType::StatusCheckStellar.concurrency_env_key(),
                QueueType::StatusCheckStellar.default_concurrency(),
            ))
            .data(app_state.clone())
            .backend(queue.transaction_status_queue_stellar.clone())
            .build_fn(apalis_transaction_status_handler);

    let notification_queue_worker = WorkerBuilder::new(NOTIFICATION_SENDER)
        .layer(ErrorHandlingLayer::new())
        .enable_tracing()
        .catch_panic()
        .retry(
            RetryPolicy::retries(QueueType::Notification.max_retries())
                .with_backoff(create_backoff_from_config(NOTIFICATION_BACKOFF)?.make_backoff()),
        )
        .concurrency(ServerConfig::get_worker_concurrency(
            QueueType::Notification.concurrency_env_key(),
            QueueType::Notification.default_concurrency(),
        ))
        .data(app_state.clone())
        .backend(queue.notification_queue.clone())
        .build_fn(apalis_notification_handler);

    let token_swap_request_queue_worker = WorkerBuilder::new(TOKEN_SWAP_REQUEST)
        .layer(ErrorHandlingLayer::new())
        .enable_tracing()
        .catch_panic()
        .retry(
            RetryPolicy::retries(QueueType::TokenSwapRequest.max_retries()).with_backoff(
                create_backoff_from_config(TOKEN_SWAP_REQUEST_BACKOFF)?.make_backoff(),
            ),
        )
        .concurrency(ServerConfig::get_worker_concurrency(
            QueueType::TokenSwapRequest.concurrency_env_key(),
            QueueType::TokenSwapRequest.default_concurrency(),
        ))
        .data(app_state.clone())
        .backend(queue.token_swap_request_queue.clone())
        .build_fn(apalis_token_swap_request_handler);

    let transaction_cleanup_queue_worker = WorkerBuilder::new(TRANSACTION_CLEANUP)
        .layer(ErrorHandlingLayer::new())
        .enable_tracing()
        .catch_panic()
        .retry(
            RetryPolicy::retries(WORKER_TRANSACTION_CLEANUP_RETRIES)
                .with_backoff(create_backoff_from_config(TX_CLEANUP_BACKOFF)?.make_backoff()),
        )
        .concurrency(ServerConfig::get_worker_concurrency(TRANSACTION_CLEANUP, 1)) // Default to 1 to avoid DB conflicts
        .data(app_state.clone())
        .backend(CronStream::new(
            apalis_cron::Schedule::from_str(TRANSACTION_CLEANUP_CRON_SCHEDULE)?,
        ))
        .build_fn(apalis_transaction_cleanup_handler);

    let system_cleanup_queue_worker = WorkerBuilder::new(SYSTEM_CLEANUP)
        .layer(ErrorHandlingLayer::new())
        .enable_tracing()
        .catch_panic()
        .retry(
            RetryPolicy::retries(WORKER_SYSTEM_CLEANUP_RETRIES)
                .with_backoff(create_backoff_from_config(SYSTEM_CLEANUP_BACKOFF)?.make_backoff()),
        )
        .concurrency(1)
        .data(app_state.clone())
        .backend(CronStream::new(apalis_cron::Schedule::from_str(
            SYSTEM_CLEANUP_CRON_SCHEDULE,
        )?))
        .build_fn(apalis_system_cleanup_handler);

    let relayer_health_check_worker = WorkerBuilder::new(RELAYER_HEALTH_CHECK)
        .layer(ErrorHandlingLayer::new())
        .enable_tracing()
        .catch_panic()
        .retry(
            RetryPolicy::retries(QueueType::RelayerHealthCheck.max_retries())
                .with_backoff(create_backoff_from_config(RELAYER_HEALTH_BACKOFF)?.make_backoff()),
        )
        .concurrency(ServerConfig::get_worker_concurrency(
            QueueType::RelayerHealthCheck.concurrency_env_key(),
            QueueType::RelayerHealthCheck.default_concurrency(),
        ))
        .data(app_state.clone())
        .backend(queue.relayer_health_check_queue.clone())
        .build_fn(apalis_relayer_health_check_handler);

    let monitor = Monitor::new()
        .register(transaction_request_queue_worker)
        .register(transaction_submission_queue_worker)
        .register(transaction_status_queue_worker)
        .register(transaction_status_queue_worker_evm)
        .register(transaction_status_queue_worker_stellar)
        .register(notification_queue_worker)
        .register(token_swap_request_queue_worker)
        .register(transaction_cleanup_queue_worker)
        .register(system_cleanup_queue_worker)
        .register(relayer_health_check_worker)
        .on_event(monitor_handle_event)
        .shutdown_timeout(Duration::from_millis(5000));

    let monitor_future = monitor.run_with_signal(async {
        let mut sigint = tokio::signal::unix::signal(SignalKind::interrupt())
            .map_err(|e| std::io::Error::other(format!("Failed to create SIGINT signal: {e}")))?;
        let mut sigterm = tokio::signal::unix::signal(SignalKind::terminate())
            .map_err(|e| std::io::Error::other(format!("Failed to create SIGTERM signal: {e}")))?;

        debug!("Workers monitor started");

        tokio::select! {
            _ = sigint.recv() => debug!("Received SIGINT."),
            _ = sigterm.recv() => debug!("Received SIGTERM."),
        };

        debug!("Workers monitor shutting down");

        Ok(())
    });
    tokio::spawn(async move {
        if let Err(e) = monitor_future.await {
            error!(error = %e, "monitor error");
        }
    });
    debug!("Workers monitor shutdown complete");

    Ok(())
}

/// Initializes swap workers for Solana and Stellar relayers.
/// This function creates and registers workers for relayers that have swap enabled and cron schedule set.
pub async fn initialize_redis_token_swap_workers<J, RR, TR, NR, NFR, SR, TCR, PR, AKR>(
    app_state: ThinDataAppState<J, RR, TR, NR, NFR, SR, TCR, PR, AKR>,
) -> Result<()>
where
    J: JobProducerTrait + Send + Sync + 'static,
    RR: RelayerRepository + Repository<RelayerRepoModel, String> + Send + Sync + 'static,
    TR: TransactionRepository + Repository<TransactionRepoModel, String> + Send + Sync + 'static,
    NR: NetworkRepository + Repository<NetworkRepoModel, String> + Send + Sync + 'static,
    NFR: Repository<NotificationRepoModel, String> + Send + Sync + 'static,
    SR: Repository<SignerRepoModel, String> + Send + Sync + 'static,
    TCR: TransactionCounterTrait + Send + Sync + 'static,
    PR: PluginRepositoryTrait + Send + Sync + 'static,
    AKR: ApiKeyRepositoryTrait + Send + Sync + 'static,
{
    let active_relayers = app_state.relayer_repository.list_active().await?;
    let relayers_with_swap_enabled = filter_relayers_for_swap(active_relayers);

    if relayers_with_swap_enabled.is_empty() {
        debug!("No relayers with swap enabled");
        return Ok(());
    }
    info!(
        "Found {} relayers with swap enabled",
        relayers_with_swap_enabled.len()
    );

    let mut workers = Vec::new();

    let swap_backoff = create_backoff_from_config(TOKEN_SWAP_CRON_BACKOFF)?.make_backoff();

    for relayer in relayers_with_swap_enabled {
        debug!(relayer = ?relayer, "found relayer with swap enabled");

        let (cron_schedule, network_type) = match &relayer.policies {
            RelayerNetworkPolicy::Solana(policy) => match policy.get_swap_config() {
                Some(config) => match config.cron_schedule {
                    Some(schedule) => (schedule, "solana".to_string()),
                    None => {
                        debug!(relayer_id = %relayer.id, "No cron schedule specified for Solana relayer; skipping");
                        continue;
                    }
                },
                None => {
                    debug!(relayer_id = %relayer.id, "No swap configuration specified for Solana relayer; skipping");
                    continue;
                }
            },
            RelayerNetworkPolicy::Stellar(policy) => match policy.get_swap_config() {
                Some(config) => match config.cron_schedule {
                    Some(schedule) => (schedule, "stellar".to_string()),
                    None => {
                        debug!(relayer_id = %relayer.id, "No cron schedule specified for Stellar relayer; skipping");
                        continue;
                    }
                },
                None => {
                    debug!(relayer_id = %relayer.id, "No swap configuration specified for Stellar relayer; skipping");
                    continue;
                }
            },
            RelayerNetworkPolicy::Evm(_) => {
                debug!(relayer_id = %relayer.id, "EVM relayers do not support swap; skipping");
                continue;
            }
        };

        let calendar_schedule = match apalis_cron::Schedule::from_str(&cron_schedule) {
            Ok(schedule) => schedule,
            Err(e) => {
                error!(relayer_id = %relayer.id, error = %e, "Failed to parse cron schedule; skipping");
                continue;
            }
        };

        // Create worker and add to the workers vector
        let worker = WorkerBuilder::new(format!(
            "{}-swap-schedule-{}",
            network_type,
            relayer.id.clone()
        ))
        .layer(ErrorHandlingLayer::new())
        .enable_tracing()
        .catch_panic()
        .retry(
            RetryPolicy::retries(WORKER_TOKEN_SWAP_REQUEST_RETRIES)
                .with_backoff(swap_backoff.clone()),
        )
        .concurrency(1)
        .data(relayer.id.clone())
        .data(app_state.clone())
        .backend(CronStream::new(calendar_schedule))
        .build_fn(apalis_token_swap_cron_handler);

        workers.push(worker);
        debug!(
            relayer_id = %relayer.id,
            network_type = %network_type,
            "Created worker for relayer with swap enabled"
        );
    }

    let mut monitor = Monitor::new()
        .on_event(monitor_handle_event)
        .shutdown_timeout(Duration::from_millis(5000));

    // Register all workers with the monitor
    for worker in workers {
        monitor = monitor.register(worker);
    }

    let monitor_future = monitor.run_with_signal(async {
        let mut sigint = tokio::signal::unix::signal(SignalKind::interrupt())
            .map_err(|e| std::io::Error::other(format!("Failed to create SIGINT signal: {e}")))?;
        let mut sigterm = tokio::signal::unix::signal(SignalKind::terminate())
            .map_err(|e| std::io::Error::other(format!("Failed to create SIGTERM signal: {e}")))?;

        debug!("Swap Monitor started");

        tokio::select! {
            _ = sigint.recv() => debug!("Received SIGINT."),
            _ = sigterm.recv() => debug!("Received SIGTERM."),
        };

        debug!("Swap Monitor shutting down");

        Ok(())
    });
    tokio::spawn(async move {
        if let Err(e) = monitor_future.await {
            error!(error = %e, "monitor error");
        }
    });
    Ok(())
}

fn monitor_handle_event(e: Worker<Event>) {
    let worker_id = e.id();
    match e.inner() {
        Event::Engage(task_id) => {
            debug!(worker_id = %worker_id, task_id = %task_id, "worker got a job");
        }
        Event::Error(e) => {
            error!(worker_id = %worker_id, error = %e, "worker encountered an error");
        }
        Event::Exit => {
            debug!(worker_id = %worker_id, "worker exited");
        }
        Event::Idle => {
            debug!(worker_id = %worker_id, "worker is idle");
        }
        Event::Start => {
            debug!(worker_id = %worker_id, "worker started");
        }
        Event::Stop => {
            debug!(worker_id = %worker_id, "worker stopped");
        }
        _ => {}
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::queues::retry_config::{
        NOTIFICATION_BACKOFF, RELAYER_HEALTH_BACKOFF, STATUS_EVM_BACKOFF, STATUS_GENERIC_BACKOFF,
        STATUS_STELLAR_BACKOFF, SYSTEM_CLEANUP_BACKOFF, TOKEN_SWAP_CRON_BACKOFF,
        TOKEN_SWAP_REQUEST_BACKOFF, TX_CLEANUP_BACKOFF, TX_REQUEST_BACKOFF, TX_SUBMISSION_BACKOFF,
    };

    // ── create_backoff tests ───────────────────────────────────────────

    #[test]
    fn test_create_backoff_with_valid_parameters() {
        let result = create_backoff(200, 5000, 0.99);
        assert!(
            result.is_ok(),
            "Should create backoff with valid parameters"
        );
    }

    #[test]
    fn test_create_backoff_with_zero_initial() {
        let result = create_backoff(0, 5000, 0.99);
        assert!(
            result.is_ok(),
            "Should handle zero initial delay (edge case)"
        );
    }

    #[test]
    fn test_create_backoff_with_equal_initial_and_max() {
        let result = create_backoff(1000, 1000, 0.5);
        assert!(result.is_ok(), "Should handle equal initial and max delays");
    }

    #[test]
    fn test_create_backoff_with_zero_jitter() {
        let result = create_backoff(500, 5000, 0.0);
        assert!(result.is_ok(), "Should handle zero jitter");
    }

    #[test]
    fn test_create_backoff_with_max_jitter() {
        let result = create_backoff(500, 5000, 1.0);
        assert!(result.is_ok(), "Should handle maximum jitter (1.0)");
    }

    #[test]
    fn test_create_backoff_with_small_values() {
        let result = create_backoff(1, 10, 0.5);
        assert!(result.is_ok(), "Should handle very small delay values");
    }

    #[test]
    fn test_create_backoff_with_large_values() {
        let result = create_backoff(10000, 60000, 0.99);
        assert!(result.is_ok(), "Should handle large delay values");
    }

    #[test]
    fn test_create_backoff_from_config_profiles() {
        let profiles = [
            TX_REQUEST_BACKOFF,
            TX_SUBMISSION_BACKOFF,
            STATUS_GENERIC_BACKOFF,
            STATUS_EVM_BACKOFF,
            STATUS_STELLAR_BACKOFF,
            NOTIFICATION_BACKOFF,
            TOKEN_SWAP_REQUEST_BACKOFF,
            TX_CLEANUP_BACKOFF,
            SYSTEM_CLEANUP_BACKOFF,
            RELAYER_HEALTH_BACKOFF,
            TOKEN_SWAP_CRON_BACKOFF,
        ];

        for cfg in profiles {
            let result = create_backoff_from_config(cfg);
            assert!(
                result.is_ok(),
                "backoff profile should be constructible: {:?}",
                cfg
            );
        }
    }

    #[test]
    fn test_create_backoff_from_config_produces_usable_backoff() {
        let profiles = [
            TX_REQUEST_BACKOFF,
            TX_SUBMISSION_BACKOFF,
            STATUS_GENERIC_BACKOFF,
            STATUS_EVM_BACKOFF,
            STATUS_STELLAR_BACKOFF,
            NOTIFICATION_BACKOFF,
            TOKEN_SWAP_REQUEST_BACKOFF,
            TX_CLEANUP_BACKOFF,
            SYSTEM_CLEANUP_BACKOFF,
            RELAYER_HEALTH_BACKOFF,
            TOKEN_SWAP_CRON_BACKOFF,
        ];

        for cfg in profiles {
            let mut maker = create_backoff_from_config(cfg).unwrap();
            // Calling make_backoff() should not panic
            let _backoff = maker.make_backoff();
        }
    }

    #[test]
    fn test_create_backoff_with_initial_greater_than_max_errors() {
        let result = create_backoff(10000, 100, 0.5);
        assert!(
            result.is_err(),
            "initial > max should be rejected by ExponentialBackoffMaker"
        );
    }

    // ── Backoff config invariant tests ─────────────────────────────────

    #[test]
    fn test_all_backoff_configs_have_valid_initial_le_max() {
        let profiles: &[(&str, RetryBackoffConfig)] = &[
            ("TX_REQUEST", TX_REQUEST_BACKOFF),
            ("TX_SUBMISSION", TX_SUBMISSION_BACKOFF),
            ("STATUS_GENERIC", STATUS_GENERIC_BACKOFF),
            ("STATUS_EVM", STATUS_EVM_BACKOFF),
            ("STATUS_STELLAR", STATUS_STELLAR_BACKOFF),
            ("NOTIFICATION", NOTIFICATION_BACKOFF),
            ("TOKEN_SWAP_REQUEST", TOKEN_SWAP_REQUEST_BACKOFF),
            ("TX_CLEANUP", TX_CLEANUP_BACKOFF),
            ("SYSTEM_CLEANUP", SYSTEM_CLEANUP_BACKOFF),
            ("RELAYER_HEALTH", RELAYER_HEALTH_BACKOFF),
            ("TOKEN_SWAP_CRON", TOKEN_SWAP_CRON_BACKOFF),
        ];

        for (name, cfg) in profiles {
            assert!(
                cfg.initial_ms <= cfg.max_ms,
                "{name}: initial_ms ({}) must be <= max_ms ({})",
                cfg.initial_ms,
                cfg.max_ms
            );
        }
    }

    #[test]
    fn test_all_backoff_configs_have_valid_jitter_range() {
        let profiles: &[(&str, RetryBackoffConfig)] = &[
            ("TX_REQUEST", TX_REQUEST_BACKOFF),
            ("TX_SUBMISSION", TX_SUBMISSION_BACKOFF),
            ("STATUS_GENERIC", STATUS_GENERIC_BACKOFF),
            ("STATUS_EVM", STATUS_EVM_BACKOFF),
            ("STATUS_STELLAR", STATUS_STELLAR_BACKOFF),
            ("NOTIFICATION", NOTIFICATION_BACKOFF),
            ("TOKEN_SWAP_REQUEST", TOKEN_SWAP_REQUEST_BACKOFF),
            ("TX_CLEANUP", TX_CLEANUP_BACKOFF),
            ("SYSTEM_CLEANUP", SYSTEM_CLEANUP_BACKOFF),
            ("RELAYER_HEALTH", RELAYER_HEALTH_BACKOFF),
            ("TOKEN_SWAP_CRON", TOKEN_SWAP_CRON_BACKOFF),
        ];

        for (name, cfg) in profiles {
            assert!(
                (0.0..=1.0).contains(&cfg.jitter),
                "{name}: jitter ({}) must be in [0.0, 1.0]",
                cfg.jitter
            );
        }
    }

    #[test]
    fn test_all_backoff_configs_have_positive_initial_ms() {
        let profiles: &[(&str, RetryBackoffConfig)] = &[
            ("TX_REQUEST", TX_REQUEST_BACKOFF),
            ("TX_SUBMISSION", TX_SUBMISSION_BACKOFF),
            ("STATUS_GENERIC", STATUS_GENERIC_BACKOFF),
            ("STATUS_EVM", STATUS_EVM_BACKOFF),
            ("STATUS_STELLAR", STATUS_STELLAR_BACKOFF),
            ("NOTIFICATION", NOTIFICATION_BACKOFF),
            ("TOKEN_SWAP_REQUEST", TOKEN_SWAP_REQUEST_BACKOFF),
            ("TX_CLEANUP", TX_CLEANUP_BACKOFF),
            ("SYSTEM_CLEANUP", SYSTEM_CLEANUP_BACKOFF),
            ("RELAYER_HEALTH", RELAYER_HEALTH_BACKOFF),
            ("TOKEN_SWAP_CRON", TOKEN_SWAP_CRON_BACKOFF),
        ];

        for (name, cfg) in profiles {
            assert!(
                cfg.initial_ms > 0,
                "{name}: initial_ms must be positive, got {}",
                cfg.initial_ms
            );
        }
    }

    // ── Worker name constant tests ─────────────────────────────────────

    #[test]
    fn test_worker_name_constants_are_nonempty() {
        let names = [
            TRANSACTION_REQUEST,
            TRANSACTION_SENDER,
            TRANSACTION_STATUS_CHECKER,
            TRANSACTION_STATUS_CHECKER_EVM,
            TRANSACTION_STATUS_CHECKER_STELLAR,
            NOTIFICATION_SENDER,
            TOKEN_SWAP_REQUEST,
            TRANSACTION_CLEANUP,
            RELAYER_HEALTH_CHECK,
            SYSTEM_CLEANUP,
        ];

        for name in &names {
            assert!(!name.is_empty(), "Worker name constant must not be empty");
        }
    }

    #[test]
    fn test_worker_name_constants_are_unique() {
        let names = [
            TRANSACTION_REQUEST,
            TRANSACTION_SENDER,
            TRANSACTION_STATUS_CHECKER,
            TRANSACTION_STATUS_CHECKER_EVM,
            TRANSACTION_STATUS_CHECKER_STELLAR,
            NOTIFICATION_SENDER,
            TOKEN_SWAP_REQUEST,
            TRANSACTION_CLEANUP,
            RELAYER_HEALTH_CHECK,
            SYSTEM_CLEANUP,
        ];

        for (i, a) in names.iter().enumerate() {
            for (j, b) in names.iter().enumerate() {
                if i != j {
                    assert_ne!(
                        a, b,
                        "Worker names must be unique: '{}' at index {} and {}",
                        a, i, j
                    );
                }
            }
        }
    }

    #[test]
    fn test_worker_names_match_concurrency_env_keys() {
        // The WorkerBuilder name for each queue-type-backed worker should match
        // the concurrency_env_key used with ServerConfig::get_worker_concurrency,
        // so that concurrency configuration picks up the correct env var.
        assert_eq!(
            TRANSACTION_REQUEST,
            QueueType::TransactionRequest.concurrency_env_key()
        );
        assert_eq!(
            TRANSACTION_SENDER,
            QueueType::TransactionSubmission.concurrency_env_key()
        );
        assert_eq!(
            TRANSACTION_STATUS_CHECKER,
            QueueType::StatusCheck.concurrency_env_key()
        );
        assert_eq!(
            TRANSACTION_STATUS_CHECKER_EVM,
            QueueType::StatusCheckEvm.concurrency_env_key()
        );
        assert_eq!(
            TRANSACTION_STATUS_CHECKER_STELLAR,
            QueueType::StatusCheckStellar.concurrency_env_key()
        );
        assert_eq!(
            NOTIFICATION_SENDER,
            QueueType::Notification.concurrency_env_key()
        );
        assert_eq!(
            TOKEN_SWAP_REQUEST,
            QueueType::TokenSwapRequest.concurrency_env_key()
        );
        assert_eq!(
            RELAYER_HEALTH_CHECK,
            QueueType::RelayerHealthCheck.concurrency_env_key()
        );
    }

    // ── monitor_handle_event tests ─────────────────────────────────────

    fn make_worker_event(event: Event) -> Worker<Event> {
        let worker_id = WorkerId::from_str("test-worker").unwrap();
        Worker::new(worker_id, event)
    }

    #[test]
    fn test_monitor_handle_event_start_does_not_panic() {
        monitor_handle_event(make_worker_event(Event::Start));
    }

    #[test]
    fn test_monitor_handle_event_engage_does_not_panic() {
        let task_id = TaskId::new();
        monitor_handle_event(make_worker_event(Event::Engage(task_id)));
    }

    #[test]
    fn test_monitor_handle_event_idle_does_not_panic() {
        monitor_handle_event(make_worker_event(Event::Idle));
    }

    #[test]
    fn test_monitor_handle_event_error_does_not_panic() {
        let error: Box<dyn std::error::Error + Send + Sync> = "test error".to_string().into();
        monitor_handle_event(make_worker_event(Event::Error(error)));
    }

    #[test]
    fn test_monitor_handle_event_stop_does_not_panic() {
        monitor_handle_event(make_worker_event(Event::Stop));
    }

    #[test]
    fn test_monitor_handle_event_exit_does_not_panic() {
        monitor_handle_event(make_worker_event(Event::Exit));
    }

    #[test]
    fn test_monitor_handle_event_custom_does_not_panic() {
        monitor_handle_event(make_worker_event(Event::Custom("test-custom".to_string())));
    }
}