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 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
//! # loop_unwrap //! Provides utility macros for unwrapping during loops. /// Works like `.unwrap`, if it's an Err or None, it calls `continue` on the loop. /// Prints an error message with `println!()` if provided. /// # Examples /// ``` /// loop { /// let input = "Not a number"; /// let parsed_input: i32 = unwrap_continue!(input.parse()); //parse returns Err for this input /// break; //<-- never reached, since `continue` is called. /// } /// ``` /// ``` /// loop { /// let input = "Not a number"; /// let parsed_input: i32 = unwrap_continue!(input.parse(), "Please Enter a Number!"); /// // "Please Enter a Number!" is printed in console with a `println!()` /// break; /// } /// ``` /// ``` /// loop { /// let some_value: i32 = unwrap_continue!(Some(32), "Please Enter a Number!"); /// assert_eq!(some_value, 32_i32) /// } /// ``` #[macro_export] macro_rules! unwrap_continue { ($x:expr, $err_msg:expr) => { match $x.to_option() { Some(v) => v, None => { println!("{}", $err_msg); continue; } } }; ($x:expr) => { match $x.to_option() { Some(v) => v, None => { continue; } } }; } /// Almost the same as `unwrap_continue()!`, just breaks the loop instead. /// Works like `.unwrap`, if it's an Err or None, it calls `break` on the loop. /// Prints an error message with `println!()` if provided. /// # Examples /// ``` /// loop { /// let input = "Not a number"; /// let parsed_input: i32 = unwrap_break!(input.parse()); //parse returns Err for this input /// } /// println!("This line will be reached."); /// ``` /// ``` /// loop { /// let input = "Not a number"; /// let parsed_input: i32 = unwrap_break!(input.parse(), "Please Enter a Number!"); /// // "Please Enter a Number!" is printed in console with a `println!()` /// //loop breaks /// } /// ``` /// ``` /// loop { /// let some_value: i32 = unwrap_break!(Some(32), "Please Enter a Number!"); /// assert_eq!(some_value, 32_i32) /// //no breakage here. /// } /// ``` #[macro_export] macro_rules! unwrap_break { ($x:expr, $err_msg:expr) => { match $x.to_option() { Some(v) => v, None => { println!("{}", $err_msg); break; } } }; ($x:expr) => { match $x.to_option() { Some(v) => v, None => { break; } } }; } pub trait ToOption<T> { fn to_option(self) -> Option<T>; } impl<T> ToOption<T> for Option<T> { fn to_option(self) -> Option<T> { self } } impl<T, U> ToOption<T> for Result<T, U> { fn to_option(self) -> Option<T> { match self { Ok(v) => Some(v), Err(_) => None, } } }