server-watchdog 0.1.2

A server monitoring and remote control tool via messenger.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use std::pin::Pin;
use std::task::{Context, Poll};
use log::warn;
use tokio::process::Child;
use tokio_stream::Stream;

/// Wrapper struct that keeps the child process alive for the lifetime of the stream
/// and ensures proper cleanup when the stream is dropped.
pub struct ChildProcessStream {
    child: Child,
    stream: Pin<Box<dyn Stream<Item = String> + Send>>,
}

impl ChildProcessStream {
    pub(crate) fn new(stream: Pin<Box<dyn Stream<Item = String> + Send>>, child: Child) -> Self {
        Self { child, stream }
    }
}

impl Stream for ChildProcessStream {
    type Item = String;

    fn poll_next(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Option<Self::Item>> {
        self.stream.as_mut().poll_next(cx)
    }
}

impl Drop for ChildProcessStream {
    fn drop(&mut self) {
        // Attempt to kill the child process when the stream is dropped
        if let Err(e) = self.child.start_kill() {
            warn!("Failed to kill child process during stream cleanup: {}", e);
        }
    }
}