teamtalk 6.0.0

TeamTalk SDK for Rust
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

use super::storage::StateStore;
use crate::client::Client;
use crate::events::Result;
use std::time::{Duration, Instant};

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum JobErrorPolicy {
    KeepRunning,
    Disable,
}

type Job = dyn FnMut(&Client, &mut dyn StateStore) -> Result<()> + Send;

struct ScheduledJob {
    interval: Duration,
    next_run: Instant,
    enabled: bool,
    on_error: JobErrorPolicy,
    job: Box<Job>,
}

#[derive(Default)]
pub struct Scheduler {
    jobs: Vec<ScheduledJob>,
}

impl Scheduler {
    pub fn new() -> Self {
        Self::default()
    }

    pub fn every<F>(&mut self, interval: Duration, on_error: JobErrorPolicy, job: F)
    where
        F: FnMut(&Client, &mut dyn StateStore) -> Result<()> + Send + 'static,
    {
        let safe_interval = interval.max(Duration::from_millis(10));
        self.jobs.push(ScheduledJob {
            interval: safe_interval,
            next_run: Instant::now() + safe_interval,
            enabled: true,
            on_error,
            job: Box::new(job),
        });
    }

    pub fn tick(&mut self, client: &Client, state: &mut dyn StateStore) -> Result<()> {
        let now = Instant::now();
        for job in &mut self.jobs {
            if !job.enabled || now < job.next_run {
                continue;
            }

            let result = (job.job)(client, state);
            job.next_run = now + job.interval;
            if let Err(err) = result {
                if matches!(job.on_error, JobErrorPolicy::Disable) {
                    job.enabled = false;
                }
                return Err(err);
            }
        }
        Ok(())
    }
}