snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-22 18:17:39 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

feat: Consider the last query word be a prefix

Browse Source 
if the last word is not followed by a space.
This commit is contained in:
Clément Renault 2018-12-11 14:49:45 +01:00
parent f97f7f93f3
commit df2b6a3e74
No known key found for this signature in database
GPG Key ID: 0151CDAB43460DAE
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/rank/query_builder.rs
View File

@ -1,7 +1,7 @@
use std::{mem, vec, str, char};
use std::ops::{Deref, Range}; use std::ops::{Deref, Range};
use std::error::Error; use std::error::Error;
use std::hash::Hash; use std::hash::Hash;
use std::{mem, vec, str};
use group_by::GroupByMut; use group_by::GroupByMut;
use hashbrown::HashMap; use hashbrown::HashMap;
@ -16,15 +16,21 @@ use crate::{Match, DocumentId};
use crate::rank::Document; use crate::rank::Document;
fn split_whitespace_automatons(query: &str) -> Vec<DfaExt> { fn split_whitespace_automatons(query: &str) -> Vec<DfaExt> {
let has_end_whitespace = query.chars().last().map_or(false, char::is_whitespace);
let mut automatons = Vec::new(); let mut automatons = Vec::new();
let mut words = query.split_whitespace().map(str::to_lowercase).peekable(); let mut words = query.split_whitespace().map(str::to_lowercase).peekable();
while let Some(word) = words.next() { while let Some(word) = words.next() {
let lev = match words.peek() {
Some(_) => automaton::build_dfa(&word), let has_following_word = words.peek().is_some();
None => automaton::build_prefix_dfa(&word), let lev = if has_following_word || has_end_whitespace {
automaton::build_dfa(&word)
} else {
automaton::build_prefix_dfa(&word)
}; };
automatons.push(lev); automatons.push(lev);
} }
automatons automatons
} }