diff options
Diffstat (limited to 'container-search/src/main/java/com/yahoo/prelude/querytransform')
-rw-r--r-- | container-search/src/main/java/com/yahoo/prelude/querytransform/RecallSearcher.java | 20 | ||||
-rw-r--r-- | container-search/src/main/java/com/yahoo/prelude/querytransform/StemmingSearcher.java | 2 |
2 files changed, 8 insertions, 14 deletions
diff --git a/container-search/src/main/java/com/yahoo/prelude/querytransform/RecallSearcher.java b/container-search/src/main/java/com/yahoo/prelude/querytransform/RecallSearcher.java index 98a8e5af00d..2c7d78c0671 100644 --- a/container-search/src/main/java/com/yahoo/prelude/querytransform/RecallSearcher.java +++ b/container-search/src/main/java/com/yahoo/prelude/querytransform/RecallSearcher.java @@ -100,22 +100,17 @@ public class RecallSearcher extends Searcher { * Returns the first word item contained in the given item tree that is an instance of {@link WordItem} with the * given word value. * - * @param root The root of the tree to check. - * @param value The word to look for. - * @return The first node found. + * @param root the root of the tree to check + * @param value the word to look for + * @return the first node found */ private static WordItem findOrigWordItem(Item root, String value) { Deque<Item> stack = new ArrayDeque<>(); stack.push(root); while (!stack.isEmpty()) { Item item = stack.pop(); - if (item.getCreator() == Item.ItemCreator.ORIG && - item instanceof WordItem word) - { - if (word.getWord().equals(value)) { - return word; - } - } + if (! item.isFilter() && item instanceof WordItem word && word.getWord().equals(value)) + return word; if (item instanceof CompositeItem lst) { for (Iterator<Item> it = lst.getItemIterator(); it.hasNext();) { stack.push(it.next()); @@ -128,16 +123,15 @@ public class RecallSearcher extends Searcher { /** * Marks all filter terms in the given query tree as unranked. * - * @param root The root of the tree to update. + * @param root the root of the tree to update */ private static void updateFilterTerms(Item root) { Deque<Item> stack = new ArrayDeque<>(); stack.push(root); while (!stack.isEmpty()) { Item item = stack.pop(); - if (item.getCreator() == Item.ItemCreator.FILTER) { + if (item.isFilter()) item.setRanked(false); - } if (item instanceof CompositeItem lst) { for (Iterator<Item> it = lst.getItemIterator(); it.hasNext();) { stack.push(it.next()); diff --git a/container-search/src/main/java/com/yahoo/prelude/querytransform/StemmingSearcher.java b/container-search/src/main/java/com/yahoo/prelude/querytransform/StemmingSearcher.java index 9bebaf98665..6aa262ca49e 100644 --- a/container-search/src/main/java/com/yahoo/prelude/querytransform/StemmingSearcher.java +++ b/container-search/src/main/java/com/yahoo/prelude/querytransform/StemmingSearcher.java @@ -375,7 +375,7 @@ public class StemmingSearcher extends Searcher { private void copyAttributes(Item blockAsItem, Item replacement) { copyWeight(blockAsItem, replacement); - replacement.setCreator(blockAsItem.getCreator()); + replacement.setFilter(blockAsItem.isFilter()); replacement.setRanked(blockAsItem.isRanked()); replacement.setPositionData(blockAsItem.usePositionData()); } |