0ct0pu5/fess
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix #1231 add addBoostFunction

Browse source
This commit is contained in:
Shinsuke Sugaya 2017-08-17 11:58:15 +09:00
parent 112c4e5c67
commit 48d6428972
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/main/java/org/codelibs/fess/helper/QueryHelper.java
View file

@ -62,6 +62,7 @@ import org.elasticsearch.index.query.QueryBuilder;
import org.elasticsearch.index.query.QueryBuilders;
import org.elasticsearch.index.query.RangeQueryBuilder;
import org.elasticsearch.index.query.functionscore.FunctionScoreQueryBuilder.FilterFunctionBuilder;
import org.elasticsearch.index.query.functionscore.ScoreFunctionBuilder;
import org.elasticsearch.index.query.functionscore.ScoreFunctionBuilders;
import org.elasticsearch.search.sort.SortBuilder;
import org.elasticsearch.search.sort.SortBuilders;
@ -125,6 +126,8 @@ public class QueryHelper {
protected Map<String, String> fieldBoostMap = new HashMap<>();
protected List<FilterFunctionBuilder> boostFunctionList = new ArrayList<>();
@PostConstruct
public void init() {
if (responseFields == null) {
@ -334,6 +337,7 @@ public class QueryHelper {
if (keyMatchHelper != null) {
keyMatchHelper.buildQuery(queryContext.getDefaultKeyword(), list);
}
list.addAll(boostFunctionList);
});
}
@ -815,4 +819,11 @@ public class QueryHelper {
this.lowercaseWildcard = lowercaseWildcard;
}
public void addBoostFunction(final ScoreFunctionBuilder<?> scoreFunction) {
boostFunctionList.add(new FilterFunctionBuilder(scoreFunction));
}
public void addBoostFunction(final QueryBuilder filter, final ScoreFunctionBuilder<?> scoreFunction) {
boostFunctionList.add(new FilterFunctionBuilder(filter, scoreFunction));
}
}