com.orientechnologies.spatial.functions.OSpatialFunctionAbstractIndexable Maven / Gradle / Ivy
Show all versions of orientdb-lucene Show documentation
/**
* Copyright 2010-2016 OrientDB LTD (http://orientdb.com)
*
* Licensed 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.
*
*
For more information: http://www.orientdb.com
*/
package com.orientechnologies.spatial.functions;
import com.orientechnologies.lucene.collections.OLuceneResultSetEmpty;
import com.orientechnologies.orient.core.command.OCommandContext;
import com.orientechnologies.orient.core.db.ODatabaseDocumentInternal;
import com.orientechnologies.orient.core.db.ODatabaseRecordThreadLocal;
import com.orientechnologies.orient.core.db.record.OIdentifiable;
import com.orientechnologies.orient.core.exception.OCommandExecutionException;
import com.orientechnologies.orient.core.index.OIndex;
import com.orientechnologies.orient.core.metadata.OMetadataInternal;
import com.orientechnologies.orient.core.record.impl.ODocument;
import com.orientechnologies.orient.core.sql.executor.OResult;
import com.orientechnologies.orient.core.sql.executor.OResultInternal;
import com.orientechnologies.orient.core.sql.functions.OIndexableSQLFunction;
import com.orientechnologies.orient.core.sql.parser.OBinaryCompareOperator;
import com.orientechnologies.orient.core.sql.parser.OExpression;
import com.orientechnologies.orient.core.sql.parser.OFromClause;
import com.orientechnologies.orient.core.sql.parser.OFromItem;
import com.orientechnologies.orient.core.sql.parser.OIdentifier;
import com.orientechnologies.orient.core.sql.parser.OJson;
import com.orientechnologies.orient.core.sql.parser.OLeOperator;
import com.orientechnologies.orient.core.sql.parser.OLtOperator;
import com.orientechnologies.spatial.index.OLuceneSpatialIndex;
import com.orientechnologies.spatial.strategy.SpatialQueryBuilderAbstract;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
/** Created by Enrico Risa on 31/08/15. */
public abstract class OSpatialFunctionAbstractIndexable extends OSpatialFunctionAbstract
implements OIndexableSQLFunction {
public OSpatialFunctionAbstractIndexable(String iName, int iMinParams, int iMaxParams) {
super(iName, iMinParams, iMaxParams);
}
protected OLuceneSpatialIndex searchForIndex(OFromClause target, OExpression[] args) {
OMetadataInternal dbMetadata = getDb().getMetadata();
OFromItem item = target.getItem();
OIdentifier identifier = item.getIdentifier();
String fieldName = args[0].toString();
String className = identifier.getStringValue();
List indices =
dbMetadata.getImmutableSchemaSnapshot().getClass(className).getIndexes().stream()
.filter(idx -> idx instanceof OLuceneSpatialIndex)
.map(idx -> (OLuceneSpatialIndex) idx)
.filter(idx -> intersect(idx.getDefinition().getFields(), Arrays.asList(fieldName)))
.collect(Collectors.toList());
if (indices.size() > 1) {
throw new IllegalArgumentException(
"too many indices matching given field name: " + String.join(",", fieldName));
}
return indices.size() == 0 ? null : indices.get(0);
}
protected ODatabaseDocumentInternal getDb() {
return ODatabaseRecordThreadLocal.instance().get();
}
protected Iterable results(
OFromClause target, OExpression[] args, OCommandContext ctx, Object rightValue) {
OIndex oIndex = searchForIndex(target, args);
if (oIndex == null) {
return null;
}
Map queryParams = new HashMap<>();
queryParams.put(SpatialQueryBuilderAbstract.GEO_FILTER, operator());
Object shape;
if (args[1].getValue() instanceof OJson) {
OJson json = (OJson) args[1].getValue();
ODocument doc = new ODocument().fromJSON(json.toString());
shape = doc.toMap();
} else {
shape = args[1].execute((OIdentifiable) null, ctx);
}
if (shape instanceof Collection) {
int size = ((Collection) shape).size();
if (size == 0) {
return new OLuceneResultSetEmpty();
}
if (size == 1) {
Object next = ((Collection) shape).iterator().next();
if (next instanceof OResult) {
OResult inner = (OResult) next;
Set propertyNames = inner.getPropertyNames();
if (propertyNames.size() == 1) {
Object property = inner.getProperty(propertyNames.iterator().next());
if (property instanceof OResult) {
shape = ((OResult) property).toElement();
}
} else {
return new OLuceneResultSetEmpty();
}
}
} else {
throw new OCommandExecutionException("The collection in input cannot be major than 1");
}
}
if (shape instanceof OResultInternal) {
shape = ((OResultInternal) shape).toElement();
}
queryParams.put(SpatialQueryBuilderAbstract.SHAPE, shape);
onAfterParsing(queryParams, args, ctx, rightValue);
Set indexes = (Set) ctx.getVariable("involvedIndexes");
if (indexes == null) {
indexes = new HashSet<>();
ctx.setVariable("involvedIndexes", indexes);
}
indexes.add(oIndex.getName());
return oIndex.getInternal().getRids(queryParams).collect(Collectors.toSet());
}
protected void onAfterParsing(
Map params, OExpression[] args, OCommandContext ctx, Object rightValue) {}
protected abstract String operator();
@Override
public boolean canExecuteInline(
OFromClause target,
OBinaryCompareOperator operator,
Object rightValue,
OCommandContext ctx,
OExpression... args) {
return true;
}
@Override
public boolean allowsIndexedExecution(
OFromClause target,
OBinaryCompareOperator operator,
Object rightValue,
OCommandContext ctx,
OExpression... args) {
if (!isValidBinaryOperator(operator)) {
return false;
}
OLuceneSpatialIndex index = searchForIndex(target, args);
return index != null;
}
@Override
public boolean shouldExecuteAfterSearch(
OFromClause target,
OBinaryCompareOperator operator,
Object rightValue,
OCommandContext ctx,
OExpression... args) {
return true;
}
@Override
public long estimate(
OFromClause target,
OBinaryCompareOperator operator,
Object rightValue,
OCommandContext ctx,
OExpression... args) {
OLuceneSpatialIndex index = searchForIndex(target, args);
return index == null ? -1 : index.size();
}
public boolean intersect(List list1, List list2) {
for (T t : list1) {
if (list2.contains(t)) {
return true;
}
}
return false;
}
protected boolean isValidBinaryOperator(OBinaryCompareOperator operator) {
return operator instanceof OLtOperator || operator instanceof OLeOperator;
}
}