gpt4 book ai didi

rust - 在 tokio::time::timeout 之后再次等待 future

转载 作者:行者123 更新时间:2023-12-03 11:35:41 26 4
gpt4 key购买 nike

背景:
我有一个进程使用 tokio::process在 tokio 运行时生成带有句柄的子进程。

它还负责在杀死 child 后释放资源,根据文档( std::process::Childtokio::process::Child ),这需要父级 wait() (或 await in tokio)的过程。

并非所有进程的行为都与 SIGINT 相同或 SIGTERM ,所以我想给 child 一些时间去死,然后我发送了SIGKILL .

想要的解决方案:

    pub async fn kill(self) {
// Close input
std::mem::drop(self.stdin);

// Send gracefull signal
let pid = nix::unistd::Pid::from_raw(self.process.id() as nix::libc::pid_t);
nix::sys::signal::kill(pid, nix::sys::signal::SIGINT);

// Give the process time to die gracefully
if let Err(_) = tokio::time::timeout(std::time::Duration::from_secs(2), self.process).await
{
// Kill forcefully
nix::sys::signal::kill(pid, nix::sys::signal::SIGKILL);
self.process.await;
}
}

但是给出了这个错误:
error[E0382]: use of moved value: `self.process`
--> src/bin/multi/process.rs:46:13
|
42 | if let Err(_) = tokio::time::timeout(std::time::Duration::from_secs(2), self.process).await
| ------------ value moved here
...
46 | self.process.await;
| ^^^^^^^^^^^^ value used here after move
|
= note: move occurs because `self.process` has type `tokio::process::Child`, which does not implement the `Copy` trait

如果我服从并删除 self.process.await , 我看到子进程仍在 ps 中占用资源.

问题:
我该怎么办 await一段时间并执行操作和 await再次如果时间到期?

注:
我通过设置一个总是发送 SIGKILL 的 tokio 计时器解决了我的直接问题。两秒后,有一个 self.process.await在底部。但是这种解决方案是不可取的,因为在计时器运行时另一个进程可能会在同一个 PID 中产生。

编辑:
添加 minimal, reproducible example ( playground )

async fn delay() {
for _ in 0..6 {
tokio::time::delay_for(std::time::Duration::from_millis(500)).await;
println!("Ping!");
}
}

async fn runner() {
let delayer = delay();
if let Err(_) = tokio::time::timeout(std::time::Duration::from_secs(2), delayer).await {
println!("Taking more than two seconds");
delayer.await;
}
}

最佳答案

您需要传递一个可变引用。但是,您首先需要固定 future ,以便其可变引用实现 Future。 pin_mutfutures 重新导出crate 是一个很好的 helper :

use futures::pin_mut;

async fn delay() {
for _ in 0..6 {
tokio::time::delay_for(std::time::Duration::from_millis(500)).await;
println!("Ping!");
}
}

async fn runner() {
let delayer = delay();
pin_mut!(delayer);
if let Err(_) = tokio::time::timeout(std::time::Duration::from_secs(2), &mut delayer).await {
println!("Taking more than two seconds");
delayer.await;
}
}

关于rust - 在 tokio::time::timeout 之后再次等待 future ,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61041856/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com