From 1552a21e77752f1baf8062b937584b80de84e0b3 Mon Sep 17 00:00:00 2001
From: fys <40801205+Fengys123@users.noreply.github.com>
Date: Wed, 10 May 2023 11:29:26 +0800
Subject: [PATCH] fix: set header (#32)

---
 src/v1/meta.rs | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/src/v1/meta.rs b/src/v1/meta.rs
index ba55931..4b70175 100644
--- a/src/v1/meta.rs
+++ b/src/v1/meta.rs
@@ -130,13 +130,20 @@ impl HeartbeatResponse {
         false
     }
 }
-
 macro_rules! gen_set_header {
     ($req: ty) => {
         impl $req {
             #[inline]
             pub fn set_header(&mut self, (cluster_id, member_id): (u64, u64)) {
-                self.header = Some(RequestHeader::new((cluster_id, member_id)));
+                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)));
+                    }
+                }
             }
         }
     };