如何在Rust程序中添加关闭钩子？

Question

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

在Java世界中，我可以使用shutdown hook来实现。有一个板条箱，恰当地称为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!");
}

struct Cleanup;

impl Drop for Cleanup {
    fn drop(&mut self) {
        eprintln!("Doing some final cleanup");
    }
}

fn main() {
    let _cleanup = Cleanup;

    panic!("Oh no!");
}

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

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？