org.apache.jackrabbit.oak.plugins.index.lucene.IndexPlanner Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.jackrabbit.oak.plugins.index.lucene;
import java.util.Collections;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicBoolean;
import javax.annotation.CheckForNull;
import com.google.common.collect.Iterables;
import org.apache.jackrabbit.JcrConstants;
import org.apache.jackrabbit.oak.api.PropertyValue;
import org.apache.jackrabbit.oak.api.Type;
import org.apache.jackrabbit.oak.commons.PathUtils;
import org.apache.jackrabbit.oak.plugins.index.lucene.IndexDefinition.IndexingRule;
import org.apache.jackrabbit.oak.plugins.index.lucene.util.FacetHelper;
import org.apache.jackrabbit.oak.query.QueryImpl;
import org.apache.jackrabbit.oak.query.fulltext.FullTextContains;
import org.apache.jackrabbit.oak.query.fulltext.FullTextExpression;
import org.apache.jackrabbit.oak.query.fulltext.FullTextTerm;
import org.apache.jackrabbit.oak.query.fulltext.FullTextVisitor;
import org.apache.jackrabbit.oak.spi.query.Filter;
import org.apache.jackrabbit.oak.spi.query.QueryConstants;
import org.apache.lucene.index.IndexReader;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import static com.google.common.collect.Lists.newArrayList;
import static com.google.common.collect.Lists.newArrayListWithCapacity;
import static com.google.common.collect.Maps.newHashMap;
import static org.apache.jackrabbit.JcrConstants.JCR_SCORE;
import static org.apache.jackrabbit.oak.commons.PathUtils.getAncestorPath;
import static org.apache.jackrabbit.oak.commons.PathUtils.getDepth;
import static org.apache.jackrabbit.oak.commons.PathUtils.getParentPath;
import static org.apache.jackrabbit.oak.spi.query.Filter.PropertyRestriction;
import static org.apache.jackrabbit.oak.spi.query.QueryIndex.IndexPlan;
import static org.apache.jackrabbit.oak.spi.query.QueryIndex.OrderEntry;
class IndexPlanner {
private static final Logger log = LoggerFactory.getLogger(IndexPlanner.class);
private final IndexDefinition definition;
private final Filter filter;
private final String indexPath;
private final List sortOrder;
private IndexNode indexNode;
private PlanResult result;
public IndexPlanner(IndexNode indexNode,
String indexPath,
Filter filter, List sortOrder) {
this.indexNode = indexNode;
this.indexPath = indexPath;
this.definition = indexNode.getDefinition();
this.filter = filter;
this.sortOrder = sortOrder;
}
IndexPlan getPlan() {
IndexPlan.Builder builder = getPlanBuilder();
if (definition.isTestMode()){
if ( builder == null) {
if (notSupportedFeature()) {
return null;
}
String msg = String.format("No plan found for filter [%s] " +
"while using definition [%s] and testMode is found to be enabled", filter, definition);
throw new IllegalStateException(msg);
} else {
builder.setEstimatedEntryCount(1)
.setCostPerExecution(1e-3)
.setCostPerEntry(1e-3);
}
}
return builder != null ? builder.build() : null;
}
@Override
public String toString() {
return "IndexPlanner{" +
"indexPath='" + indexPath + '\'' +
", filter=" + filter +
", sortOrder=" + sortOrder +
'}';
}
private IndexPlan.Builder getPlanBuilder() {
log.trace("Evaluating plan with index definition {}", definition);
FullTextExpression ft = filter.getFullTextConstraint();
if (!definition.getVersion().isAtLeast(IndexFormatVersion.V2)){
log.trace("Index is old format. Not supported");
return null;
}
//Query Fulltext and Index does not support fulltext
if (ft != null && !definition.isFullTextEnabled()) {
return null;
}
IndexingRule indexingRule = getApplicableRule();
if (indexingRule == null){
return null;
}
//Query Fulltext and indexing rule does not support fulltext
if (ft != null && !indexingRule.isFulltextEnabled()){
return null;
}
if (!checkForQueryPaths()) {
log.trace("Opting out due mismatch between path restriction {} and query paths {}",
filter.getPath(), definition.getQueryPaths());
return null;
}
result = new PlanResult(indexPath, definition, indexingRule);
if (definition.hasFunctionDefined()
&& filter.getPropertyRestriction(definition.getFunctionName()) != null) {
return getNativeFunctionPlanBuilder(indexingRule.getBaseNodeType());
}
List indexedProps = newArrayListWithCapacity(filter.getPropertyRestrictions().size());
//Optimization - Go further only if any of the property is configured
//for property index
List facetFields = new LinkedList();
if (indexingRule.propertyIndexEnabled) {
for (PropertyRestriction pr : filter.getPropertyRestrictions()) {
String name = pr.propertyName;
if (QueryConstants.RESTRICTION_LOCAL_NAME.equals(name)) {
continue;
}
if (QueryImpl.REP_FACET.equals(pr.propertyName)) {
String value = pr.first.getValue(Type.STRING);
facetFields.add(FacetHelper.parseFacetField(value));
}
PropertyDefinition pd = indexingRule.getConfig(pr.propertyName);
if (pd != null && pd.propertyIndexEnabled()) {
if (pr.isNullRestriction() && !pd.nullCheckEnabled){
continue;
}
indexedProps.add(name);
result.propDefns.put(name, pd);
}
}
}
boolean evalNodeTypeRestrictions = canEvalNodeTypeRestrictions(indexingRule);
boolean evalPathRestrictions = canEvalPathRestrictions(indexingRule);
boolean canEvalAlFullText = canEvalAllFullText(indexingRule, ft);
boolean canEvalNodeNameRestriction = canEvalNodeNameRestriction(indexingRule);
if (ft != null && !canEvalAlFullText){
return null;
}
//Fulltext expression can also be like jcr:contains(jcr:content/metadata/@format, 'image')
List sortOrder = createSortOrder(indexingRule);
boolean canSort = canSortByProperty(sortOrder);
if (!indexedProps.isEmpty() || canSort || ft != null
|| evalPathRestrictions || evalNodeTypeRestrictions || canEvalNodeNameRestriction) {
//TODO Need a way to have better cost estimate to indicate that
//this index can evaluate more propertyRestrictions natively (if more props are indexed)
//For now we reduce cost per entry
int costPerEntryFactor = indexedProps.size();
costPerEntryFactor += sortOrder.size();
//this index can evaluate more propertyRestrictions natively (if more props are indexed)
//For now we reduce cost per entry
IndexPlan.Builder plan = defaultPlan();
if (!sortOrder.isEmpty()) {
plan.setSortOrder(sortOrder);
}
if (costPerEntryFactor == 0){
costPerEntryFactor = 1;
}
if (facetFields.size() > 0) {
plan.setAttribute(FacetHelper.ATTR_FACET_FIELDS, facetFields);
}
if (ft == null){
result.enableNonFullTextConstraints();
}
if (evalNodeTypeRestrictions){
result.enableNodeTypeEvaluation();
}
if (canEvalNodeNameRestriction){
result.enableNodeNameRestriction();
}
return plan.setCostPerEntry(definition.getCostPerEntry() / costPerEntryFactor);
}
//TODO Support for property existence queries
return null;
}
private IndexPlan.Builder getNativeFunctionPlanBuilder(String indexingRuleBaseNodeType) {
boolean canHandleNativeFunction = true;
PropertyValue pv = filter.getPropertyRestriction(definition.getFunctionName()).first;
String query = pv.getValue(Type.STRING);
if (query.startsWith("suggest?term=")) {
if (definition.isSuggestEnabled()) {
canHandleNativeFunction = indexingRuleBaseNodeType.equals(filter.getNodeType());
} else {
canHandleNativeFunction = false;
}
} else if (query.startsWith("spellcheck?term=")) {
if (definition.isSpellcheckEnabled()) {
canHandleNativeFunction = indexingRuleBaseNodeType.equals(filter.getNodeType());
} else {
canHandleNativeFunction = false;
}
}
//Suggestion and SpellCheck use virtual paths which is same for all results
if (canHandleNativeFunction) {
result.disableUniquePaths();
}
//If native function can be handled by this index then ensure
// that lowest cost if returned
return canHandleNativeFunction ? defaultPlan().setEstimatedEntryCount(1) : null;
}
/**
* Check if there is a mismatch between QueryPaths associated with index
* and path restriction specified in query
* @return true if QueryPaths and path restrictions do not have any conflict
*/
private boolean checkForQueryPaths() {
String[] queryPaths = definition.getQueryPaths();
if (queryPaths == null){
//No explicit value specified. Assume '/' which results in true
return true;
}
String pathRestriction = filter.getPath();
for (String queryPath : queryPaths){
if (queryPath.equals(pathRestriction) || PathUtils.isAncestor(queryPath, pathRestriction)){
return true;
}
}
return false;
}
private boolean canEvalNodeNameRestriction(IndexingRule indexingRule) {
PropertyRestriction pr = filter.getPropertyRestriction(QueryConstants.RESTRICTION_LOCAL_NAME);
if (pr == null){
return false;
}
return indexingRule.isNodeNameIndexed();
}
private static boolean canSortByProperty(List sortOrder) {
if (sortOrder.isEmpty()) {
return false;
}
// If jcr:score is the only sort order then opt out
if (sortOrder.size() == 1 &&
JCR_SCORE.equals(sortOrder.get(0).getPropertyName())) {
return false;
}
return true;
}
private boolean canEvalAllFullText(final IndexingRule indexingRule, FullTextExpression ft) {
if (ft == null){
return false;
}
final HashSet relPaths = new HashSet();
final HashSet nonIndexedPaths = new HashSet();
final AtomicBoolean relativeParentsFound = new AtomicBoolean();
final AtomicBoolean nodeScopedCondition = new AtomicBoolean();
ft.accept(new FullTextVisitor.FullTextVisitorBase() {
@Override
public boolean visit(FullTextContains contains) {
visitTerm(contains.getPropertyName());
return true;
}
@Override
public boolean visit(FullTextTerm term) {
visitTerm(term.getPropertyName());
return true;
}
private void visitTerm(String propertyName) {
String p = propertyName;
String propertyPath = null;
String nodePath = null;
if (p == null) {
relPaths.add("");
} else if (p.startsWith("../") || p.startsWith("./")) {
relPaths.add(p);
relativeParentsFound.set(true);
} else if (getDepth(p) > 1) {
String parent = getParentPath(p);
if (LucenePropertyIndex.isNodePath(p)){
nodePath = parent;
} else {
propertyPath = p;
}
relPaths.add(parent);
} else {
propertyPath = p;
relPaths.add("");
}
if (nodePath != null
&& !indexingRule.isAggregated(nodePath)){
nonIndexedPaths.add(p);
} else if (propertyPath != null
&& !indexingRule.isIndexed(propertyPath)){
nonIndexedPaths.add(p);
}
if (nodeScopedTerm(propertyName)){
nodeScopedCondition.set(true);
}
}
});
if (nodeScopedCondition.get() && !indexingRule.isNodeFullTextIndexed()){
return false;
}
if (relativeParentsFound.get()){
log.debug("Relative parents found {} which are not supported", relPaths);
return false;
}
if (!nonIndexedPaths.isEmpty()){
if (relPaths.size() > 1){
log.debug("Following relative property paths are not index", relPaths);
return false;
}
result.setParentPath(Iterables.getOnlyElement(relPaths, ""));
//Such path translation would only work if index contains
//all the nodes
return definition.indexesAllTypes();
} else {
result.setParentPath("");
}
return true;
}
private boolean canEvalPathRestrictions(IndexingRule rule) {
//Opt out if one is looking for all children for '/' as its equivalent to
//NO_RESTRICTION
if (filter.getPathRestriction() == Filter.PathRestriction.NO_RESTRICTION
|| (filter.getPathRestriction() == Filter.PathRestriction.ALL_CHILDREN
&& PathUtils.denotesRoot(filter.getPath()))
){
return false;
}
//If no other restrictions is provided and query is pure
//path restriction based then need to be sure that index definition at least
//allows indexing all the path for given nodeType
return definition.evaluatePathRestrictions() && rule.indexesAllNodesOfMatchingType();
}
private boolean canEvalNodeTypeRestrictions(IndexingRule rule) {
//No need to handle nt:base
if (filter.matchesAllTypes()){
return false;
}
//Only opt in if rule is not derived from nt:base otherwise it would
//get used when there a full text index on all nodes
return rule.indexesAllNodesOfMatchingType() && !rule.isBasedOnNtBase();
}
private IndexPlan.Builder defaultPlan() {
return new IndexPlan.Builder()
.setCostPerExecution(definition.getCostPerExecution())
.setCostPerEntry(definition.getCostPerEntry())
.setFulltextIndex(definition.isFullTextEnabled())
.setIncludesNodeData(false) // we should not include node data
.setFilter(filter)
.setPathPrefix(getPathPrefix())
.setDelayed(true) //Lucene is always async
.setAttribute(LucenePropertyIndex.ATTR_PLAN_RESULT, result)
.setEstimatedEntryCount(estimatedEntryCount())
.setPlanName(indexPath);
}
private long estimatedEntryCount() {
//Other index only compete in case of property indexes. For fulltext
//index return true count so as to allow multiple property indexes
//to be compared fairly
FullTextExpression ft = filter.getFullTextConstraint();
if (ft != null && definition.isFullTextEnabled()){
return definition.getFulltextEntryCount(getReader().numDocs());
}
return Math.min(definition.getEntryCount(), getReader().numDocs());
}
private String getPathPrefix() {
// 2 = /oak:index/
String parentPath = PathUtils.getAncestorPath(indexPath, 2);
return PathUtils.denotesRoot(parentPath) ? "" : parentPath;
}
private IndexReader getReader() {
return indexNode.getSearcher().getIndexReader();
}
private List createSortOrder(IndexingRule rule) {
if (sortOrder == null) {
return Collections.emptyList();
}
List orderEntries = newArrayListWithCapacity(sortOrder.size());
for (OrderEntry o : sortOrder) {
PropertyDefinition pd = rule.getConfig(o.getPropertyName());
if (pd != null
&& pd.ordered
&& o.getPropertyType() != null
&& !o.getPropertyType().isArray()) {
orderEntries.add(o); //Lucene can manage any order desc/asc
result.sortedProperties.add(pd);
} else if (o.getPropertyName().equals(IndexDefinition.NATIVE_SORT_ORDER.getPropertyName())) {
// Supports jcr:score descending natively
orderEntries.add(IndexDefinition.NATIVE_SORT_ORDER);
}
}
//TODO Should we return order entries only when all order clauses are satisfied
return orderEntries;
}
@CheckForNull
private IndexingRule getApplicableRule() {
if (filter.matchesAllTypes()){
return definition.getApplicableIndexingRule(JcrConstants.NT_BASE);
} else {
//TODO May be better if filter.getSuperTypes returned a list which maintains
//inheritance order and then we iterate over that
for (IndexingRule rule : definition.getDefinedRules()){
if (filter.getSupertypes().contains(rule.getNodeTypeName())){
//Theoretically there may be multiple rules for same nodeType with
//some condition defined. So again find a rule which applies
IndexingRule matchingRule = definition.getApplicableIndexingRule(rule.getNodeTypeName());
if (matchingRule != null){
log.debug("Applicable IndexingRule found {}", matchingRule);
return rule;
}
}
//nt:base is applicable for all. This specific condition is
//required to support mixin case as filter.getSupertypes() for mixin based
//query only includes the mixin type and not nt:base
if (rule.getNodeTypeName().equals(JcrConstants.NT_BASE)){
return rule;
}
}
log.trace("No applicable IndexingRule found for any of the superTypes {}",
filter.getSupertypes());
}
return null;
}
private boolean notSupportedFeature() {
if(filter.getPathRestriction() == Filter.PathRestriction.NO_RESTRICTION
&& filter.matchesAllTypes()
&& filter.getPropertyRestrictions().isEmpty()) {
//This mode includes name(), localname() queries
//OrImpl [a/name] = 'Hello' or [b/name] = 'World'
//Relative parent properties where [../foo1] is not null
return true;
}
return false;
}
/**
* Determine if the propertyName of a fulltext term indicates current node
* @param propertyName property name in the full text term clause
*/
private static boolean nodeScopedTerm(String propertyName) {
return propertyName == null || ".".equals(propertyName) || "*".equals(propertyName);
}
//~--------------------------------------------------------< PlanResult >
public static class PlanResult {
final String indexPath;
final IndexDefinition indexDefinition;
final IndexingRule indexingRule;
private List sortedProperties = newArrayList();
private Map propDefns = newHashMap();
private boolean nonFullTextConstraints;
private int parentDepth;
private String parentPathSegment;
private boolean relativize;
private boolean nodeTypeRestrictions;
private boolean nodeNameRestriction;
private boolean uniquePathsRequired = true;
public PlanResult(String indexPath, IndexDefinition defn, IndexingRule indexingRule) {
this.indexPath = indexPath;
this.indexDefinition = defn;
this.indexingRule = indexingRule;
}
public PropertyDefinition getPropDefn(PropertyRestriction pr){
return propDefns.get(pr.propertyName);
}
public PropertyDefinition getOrderedProperty(int index){
return sortedProperties.get(index);
}
public boolean isPathTransformed(){
return relativize;
}
public boolean isUniquePathsRequired() {
return uniquePathsRequired;
}
/**
* Transforms the given path if the query involved relative properties and index
* is not making use of aggregated properties. If the path
*
* @param path path to transform
* @return transformed path. Returns null if the path does not confirm to relative
* path requirements
*/
@CheckForNull
public String transformPath(String path){
if (isPathTransformed()){
// get the base path
// ensure the path ends with the given
// relative path
if (!path.endsWith(parentPathSegment)) {
return null;
}
return getAncestorPath(path, parentDepth);
}
return path;
}
public boolean evaluateNonFullTextConstraints(){
return nonFullTextConstraints;
}
public boolean evaluateNodeTypeRestriction() {
return nodeTypeRestrictions;
}
public boolean evaluateNodeNameRestriction() {return nodeNameRestriction;}
private void setParentPath(String relativePath){
parentPathSegment = "/" + relativePath;
if (relativePath.isEmpty()){
// we only restrict non-full-text conditions if there is
// no relative property in the full-text constraint
enableNonFullTextConstraints();
} else {
relativize = true;
parentDepth = getDepth(relativePath);
}
}
private void enableNonFullTextConstraints(){
nonFullTextConstraints = true;
}
private void enableNodeTypeEvaluation() {
nodeTypeRestrictions = true;
}
private void enableNodeNameRestriction(){
nodeNameRestriction = true;
}
private void disableUniquePaths(){
uniquePathsRequired = false;
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy