use bytes::{Bytes, TaggedBytes}; use common::Command; use ser::Stream; use serialization::serialize_payload; use zebra_network::Magic; use {MessageHeader, MessageResult, Payload}; pub fn to_raw_message(magic: Magic, command: Command, payload: &Bytes) -> Bytes { let header = MessageHeader::for_data(magic, command, payload); let mut stream = Stream::default(); stream.append(&header); stream.append_slice(payload); stream.out() } pub struct Message { bytes: TaggedBytes, } impl Message where T: Payload, { pub fn new(magic: Magic, version: u32, payload: &T) -> MessageResult { let serialized = try!(serialize_payload(payload, version)); let message = Message { bytes: TaggedBytes::new(to_raw_message(magic, T::command().into(), &serialized)), }; Ok(message) } pub fn len(&self) -> usize { self.bytes.len() } } impl AsRef<[u8]> for Message { fn as_ref(&self) -> &[u8] { self.bytes.as_ref() } } impl From> for Bytes { fn from(m: Message) -> Self { m.bytes.into_raw() } }