Rust Rocket如何获取当前工作线程ID排查高并发请求超时问题

线程ID获取方法 #

你不需要依赖Rocket框架的专属API，直接使用Rust标准库即可获取当前处理请求的线程ID，同时可以搭配标准库的时间工具统计请求处理耗时，示例实现如下：

use std::thread;
use std::time::Instant;
// 保留你原有其他导入

#[post("/v1",data = "<record>")]
pub fn word_search(record: Json<WordSearchRequest>, login_user_info: LoginUserInfo) -> content::Json<String> {
    // 记录请求开始信息
    let start = Instant::now();
    let current_thread_id = thread::current().id();
    println!("新请求进入，处理线程ID: {:?}", current_thread_id);

    // 此处保留你原有从数据库拉取数据的相关逻辑

    // 处理完成打印耗时
    let cost = start.elapsed();
    println!("请求处理完成，线程ID: {:?}，总耗时: {:?}", current_thread_id, cost);
    // 保留你原有返回逻辑
}

单线程怀疑相关说明 #

Rocket 0.5.0-rc.1 版本默认采用多线程异步 runtime 处理请求，默认worker线程数等于当前服务器的CPU逻辑核心数，不会默认运行在单线程模式。你可以通过上面代码打印的线程ID，确认实际处理请求的线程数量是否符合预期。
如果确认是多线程仍出现超时问题，可优先排查以下点：

• 数据库查询本身的执行耗时是否过高
• 数据库连接池的最大连接数配置是否足够，连接数不足会导致大量请求阻塞在等待数据库连接的环节
• 可通过Rocket配置手动调整worker线程数，在Rocket.toml配置文件中添加workers = 自定义线程数即可，也可以通过代码手动指定配置：

  use rocket::Config;
  
  #[launch]
  fn rocket() -> _ {
      rocket::build()
          .configure(Config {
              workers: 8, // 自定义worker线程数
              ..Default::default()
          })
          // 保留你原有挂载路由、挂载状态等逻辑
  }
  

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