reifydb-sub-task 0.4.9

Task execution subsystem for ReifyDB
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2025 ReifyDB

use reifydb_core::util::ioc::IocContainer;
use reifydb_sub_api::subsystem::{Subsystem, SubsystemFactory};
use reifydb_transaction::interceptor::builder::InterceptorBuilder;
use reifydb_type::Result;

use crate::{subsystem::TaskSubsystem, task::ScheduledTask};

/// Configuration for the task scheduler subsystem
#[derive(Default)]
pub struct TaskConfig {
	/// Tasks to register at startup
	tasks: Vec<ScheduledTask>,
}

impl TaskConfig {
	/// Create a new task configuration
	pub fn new(tasks: Vec<ScheduledTask>) -> Self {
		Self {
			tasks,
		}
	}
}

/// Factory for creating TaskSubsystem instances
pub struct TaskSubsystemFactory {
	config: TaskConfig,
}

impl TaskSubsystemFactory {
	/// Create a new factory with default configuration
	pub fn new() -> Self {
		Self {
			config: TaskConfig::default(),
		}
	}

	/// Create a factory with custom configuration
	pub fn with_config(config: TaskConfig) -> Self {
		Self {
			config,
		}
	}
}

impl Default for TaskSubsystemFactory {
	fn default() -> Self {
		Self::new()
	}
}

impl SubsystemFactory for TaskSubsystemFactory {
	fn provide_interceptors(&self, builder: InterceptorBuilder, _ioc: &IocContainer) -> InterceptorBuilder {
		// Task subsystem doesn't need any special interceptors
		builder
	}

	fn create(self: Box<Self>, ioc: &IocContainer) -> Result<Box<dyn Subsystem>> {
		Ok(Box::new(TaskSubsystem::new(ioc, self.config.tasks)))
	}
}