asp/keep
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add convenience methods to filter service and improve filter chain processing

Browse Source 
Co-authored-by: aider (openai/andrew/openrouter/deepseek/deepseek-chat-v3.1) <aider@aider.chat>
This commit is contained in:
Andrew Phillips
2025-08-28 22:16:59 -03:00
parent 01b7046970
commit 2e40ab7a45
2 changed files with 35 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/filter_plugin/mod.rs
View File

@@ -31,7 +31,10 @@ impl FilterChain {
pub fn process(&mut self, data: &[u8]) -> Result<Vec<u8>> {
let mut current_data = data.to_vec();
for plugin in &mut self.plugins {
current_data = plugin.process(&current_data)?;
// Process the current data through the plugin
let processed = plugin.process(&current_data)?;
current_data = processed;
// Early exit if no data remains
if current_data.is_empty() {
break;
@@ -42,20 +45,15 @@ impl FilterChain {
pub fn finish(&mut self) -> Result<Vec<u8>> {
let mut result = Vec::new();
let mut all_data = Vec::new();
// Process each plugin's finish method and collect results
for plugin in &mut self.plugins {
let processed = plugin.finish()?;
if !processed.is_empty() {
all_data.extend(processed);
let finished_data = plugin.finish()?;
if !finished_data.is_empty() {
result.extend(finished_data);
}
}
// If we have any data from finish, use it
if !all_data.is_empty() {
result = all_data;
}
Ok(result)
}
}