p2p/network/kad/
p2p_network_kad_message.rs

1// Automatically generated rust module for 'p2p_network_kad_message.proto' file
2
3#![allow(non_snake_case)]
4#![allow(non_upper_case_globals)]
5#![allow(non_camel_case_types)]
6#![allow(unused_imports)]
7#![allow(unknown_lints)]
8#![allow(clippy::all)]
9#![cfg_attr(rustfmt, rustfmt_skip)]
10
11
12use std::borrow::Cow;
13use quick_protobuf::{MessageInfo, MessageRead, MessageWrite, BytesReader, Writer, WriterBackend, Result};
14use quick_protobuf::sizeofs::*;
15use super::*;
16
17#[allow(clippy::derive_partial_eq_without_eq)]
18#[derive(Debug, Default, PartialEq, Clone)]
19pub struct Record<'a> {
20    pub key: Cow<'a, [u8]>,
21    pub value: Cow<'a, [u8]>,
22    pub timeReceived: Cow<'a, str>,
23}
24
25impl<'a> MessageRead<'a> for Record<'a> {
26    fn from_reader(r: &mut BytesReader, bytes: &'a [u8]) -> Result<Self> {
27        let mut msg = Self::default();
28        while !r.is_eof() {
29            match r.next_tag(bytes) {
30                Ok(10) => msg.key = r.read_bytes(bytes).map(Cow::Borrowed)?,
31                Ok(18) => msg.value = r.read_bytes(bytes).map(Cow::Borrowed)?,
32                Ok(42) => msg.timeReceived = r.read_string(bytes).map(Cow::Borrowed)?,
33                Ok(t) => { r.read_unknown(bytes, t)?; }
34                Err(e) => return Err(e),
35            }
36        }
37        Ok(msg)
38    }
39}
40
41impl<'a> MessageWrite for Record<'a> {
42    fn get_size(&self) -> usize {
43        0
44        + if self.key == Cow::Borrowed(b"") { 0 } else { 1 + sizeof_len((&self.key).len()) }
45        + if self.value == Cow::Borrowed(b"") { 0 } else { 1 + sizeof_len((&self.value).len()) }
46        + if self.timeReceived == "" { 0 } else { 1 + sizeof_len((&self.timeReceived).len()) }
47    }
48
49    fn write_message<W: WriterBackend>(&self, w: &mut Writer<W>) -> Result<()> {
50        if self.key != Cow::Borrowed(b"") { w.write_with_tag(10, |w| w.write_bytes(&**&self.key))?; }
51        if self.value != Cow::Borrowed(b"") { w.write_with_tag(18, |w| w.write_bytes(&**&self.value))?; }
52        if self.timeReceived != "" { w.write_with_tag(42, |w| w.write_string(&**&self.timeReceived))?; }
53        Ok(())
54    }
55}
56
57#[allow(clippy::derive_partial_eq_without_eq)]
58#[derive(Debug, Default, PartialEq, Clone)]
59pub struct Message<'a> {
60    pub type_pb: mod_Message::MessageType,
61    pub clusterLevelRaw: i32,
62    pub key: Cow<'a, [u8]>,
63    pub record: Option<Record<'a>>,
64    pub closerPeers: Vec<mod_Message::Peer<'a>>,
65    pub providerPeers: Vec<mod_Message::Peer<'a>>,
66}
67
68impl<'a> MessageRead<'a> for Message<'a> {
69    fn from_reader(r: &mut BytesReader, bytes: &'a [u8]) -> Result<Self> {
70        let mut msg = Self::default();
71        while !r.is_eof() {
72            match r.next_tag(bytes) {
73                Ok(8) => msg.type_pb = r.read_enum(bytes)?,
74                Ok(80) => msg.clusterLevelRaw = r.read_int32(bytes)?,
75                Ok(18) => msg.key = r.read_bytes(bytes).map(Cow::Borrowed)?,
76                Ok(26) => msg.record = Some(r.read_message::<Record>(bytes)?),
77                Ok(66) => msg.closerPeers.push(r.read_message::<mod_Message::Peer>(bytes)?),
78                Ok(74) => msg.providerPeers.push(r.read_message::<mod_Message::Peer>(bytes)?),
79                Ok(t) => { r.read_unknown(bytes, t)?; }
80                Err(e) => return Err(e),
81            }
82        }
83        Ok(msg)
84    }
85}
86
87impl<'a> MessageWrite for Message<'a> {
88    fn get_size(&self) -> usize {
89        0
90        + if self.type_pb == p2p_network_kad_message::mod_Message::MessageType::PUT_VALUE { 0 } else { 1 + sizeof_varint(*(&self.type_pb) as u64) }
91        + if self.clusterLevelRaw == 0i32 { 0 } else { 1 + sizeof_varint(*(&self.clusterLevelRaw) as u64) }
92        + if self.key == Cow::Borrowed(b"") { 0 } else { 1 + sizeof_len((&self.key).len()) }
93        + self.record.as_ref().map_or(0, |m| 1 + sizeof_len((m).get_size()))
94        + self.closerPeers.iter().map(|s| 1 + sizeof_len((s).get_size())).sum::<usize>()
95        + self.providerPeers.iter().map(|s| 1 + sizeof_len((s).get_size())).sum::<usize>()
96    }
97
98    fn write_message<W: WriterBackend>(&self, w: &mut Writer<W>) -> Result<()> {
99        if self.type_pb != p2p_network_kad_message::mod_Message::MessageType::PUT_VALUE { w.write_with_tag(8, |w| w.write_enum(*&self.type_pb as i32))?; }
100        if self.clusterLevelRaw != 0i32 { w.write_with_tag(80, |w| w.write_int32(*&self.clusterLevelRaw))?; }
101        if self.key != Cow::Borrowed(b"") { w.write_with_tag(18, |w| w.write_bytes(&**&self.key))?; }
102        if let Some(ref s) = self.record { w.write_with_tag(26, |w| w.write_message(s))?; }
103        for s in &self.closerPeers { w.write_with_tag(66, |w| w.write_message(s))?; }
104        for s in &self.providerPeers { w.write_with_tag(74, |w| w.write_message(s))?; }
105        Ok(())
106    }
107}
108
109pub mod mod_Message {
110
111use std::borrow::Cow;
112use super::*;
113
114#[allow(clippy::derive_partial_eq_without_eq)]
115#[derive(Debug, Default, PartialEq, Clone)]
116pub struct Peer<'a> {
117    pub id: Cow<'a, [u8]>,
118    pub addrs: Vec<Cow<'a, [u8]>>,
119    pub connection: mod_Message::ConnectionType,
120}
121
122impl<'a> MessageRead<'a> for Peer<'a> {
123    fn from_reader(r: &mut BytesReader, bytes: &'a [u8]) -> Result<Self> {
124        let mut msg = Self::default();
125        while !r.is_eof() {
126            match r.next_tag(bytes) {
127                Ok(10) => msg.id = r.read_bytes(bytes).map(Cow::Borrowed)?,
128                Ok(18) => msg.addrs.push(r.read_bytes(bytes).map(Cow::Borrowed)?),
129                Ok(24) => msg.connection = r.read_enum(bytes)?,
130                Ok(t) => { r.read_unknown(bytes, t)?; }
131                Err(e) => return Err(e),
132            }
133        }
134        Ok(msg)
135    }
136}
137
138impl<'a> MessageWrite for Peer<'a> {
139    fn get_size(&self) -> usize {
140        0
141        + if self.id == Cow::Borrowed(b"") { 0 } else { 1 + sizeof_len((&self.id).len()) }
142        + self.addrs.iter().map(|s| 1 + sizeof_len((s).len())).sum::<usize>()
143        + if self.connection == p2p_network_kad_message::mod_Message::ConnectionType::NOT_CONNECTED { 0 } else { 1 + sizeof_varint(*(&self.connection) as u64) }
144    }
145
146    fn write_message<W: WriterBackend>(&self, w: &mut Writer<W>) -> Result<()> {
147        if self.id != Cow::Borrowed(b"") { w.write_with_tag(10, |w| w.write_bytes(&**&self.id))?; }
148        for s in &self.addrs { w.write_with_tag(18, |w| w.write_bytes(&**s))?; }
149        if self.connection != p2p_network_kad_message::mod_Message::ConnectionType::NOT_CONNECTED { w.write_with_tag(24, |w| w.write_enum(*&self.connection as i32))?; }
150        Ok(())
151    }
152}
153
154#[derive(Debug, PartialEq, Eq, Clone, Copy)]
155pub enum MessageType {
156    PUT_VALUE = 0,
157    GET_VALUE = 1,
158    ADD_PROVIDER = 2,
159    GET_PROVIDERS = 3,
160    FIND_NODE = 4,
161    PING = 5,
162}
163
164impl Default for MessageType {
165    fn default() -> Self {
166        MessageType::PUT_VALUE
167    }
168}
169
170impl From<i32> for MessageType {
171    fn from(i: i32) -> Self {
172        match i {
173            0 => MessageType::PUT_VALUE,
174            1 => MessageType::GET_VALUE,
175            2 => MessageType::ADD_PROVIDER,
176            3 => MessageType::GET_PROVIDERS,
177            4 => MessageType::FIND_NODE,
178            5 => MessageType::PING,
179            _ => Self::default(),
180        }
181    }
182}
183
184impl<'a> From<&'a str> for MessageType {
185    fn from(s: &'a str) -> Self {
186        match s {
187            "PUT_VALUE" => MessageType::PUT_VALUE,
188            "GET_VALUE" => MessageType::GET_VALUE,
189            "ADD_PROVIDER" => MessageType::ADD_PROVIDER,
190            "GET_PROVIDERS" => MessageType::GET_PROVIDERS,
191            "FIND_NODE" => MessageType::FIND_NODE,
192            "PING" => MessageType::PING,
193            _ => Self::default(),
194        }
195    }
196}
197
198#[derive(Debug, PartialEq, Eq, Clone, Copy)]
199pub enum ConnectionType {
200    NOT_CONNECTED = 0,
201    CONNECTED = 1,
202    CAN_CONNECT = 2,
203    CANNOT_CONNECT = 3,
204}
205
206impl Default for ConnectionType {
207    fn default() -> Self {
208        ConnectionType::NOT_CONNECTED
209    }
210}
211
212impl From<i32> for ConnectionType {
213    fn from(i: i32) -> Self {
214        match i {
215            0 => ConnectionType::NOT_CONNECTED,
216            1 => ConnectionType::CONNECTED,
217            2 => ConnectionType::CAN_CONNECT,
218            3 => ConnectionType::CANNOT_CONNECT,
219            _ => Self::default(),
220        }
221    }
222}
223
224impl<'a> From<&'a str> for ConnectionType {
225    fn from(s: &'a str) -> Self {
226        match s {
227            "NOT_CONNECTED" => ConnectionType::NOT_CONNECTED,
228            "CONNECTED" => ConnectionType::CONNECTED,
229            "CAN_CONNECT" => ConnectionType::CAN_CONNECT,
230            "CANNOT_CONNECT" => ConnectionType::CANNOT_CONNECT,
231            _ => Self::default(),
232        }
233    }
234}
235
236}
237
p2p/network/kad/p2p_network_kad_message.rs

p2p/network/kad/
p2p_network_kad_message.rs
