MeiliSearch/src/levenshtein.rs
2018-08-25 17:00:05 +02:00

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)
}
}
}