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
#[cfg(test)]
mod tests {
    #[test]
    fn it_works() {
        assert_eq!(
            crate::partition_eithers(vec![Ok(1), Err("foo"), Ok(2), Ok(3), Err("bar")]),
            (vec![1, 2, 3], vec!["foo", "bar"])
        );
    }

    #[test]
    fn it_works2() {
        assert_eq!(
            crate::collect_any_errors(vec![Ok(100), Ok(200), Ok(300), Err("bar")]),
            Err(vec!["bar"])
        );
    }

    #[test]
    fn it_works3() {
        assert_eq!(
            crate::collect_any_errors::<u32, String>(vec![Ok(100), Ok(200), Ok(300)]),
            Ok(vec![100, 200, 300])
        )
    }
}

pub fn partition_eithers<T, E>(a: Vec<Result<T, E>>) -> (Vec<T>, Vec<E>) {
    let mut ts = Vec::new();
    let mut es = Vec::new();
    for q in a {
        match q {
            Ok(t) => ts.push(t),
            Err(e) => es.push(e),
        }
    }
    (ts, es)
}

// return Ok if all are Ok
pub fn collect_any_errors<T, E>(a: Vec<Result<T, E>>) -> Result<Vec<T>, Vec<E>> {
    let (ts, es) = partition_eithers(a);
    if es.is_empty() {
        Ok(ts)
    } else {
        Err(es)
    }
}