5

我想在 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!");
}
4

2 回答 2

9

由于没有简单的方法来解决这个问题（至少，我没有找到），我将我的异步方法转换为同步方法。并在 Python 方面将其称为，

async fn my_method(s: &str) -> Result<String, Error> {
    // do something
}

#[pyfunction]
fn my_sync_method(s: String) -> PyResult<String> {
    let mut rt = tokio::runtime::Runtime::new().unwrap();
    let mut contents = String::new();
    rt.block_on(async {
        result = format!("{}", my_sync_method(&s).await.unwrap()).to_string();
    });
   Ok((result))
}

#[pymodule]
fn MyModule(py: Python, m: &PyModule) -> PyResult<()> {
    m.add_wrapped(wrap_pyfunction!(my_sync_method))?;
    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 MyModule
result = MyModule.my_sync_method("hello")

使用setuptools-rust，我可以将它捆绑为一个普通的 Python 包。

以上所有代码和命令都在新发布的 Linux Mint 20 上进行了测试。在 MacOS 上，二进制文件将为libMyModule.dylib.

于 2020-06-28T13:54:55.073 回答
4

如果你想用 Python 来控制 Rust 的 async 功能，我认为它不会起作用（或者至少它很复杂，因为你需要连接两种不同的future机制）。对于异步函数，Rust 编译器将维护一个状态机来管理在 await 控制下正确运行的协程。这是 Rust 应用程序的内部状态，Python 无法触及它。同样，Python 解释器也有一个 Rust 无法触及的状态机。

我确实找到了有关如何使用 FFI 导出异步函数的主题。主要思想是将 async 包装在 a 中BoxFuture，让 C 控制将其返回给 Rust 的时间。但是，您不能BoxFuture在 PyO3 中使用，因为它的pyfunction宏无法将函数返回转换BoxFuture为 Python 回调。您可以尝试使用 FFI 创建一个库并使用 python 的 cffi 模块来加载它。

于 2020-06-28T10:20:59.430 回答