如何从 Python 调用 Rust 异步方法？

Question

我想在 Python 中使用 Rust 异步方法。我正在尝试使用PyO3或rust-cpython。

例如，对于同步 Rust 函数，我可以使用，

#[pyfunction]

fn myfunc(a: String) -> PyResult<String> {

   let mut contents = String::new();

   contents = a.to_string() + " appended";

   Ok((contents))

}

#[pymodule]

fn MyModule(py: Python, m: &PyModule) -> PyResult<()> {

    m.add_wrapped(wrap_pyfunction!(urlshot))?;

    Ok(())

}

对于异步方法，我该怎么做？例如，我想在 Python 中调用以下方法，

async fn hello_world() {

    println!("hello, world!");

}

Answer 1

由于没有解决此问题的简单方法（至少，我还没有找到），我将我的异步方法转换为同步方法。并在 Python 方面将其称为，async fn my_method(s: &str) -> Result<String, Error> {&nbsp; &nbsp; // do something}#[pyfunction]fn my_sync_method(s: String) -> PyResult<String> {&nbsp; &nbsp; let mut rt = tokio::runtime::Runtime::new().unwrap();&nbsp; &nbsp; let mut contents = String::new();&nbsp; &nbsp; rt.block_on(async {&nbsp; &nbsp; &nbsp; &nbsp; result = format!("{}", my_sync_method(&s).await.unwrap()).to_string();&nbsp; &nbsp; });&nbsp; &nbsp;Ok((result))}#[pymodule]fn MyModule(py: Python, m: &PyModule) -> PyResult<()> {&nbsp; &nbsp; m.add_wrapped(wrap_pyfunction!(my_sync_method))?;&nbsp; &nbsp; Ok(())}已编辑在Cargo.toml文件中，我添加了以下依赖项，[dependencies.pyo3]git = "https://github.com/PyO3/pyo3"features = ["extension-module"]运行后cargo build --release生成target/release/libMyModule.so二进制文件。将其重命名为MyModule.so，现在可以从 Python 导入它。import MyModuleresult = MyModule.my_sync_method("hello")使用setuptools-rust，我可以将它捆绑为一个普通的 Python 包。以上所有代码和命令都在新发布的 Linux Mint 20 上进行了测试。在 MacOS 上，二进制文件将为libMyModule.dylib.

Answer 2

如果你想使用 Python 来控制 Rust 的异步功能，我认为它不会起作用（或者至少它非常复杂，因为你需要连接两种不同的future机制）。对于异步函数，Rust 编译器将维护一个状态机来管理在 await 的控制下正确运行的协程。这是 Rust 应用程序的内部状态，Python 无法触及它。同样，Python 解释器也有 Rust 无法触及的状态机。我确实找到了有关如何使用 FFI 导出异步函数的主题。主要思想是将异步包装在 a 中BoxFuture，让 C 控制将其返回给 Rust 的时间。但是，您不能BoxFuture在 PyO3 中使用，因为它的pyfunction宏不能将函数返回转换BoxFuture为 Python 回调。您可以尝试使用 FFI 创建一个库并使用 python 的 cffi 模块来加载它。