fix: add role paramter in meta RequestHeader::new() (#34)
* fix: add role in meta RequestHeader
This commit is contained in:
+5
-4
@@ -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));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user