rskit-worker 0.1.0-alpha.2

Task worker pool with JoinSet, typed events, and provider bridges
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

use std::sync::Arc;
use std::sync::atomic::{AtomicUsize, Ordering};

/// Strategy used to pick a worker slot in the pool.
#[derive(Debug, Clone, Default)]
#[non_exhaustive]
pub enum DispatchStrategy {
    /// Assign tasks round-robin across available worker slots.
    #[default]
    RoundRobin,
    /// Prefer slots with the lowest active task count (future extension).
    LeastLoaded,
}

/// Simple round-robin counter shared across the pool.
#[derive(Clone)]
pub struct RoundRobinDispatcher {
    counter: Arc<AtomicUsize>,
    slots: usize,
}

impl RoundRobinDispatcher {
    /// Create a new dispatcher that distributes work across `slots` worker slots.
    pub fn new(slots: usize) -> Self {
        Self {
            counter: Arc::new(AtomicUsize::new(0)),
            slots,
        }
    }

    /// Return the next worker index.
    pub fn next(&self) -> usize {
        if self.slots == 0 {
            return 0;
        }
        self.counter.fetch_add(1, Ordering::Relaxed) % self.slots
    }
}