fix: add role paramter in meta RequestHeader::new() (#34)

* fix: add role in meta RequestHeader
This commit is contained in:
fys
2023-05-12 15:43:47 +08:00
committed by GitHub
parent 1552a21e77
commit 6bfb02057c
+5 -4
View File
@@ -57,12 +57,12 @@ impl Eq for Peer {}
impl RequestHeader { impl RequestHeader {
#[inline] #[inline]
pub fn new((cluster_id, member_id): (u64, u64)) -> Self { pub fn new((cluster_id, member_id): (u64, u64), role: Role) -> Self {
Self { Self {
protocol_version: PROTOCOL_VERSION, protocol_version: PROTOCOL_VERSION,
cluster_id, cluster_id,
member_id, member_id,
role: 0, role: role.into(),
} }
} }
} }
@@ -134,14 +134,15 @@ 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), role: Role) {
match self.header.as_mut() { match self.header.as_mut() {
Some(header) => { Some(header) => {
header.cluster_id = cluster_id; header.cluster_id = cluster_id;
header.member_id = member_id; header.member_id = member_id;
header.role = role.into();
} }
None => { None => {
self.header = Some(RequestHeader::new((cluster_id, member_id))); self.header = Some(RequestHeader::new((cluster_id, member_id), role));
} }
} }
} }