如何为 Rust 程序添加关机钩子?

Question

我正在编写一个 Rust 程序，无论发生什么，它都需要在执行结束时保存一些数据。

在 Java 世界中，我会使用关闭钩子来做到这一点。有一个板条箱，恰如其分地称为shutdown_hooks，但它似乎只能注册extern "C"函数(我不完全确定它会在panic!(...)上运行(。

如何实现在正常退出时以及在恐慌时触发的关机钩子？

Answer 1

在一般情况下，这是不可能的。即使忽略来自程序外部的影响(如 mcarton 所述(，编译最终二进制文件的人也可以选择是否真的触发了堆栈展开，或者它是否只是中止了程序。在后一种情况下，您无能为力。

在解除恐慌或正常退出的情况下，您可以实施Drop并使用 RAII 的常规方面：

struct Cleanup;
impl Drop for Cleanup {
fn drop(&mut self) {
eprintln!("Doing some final cleanup");
}
}
fn main() {
let _cleanup = Cleanup;
panic!("Oh no!");
}

thread 'main' panicked at 'Oh no!', src/main.rs:12:5
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace.
Doing some final cleanup

看起来Java的关闭钩子允许在线程中并行运行多段代码。您可以通过一些小的修改来做类似的事情：

use std::{
sync::{Arc, Condvar, Mutex},
thread,
};
#[derive(Debug, Default)]
struct Cleanup {
hooks: Vec<thread::JoinHandle<()>>,
run: Arc<Mutex<bool>>,
go: Arc<Condvar>,
}
impl Cleanup {
fn add(&mut self, f: impl FnOnce() + Send + 'static) {
let run = self.run.clone();
let go = self.go.clone();
let t = thread::spawn(move || {
let mut run = run.lock().unwrap();
while !*run {
run = go.wait(run).unwrap();
}
f();
});
self.hooks.push(t);
}
}
impl Drop for Cleanup {
fn drop(&mut self) {
eprintln!("Starting final cleanup");
*self.run.lock().unwrap() = true;
self.go.notify_all();
for h in self.hooks.drain(..) {
h.join().unwrap();
}
eprintln!("Final cleanup complete");
}
}
fn main() {
let mut cleanup = Cleanup::default();
cleanup.add(|| {
eprintln!("Cleanup #1");
});
cleanup.add(|| {
eprintln!("Cleanup #2");
});
panic!("Oh no!");
}

另请参阅：

• 注册函数以在 Rust 程序意外退出期间运行的最佳方法是什么？
• 有没有办法告诉 Rust 编译器在处理恐慌时对部分初始化的数组元素调用 drop？