Dedup StdOut/StdErr observer pt.2 (#2876)
* dedup stdout stdin * mm * Claude was not enough * but was good enough * ok * okok * okokok * shorter now * lol * fixer
This commit is contained in:
parent
6eabb79920
commit
2842f95e2a
@ -46,7 +46,7 @@ impl StdOutToMetadataFeedback {
|
|||||||
.get(&self.o_ref)
|
.get(&self.o_ref)
|
||||||
.ok_or(Error::illegal_state("StdOutObserver is missing"))?;
|
.ok_or(Error::illegal_state("StdOutObserver is missing"))?;
|
||||||
let buffer = observer
|
let buffer = observer
|
||||||
.data
|
.output
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.ok_or(Error::illegal_state("StdOutObserver has no stdout"))?;
|
.ok_or(Error::illegal_state("StdOutObserver has no stdout"))?;
|
||||||
let stdout = String::from_utf8_lossy(buffer).into_owned();
|
let stdout = String::from_utf8_lossy(buffer).into_owned();
|
||||||
@ -139,7 +139,7 @@ where
|
|||||||
.get(&self.o_ref)
|
.get(&self.o_ref)
|
||||||
.ok_or(Error::illegal_state("StdErrObserver is missing"))?;
|
.ok_or(Error::illegal_state("StdErrObserver is missing"))?;
|
||||||
let buffer = observer
|
let buffer = observer
|
||||||
.data
|
.output
|
||||||
.as_ref()
|
.as_ref()
|
||||||
.ok_or(Error::illegal_state("StdErrObserver has no stderr"))?;
|
.ok_or(Error::illegal_state("StdErrObserver has no stderr"))?;
|
||||||
let stderr = String::from_utf8_lossy(buffer).into_owned();
|
let stderr = String::from_utf8_lossy(buffer).into_owned();
|
||||||
|
@ -71,8 +71,8 @@ use crate::{observers::Observer, Error};
|
|||||||
/// ) -> Result<bool, Error>
|
/// ) -> Result<bool, Error>
|
||||||
/// {
|
/// {
|
||||||
/// unsafe {
|
/// unsafe {
|
||||||
/// STDOUT = observers.get(&self.stdout_observer).unwrap().data.clone();
|
/// STDOUT = observers.get(&self.stdout_observer).unwrap().output.clone();
|
||||||
/// STDERR = observers.get(&self.stderr_observer).unwrap().data.clone();
|
/// STDERR = observers.get(&self.stderr_observer).unwrap().output.clone();
|
||||||
/// }
|
/// }
|
||||||
/// Ok(true)
|
/// Ok(true)
|
||||||
/// }
|
/// }
|
||||||
@ -169,58 +169,59 @@ use crate::{observers::Observer, Error};
|
|||||||
/// ```
|
/// ```
|
||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
|
#[derive(Debug, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
|
||||||
pub struct StreamObserver<T> {
|
pub struct OutputObserver<T> {
|
||||||
/// The name of the observer.
|
/// The name of the observer.
|
||||||
pub name: Cow<'static, str>,
|
pub name: Cow<'static, str>,
|
||||||
/// The captured stdout/stderr data during last execution.
|
/// The captured stdout/stderr data during last execution.
|
||||||
pub data: Option<Vec<u8>>,
|
pub output: Option<Vec<u8>>,
|
||||||
/// Phantom data to hold the stream type
|
/// Phantom data to hold the stream type
|
||||||
phantom: PhantomData<T>,
|
phantom: PhantomData<T>,
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Marker traits to distinguish between stdout and stderr
|
/// Marker traits to mark stdout for the `OutputObserver`
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct StdOutMarker;
|
pub struct StdOutMarker;
|
||||||
/// Marker traits to distinguish between stdout and stderr
|
|
||||||
|
/// Marker traits to mark stderr for the `OutputObserver`
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct StdErrMarker;
|
pub struct StdErrMarker;
|
||||||
|
|
||||||
impl<T> StreamObserver<T> {
|
impl<T> OutputObserver<T> {
|
||||||
/// Create a new `StreamObserver` with the given name.
|
/// Create a new `OutputObserver` with the given name.
|
||||||
#[must_use]
|
#[must_use]
|
||||||
pub fn new(name: &'static str) -> Self {
|
pub fn new(name: &'static str) -> Self {
|
||||||
Self {
|
Self {
|
||||||
name: Cow::from(name),
|
name: Cow::from(name),
|
||||||
data: None,
|
output: None,
|
||||||
phantom: PhantomData,
|
phantom: PhantomData,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// React to new stream data
|
/// React to new stream data
|
||||||
pub fn observe(&mut self, data: &[u8]) {
|
pub fn observe(&mut self, data: &[u8]) {
|
||||||
self.data = Some(data.into());
|
self.output = Some(data.into());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<T> Named for StreamObserver<T> {
|
impl<T> Named for OutputObserver<T> {
|
||||||
fn name(&self) -> &Cow<'static, str> {
|
fn name(&self) -> &Cow<'static, str> {
|
||||||
&self.name
|
&self.name
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl<I, S, T> Observer<I, S> for StreamObserver<T> {
|
impl<I, S, T> Observer<I, S> for OutputObserver<T> {
|
||||||
fn pre_exec_child(&mut self, _state: &mut S, _input: &I) -> Result<(), Error> {
|
fn pre_exec_child(&mut self, _state: &mut S, _input: &I) -> Result<(), Error> {
|
||||||
self.data = None;
|
self.output = None;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn pre_exec(&mut self, _state: &mut S, _input: &I) -> Result<(), Error> {
|
fn pre_exec(&mut self, _state: &mut S, _input: &I) -> Result<(), Error> {
|
||||||
self.data = None;
|
self.output = None;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// An observer that captures stdout of a target.
|
/// An observer that captures stdout of a target.
|
||||||
pub type StdOutObserver = StreamObserver<StdOutMarker>;
|
pub type StdOutObserver = OutputObserver<StdOutMarker>;
|
||||||
/// An observer that captures stderr of a target.
|
/// An observer that captures stderr of a target.
|
||||||
pub type StdErrObserver = StreamObserver<StdErrMarker>;
|
pub type StdErrObserver = OutputObserver<StdErrMarker>;
|
||||||
|
Loading…
x
Reference in New Issue
Block a user