/
ConnectedAllContext.java
74 lines (63 loc) · 1.94 KB
/
ConnectedAllContext.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
/*
* Hibernate Search, full-text search for your domain model
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
*/
package org.hibernate.search.query.dsl.impl;
import java.util.ArrayList;
import java.util.List;
import org.apache.lucene.search.BooleanClause;
import org.apache.lucene.search.BooleanQuery;
import org.apache.lucene.search.MatchAllDocsQuery;
import org.apache.lucene.search.Query;
import org.hibernate.search.query.dsl.AllContext;
/**
* @author Emmanuel Bernard
*/
public class ConnectedAllContext implements AllContext {
private final List<BooleanClause> clauses;
private final QueryCustomizer queryCustomizer;
public ConnectedAllContext() {
this.queryCustomizer = new QueryCustomizer();
this.clauses = new ArrayList<BooleanClause>( 5 );
this.clauses.add( new BooleanClause( new MatchAllDocsQuery(), BooleanClause.Occur.SHOULD ) );
}
@Override
public Query createQuery() {
Query query;
if ( clauses.size() == 1 ) {
query = clauses.get( 0 ).getQuery();
}
else {
BooleanQuery.Builder booleanQueryBuilder = new BooleanQuery.Builder();
for ( BooleanClause clause : clauses ) {
booleanQueryBuilder.add( clause );
}
query = booleanQueryBuilder.build();
}
return queryCustomizer.setWrappedQuery( query ).createQuery();
}
@Override
public AllContext except(Query... queriesMatchingExcludedDocuments) {
for ( Query query : queriesMatchingExcludedDocuments ) {
clauses.add( new BooleanClause( query, BooleanClause.Occur.MUST_NOT ) );
}
return this;
}
@Override
public AllContext boostedTo(float boost) {
queryCustomizer.boostedTo( boost );
return this;
}
@Override
public AllContext withConstantScore() {
queryCustomizer.withConstantScore();
return this;
}
@Override
public AllContext filteredBy(Query filter) {
queryCustomizer.filteredBy( filter );
return this;
}
}