如何在同步异步上下文使用Tokio Mutex？遇阻塞锁报错求解答

问题描述 #

使用第三方库的Fold trait（不支持异步方法），将状态封装在Arc<tokio::sync::Mutex<T>>中并在库上下文内使用。

代码示例 #

状态结构体实现Fold trait：

struct MyStruct {
  state: Arc<tokio::sync::Mutex<usize>>
}

impl Fold for MyStruct {
  fn fold_xyz(&self, node: Node) {
    let state = self.state.blocking_lock();
  }
}

库调用逻辑：

async fn fold_thing(state: Arc<Mutex<usize>>) {
  let mut my_struct = MyStruct { state };

  let result = library.globals.set(move || {
    target.fold_with(&mut my_struct)
  });
}

报错信息 #

Cannot block the current thread from within a runtime. This happens because
a function attempted to block the current thread while the thread is being
used to drive asynchronous tasks

疑问 #

报错是因为在async函数的闭包中使用互斥锁？还是第三方库内部用了异步运行时？

错误原因 #

核心问题是你在Tokio异步运行时的工作线程上执行了阻塞操作：

• tokio::sync::Mutex::blocking_lock()是阻塞方法，会暂停当前线程直到获取锁。
• async fn fold_thing运行在Tokio的异步工作线程上，而Tokio要求工作线程不能被阻塞——这些线程需要持续调度其他异步任务，阻塞会导致整个运行时性能下降甚至死锁。
• 报错和闭包本身无关，也不是第三方库内部使用了异步运行时，本质是你在Tokio管理的线程上调用了阻塞锁操作。

解决方案 #

方案1：利用Tokio Runtime Handle执行异步锁操作 #

因为Fold trait方法是同步的，无法直接使用.lock().await，可以提前保存Tokio Runtime的Handle，在同步方法中通过它执行异步锁逻辑：

// 修改MyStruct，添加Runtime Handle字段
struct MyStruct {
  state: Arc<tokio::sync::Mutex<usize>>,
  rt_handle: tokio::runtime::Handle,
}

impl Fold for MyStruct {
  fn fold_xyz(&self, node: Node) {
    // 通过Runtime Handle在同步方法中执行异步逻辑
    self.rt_handle.block_on(async {
      let mut state = self.state.lock().await;
      // 在这里处理状态逻辑
    });
  }
}

// 在调用处传递Runtime Handle
async fn fold_thing(state: Arc<tokio::sync::Mutex<usize>>) {
  let rt_handle = tokio::runtime::Handle::current();
  let mut my_struct = MyStruct { state, rt_handle };

  let result = library.globals.set(move || {
    target.fold_with(&mut my_struct)
  });
}

方案2：替换为同步互斥锁 #

如果状态不需要跨异步任务高效共享，或者可以接受同步锁的性能开销，直接将tokio::sync::Mutex替换为标准库的std::sync::Mutex：

struct MyStruct {
  state: Arc<std::sync::Mutex<usize>>
}

impl Fold for MyStruct {
  fn fold_xyz(&self, node: Node) {
    // 直接使用同步锁，不会触发Tokio的阻塞检测
    let mut state = self.state.lock().unwrap();
    // 处理状态逻辑
  }
}

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