Struct TaskQueue

pub struct TaskQueue {
    db: Arc<Database>,
    notify: Arc<Notify>,
}

Pure storage layer for task queue - only handles data persistence

Fields

db: Arc<Database>

notify: Arc<Notify>

Implementations

impl TaskQueue

pub fn new(db: Arc<Database>) -> Result<Self>

Create a new task queue instance

pub fn insert_pending( &self, priority: u64, task_id: &str, data: &[u8], ) -> Result<()>

Insert a task into the pending queue with composite key for uniqueness

pub fn atomic_pop_pending<F>(&self, on_task: F) -> Result<Option<Task>>

where F: FnOnce(&mut Task),

Atomically pop the first pending task and move it to processing Accepts a callback to update task state within the same transaction This prevents race conditions and ensures atomicity of pop→update→save

pub fn get_first_pending(&self) -> Result<Option<(u64, Vec<u8>)>>

Get the first pending task without removing it Returns (priority, data) - note: priority extracted from composite key

pub fn move_to_processing( &self, priority: u64, task_id: &str, data: &[u8], ) -> Result<()>

Move a task from pending to processing (legacy method for tests) Prefer atomic_pop_pending() for production code to avoid race conditions

pub fn move_to_completed(&self, task_id: &str, data: &[u8]) -> Result<()>

Move a task from processing to completed

pub fn get_from_processing(&self, task_id: &str) -> Result<Option<Vec<u8>>>

Get a task from processing table

pub fn remove_from_processing(&self, task_id: &str) -> Result<()>

Remove a task from processing table

pub fn get_from_any_table(&self, task_id: &str) -> Result<Option<Vec<u8>>>

Get a task from any table

pub fn get_all_pending(&self) -> Result<Vec<Vec<u8>>>

Get all tasks from pending table

pub fn get_all_processing(&self) -> Result<Vec<Vec<u8>>>

Get all tasks from processing table

pub fn get_all_completed(&self) -> Result<Vec<Vec<u8>>>

Get all tasks from completed table

pub async fn wait_for_task(&self)

Wait for a task to be available

pub fn notify_task_available(&self)

Notify that a task is available

Trait Implementations

impl Clone for TaskQueue

fn clone(&self) -> TaskQueue

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.