feat: support rewrite config after parse_rule_config invoked (aliba…

…ba#401)
panxiaochao · Jun 28, 2023 · fc05a3b · fc05a3b
1 parent 9fc2760
commit fc05a3b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/plugins/wasm-rust/src/rule_matcher.rs b/plugins/wasm-rust/src/rule_matcher.rs
@@ -163,6 +163,17 @@ where
         return self.global_config.as_ref();
     }
 
+    pub fn rewrite_config(&mut self, rewrite: fn(config: &PluginConfig) -> PluginConfig) {
+        self.global_config = match self.global_config.as_ref() {
+            None => None,
+            Some(config) => Some(rewrite(config))
+        };
+
+        for rule_config in &mut self.rule_config {
+            rule_config.config = rewrite(&rule_config.config);
+        }
+    }
+
     fn parse_route_match_config(config: &Value) -> HashSet<String> {
         let empty_vec = Vec::new();
         let keys = config[MATCH_ROUTE_KEY].as_array().unwrap_or(&empty_vec);