src/main.rs
author Sunil Nimmagadda <sunil@nimmagadda.net>
Mon, 28 Nov 2022 14:14:39 +0530
changeset 1 80281b8a2707
parent 0 32171fc0e41f
permissions -rw-r--r--
expect instead of unwrap.
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
     1
use tokio::signal::unix::{signal, SignalKind};
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
     2
use tokio::time;
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
     3
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
     4
const ADV_INTERVAL: u64 = 3;
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
     5
const SKEW_INTERVAL: u64 = 1;
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
     6
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
     7
async fn advert_handler() {
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
     8
    println!("3 second timer fired");
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
     9
}
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    10
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    11
async fn skew_handler() {
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    12
    println!("1 second timer fired");
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    13
}
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    14
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    15
async fn sighup_handler() {
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    16
    println!("got SIGHUP");
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    17
}
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    18
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    19
#[tokio::main(flavor = "current_thread")]
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    20
async fn main() {
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    21
    let advert_handle = tokio::spawn(async move {
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    22
        let mut advert_interval = time::interval(time::Duration::from_secs(ADV_INTERVAL));
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    23
        loop {
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    24
            advert_interval.tick().await;
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    25
            advert_handler().await;
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    26
        }
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    27
    });
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    28
    let skew_handle = tokio::spawn(async move {
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    29
        let mut skew_interval = time::interval(time::Duration::from_secs(SKEW_INTERVAL));
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    30
        loop {
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    31
            skew_interval.tick().await;
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    32
            skew_handler().await;
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    33
        }
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    34
    });
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    35
    let sighup_handle = tokio::spawn(async move {
1
80281b8a2707 expect instead of unwrap.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents: 0
diff changeset
    36
        let mut stream = signal(SignalKind::hangup()).expect("Should be a signal stream");
0
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    37
        loop {
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    38
            stream.recv().await;
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    39
            sighup_handler().await;
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    40
        }
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    41
    });
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    42
    let (_, _, _) = tokio::join!(advert_handle, skew_handle, sighup_handle);
32171fc0e41f Let's begin.
Sunil Nimmagadda <sunil@nimmagadda.net>
parents:
diff changeset
    43
}