pg-setup 0.2.0

Helper to create and drop postgres DBs. Useful for testing.
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
54
55
56
57
58
59
60
61

use std::io::{BufRead, Read};
use std::process;

use crate::error::Error;

use super::error::Result;

#[allow(unused)]
pub fn cmd(cmd: impl AsRef<str>) -> Result<process::Child> {
    trace!("running command {}", cmd.as_ref());
    let mut args = cmd.as_ref().split(' ');
    let cmd = args.next().unwrap();
    cmd_with_args(cmd, args)
}

pub fn cmd_with_args<I, S>(cmd: impl AsRef<str>, args: I) -> Result<process::Child>
where
    I: IntoIterator<Item = S> + Clone,
    S: AsRef<std::ffi::OsStr> + std::fmt::Debug,
{
    trace!(
        "running command {} {:?}",
        cmd.as_ref(),
        args.clone().into_iter().collect::<Vec<_>>()
    );

    let mut proc = std::process::Command::new(cmd.as_ref())
        .args(args)
        .stdout(process::Stdio::piped())
        .stderr(process::Stdio::piped())
        .spawn()
        .map_err(Error::ProcessError)?;

    fn fwd_stream(stream: Option<impl Read + Send + 'static>) {
        if let Some(stream) = stream {
            std::thread::spawn(move || {
                let mut reader = std::io::BufReader::new(stream);
                let mut line = String::new();
                loop {
                    match reader.read_line(&mut line) {
                        Err(err) => {
                            eprintln!("error reading line: {err}");
                        }
                        Ok(0) => {
                            break;
                        }
                        Ok(_) => {
                            debug!("{}", line.trim_end());
                            line.clear();
                        }
                    }
                }
            });
        }
    }

    fwd_stream(proc.stdout.take());
    fwd_stream(proc.stderr.take());

    Ok(proc)
}