如何在Rust多线程中通过PyO3仅导入一次Python模块并复用？

你的代码编译失败的核心原因是：&PyModule是绑定到GIL作用域的临时借用，生命周期无法跨越线程。要实现跨线程复用模块，需要使用拥有所有权且线程安全的类型来存储模块实例。

正确实现步骤 #

1. 将导入的PyModule转换为PyObject：PyObject实现了Send和Sync，且持有Python对象的所有权，可安全跨线程传递。
2. 用Arc包裹PyObject：实现多线程共享所有权，避免所有权转移问题。
3. 在每个线程的GIL作用域内，将PyObject转换回PyModule使用。

修改后的代码示例 #

use pyo3::{prelude::*, types::PyModule};
use std::sync::Arc;
use std::thread;

fn main() -> PyResult<()> {
    // 一次性导入模块，转换为Arc<PyObject>
    let my_module = Arc::new(Python::with_gil(|py| -> PyResult<PyObject> {
        let pm = py.import("time")?;
        Ok(pm.into())
    })?);

    let mut handles = vec![];
    for i in 0..2 {
        let module_clone = Arc::clone(&my_module);
        let t = thread::spawn(move || -> PyResult<()> {
            Python::with_gil(|py| {
                // 将PyObject转换回PyModule
                let pm = module_clone.as_ref(py).downcast::<PyModule>()?;
                // 调用模块函数示例：time.sleep(1)
                pm.call_method1("sleep", (1,))?;
                println!("Thread {} finished sleeping", i);
                Ok(())
            })
        });
        handles.push(t);
    }

    // 等待所有线程完成
    for h in handles {
        h.join().unwrap()?;
    }

    Ok(())
}

关键细节说明 #

• Arc<PyObject>：Arc提供线程安全的引用计数，PyObject内部维护Python对象的引用计数，双重计数确保对象在所有线程使用完毕后才会被释放。
• as_ref(py)：在GIL作用域内将PyObject转换为PyAny引用，再通过downcast转换为PyModule，确保类型安全。
• 线程内的PyResult：每个线程的Python操作需要返回PyResult，统一处理可能的Python错误。

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