如何从另一个线程终止或挂起一个 Rust 线程

发布于10月05日

Editor's note-此示例创建于Rust 1.0之前，自那时起，特定类型已更改或删除.一般问题和概念仍然有效.

我产生了一个线程，里面有无限循环和计时器.

thread::spawn(|| {
    let mut timer = Timer::new().unwrap();
    let periodic = timer.periodic(Duration::milliseconds(200));
    loop {
        periodic.recv();

        // Do my work here
    }
});

基于某些条件，经过一段时间后，我需要从程序的另一部分终止该线程.换句话说，我想退出无限循环.我怎样才能正确地做到这一点？另外，我怎样才能暂停这个线程并在以后继续呢？

我试图使用一个全局不安全标志来打破循环，但我认为这个解决方案看起来不太好.

Terminated externally

在worker循环的每次迭代中，我们判断是否有人通过渠道通知我们.如果是，或者通道的另一端超出了范围，我们就打破循环.

use std::io::{self, BufRead};
use std::sync::mpsc::{self, TryRecvError};
use std::thread;
use std::time::Duration;

fn main() {
    println!("Press enter to terminate the child thread");
    let (tx, rx) = mpsc::channel();

    thread::spawn(move || loop {
        println!("Working...");
        thread::sleep(Duration::from_millis(500));
        match rx.try_recv() {
            Ok(_) | Err(TryRecvError::Disconnected) => {
                println!("Terminating.");
                break;
            }
            Err(TryRecvError::Empty) => {}
        }
    });

    let mut line = String::new();
    let stdin = io::stdin();
    let _ = stdin.lock().read_line(&mut line);

    let _ = tx.send(());
}

Suspending and resuming

我们使用recv()，它会暂停线程，直到有东西到达通道.为了恢复线程，需要通过通道发送一些内容；在这种情况下，单位值为().如果信道的传输端被丢弃，recv()将返回Err(())——我们使用它退出环路.

use std::io::{self, BufRead};
use std::sync::mpsc;
use std::thread;
use std::time::Duration;

fn main() {
    println!("Press enter to wake up the child thread");
    let (tx, rx) = mpsc::channel();
    thread::spawn(move || loop {
        println!("Suspending...");
        match rx.recv() {
            Ok(_) => {
                println!("Working...");
                thread::sleep(Duration::from_millis(500));
            }
            Err(_) => {
                println!("Terminating.");
                break;
            }
        }
    });

    let mut line = String::new();
    let stdin = io::stdin();
    for _ in 0..4 {
        let _ = stdin.lock().read_line(&mut line);
        let _ = tx.send(());
    }
}