use cheetah_string::CheetahString;
use rocketmq_macros::RequestHeaderCodec;
use serde::Deserialize;
use serde::Serialize;
#[derive(Clone, Debug, Default, Serialize, Deserialize, RequestHeaderCodec)]
#[serde(rename_all = "camelCase")]
#[header(
type_id = "rocketmq_protocol::protocol::header::add_broker_request_header::AddBrokerRequestHeader",
java_class = "org.apache.rocketmq.remoting.protocol.header.AddBrokerRequestHeader"
)]
pub struct AddBrokerRequestHeader {
pub config_path: Option<CheetahString>,
}
#[cfg(test)]
mod tests {
use std::collections::HashMap;
use super::*;
use crate::protocol::command_custom_header::CommandCustomHeader;
use crate::protocol::command_custom_header::FromMap;
#[test]
fn add_broker_request_header_serializes_config_path() {
let header = AddBrokerRequestHeader {
config_path: Some(CheetahString::from_static_str("/tmp/broker.conf")),
};
let map = header.to_map().expect("header should encode");
assert_eq!(
map.get(&CheetahString::from_static_str("configPath")),
Some(&CheetahString::from_static_str("/tmp/broker.conf"))
);
}
#[test]
fn add_broker_request_header_deserializes_config_path() {
let mut map = HashMap::new();
map.insert(
CheetahString::from_static_str("configPath"),
CheetahString::from_static_str("/tmp/broker.conf"),
);
let header = <AddBrokerRequestHeader as FromMap>::from(&map).expect("header should decode");
assert_eq!(
header.config_path,
Some(CheetahString::from_static_str("/tmp/broker.conf"))
);
}
}