zenoh_protocol/network/
push.rs

1use zenoh_buffers::buffer::Buffer;
2
3//
4// Copyright (c) 2022 ZettaScale Technology
5//
6// This program and the accompanying materials are made available under the
7// terms of the Eclipse Public License 2.0 which is available at
8// http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
9// which is available at https://www.apache.org/licenses/LICENSE-2.0.
10//
11// SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
12//
13// Contributors:
14//   ZettaScale Zenoh Team, <zenoh@zettascale.tech>
15//
16#[cfg(feature = "test")]
17use crate::zenoh::Put;
18use crate::{core::WireExpr, zenoh::PushBody};
19
20pub mod flag {
21    pub const N: u8 = 1 << 5; // 0x20 Named         if N==1 then the key expr has name/suffix
22    pub const M: u8 = 1 << 6; // 0x40 Mapping       if M==1 then key expr mapping is the one declared by the sender, else it is the one declared by the receiver
23    pub const Z: u8 = 1 << 7; // 0x80 Extensions    if Z==1 then an extension will follow
24}
25
26/// ```text
27/// Flags:
28/// - N: Named          If N==1 then the key expr has name/suffix
29/// - M: Mapping        if M==1 then key expr mapping is the one declared by the sender, else it is the one declared by the receiver
30/// - Z: Extension      If Z==1 then at least one extension is present
31///
32///  7 6 5 4 3 2 1 0
33/// +-+-+-+-+-+-+-+-+
34/// |Z|M|N|  PUSH   |
35/// +-+-+-+---------+
36/// ~ key_scope:z16 ~
37/// +---------------+
38/// ~  key_suffix   ~  if N==1 -- <u8;z16>
39/// +---------------+
40/// ~  [push_exts]  ~  if Z==1
41/// +---------------+
42/// ~   PushBody    ~
43/// +---------------+
44/// ```
45#[derive(Debug, Clone, PartialEq, Eq)]
46pub struct Push {
47    pub wire_expr: WireExpr<'static>,
48    pub ext_qos: ext::QoSType,
49    pub ext_tstamp: Option<ext::TimestampType>,
50    pub ext_nodeid: ext::NodeIdType,
51    pub payload: PushBody,
52}
53
54pub mod ext {
55    use crate::{zextz64, zextzbuf};
56
57    pub type QoS = zextz64!(0x1, false);
58    pub type QoSType = crate::network::ext::QoSType<{ QoS::ID }>;
59
60    pub type Timestamp = zextzbuf!(0x2, false);
61    pub type TimestampType = crate::network::ext::TimestampType<{ Timestamp::ID }>;
62
63    pub type NodeId = zextz64!(0x3, true);
64    pub type NodeIdType = crate::network::ext::NodeIdType<{ NodeId::ID }>;
65}
66
67impl Push {
68    pub fn payload_size(&self) -> usize {
69        match &self.payload {
70            PushBody::Put(p) => {
71                p.payload.len() + p.ext_attachment.as_ref().map_or(0, |a| a.buffer.len())
72            }
73            PushBody::Del(d) => d.ext_attachment.as_ref().map_or(0, |a| a.buffer.len()),
74        }
75    }
76
77    #[cfg(feature = "test")]
78    #[doc(hidden)]
79    pub fn rand() -> Self {
80        use rand::Rng;
81
82        let mut rng = rand::thread_rng();
83        let wire_expr = WireExpr::rand();
84        let payload = PushBody::rand();
85        let ext_qos = ext::QoSType::rand();
86        let ext_tstamp = rng.gen_bool(0.5).then(ext::TimestampType::rand);
87        let ext_nodeid = ext::NodeIdType::rand();
88
89        Self {
90            wire_expr,
91            payload,
92            ext_tstamp,
93            ext_qos,
94            ext_nodeid,
95        }
96    }
97}
98
99impl From<PushBody> for Push {
100    fn from(value: PushBody) -> Self {
101        Self {
102            wire_expr: WireExpr::empty(),
103            ext_qos: ext::QoSType::DEFAULT,
104            ext_tstamp: None,
105            ext_nodeid: ext::NodeIdType::DEFAULT,
106            payload: value,
107        }
108    }
109}
110
111#[cfg(feature = "test")]
112impl From<Put> for Push {
113    fn from(value: Put) -> Self {
114        PushBody::from(value).into()
115    }
116}
117
118#[cfg(feature = "test")]
119impl From<Vec<u8>> for Push {
120    fn from(value: Vec<u8>) -> Self {
121        Self::from(Put {
122            payload: value.into(),
123            ..Put::default()
124        })
125    }
126}
zenoh_protocol/network/push.rs

zenoh_protocol/network/
push.rs
