fix(lust-thrift): client RpcCx didn't clear extension

cloudwego · Nov 29, 2023 · ac5d9fe · ac5d9fe
1 parent 6145eed
commit ac5d9fe
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/volo-thrift/Cargo.toml b/volo-thrift/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "volo-thrift"
-version = "0.8.1"
+version = "0.8.2"
 edition.workspace = true
 homepage.workspace = true
 repository.workspace = true

diff --git a/volo-thrift/src/context.rs b/volo-thrift/src/context.rs
@@ -209,12 +209,13 @@ impl ClientContext {
 
     #[inline]
     pub fn reset(&mut self, seq_id: i32, msg_type: TMessageType) {
-        self.rpc_info.clear();
         self.seq_id = seq_id;
         self.message_type = msg_type;
         self.transport.should_reuse = true;
         self.stats.reset();
         self.common_stats.reset();
+        // self.0 is RpcCx, this reset will clear rpcinfo and extension
+        self.0.reset(self.0.inner);
     }
 }