mirror of
https://gitlab.com/fabinfra/fabaccess/bffh.git
synced 2024-12-04 20:04:56 +01:00
89 lines
2.9 KiB
Rust
89 lines
2.9 KiB
Rust
use std::collections::HashMap;
|
|
use std::process::{Command, Stdio};
|
|
use futures_util::future::BoxFuture;
|
|
|
|
use crate::actors::Actor;
|
|
use crate::db::ArchivedValue;
|
|
use crate::resources::modules::fabaccess::ArchivedStatus;
|
|
use crate::resources::state::State;
|
|
|
|
pub struct Process {
|
|
name: String,
|
|
cmd: String,
|
|
args: Vec<String>,
|
|
}
|
|
|
|
impl Process {
|
|
pub fn new(name: String, params: &HashMap<String, String>) -> Option<Self> {
|
|
let cmd = params.get("cmd").map(|s| s.to_string())?;
|
|
let args = params.get("args").map(|argv|
|
|
argv.split_whitespace()
|
|
.map(|s| s.to_string())
|
|
.collect())
|
|
.unwrap_or_else(Vec::new);
|
|
|
|
Some(Self { name, cmd, args })
|
|
}
|
|
|
|
pub fn into_boxed_actuator(self) -> Box<dyn Actor + Sync + Send> {
|
|
Box::new(self)
|
|
}
|
|
}
|
|
|
|
impl Actor for Process {
|
|
fn apply(&mut self, state: ArchivedValue<State>) -> BoxFuture<'static, ()> {
|
|
tracing::debug!(name=%self.name, cmd=%self.cmd, ?state,
|
|
"Process actor updating state");
|
|
let mut command = Command::new(&self.cmd);
|
|
command
|
|
.stdin(Stdio::null())
|
|
.args(self.args.iter())
|
|
.arg(&self.name);
|
|
|
|
match &state.as_ref().inner.state {
|
|
ArchivedStatus::Free => {
|
|
command.arg("free");
|
|
}
|
|
ArchivedStatus::InUse(by) => {
|
|
command.arg("inuse").arg(by.id.as_str());
|
|
}
|
|
ArchivedStatus::ToCheck(by) => {
|
|
command.arg("tocheck")
|
|
.arg(by.id.as_str());
|
|
}
|
|
ArchivedStatus::Blocked(by) => {
|
|
command.arg("blocked")
|
|
.arg(by.id.as_str());
|
|
}
|
|
ArchivedStatus::Disabled => { command.arg("disabled"); },
|
|
ArchivedStatus::Reserved(by) => {
|
|
command.arg("reserved")
|
|
.arg(by.id.as_str());
|
|
}
|
|
}
|
|
|
|
let name = self.name.clone();
|
|
Box::pin(async move { match command.output() {
|
|
Ok(retv) if retv.status.success() => {
|
|
tracing::trace!("Actor was successful");
|
|
let outstr = String::from_utf8_lossy(&retv.stdout);
|
|
for line in outstr.lines() {
|
|
tracing::debug!(%name, %line, "actor stdout");
|
|
}
|
|
}
|
|
Ok(retv) => {
|
|
tracing::warn!(%name, ?state, code=?retv.status,
|
|
"Actor returned nonzero exitcode"
|
|
);
|
|
if !retv.stderr.is_empty() {
|
|
let errstr = String::from_utf8_lossy(&retv.stderr);
|
|
for line in errstr.lines() {
|
|
tracing::warn!(%name, %line, "actor stderr");
|
|
}
|
|
}
|
|
}
|
|
Err(error) => tracing::warn!(%name, ?error, "process actor failed to run cmd"),
|
|
}})
|
|
}
|
|
}
|