Skip to main content

rocketmq_remoting/protocol/header/
get_all_subscription_group_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
20#[derive(Clone, Debug, Serialize, Deserialize, Default, RequestHeaderCodecV2)]
21#[serde(rename_all = "camelCase")]
22pub struct GetAllSubscriptionGroupRequestHeader {
23    #[required]
24    pub group_seq: i32,
25    #[serde(skip_serializing_if = "Option::is_none")]
26    pub data_version: Option<CheetahString>,
27    #[serde(skip_serializing_if = "Option::is_none")]
28    pub max_group_num: Option<i32>,
29}
30
31#[derive(Clone, Debug, Serialize, Deserialize, Default, RequestHeaderCodecV2)]
32#[serde(rename_all = "camelCase")]
33pub struct GetAllSubscriptionGroupResponseHeader {
34    #[required]
35    pub total_group_num: i32,
36}
37
38#[cfg(test)]
39mod tests {
40    use std::collections::HashMap;
41
42    use super::*;
43    use crate::protocol::command_custom_header::CommandCustomHeader;
44    use crate::protocol::command_custom_header::FromMap;
45
46    #[test]
47    fn get_all_subscription_group_request_header_maps_java_fields() {
48        let header = GetAllSubscriptionGroupRequestHeader {
49            group_seq: 7,
50            data_version: Some(CheetahString::from_static_str("dv-1")),
51            max_group_num: Some(32),
52        };
53
54        let map = header.to_map().expect("header should encode");
55        assert_eq!(map.get("groupSeq").map(|value| value.as_str()), Some("7"));
56        assert_eq!(map.get("dataVersion").map(|value| value.as_str()), Some("dv-1"));
57        assert_eq!(map.get("maxGroupNum").map(|value| value.as_str()), Some("32"));
58
59        let decoded = <GetAllSubscriptionGroupRequestHeader as FromMap>::from(&map).expect("header should decode");
60        assert_eq!(decoded.group_seq, 7);
61        assert_eq!(decoded.data_version.as_deref(), Some("dv-1"));
62        assert_eq!(decoded.max_group_num, Some(32));
63    }
64
65    #[test]
66    fn get_all_subscription_group_request_header_requires_group_seq_like_java_header() {
67        let result = <GetAllSubscriptionGroupRequestHeader as FromMap>::from(&HashMap::new());
68
69        assert!(result.is_err());
70    }
71
72    #[test]
73    fn get_all_subscription_group_response_header_maps_total_group_num() {
74        let header = GetAllSubscriptionGroupResponseHeader { total_group_num: 17 };
75
76        let map = header.to_map().expect("header should encode");
77        assert_eq!(map.get("totalGroupNum").map(|value| value.as_str()), Some("17"));
78
79        let decoded = <GetAllSubscriptionGroupResponseHeader as FromMap>::from(&map).expect("header should decode");
80        assert_eq!(decoded.total_group_num, 17);
81    }
82}