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

* fix: add role in meta RequestHeader
GreptimeTeam · May 12, 2023 · 6bfb020 · 6bfb020
1 parent 1552a21
commit 6bfb020
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/v1/meta.rs b/src/v1/meta.rs
@@ -57,12 +57,12 @@ impl Eq for Peer {}
 
 impl RequestHeader {
     #[inline]
-    pub fn new((cluster_id, member_id): (u64, u64)) -> Self {
+    pub fn new((cluster_id, member_id): (u64, u64), role: Role) -> Self {
         Self {
             protocol_version: PROTOCOL_VERSION,
             cluster_id,
             member_id,
-            role: 0,
+            role: role.into(),
         }
     }
 }
@@ -134,14 +134,15 @@ macro_rules! gen_set_header {
     ($req: ty) => {
         impl $req {
             #[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() {
                     Some(header) => {
                         header.cluster_id = cluster_id;
                         header.member_id = member_id;
+                        header.role = role.into();
                     }
                     None => {
-                        self.header = Some(RequestHeader::new((cluster_id, member_id)));
+                        self.header = Some(RequestHeader::new((cluster_id, member_id), role));
                     }
                 }
             }