rocketmq_remoting/rpc/
topic_request_header.rs

1// Copyright 2023 The RocketMQ Rust Authors
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use cheetah_string::CheetahString;
16use rocketmq_macros::RequestHeaderCodecV2;
17use serde::Deserialize;
18use serde::Serialize;
19
20use crate::rpc::rpc_request_header::RpcRequestHeader;
21
22#[derive(Clone, Debug, Serialize, Deserialize, Default, RequestHeaderCodecV2)]
23pub struct TopicRequestHeader {
24    #[serde(flatten)]
25    pub rpc_request_header: Option<RpcRequestHeader>,
26    pub lo: Option<bool>,
27}
28
29impl TopicRequestHeader {
30    pub fn get_lo(&self) -> Option<&bool> {
31        self.lo.as_ref()
32    }
33
34    pub fn set_lo(&mut self, lo: bool) {
35        self.lo = Some(lo);
36    }
37
38    pub fn get_broker_name(&self) -> Option<&CheetahString> {
39        self.rpc_request_header.as_ref().and_then(|v| v.broker_name.as_ref())
40    }
41}
42
43#[cfg(test)]
44mod tests {
45    use cheetah_string::CheetahString;
46
47    use super::*;
48
49    #[test]
50    fn get_lo_returns_none_when_lo_is_none() {
51        let header = TopicRequestHeader::default();
52        assert_eq!(header.get_lo(), None);
53    }
54
55    #[test]
56    fn get_lo_returns_some_when_lo_is_set() {
57        let mut header = TopicRequestHeader::default();
58        header.set_lo(true);
59        assert_eq!(header.get_lo(), Some(&true));
60    }
61
62    #[test]
63    fn set_lo_updates_lo_value() {
64        let mut header = TopicRequestHeader::default();
65        header.set_lo(false);
66        assert_eq!(header.get_lo(), Some(&false));
67    }
68
69    #[test]
70    fn get_broker_name_returns_none_when_rpc_request_header_is_none() {
71        let header = TopicRequestHeader::default();
72        assert_eq!(header.get_broker_name(), None);
73    }
74
75    #[test]
76    fn get_broker_name_returns_none_when_broker_name_is_none() {
77        let rpc_header = RpcRequestHeader {
78            broker_name: None,
79            ..Default::default()
80        };
81        let header = TopicRequestHeader {
82            rpc_request_header: Some(rpc_header),
83            ..Default::default()
84        };
85        assert_eq!(header.get_broker_name(), None);
86    }
87
88    #[test]
89    fn get_broker_name_returns_some_when_broker_name_is_set() {
90        let broker_name = CheetahString::from("TestBroker");
91        let rpc_header = RpcRequestHeader {
92            broker_name: Some(broker_name.clone()),
93            ..Default::default()
94        };
95        let header = TopicRequestHeader {
96            rpc_request_header: Some(rpc_header),
97            ..Default::default()
98        };
99        assert_eq!(header.get_broker_name(), Some(&broker_name));
100    }
101}
rocketmq_remoting/rpc/topic_request_header.rs

rocketmq_remoting/rpc/
topic_request_header.rs
