#[derive(Debug, Clone)]
pub enum Frame<T, B, E> {
Message {
message: T,
body: bool,
},
Body {
chunk: Option<B>,
},
Error {
error: E,
},
}
impl<T, B, E> Frame<T, B, E> {
pub fn unwrap_msg(self) -> T {
match self {
Frame::Message { message, .. } => message,
Frame::Body { .. } => panic!("called `Frame::unwrap_msg()` on a `Body` value"),
Frame::Error { .. } => panic!("called `Frame::unwrap_msg()` on an `Error` value"),
}
}
pub fn unwrap_body(self) -> Option<B> {
match self {
Frame::Body { chunk } => chunk,
Frame::Message { .. } => panic!("called `Frame::unwrap_body()` on a `Message` value"),
Frame::Error { .. } => panic!("called `Frame::unwrap_body()` on an `Error` value"),
}
}
pub fn unwrap_err(self) -> E {
match self {
Frame::Error { error } => error,
Frame::Body { .. } => panic!("called `Frame::unwrap_err()` on a `Body` value"),
Frame::Message { .. } => panic!("called `Frame::unwrap_err()` on a `Message` value"),
}
}
}