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

#![cfg_attr(feature = "read-initializer", feature(read_initializer))]
#[macro_use]
extern crate enum_primitive_derive;

pub mod decode;
pub mod encode;
pub mod rpc;

/// Used when iterating over collections, to return either the next item or
/// indicate end of the collection, returning the underlying reader.
pub enum MsgPackOption<T, U> {
    Some(T),
    End(U),
}

impl<T, U> MsgPackOption<T, U> {
    /// Convert to an `Option`, dropping U
    pub fn into_option(self) -> Option<T> {
        match self {
            MsgPackOption::Some(t) => Some(t),
            MsgPackOption::End(_u) => None,
        }
    }

    pub fn unwrap(self) -> T {
        self.into_option().unwrap()
    }

    pub fn unwrap_end(self) -> U {
        if let MsgPackOption::End(u) = self {
            Some(u)
        } else {
            None
        }
        .unwrap()
    }
}