45 lines
1.2 KiB
Markdown
45 lines
1.2 KiB
Markdown
|
```rust title="Rust"
|
||
|
|
use kreuzberg::plugins::{Plugin, PostProcessor, ProcessingStage};
|
||
|
|
use kreuzberg::{Result, ExtractionResult, ExtractionConfig};
|
||
|
|
use async_trait::async_trait;
|
||
|
|
|
||
|
|
struct WordCountProcessor;
|
||
|
|
|
||
|
|
impl Plugin for WordCountProcessor {
|
||
|
|
fn name(&self) -> &str { "word-count" }
|
||
|
|
fn version(&self) -> String { "1.0.0".to_string() }
|
||
|
|
fn initialize(&self) -> Result<()> { Ok(()) }
|
||
|
|
fn shutdown(&self) -> Result<()> { Ok(()) }
|
||
|
|
}
|
||
|
|
|
||
|
|
#[async_trait]
|
||
|
|
impl PostProcessor for WordCountProcessor {
|
||
|
|
async fn process(
|
||
|
|
&self,
|
||
|
|
result: &mut ExtractionResult,
|
||
|
|
_config: &ExtractionConfig
|
||
|
|
) -> Result<()> {
|
||
|
|
let word_count = result.content.split_whitespace().count();
|
||
|
|
|
||
|
|
result.processing_warnings.push(ProcessingWarning {
|
||
|
|
source: "word-count".to_string(),
|
||
|
|
message: format!("Processed with word count: {}", word_count)
|
||
|
|
});
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
|
||
|
|
fn processing_stage(&self) -> ProcessingStage {
|
||
|
|
ProcessingStage::Early
|
||
|
|
}
|
||
|
|
|
||
|
|
fn should_process(
|
||
|
|
&self,
|
||
|
|
result: &ExtractionResult,
|
||
|
|
_config: &ExtractionConfig
|
||
|
|
) -> bool {
|
||
|
|
!result.content.is_empty()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
```
|