feat(wasm): 优化 PluginHttpWrapper 中 res_headers 获取

alibaba · Sep 10, 2024 · cea757a · cea757a
1 parent ebf482c
commit cea757a
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/plugins/wasm-rust/src/plugin_wrapper.rs b/plugins/wasm-rust/src/plugin_wrapper.rs
@@ -155,7 +155,7 @@ where
                 Err(_) => {
                     log(
                         LogLevel::Warn,
-                        format!("http header contains non-ASCII characters header: {}", k ).as_str(),
+                        format!("request http header contains non-ASCII characters header: {}", k ).as_str(),
                     ).unwrap();
                 }
             }
@@ -202,9 +202,20 @@ where
         num_headers: usize,
         end_of_stream: bool,
     ) -> HeaderAction {
-        for (k, v) in self.get_http_response_headers() {
-            self.res_headers.insert(k, v);
+        for (k, v) in self.get_http_response_headers_bytes() {
+            match String::from_utf8(v) {
+                Ok(header_value) => {
+                    self.res_headers.insert(k, header_value);
+                }
+                Err(_) => {
+                    log(
+                        LogLevel::Warn,
+                        format!("response http header contains non-ASCII characters header: {}", k ).as_str(),
+                    ).unwrap();
+                }
+            }
         }
+
         let ret = self
             .http_content
             .on_http_response_headers(num_headers, end_of_stream);