如何在Rust中重命名主线程？解决多进程panic识别难题

问题原因 #

你直接调用pthread_setname_np修改的是系统层面的线程名称，但Rust的panic日志输出的是Rust内部线程结构体中存储的名称，而非直接读取系统线程名。主线程启动时Rust已经将其名称设为main，仅修改系统线程名不会同步到Rust的内部存储，所以panic信息仍显示main。

解决方案：使用Rust标准库的set_name方法 #

从Rust 1.63版本开始，std::thread::Thread提供了set_name方法，支持修改当前线程（包括主线程）的名称，这个修改会同步到Rust的内部存储，panic时会正确显示自定义名称。

示例代码 #

fn main() {
    // 为主线程设置自定义名称
    std::thread::current().set_name("process_A_main").unwrap();

    panic!("explicit panic");
}

运行cargo run后，panic日志会输出：

thread 'process_A_main' panicked at 'explicit panic', src/main.rs:5:5

旧版本Rust兼容方案（1.63之前） #

如果你的项目使用的是1.63之前的Rust版本，无法直接使用set_name，可以通过以下方式实现：

1. 保留pthread_setname_np调用修改系统线程名，用于系统工具（如ps、top）识别。
2. 自定义panic钩子，在panic时读取系统线程名并输出：

use libc::{pthread_getname_np, pthread_self};
use std::{ffi::CString, ffi::OsStr, os::unix::ffi::OsStrExt};

fn main() {
    // 设置系统线程名
    let name = CString::new("process_B_main").unwrap();
    unsafe {
        libc::pthread_setname_np(pthread_self(), name.as_ptr());
    }

    // 自定义panic钩子
    std::panic::set_hook(Box::new(|panic_info| {
        // 读取当前线程的系统名称
        let mut buf = [0u8; 16];
        unsafe {
            pthread_getname_np(pthread_self(), buf.as_mut_ptr() as *mut i8, buf.len());
        }
        let thread_name = OsStr::from_bytes(&buf).to_string_lossy().trim_end_matches('\0');
        
        // 输出自定义格式的panic信息
        eprintln!("thread '{}' panicked at '{}'", thread_name, panic_info.message().unwrap_or(&format_args!("unknown panic")));
        if let Some(location) = panic_info.location() {
            eprintln!("{}:{}", location.file(), location.line());
        }
    }));

    panic!("explicit panic");
}

这个方案会让panic日志显示你设置的系统线程名，同时兼容旧版Rust。

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