rocketmq_protocol/protocol/header/
update_user_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(Debug, Clone, Serialize, Deserialize, Default, RequestHeaderCodec)]
21#[serde(rename_all = "camelCase")]
22#[header(
23    type_id = "rocketmq_protocol::protocol::header::update_user_request_header::UpdateUserRequestHeader",
24    java_class = "org.apache.rocketmq.remoting.protocol.header.UpdateUserRequestHeader"
25)]
26pub struct UpdateUserRequestHeader {
27    #[header(default, default_semantic = "literal:")]
28    pub username: CheetahString,
29}
30impl UpdateUserRequestHeader {
31    pub fn set_username(&mut self, username: CheetahString) {
32        self.username = username;
33    }
34}
35
36#[cfg(test)]
37mod tests {
38    use std::collections::HashMap;
39
40    use super::*;
41    use crate::protocol::command_custom_header::{CommandCustomHeader, FromMap};
42
43    #[test]
44    fn serde_and_request_header_codec_preserve_the_username() {
45        let mut header = UpdateUserRequestHeader::default();
46        header.set_username(CheetahString::from("user-a"));
47        let json = serde_json::to_string(&header).unwrap();
48        let map = header.to_map().unwrap();
49
50        assert_eq!(json, r#"{"username":"user-a"}"#);
51        assert_eq!(
52            serde_json::from_str::<UpdateUserRequestHeader>(&json).unwrap().username,
53            "user-a"
54        );
55        assert_eq!(
56            <UpdateUserRequestHeader as FromMap>::from(&map).unwrap().username,
57            "user-a"
58        );
59        assert!(<UpdateUserRequestHeader as FromMap>::from(&HashMap::new())
60            .unwrap()
61            .username
62            .is_empty());
63    }
64}
rocketmq_protocol/protocol/header/update_user_request_header.rs

rocketmq_protocol/protocol/header/
update_user_request_header.rs
