camel-core 0.5.8

use std::sync::Arc;

use async_trait::async_trait;
use tokio::sync::Mutex;

use camel_api::{CamelError, RuntimeQueryResult};

use crate::lifecycle::adapters::route_controller::RouteControllerInternal;
use crate::lifecycle::application::RouteDefinition;
use crate::lifecycle::ports::RuntimeExecutionPort;

/// Runtime side-effect adapter backed by the technical route controller.
#[derive(Clone)]
pub struct RuntimeExecutionAdapter {
    controller: Arc<Mutex<dyn RouteControllerInternal>>,
}

impl RuntimeExecutionAdapter {
    pub fn new(controller: Arc<Mutex<dyn RouteControllerInternal>>) -> Self {
        Self { controller }
    }
}

#[async_trait]
impl RuntimeExecutionPort for RuntimeExecutionAdapter {
    async fn register_route(&self, definition: RouteDefinition) -> Result<(), CamelError> {
        let mut controller = self.controller.lock().await;
        controller.add_route(definition)
    }

    async fn start_route(&self, route_id: &str) -> Result<(), CamelError> {
        let mut controller = self.controller.lock().await;
        controller.start_route(route_id).await
    }

    async fn stop_route(&self, route_id: &str) -> Result<(), CamelError> {
        let mut controller = self.controller.lock().await;
        controller.stop_route(route_id).await
    }

    async fn suspend_route(&self, route_id: &str) -> Result<(), CamelError> {
        let mut controller = self.controller.lock().await;
        controller.suspend_route(route_id).await
    }

    async fn resume_route(&self, route_id: &str) -> Result<(), CamelError> {
        let mut controller = self.controller.lock().await;
        controller.resume_route(route_id).await
    }

    async fn reload_route(&self, route_id: &str) -> Result<(), CamelError> {
        let mut controller = self.controller.lock().await;
        controller.restart_route(route_id).await
    }

    async fn remove_route(&self, route_id: &str) -> Result<(), CamelError> {
        let mut controller = self.controller.lock().await;
        controller.remove_route(route_id)
    }

    async fn in_flight_count(&self, route_id: &str) -> Result<RuntimeQueryResult, CamelError> {
        let controller = self.controller.lock().await;
        Ok(match controller.in_flight_count(route_id) {
            Some(count) => RuntimeQueryResult::InFlightCount {
                route_id: route_id.to_string(),
                count,
            },
            None => RuntimeQueryResult::RouteNotFound {
                route_id: route_id.to_string(),
            },
        })
    }
}