Skip to main content

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