fix: set header (#32)

This commit is contained in:
fys
2023-05-10 11:29:26 +08:00
committed by GitHub
parent b450b690ef
commit 1552a21e77
+8 -1
View File
@@ -130,15 +130,22 @@ impl HeartbeatResponse {
false false
} }
} }
macro_rules! gen_set_header { macro_rules! gen_set_header {
($req: ty) => { ($req: ty) => {
impl $req { impl $req {
#[inline] #[inline]
pub fn set_header(&mut self, (cluster_id, member_id): (u64, u64)) { pub fn set_header(&mut self, (cluster_id, member_id): (u64, u64)) {
match self.header.as_mut() {
Some(header) => {
header.cluster_id = cluster_id;
header.member_id = member_id;
}
None => {
self.header = Some(RequestHeader::new((cluster_id, member_id))); self.header = Some(RequestHeader::new((cluster_id, member_id)));
} }
} }
}
}
}; };
} }