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

use crate::MemoryState;
use core::fmt;

pub struct Error<T> {
    pub state: MemoryState,
    pub input: T,
    pub retry: bool,
}
impl<T> fmt::Debug for Error<T> {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        fmt::Debug::fmt(&self.state, f)
    }
}

pub fn retry<I, O, F>(mut f: F, mut input: I) -> Result<O, Error<I>>
where
    F: FnMut(I) -> Result<O, Error<I>>,
{
    loop {
        match f(input) {
            Ok(val) => return Ok(val),
            Err(err) if err.retry => {
                input = err.input;
                spin_loop::spin();
                continue;
            }
            Err(err) => return Err(err),
        }
    }
}