private ListMultimap<String, TextHit> query(
Node property, String queryString, int limit, ExecutionContext execCxt) {
// use the graph information in the text index if possible
if (textIndex.getDocDef().getGraphField() != null
&& execCxt.getActiveGraph() instanceof GraphView) {
GraphView activeGraph = (GraphView) execCxt.getActiveGraph();
if (!Quad.isUnionGraph(activeGraph.getGraphName())) {
String uri =
activeGraph.getGraphName() != null
? TextQueryFuncs.graphNodeToString(activeGraph.getGraphName())
: Quad.defaultGraphNodeGenerated.getURI();
String escaped = QueryParserBase.escape(uri);
String qs2 = textIndex.getDocDef().getGraphField() + ":" + escaped;
queryString = "(" + queryString + ") AND " + qs2;
}
}
// for language-based search extension
if (textIndex.getDocDef().getLangField() != null) {
String field = textIndex.getDocDef().getLangField();
if (langArg != null) {
String qs2 = !"none".equals(langArg) ? field + ":" + langArg : "-" + field + ":*";
queryString = "(" + queryString + ") AND " + qs2;
}
}
Explain.explain(execCxt.getContext(), "Text query: " + queryString);
if (log.isDebugEnabled()) log.debug("Text query: {} ({})", queryString, limit);
String cacheKey = limit + " " + property + " " + queryString;
Cache<String, ListMultimap<String, TextHit>> queryCache =
(Cache<String, ListMultimap<String, TextHit>>) execCxt.getContext().get(cacheSymbol);
if (queryCache == null) {
/* doesn't yet exist, need to create it */
queryCache = CacheFactory.createCache(CACHE_SIZE);
execCxt.getContext().put(cacheSymbol, queryCache);
}
final String queryStr = queryString; // final needed for the lambda function
ListMultimap<String, TextHit> results =
queryCache.getOrFill(
cacheKey,
() -> {
List<TextHit> resultList = textIndex.query(property, queryStr, limit);
ListMultimap<String, TextHit> resultMultimap = LinkedListMultimap.create();
for (TextHit result : resultList) {
resultMultimap.put(result.getNode().getURI(), result);
}
return resultMultimap;
});
return results;
}
@Override
public void build(
PropFuncArg argSubject, Node predicate, PropFuncArg argObject, ExecutionContext execCxt) {
super.build(argSubject, predicate, argObject, execCxt);
DatasetGraph dsg = execCxt.getDataset();
textIndex = chooseTextIndex(dsg);
if (argSubject.isList()) {
int size = argSubject.getArgListSize();
if (size != 2 && size != 3) {
throw new QueryBuildException(
"Subject has "
+ argSubject.getArgList().size()
+ " elements, not 2 or 3: "
+ argSubject);
}
}
if (argObject.isList()) {
List<Node> list = argObject.getArgList();
if (list.size() == 0) throw new QueryBuildException("Zero-length argument list");
if (list.size() > 4) throw new QueryBuildException("Too many arguments in list : " + list);
// extract of extra lang arg if present and if is usable.
// arg is removed from the list to avoid conflict with order and args length
langArg = extractArg("lang", list);
if (langArg != null && textIndex.getDocDef().getLangField() == null)
log.warn("lang argument is ignored if langField not set in the index configuration");
}
}
/**
* Deconstruct the node or list object argument and make a StrMatch The 'executionTime' flag
* indciates whether this is for a build time static check, or for runtime execution.
*/
private StrMatch objectToStruct(PropFuncArg argObject, boolean executionTime) {
EntityDefinition docDef = textIndex.getDocDef();
if (argObject.isNode()) {
Node o = argObject.getArg();
if (!o.isLiteral()) {
if (executionTime) log.warn("Object to text query is not a literal");
return null;
}
RDFDatatype dt = o.getLiteralDatatype();
if (dt != null && dt != XSDDatatype.XSDstring) {
log.warn("Object to text query is not a string");
return null;
}
String qs = o.getLiteralLexicalForm();
return new StrMatch(null, qs, -1, 0);
}
List<Node> list = argObject.getArgList();
if (list.size() == 0 || list.size() > 3)
throw new TextIndexException("Change in object list size");
Node predicate = null;
String field = null; // Do not prepend the field name - rely on default field
int idx = 0;
Node x = list.get(0);
// Property?
if (x.isURI()) {
predicate = x;
idx++;
if (idx >= list.size())
throw new TextIndexException("Property specificed but no query string : " + list);
x = list.get(idx);
field = docDef.getField(predicate);
if (field == null) {
log.warn("Predicate not indexed: " + predicate);
return null;
}
}
// String!
if (!x.isLiteral()) {
if (executionTime) log.warn("Text query string is not a literal " + list);
return null;
}
if (x.getLiteralDatatype() != null && !x.getLiteralDatatype().equals(XSDDatatype.XSDstring)) {
log.warn("Text query is not a string " + list);
return null;
}
String queryString = x.getLiteralLexicalForm();
idx++;
int limit = -1;
float score = 0;
if (idx < list.size()) {
// Limit?
x = list.get(idx);
idx++;
if (!x.isLiteral()) {
if (executionTime) log.warn("Text query limit is not an integer " + x);
return null;
}
int v = NodeFactoryExtra.nodeToInt(x);
limit = (v < 0) ? -1 : v;
}
String qs = queryString;
if (field != null) qs = field + ":" + qs;
return new StrMatch(predicate, qs, limit, score);
}
