mirror of
https://github.com/meilisearch/MeiliSearch
synced 2024-06-07 21:37:48 +02:00
28 lines
740 B
Rust
28 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)
|
|
}
|
|
}
|
|
}
|