rocketmq_remoting/protocol/header/
get_topic_stats_info_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::protocol::header::message_operation_header::TopicRequestHeaderTrait;
21use crate::protocol::header::namesrv::topic_operation_header::TopicRequestHeader;
22use crate::rpc::rpc_request_header::RpcRequestHeader;
23
24#[derive(Clone, Debug, Serialize, Deserialize, Default, RequestHeaderCodecV2)]
25#[serde(rename_all = "camelCase")]
26pub struct GetTopicStatsInfoRequestHeader {
27    #[required]
28    pub topic: CheetahString,
29
30    #[serde(flatten)]
31    pub topic_request_header: Option<TopicRequestHeader>,
32}
33
34impl TopicRequestHeaderTrait for GetTopicStatsInfoRequestHeader {
35    fn set_lo(&mut self, lo: Option<bool>) {
36        self.topic_request_header
37            .get_or_insert_with(TopicRequestHeader::default)
38            .lo = lo;
39    }
40
41    fn lo(&self) -> Option<bool> {
42        self.topic_request_header.as_ref().and_then(|header| header.lo)
43    }
44
45    fn set_topic(&mut self, topic: CheetahString) {
46        self.topic = topic;
47    }
48
49    fn topic(&self) -> &CheetahString {
50        &self.topic
51    }
52
53    fn broker_name(&self) -> Option<&CheetahString> {
54        self.topic_request_header
55            .as_ref()
56            .and_then(|header| header.rpc.as_ref())
57            .and_then(|rpc| rpc.broker_name.as_ref())
58    }
59
60    fn set_broker_name(&mut self, broker_name: CheetahString) {
61        self.topic_request_header
62            .get_or_insert_with(TopicRequestHeader::default)
63            .rpc
64            .get_or_insert_with(RpcRequestHeader::default)
65            .broker_name = Some(broker_name);
66    }
67
68    fn namespace(&self) -> Option<&str> {
69        self.topic_request_header
70            .as_ref()
71            .and_then(|header| header.rpc.as_ref())
72            .and_then(|rpc| rpc.namespace.as_deref())
73    }
74
75    fn set_namespace(&mut self, namespace: CheetahString) {
76        self.topic_request_header
77            .get_or_insert_with(TopicRequestHeader::default)
78            .rpc
79            .get_or_insert_with(RpcRequestHeader::default)
80            .namespace = Some(namespace);
81    }
82
83    fn namespaced(&self) -> Option<bool> {
84        self.topic_request_header
85            .as_ref()
86            .and_then(|header| header.rpc.as_ref())
87            .and_then(|rpc| rpc.namespaced)
88    }
89
90    fn set_namespaced(&mut self, namespaced: bool) {
91        self.topic_request_header
92            .get_or_insert_with(TopicRequestHeader::default)
93            .rpc
94            .get_or_insert_with(RpcRequestHeader::default)
95            .namespaced = Some(namespaced);
96    }
97
98    fn oneway(&self) -> Option<bool> {
99        self.topic_request_header
100            .as_ref()
101            .and_then(|header| header.rpc.as_ref())
102            .and_then(|rpc| rpc.oneway)
103    }
104
105    fn set_oneway(&mut self, oneway: bool) {
106        self.topic_request_header
107            .get_or_insert_with(TopicRequestHeader::default)
108            .rpc
109            .get_or_insert_with(RpcRequestHeader::default)
110            .oneway = Some(oneway);
111    }
112
113    fn queue_id(&self) -> i32 {
114        -1
115    }
116
117    fn set_queue_id(&mut self, _queue_id: i32) {}
118}
119
120#[cfg(test)]
121mod tests {
122    use cheetah_string::CheetahString;
123
124    use super::*;
125
126    #[test]
127    fn get_topic_stats_info_request_header_serializes_correctly() {
128        let header = GetTopicStatsInfoRequestHeader {
129            topic: CheetahString::from_static_str("test_topic"),
130            topic_request_header: None,
131        };
132        let serialized = serde_json::to_string(&header).unwrap();
133        let expected = r#"{"topic":"test_topic"}"#;
134        assert_eq!(serialized, expected);
135    }
136
137    #[test]
138    fn get_topic_stats_info_request_header_deserializes_correctly() {
139        let data = r#"{"topic":"test_topic"}"#;
140        let header: GetTopicStatsInfoRequestHeader = serde_json::from_str(data).unwrap();
141        assert_eq!(header.topic, CheetahString::from_static_str("test_topic"));
142        assert!(header.topic_request_header.is_some());
143    }
144
145    #[test]
146    fn get_topic_stats_info_request_header_handles_missing_optional_fields() {
147        let data = r#"{"topic":"test_topic"}"#;
148        let header: GetTopicStatsInfoRequestHeader = serde_json::from_str(data).unwrap();
149        assert_eq!(header.topic, CheetahString::from_static_str("test_topic"));
150        assert!(header.topic_request_header.is_some());
151    }
152
153    #[test]
154    fn get_topic_stats_info_request_header_handles_invalid_data() {
155        let data = r#"{"topic":12345}"#;
156        let result: Result<GetTopicStatsInfoRequestHeader, _> = serde_json::from_str(data);
157        assert!(result.is_err());
158    }
159}
rocketmq_remoting/protocol/header/get_topic_stats_info_request_header.rs

rocketmq_remoting/protocol/header/
get_topic_stats_info_request_header.rs
