@@ -1,75 +0,0 @@
|
||||
use std::path::PathBuf;
|
||||
use std::sync::Arc;
|
||||
|
||||
use crate::config::Config;
|
||||
use crate::scheduler::{Scheduler, SchedulerError};
|
||||
use crate::scheduler_trait::SchedulerTrait;
|
||||
use crate::temporal_scheduler::TemporalScheduler;
|
||||
|
||||
pub enum SchedulerType {
|
||||
Legacy,
|
||||
Temporal,
|
||||
}
|
||||
|
||||
impl SchedulerType {
|
||||
/// Determine scheduler type from configuration
|
||||
pub fn from_config() -> Self {
|
||||
let config = Config::global();
|
||||
match config.get_param::<String>("GOOSE_SCHEDULER_TYPE") {
|
||||
Ok(scheduler_type) => match scheduler_type.to_lowercase().as_str() {
|
||||
"temporal" => SchedulerType::Temporal,
|
||||
"legacy" => SchedulerType::Legacy,
|
||||
_ => {
|
||||
tracing::warn!(
|
||||
"Unknown scheduler type '{}', defaulting to legacy",
|
||||
scheduler_type
|
||||
);
|
||||
SchedulerType::Legacy
|
||||
}
|
||||
},
|
||||
Err(_) => {
|
||||
// Default to temporal scheduler
|
||||
SchedulerType::Temporal
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Factory for creating scheduler instances
|
||||
pub struct SchedulerFactory;
|
||||
|
||||
impl SchedulerFactory {
|
||||
/// Create a scheduler instance based on configuration
|
||||
pub async fn create(storage_path: PathBuf) -> Result<Arc<dyn SchedulerTrait>, SchedulerError> {
|
||||
let scheduler_type = SchedulerType::from_config();
|
||||
|
||||
match scheduler_type {
|
||||
SchedulerType::Legacy => {
|
||||
tracing::info!("Creating legacy scheduler");
|
||||
let scheduler = Scheduler::new(storage_path).await?;
|
||||
Ok(scheduler as Arc<dyn SchedulerTrait>)
|
||||
}
|
||||
SchedulerType::Temporal => {
|
||||
tracing::info!("Creating Temporal scheduler");
|
||||
let scheduler = TemporalScheduler::new().await?;
|
||||
Ok(scheduler as Arc<dyn SchedulerTrait>)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Create a specific scheduler type (for testing or explicit use)
|
||||
pub async fn create_legacy(
|
||||
storage_path: PathBuf,
|
||||
) -> Result<Arc<dyn SchedulerTrait>, SchedulerError> {
|
||||
tracing::info!("Creating legacy scheduler (explicit)");
|
||||
let scheduler = Scheduler::new(storage_path).await?;
|
||||
Ok(scheduler as Arc<dyn SchedulerTrait>)
|
||||
}
|
||||
|
||||
/// Create a Temporal scheduler (for testing or explicit use)
|
||||
pub async fn create_temporal() -> Result<Arc<dyn SchedulerTrait>, SchedulerError> {
|
||||
tracing::info!("Creating Temporal scheduler (explicit)");
|
||||
let scheduler = TemporalScheduler::new().await?;
|
||||
Ok(scheduler as Arc<dyn SchedulerTrait>)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user