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
use std::io::{Error, ErrorKind, Result};
use crate::ids;
use crate::message::{self, Compressor, Outbound};
#[derive(
std::clone::Clone,
std::cmp::Eq,
std::cmp::Ord,
std::cmp::PartialEq,
std::cmp::PartialOrd,
std::fmt::Debug,
std::hash::Hash,
)]
pub struct Message {
pub chain_id: ids::Id,
pub request_id: u32,
pub summary: Vec<u8>,
}
impl Message {
pub fn create(
chain_id: ids::Id,
request_id: u32,
summary: Vec<u8>,
) -> impl Outbound + Compressor {
Self {
chain_id,
request_id,
summary,
}
}
}
impl std::fmt::Display for Message {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(f, "msg state_summary_frontier")
}
}
impl Outbound for Message {
fn serialize_with_header(&self) -> Result<bytes::Bytes> {
let type_id = message::TYPES
.get("state_summary_frontier")
.ok_or_else(|| Error::new(ErrorKind::InvalidInput, "unknown type name"))?;
let packer = message::default_packer_with_header();
packer.pack_byte(*type_id)?;
packer.pack_bool(false)?; packer.pack_bytes(self.chain_id.as_ref())?;
packer.pack_u32(self.request_id)?;
packer.pack_bytes_with_header(self.summary.as_ref())?;
Ok(packer.take_bytes())
}
}
#[test]
fn test_message() {
let msg = Message::create(ids::Id::empty(), 7, vec![0x01, 0x02, 0x03]);
let data_with_header = msg.serialize_with_header().unwrap();
let expected_data: &[u8] = &[
0x00, 0x00, 0x00, 0x2d, 0x18, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x07, 0x00, 0x00, 0x00, 0x03, 0x01, 0x02, 0x03, ];
assert!(cmp_manager::eq_vectors(&expected_data, &data_with_header));
}