基于Rust实现外部监控重启主程序的技术咨询

1. 进程名获取与写入文件的方案 #

在Rust中，有两种常见方式获取进程名：

• 最简单的方式是读取启动命令的第一个参数：std::env::args().next().unwrap_or_default()，这个值通常是进程的可执行文件路径或名称（取决于启动方式）。
• 更准确获取进程的短名称，可以通过Linux的/proc/self/comm文件读取，或者使用procfs crate简化操作：

  use procfs::process::Process;
  
  fn get_process_name() -> String {
      let proc = Process::myself().unwrap();
      proc.comm().unwrap_or_default()
  }
  

完全可以将进程名（更推荐同时写入PID，因为进程名可能重复，而PID是系统唯一标识）和时间戳一起写入监控文件。比如可以用简单的文本格式：

pid:12345
timestamp:1699999999
name:my_main_app

或者JSON格式（更便于程序解析）：

{"pid":12345,"timestamp":1699999999,"name":"my_main_app"}

控制器读取文件后，就能通过PID精准定位要重启的主进程，避免误操作其他同名进程。

2. 确保主程序线程终止的方法 #

内部重启的情况 #

你提到的Command::new("/proc/self/exe").exec()调用后，原进程的所有线程会立即终止。因为exec系统调用会直接用新的程序镜像替换当前进程的整个地址空间，原进程的所有线程、资源都会被内核清理，不存在残留线程的问题。

外部重启的情况 #

控制器重启主程序时，需要分两步确保所有线程终止：

1. 优雅终止：先向主进程发送SIGTERM信号，主程序可以注册信号处理函数，收到信号后主动关闭所有工作线程（比如通过Arc<AtomicBool>标记终止状态，让线程循环检查并退出，或者用异步框架的取消机制）。
   Rust中注册信号处理的示例：

   use std::sync::Arc;
   use std::sync::atomic::{AtomicBool, Ordering};
   use signal_hook::{consts::SIGTERM, iterator::Signals};
   
   fn setup_signal_handler(terminate: Arc<AtomicBool>) {
       let mut signals = Signals::new(&[SIGTERM]).unwrap();
       std::thread::spawn(move || {
           for _ in signals.forever() {
               terminate.store(true, Ordering::SeqCst);
               // 这里可以添加等待线程退出的逻辑
           }
       });
   }
   

2. 强制终止：如果主程序没有响应SIGTERM（比如卡死），控制器可以在等待一段时间（比如10秒）后发送SIGKILL信号，这个信号无法被进程捕获或忽略，内核会直接终止进程及其所有线程，确保彻底清理。

内部重启和外部重启的核心逻辑都是终止原进程（连带所有线程）再启动新实例，但外部重启需要额外处理进程定位、信号发送的逻辑，而内部重启是进程自我替换，更直接，但可靠性不如外部监控的方式。

内容的提问来源于stack exchange，提问作者tommy1213