mirror of
https://github.com/meilisearch/MeiliSearch
synced 2025-07-04 20:37:15 +02:00
27 lines
740 B
Rust
27 lines
740 B
Rust
use levenshtein_automata::{LevenshteinAutomatonBuilder, DFA};
|
|
|
|
pub struct LevBuilder {
|
|
automatons: [LevenshteinAutomatonBuilder; 3],
|
|
}
|
|
|
|
impl LevBuilder {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
automatons: [
|
|
LevenshteinAutomatonBuilder::new(0, false),
|
|
LevenshteinAutomatonBuilder::new(1, false),
|
|
LevenshteinAutomatonBuilder::new(2, false),
|
|
],
|
|
}
|
|
}
|
|
|
|
pub fn build_automaton(&self, query: &str) -> DFA {
|
|
if query.len() <= 4 {
|
|
self.automatons[0].build_dfa(query)
|
|
} else if query.len() <= 8 {
|
|
self.automatons[1].build_dfa(query)
|
|
} else {
|
|
self.automatons[2].build_dfa(query)
|
|
}
|
|
}
|
|
}
|