package org.exist.xquery.modules.lucene;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import org.apache.log4j.Logger;
import org.apache.lucene.queryParser.ParseException;
import org.exist.dom.DocumentSet;
import org.exist.dom.NodeSet;
import org.exist.dom.QName;
import org.exist.dom.VirtualNodeSet;
import org.exist.indexing.lucene.LuceneIndex;
import org.exist.indexing.lucene.LuceneIndexWorker;
import org.exist.storage.ElementValue;
import org.exist.xquery.*;
import org.exist.xquery.value.FunctionParameterSequenceType;
import org.exist.xquery.value.FunctionReturnSequenceType;
import org.exist.xquery.value.Item;
import org.exist.xquery.value.NodeValue;
import org.exist.xquery.value.Sequence;
import org.exist.xquery.value.SequenceType;
import org.exist.xquery.value.Type;
import org.w3c.dom.Element;
public class Query extends Function implements Optimizable {
protected static final Logger logger = Logger.getLogger(Query.class);
public final static FunctionSignature[] signatures = {
new FunctionSignature(
new QName("query", LuceneModule.NAMESPACE_URI, LuceneModule.PREFIX),
"Queries a node set using a Lucene full text index; a lucene index " +
"must already be defined on the nodes, because if no index is available " +
"on a node, nothing will be found. Indexes on descendant nodes are not " +
"used. The context of the Lucene query is determined by the given input " +
"node set. The query is specified either as a query string based on " +
"Lucene's default query syntax or as an XML fragment. " +
"See http://exist-db.org/lucene.html#N1029E for complete documentation.",
new SequenceType[] {
new FunctionParameterSequenceType("nodes", Type.NODE, Cardinality.ZERO_OR_MORE,
"The node set to search using a Lucene full text index which is defined on those nodes"),
new FunctionParameterSequenceType("query", Type.ITEM, Cardinality.EXACTLY_ONE,
"The query to search for, provided either as a string or text in Lucene's default query " +
"syntax or as an XML fragment to bypass Lucene's default query parser")
},
new FunctionReturnSequenceType(Type.NODE, Cardinality.ZERO_OR_MORE,
"all nodes from the input node set matching the query. match highlighting information " +
"will be available for all returned nodes. Lucene's match score can be retrieved via " +
"the ft:score function.")
),
new FunctionSignature(
new QName("query", LuceneModule.NAMESPACE_URI, LuceneModule.PREFIX),
"Queries a node set using a Lucene full text index; a lucene index " +
"must already be defined on the nodes, because if no index is available " +
"on a node, nothing will be found. Indexes on descendant nodes are not " +
"used. The context of the Lucene query is determined by the given input " +
"node set. The query is specified either as a query string based on " +
"Lucene's default query syntax or as an XML fragment. " +
"See http://exist-db.org/lucene.html#N1029E for complete documentation.",
new SequenceType[] {
new FunctionParameterSequenceType("nodes", Type.NODE, Cardinality.ZERO_OR_MORE,
"The node set to search using a Lucene full text index which is defined on those nodes"),
new FunctionParameterSequenceType("query", Type.ITEM, Cardinality.EXACTLY_ONE,
"The query to search for, provided either as a string or text in Lucene's default query " +
"syntax or as an XML fragment to bypass Lucene's default query parser"),
new FunctionParameterSequenceType("options", Type.NODE, Cardinality.ZERO_OR_ONE,
"An XML fragment containing options to be passed to Lucene's query parser. The following " +
"options are supported (a description can be found in the docs):\n" +
"<options>\n" +
" <default-operator>and|or</default-operator>\n" +
" <phrase-slop>number</phrase-slop>\n" +
" <leading-wildcard>yes|no</leading-wildcard>\n" +
" <filter-rewrite>yes|no</filter-rewrite>\n" +
"</options>")
},
new FunctionReturnSequenceType(Type.NODE, Cardinality.ZERO_OR_MORE,
"all nodes from the input node set matching the query. match highlighting information " +
"will be available for all returned nodes. Lucene's match score can be retrieved via " +
"the ft:score function.")
)
};
private LocationStep contextStep = null;
protected QName contextQName = null;
protected int axis = Constants.UNKNOWN_AXIS;
private NodeSet preselectResult = null;
protected boolean optimizeSelf = false;
protected boolean optimizeChild = false;
public Query(XQueryContext context, FunctionSignature signature) {
super(context, signature);
}
public void setArguments(List arguments) throws XPathException {
Expression path = (Expression) arguments.get(0);
steps.add(path);
Expression arg = (Expression) arguments.get(1);
arg = new DynamicCardinalityCheck(context, Cardinality.EXACTLY_ONE, arg,
new org.exist.xquery.util.Error(org.exist.xquery.util.Error.FUNC_PARAM_CARDINALITY, "2", mySignature));
steps.add(arg);
if (arguments.size() == 3) {
arg = (Expression) arguments.get(2);
arg = new DynamicCardinalityCheck(context, Cardinality.EXACTLY_ONE, arg,
new org.exist.xquery.util.Error(org.exist.xquery.util.Error.FUNC_PARAM_CARDINALITY, "2", mySignature));
arg = new DynamicTypeCheck(context, Type.ELEMENT, arg);
steps.add(arg);
}
}
/* (non-Javadoc)
* @see org.exist.xquery.PathExpr#analyze(org.exist.xquery.Expression)
*/
public void analyze(AnalyzeContextInfo contextInfo) throws XPathException {
super.analyze(new AnalyzeContextInfo(contextInfo));
List<LocationStep> steps = BasicExpressionVisitor.findLocationSteps(getArgument(0));
if (!steps.isEmpty()) {
LocationStep firstStep = steps.get(0);
LocationStep lastStep = steps.get(steps.size() - 1);
if (steps.size() == 1 && firstStep.getAxis() == Constants.SELF_AXIS) {
Expression outerExpr = contextInfo.getContextStep();
if (outerExpr != null && outerExpr instanceof LocationStep) {
LocationStep outerStep = (LocationStep) outerExpr;
NodeTest test = outerStep.getTest();
if (test.getName() == null)
contextQName = new QName(null, null, null);
else if (test.isWildcardTest())
contextQName = test.getName();
else
contextQName = new QName(test.getName());
if (outerStep.getAxis() == Constants.ATTRIBUTE_AXIS || outerStep.getAxis() == Constants.DESCENDANT_ATTRIBUTE_AXIS)
contextQName.setNameType(ElementValue.ATTRIBUTE);
contextStep = firstStep;
axis = outerStep.getAxis();
optimizeSelf = true;
}
} else {
NodeTest test = lastStep.getTest();
if (test.getName() == null)
contextQName = new QName(null, null, null);
else if (test.isWildcardTest())
contextQName = test.getName();
else
contextQName = new QName(test.getName());
if (lastStep.getAxis() == Constants.ATTRIBUTE_AXIS || lastStep.getAxis() == Constants.DESCENDANT_ATTRIBUTE_AXIS)
contextQName.setNameType(ElementValue.ATTRIBUTE);
axis = firstStep.getAxis();
optimizeChild = steps.size() == 1 &&
(axis == Constants.CHILD_AXIS || axis == Constants.ATTRIBUTE_AXIS);
contextStep = lastStep;
}
}
}
public boolean canOptimize(Sequence contextSequence) {
return contextQName != null;
}
public boolean optimizeOnSelf() {
return optimizeSelf;
}
public boolean optimizeOnChild() {
return optimizeChild;
}
public int getOptimizeAxis() {
return axis;
}
public NodeSet preSelect(Sequence contextSequence, boolean useContext) throws XPathException {
if (contextSequence != null && !contextSequence.isPersistentSet())
// in-memory docs won't have an index
return NodeSet.EMPTY_SET;
long start = System.currentTimeMillis();
// the expression can be called multiple times, so we need to clear the previous preselectResult
preselectResult = null;
LuceneIndexWorker index = (LuceneIndexWorker)
context.getBroker().getIndexController().getWorkerByIndexId(LuceneIndex.ID);
DocumentSet docs = contextSequence.getDocumentSet();
Item key = getKey(contextSequence, null);
List<QName> qnames = new ArrayList<QName>(1);
qnames.add(contextQName);
Properties options = parseOptions(contextSequence, null);
try {
if (Type.subTypeOf(key.getType(), Type.ELEMENT))
preselectResult = index.query(context, getExpressionId(), docs, useContext ? contextSequence.toNodeSet() : null,
qnames, (Element) ((NodeValue)key).getNode(), NodeSet.DESCENDANT, options);
else
preselectResult = index.query(context, getExpressionId(), docs, useContext ? contextSequence.toNodeSet() : null,
qnames, key.getStringValue(), NodeSet.DESCENDANT, options);
} catch (IOException e) {
throw new XPathException(this, "Error while querying full text index: " + e.getMessage(), e);
} catch (ParseException e) {
throw new XPathException(this, "Error while querying full text index: " + e.getMessage(), e);
}
LOG.debug("Lucene query took " + (System.currentTimeMillis() - start));
if( context.getProfiler().traceFunctions() ) {
context.getProfiler().traceIndexUsage( context, "lucene", this, PerformanceStats.OPTIMIZED_INDEX, System.currentTimeMillis() - start );
}
return preselectResult;
}
public Sequence eval(Sequence contextSequence, Item contextItem) throws XPathException {
if (contextItem != null)
contextSequence = contextItem.toSequence();
if (contextSequence != null && !contextSequence.isPersistentSet())
// in-memory docs won't have an index
return Sequence.EMPTY_SEQUENCE;
NodeSet result;
if (preselectResult == null) {
long start = System.currentTimeMillis();
Sequence input = getArgument(0).eval(contextSequence);
if (!(input instanceof VirtualNodeSet) && input.isEmpty())
result = NodeSet.EMPTY_SET;
else {
NodeSet inNodes = input.toNodeSet();
DocumentSet docs = inNodes.getDocumentSet();
LuceneIndexWorker index = (LuceneIndexWorker)
context.getBroker().getIndexController().getWorkerByIndexId(LuceneIndex.ID);
Item key = getKey(contextSequence, contextItem);
List<QName> qnames = null;
if (contextQName != null) {
qnames = new ArrayList<QName>(1);
qnames.add(contextQName);
}
Properties options = parseOptions(contextSequence, contextItem);
try {
if (Type.subTypeOf(key.getType(), Type.ELEMENT))
result = index.query(context, getExpressionId(), docs, inNodes, qnames,
(Element)((NodeValue)key).getNode(), NodeSet.ANCESTOR, options);
else
result = index.query(context, getExpressionId(), docs, inNodes, qnames,
key.getStringValue(), NodeSet.ANCESTOR, options);
} catch (IOException e) {
throw new XPathException(this, e.getMessage());
} catch (ParseException e) {
throw new XPathException(this, e.getMessage());
}
}
if( context.getProfiler().traceFunctions() ) {
context.getProfiler().traceIndexUsage( context, "lucene", this, PerformanceStats.BASIC_INDEX, System.currentTimeMillis() - start );
}
} else {
contextStep.setPreloadedData(contextSequence.getDocumentSet(), preselectResult);
result = getArgument(0).eval(contextSequence).toNodeSet();
}
return result;
}
protected Item getKey(Sequence contextSequence, Item contextItem) throws XPathException {
Sequence keySeq = getArgument(1).eval(contextSequence, contextItem);
Item key = keySeq.itemAt(0);
if (!(Type.subTypeOf(key.getType(), Type.STRING) || Type.subTypeOf(key.getType(), Type.NODE)))
throw new XPathException(this, "Second argument to ft:query should either be a query string or " +
"an XML element describing the query. Found: " + Type.getTypeName(key.getType()));
return key;
}
public int getDependencies() {
final Expression stringArg = getArgument(0);
if (Type.subTypeOf(stringArg.returnsType(), Type.NODE) &&
!Dependency.dependsOn(stringArg, Dependency.CONTEXT_ITEM)) {
return Dependency.CONTEXT_SET;
} else {
return Dependency.CONTEXT_SET + Dependency.CONTEXT_ITEM;
}
}
public int returnsType() {
return Type.NODE;
}
protected Properties parseOptions(Sequence contextSequence, Item contextItem) throws XPathException {
if (getArgumentCount() < 3)
return null;
Properties options = new Properties();
Sequence optSeq = getArgument(2).eval(contextSequence, contextItem);
NodeValue optRoot = (NodeValue) optSeq.itemAt(0);
try {
XMLStreamReader reader = context.getXMLStreamReader(optRoot);
reader.next();
reader.next();
while (reader.hasNext()) {
int status = reader.next();
if (status == XMLStreamReader.START_ELEMENT) {
options.put(reader.getLocalName(), reader.getElementText());
}
}
return options;
} catch (XMLStreamException e) {
throw new XPathException(this, "Error while parsing options to ft:query: " + e.getMessage(), e);
} catch (IOException e) {
throw new XPathException(this, "Error while parsing options to ft:query: " + e.getMessage(), e);
}
}
}