depl 2.4.3

Toolkit for a bunch of local and remote CI/CD actions
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

//! Daemons module.

use async_process::Child;
use std::sync::Arc;
use tokio::sync::Mutex;

/// Current Pipeline's daemons.
#[derive(Clone)]
pub struct Daemons(Arc<Mutex<Option<Vec<Daemon>>>>);

/// Single daemon.
pub struct Daemon(Child);

impl Default for Daemons {
  fn default() -> Self {
    Self(Arc::new(Mutex::new(Some(Vec::new()))))
  }
}

impl Daemons {
  /// Adds new daemon.
  pub async fn add_daemon(&self, child: Child) {
    let mut guard = self.0.lock().await;
    if let Some(v) = guard.as_mut() {
      v.push(Daemon(child));
    } else {
      *guard = Some(vec![Daemon(child)])
    }
  }

  /// Shutdowns all daemons.
  pub async fn shutdown(&mut self) {
    let daemons = {
      let mut guard = self.0.lock().await;
      guard.take().unwrap_or_default()
    };

    for mut daemon in daemons.into_iter() {
      daemon.shutdown().await;
    }
  }
}

impl Daemon {
  async fn shutdown(&mut self) {
    tokio::select! {
      _ = tokio::time::sleep(tokio::time::Duration::from_secs(10)) => {
        let _ = self.0.kill();
      }
      _ = self.0.status() => {},
    }
  }
}