Rust channel threads with cancellation sample

main.rs

use anyhow::Result;
use std::sync::Arc;
use std::time::Duration;

#[tokio::main]
async fn main() -> Result<()> {

    let (tx, rx) = std::sync::mpsc::channel();
    let cts = Arc::new(cancellation::CancellationTokenSource::new());
    let cts_clone = cts.clone();

    let t1 = tokio::spawn(async move {
       let mut counter: u64 = 1;
       while !cts_clone.is_canceled() {
            tx.send(format!("Sending message {} to channel", counter)).unwrap();
            counter += 1;
            tokio::time::sleep(Duration::from_millis(2000)).await;
        }

        drop(tx);
        println!("Dropping writing channel due to cancellation");
    });

    let t2 = tokio::spawn(async move {
        while let Ok(message) = rx.recv() {
            println!("Received message: {}", message);
        }

        println!("Existing reader thread. Channel closed");
    });

    ctrlc::set_handler(move || {
        println!("Cancelling");
        cts.cancel();
    })?;

    let _ = tokio::join!(t1, t2);

    Ok(())

}