package org.apache.lucene.queryparser.surround.query;
import java.util.ArrayList;
import java.util.List;
import java.util.Iterator;
import org.apache.lucene.search.Query;
public class FieldsQuery extends SrndQuery {
private SrndQuery q;
private List<String> fieldNames;
private final char fieldOp;
private static final String OR_OPERATOR_NAME = "OR";
public FieldsQuery(SrndQuery q, List<String> fieldNames, char fieldOp) {
this.q = q;
this.fieldNames = fieldNames;
this.fieldOp = fieldOp;
}
public FieldsQuery(SrndQuery q, String fieldName, char fieldOp) {
this.q = q;
fieldNames = new ArrayList<>();
fieldNames.add(fieldName);
this.fieldOp = fieldOp;
}
@Override
public boolean isFieldsSubQueryAcceptable() {
return false;
}
public Query makeLuceneQueryNoBoost(BasicQueryFactory qf) {
if (fieldNames.size() == 1) {
return q.makeLuceneQueryFieldNoBoost(fieldNames.get(0), qf);
} else {
List<SrndQuery> queries = new ArrayList<>();
Iterator<String> fni = getFieldNames().listIterator();
SrndQuery qc;
while (fni.hasNext()) {
qc = q.clone();
queries.add( new FieldsQuery( qc, fni.next(), fieldOp));
}
OrQuery oq = new OrQuery(queries,
true ,
OR_OPERATOR_NAME);
return oq.makeLuceneQueryField(null, qf);
}
}
@Override
public Query makeLuceneQueryFieldNoBoost(String fieldName, BasicQueryFactory qf) {
return makeLuceneQueryNoBoost(qf);
}
public List<String> getFieldNames() {return fieldNames;}
public char getFieldOperator() { return fieldOp;}
@Override
public String toString() {
StringBuilder r = new StringBuilder();
r.append("(");
fieldNamesToString(r);
r.append(q.toString());
r.append(")");
return r.toString();
}
protected void fieldNamesToString(StringBuilder r) {
Iterator<String> fni = getFieldNames().listIterator();
while (fni.hasNext()) {
r.append(fni.next());
r.append(getFieldOperator());
}
}
}