the-distro/ofborg
2
0
Fork 1
Code Issues 1 Pull requests 1 Packages Projects Releases Wiki Activity

clean up whitespace

Browse source
This commit is contained in:
Graham Christensen 2018-01-27 14:41:39 -05:00
parent b2e34d91d9
commit 7511eb771b
No known key found for this signature in database
GPG key ID: ACA1C1D120C83D5C
3 changed files with 77 additions and 55 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

83
ofborg/src/asynccmd.rs
View file

@ -47,8 +47,7 @@ enum WaitResult<T> {
fn reader_tx<R: 'static + Read + Send>(read: R, tx: SyncSender<String>) -> thread::JoinHandle<()> {
let read = BufReader::new(read);
thread::spawn(move || {
for line in read.lines() {
thread::spawn(move || for line in read.lines() {
let to_send: String = match line {
Ok(line) => line,
Err(e) => {
@ -60,24 +59,39 @@ fn reader_tx<R: 'static + Read + Send>(read: R, tx: SyncSender<String>) -> threa
if let Err(e) = tx.send(to_send) {
error!("Failed to send log line: {:?}", e);
}
}
})
}
fn spawn_join<T: Send + 'static>(id: WaitTarget, tx: SyncSender<(WaitTarget, WaitResult<T>)>, waiting_on: thread::JoinHandle<T>) -> thread::JoinHandle<()> {
thread::spawn(move || {
if let Err(e) = tx.send((id, WaitResult::Thread(waiting_on.join()))) {
fn spawn_join<T: Send + 'static>(
id: WaitTarget,
tx: SyncSender<(WaitTarget, WaitResult<T>)>,
waiting_on: thread::JoinHandle<T>,
) -> thread::JoinHandle<()> {
thread::spawn(move || if let Err(e) = tx.send((
id,
WaitResult::Thread(
waiting_on.join(),
),
))
{
error!("Failed to send message to the thread waiter: {:?}", e);
}
})
}
fn child_wait<T: Send + 'static>(id: WaitTarget, tx: SyncSender<(WaitTarget, WaitResult<T>)>, mut waiting_on: Child) -> thread::JoinHandle<()> {
thread::spawn(move || {
if let Err(e) = tx.send((id, WaitResult::Process(waiting_on.wait()))) {
fn child_wait<T: Send + 'static>(
id: WaitTarget,
tx: SyncSender<(WaitTarget, WaitResult<T>)>,
mut waiting_on: Child,
) -> thread::JoinHandle<()> {
thread::spawn(move || if let Err(e) = tx.send((
id,
WaitResult::Process(
waiting_on.wait(),
),
))
{
error!("Failed to send message to the thread waiter: {:?}", e);
}
})
}
@ -103,8 +117,8 @@ impl AsyncCmd {
spawn_join(
WaitTarget::Stderr,
monitor_tx.clone(),
reader_tx(child.stderr.take().unwrap(), proc_tx.clone())
)
reader_tx(child.stderr.take().unwrap(), proc_tx.clone()),
),
);
waiters.insert(
@ -112,17 +126,13 @@ impl AsyncCmd {
spawn_join(
WaitTarget::Stdout,
monitor_tx.clone(),
reader_tx(child.stdout.take().unwrap(), proc_tx.clone())
)
reader_tx(child.stdout.take().unwrap(), proc_tx.clone()),
),
);
waiters.insert(
WaitTarget::Child,
child_wait(
WaitTarget::Child,
monitor_tx.clone(),
child
)
child_wait(WaitTarget::Child, monitor_tx.clone(), child),
);
let head_waiter = thread::spawn(move || {
@ -147,7 +157,12 @@ impl AsyncCmd {
}
}
None => { error!("Received notice that {:?} finished, but it isn't being waited on?", id); }
None => {
error!(
"Received notice that {:?} finished, but it isn't being waited on?",
id
);
}
}
if waiters.len() == 0 {
@ -156,7 +171,10 @@ impl AsyncCmd {
}
}
info!("Out of the child waiter recv, with {:?} remaining waits", waiters.len());
info!(
"Out of the child waiter recv, with {:?} remaining waits",
waiters.len()
);
return return_status;
});
@ -226,7 +244,9 @@ mod tests {
fn lots_of_small_ios_test() {
let mut cmd = Command::new("/bin/sh");
cmd.arg("-c");
cmd.arg("for i in `seq 1 100`; do (seq 1 100)& (seq 1 100 >&2)& wait; wait; done");
cmd.arg(
"for i in `seq 1 100`; do (seq 1 100)& (seq 1 100 >&2)& wait; wait; done",
);
let acmd = AsyncCmd::new(cmd);
let mut spawned = acmd.spawn();
@ -234,7 +254,9 @@ mod tests {
assert_eq!(lines.len(), 20000);
let thread_result = spawned.wait();
let child_result_opt = thread_result.expect("Thread should exit correctly");
let child_result = child_result_opt.expect("Thread should have properly properly returned the child's status");
let child_result = child_result_opt.expect(
"Thread should have properly properly returned the child's status",
);
let exit_status = child_result.expect("The child should have no problem exiting");
assert_eq!(true, exit_status.success());
}
@ -252,7 +274,9 @@ mod tests {
assert_eq!(lines.len(), 200000);
let thread_result = spawned.wait();
let child_result_opt = thread_result.expect("Thread should exit correctly");
let child_result = child_result_opt.expect("Thread should have properly properly returned the child's status");
let child_result = child_result_opt.expect(
"Thread should have properly properly returned the child's status",
);
let exit_status = child_result.expect("The child should have no problem exiting");
assert_eq!(true, exit_status.success());
}
@ -270,11 +294,10 @@ mod tests {
let mut spawned = acmd.spawn();
let lines: Vec<String> = spawned.lines().into_iter().collect();
assert_eq!(lines, vec![
"hi",
"Non-UTF8 data omitted from the log.",
"there",
]);
assert_eq!(
lines,
vec!["hi", "Non-UTF8 data omitted from the log.", "there"]
);
let ret = spawned.wait().unwrap().unwrap().unwrap().success();
assert_eq!(true, ret);
}

3
ofborg/src/test_scratch.rs
View file

@ -3,7 +3,7 @@ use std::path::{Path,PathBuf};
use std::process::Command;
pub struct TestScratch {
root: PathBuf
root: PathBuf,
}
impl TestScratch {
@ -34,7 +34,6 @@ impl TestScratch {
pub fn path(&self) -> PathBuf {
self.root.clone()
}
}
impl Drop for TestScratch {