From 6bfb02057c40da0e397c0cb4f6b87bd769669d50 Mon Sep 17 00:00:00 2001
From: fys <40801205+Fengys123@users.noreply.github.com>
Date: Fri, 12 May 2023 15:43:47 +0800
Subject: [PATCH] fix: add role paramter in meta RequestHeader::new() (#34)

* fix: add role in meta RequestHeader
---
 src/v1/meta.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/v1/meta.rs b/src/v1/meta.rs
index 4b701756..038946e9 100644
--- 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));
                     }
                 }
             }