forge-jobs 0.2.0

Sidekiq-style job queue with embedded SQLite and pluggable Postgres. Per-queue workers + cron + cluster-wide rate-limit budget + cancellation that survives across replicas.
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
64
65
66
67
68
69
70
71
72
73
74
75
76

//! Job-kind → queue-name routing.
//!
//! The runtime asks the configured router for the destination queue
//! whenever an enqueue request omits `queue_name`.

/// Strategy for routing a job to one of the configured queues.
pub trait Router: Send + Sync + 'static {
    /// Pick the queue that should receive a job of this kind. The
    /// returned name must match a `queue.name` row registered via
    /// [`crate::runtime::QueueRuntime::ensure_queue`] or the job
    /// will languish in `pending`.
    fn route(&self, kind: &str) -> &'static str;
}

/// Routes every kind to the `"default"` queue. Useful for tests,
/// demos, and apps with a single shared queue.
#[derive(Debug, Default, Clone, Copy)]
pub struct DefaultRouter;

impl Router for DefaultRouter {
    fn route(&self, _kind: &str) -> &'static str {
        "default"
    }
}

/// Routes by `kind`'s prefix to a per-source queue.
///
/// Used by tech-admin to split GH and Slack jobs across two queues
/// so each can be paused independently (e.g. when its source's
/// secret is missing) and throttled without affecting the other.
///
/// Routing rules:
///   - `gh_*`    → `"gh"`
///   - `slack_*` → `"slack"`
///   - anything else → `"default"`
#[derive(Debug, Default, Clone, Copy)]
pub struct KindPrefixRouter;

impl Router for KindPrefixRouter {
    fn route(&self, kind: &str) -> &'static str {
        if kind.starts_with("gh_") {
            "gh"
        } else if kind.starts_with("slack_") {
            "slack"
        } else {
            "default"
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn routes_gh_prefix_to_gh_queue() {
        let r = KindPrefixRouter;
        assert_eq!(r.route("gh_issue_refresh"), "gh");
        assert_eq!(r.route("gh_board_scan"), "gh");
    }

    #[test]
    fn routes_slack_prefix_to_slack_queue() {
        let r = KindPrefixRouter;
        assert_eq!(r.route("slack_channel_scan"), "slack");
        assert_eq!(r.route("slack_thread_refresh"), "slack");
    }

    #[test]
    fn unknown_prefix_routes_to_default() {
        let r = KindPrefixRouter;
        assert_eq!(r.route("noop_echo"), "default");
        assert_eq!(r.route("tickets_sync"), "default");
        assert_eq!(r.route(""), "default");
    }
}