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

#![doc = include_str!("channel.md")]

use super::runner::Runner;
use super::runner::{self};
use async_channel as achannel;
use std::time::Duration;

/// just pure empty unit
pub const TASK_SIGNAL: () = ();

/// Create a [`runner::Runner`].
/// Signals every `interval`.
pub fn new(interval: Duration, block: bool) -> (achannel::Receiver<()>, Runner<()>) {
    let (sub_sender, main_receiver) = achannel::bounded::<()>(1);
    let runner = if block {
        runner::new_runner(
            interval,
            || (),
            move |_| {
                if let Err(e) = sub_sender.send_blocking(TASK_SIGNAL) {
                    panic!("[task-controler] error sending start signal: {}", e);
                }
                false
            },
        )
    } else {
        runner::new_runner(
            interval,
            || (),
            move |_| {
                if let Err(e) = sub_sender.force_send(TASK_SIGNAL) {
                    panic!("[task-controler] error sending start signal: {}", e);
                }
                false
            },
        )
    };
    (main_receiver, runner)
}

/// create unbouned channel that wraps a runner
pub fn new_unbound(interval: Duration) -> (achannel::Receiver<()>, Runner<()>) {
    let (sub_sender, main_receiver) = achannel::unbounded::<()>();
    let runner = runner::new_runner(
        interval,
        || (),
        move |_| {
            if let Err(e) = sub_sender.send_blocking(TASK_SIGNAL) {
                panic!("[task-controler] error sending start signal: {}", e);
            }
            false
        },
    );
    (main_receiver, runner)
}