Search in sources :

Example 11 with Distance

use of org.springframework.data.geo.Distance in project spring-data-mongodb by spring-projects.

the class GeoConvertersUnitTests method convertsCircleToDocumentAndBackCorrectlyMilesDistance.

// DATAMONGO-858
@Test
public void convertsCircleToDocumentAndBackCorrectlyMilesDistance() {
    Distance radius = new Distance(3, Metrics.MILES);
    Circle circle = new Circle(new Point(1, 2), radius);
    Document document = CircleToDocumentConverter.INSTANCE.convert(circle);
    Circle result = DocumentToCircleConverter.INSTANCE.convert(document);
    assertThat(result, is(circle));
    assertThat(result.getRadius(), is(radius));
}
Also used : Circle(org.springframework.data.geo.Circle) Point(org.springframework.data.geo.Point) Document(org.bson.Document) Distance(org.springframework.data.geo.Distance) Test(org.junit.Test)

Example 12 with Distance

use of org.springframework.data.geo.Distance in project spring-data-mongodb by spring-projects.

the class GeoConvertersUnitTests method convertsSphereCorrectlyWhenUsingNonDoubleForCoordinates.

// DATAMONGO-1607
@Test
public void convertsSphereCorrectlyWhenUsingNonDoubleForCoordinates() {
    Document sphere = new Document();
    sphere.put("center", new Document().append("x", 1).append("y", 2));
    sphere.put("radius", 3L);
    assertThat(DocumentToSphereConverter.INSTANCE.convert(sphere), is(new Sphere(new Point(1, 2), new Distance(3))));
}
Also used : Sphere(org.springframework.data.mongodb.core.geo.Sphere) Point(org.springframework.data.geo.Point) Document(org.bson.Document) Distance(org.springframework.data.geo.Distance) Test(org.junit.Test)

Example 13 with Distance

use of org.springframework.data.geo.Distance in project spring-data-mongodb by spring-projects.

the class GeoConvertersUnitTests method convertsCircleCorrectlyWhenUsingNonDoubleForCoordinates.

// DATAMONGO-1607
@Test
public void convertsCircleCorrectlyWhenUsingNonDoubleForCoordinates() {
    Document circle = new Document();
    circle.put("center", new Document().append("x", 1).append("y", 2));
    circle.put("radius", 3L);
    assertThat(DocumentToCircleConverter.INSTANCE.convert(circle), is(new Circle(new Point(1, 2), new Distance(3))));
}
Also used : Circle(org.springframework.data.geo.Circle) Point(org.springframework.data.geo.Point) Document(org.bson.Document) Distance(org.springframework.data.geo.Distance) Test(org.junit.Test)

Example 14 with Distance

use of org.springframework.data.geo.Distance in project spring-data-mongodb by spring-projects.

the class MongoParametersParameterAccessor method getDistanceRange.

public Range<Distance> getDistanceRange() {
    MongoParameters mongoParameters = method.getParameters();
    int rangeIndex = mongoParameters.getRangeIndex();
    if (rangeIndex != -1) {
        return getValue(rangeIndex);
    }
    int maxDistanceIndex = mongoParameters.getMaxDistanceIndex();
    Distance maxDistance = maxDistanceIndex == -1 ? null : (Distance) getValue(maxDistanceIndex);
    return new Range<Distance>(null, maxDistance);
}
Also used : Range(org.springframework.data.domain.Range) Point(org.springframework.data.geo.Point) Distance(org.springframework.data.geo.Distance)

Example 15 with Distance

use of org.springframework.data.geo.Distance in project spring-data-mongodb by spring-projects.

the class MongoQueryCreator method from.

/**
 * Populates the given {@link CriteriaDefinition} depending on the {@link Part} given.
 *
 * @param part
 * @param property
 * @param criteria
 * @param parameters
 * @return
 */
private Criteria from(Part part, MongoPersistentProperty property, Criteria criteria, Iterator<Object> parameters) {
    Type type = part.getType();
    switch(type) {
        case AFTER:
        case GREATER_THAN:
            return criteria.gt(parameters.next());
        case GREATER_THAN_EQUAL:
            return criteria.gte(parameters.next());
        case BEFORE:
        case LESS_THAN:
            return criteria.lt(parameters.next());
        case LESS_THAN_EQUAL:
            return criteria.lte(parameters.next());
        case BETWEEN:
            return criteria.gt(parameters.next()).lt(parameters.next());
        case IS_NOT_NULL:
            return criteria.ne(null);
        case IS_NULL:
            return criteria.is(null);
        case NOT_IN:
            return criteria.nin(nextAsArray(parameters));
        case IN:
            return criteria.in(nextAsArray(parameters));
        case LIKE:
        case STARTING_WITH:
        case ENDING_WITH:
        case CONTAINING:
            return createContainingCriteria(part, property, criteria, parameters);
        case NOT_LIKE:
            return createContainingCriteria(part, property, criteria.not(), parameters);
        case NOT_CONTAINING:
            return createContainingCriteria(part, property, criteria.not(), parameters);
        case REGEX:
            return criteria.regex(parameters.next().toString());
        case EXISTS:
            return criteria.exists((Boolean) parameters.next());
        case TRUE:
            return criteria.is(true);
        case FALSE:
            return criteria.is(false);
        case NEAR:
            Range<Distance> range = accessor.getDistanceRange();
            Optional<Distance> distance = range.getUpperBound().getValue();
            Optional<Distance> minDistance = range.getLowerBound().getValue();
            Point point = accessor.getGeoNearLocation();
            Point pointToUse = point == null ? nextAs(parameters, Point.class) : point;
            boolean isSpherical = isSpherical(property);
            return distance.map(it -> {
                if (isSpherical || !Metrics.NEUTRAL.equals(it.getMetric())) {
                    criteria.nearSphere(pointToUse);
                } else {
                    criteria.near(pointToUse);
                }
                criteria.maxDistance(it.getNormalizedValue());
                minDistance.ifPresent(min -> criteria.minDistance(min.getNormalizedValue()));
                return criteria;
            }).orElseGet(() -> isSpherical ? criteria.nearSphere(pointToUse) : criteria.near(pointToUse));
        case WITHIN:
            Object parameter = parameters.next();
            return criteria.within((Shape) parameter);
        case SIMPLE_PROPERTY:
            return isSimpleComparisionPossible(part) ? criteria.is(parameters.next()) : createLikeRegexCriteriaOrThrow(part, property, criteria, parameters, false);
        case NEGATING_SIMPLE_PROPERTY:
            return isSimpleComparisionPossible(part) ? criteria.ne(parameters.next()) : createLikeRegexCriteriaOrThrow(part, property, criteria, parameters, true);
        default:
            throw new IllegalArgumentException("Unsupported keyword!");
    }
}
Also used : MongoRegexCreator(org.springframework.data.mongodb.core.query.MongoRegexCreator) Arrays(java.util.Arrays) MongoPersistentProperty(org.springframework.data.mongodb.core.mapping.MongoPersistentProperty) IgnoreCaseType(org.springframework.data.repository.query.parser.Part.IgnoreCaseType) Metrics(org.springframework.data.geo.Metrics) LoggerFactory(org.slf4j.LoggerFactory) Shape(org.springframework.data.geo.Shape) Type(org.springframework.data.repository.query.parser.Part.Type) MappingContext(org.springframework.data.mapping.context.MappingContext) Part(org.springframework.data.repository.query.parser.Part) Distance(org.springframework.data.geo.Distance) AbstractQueryCreator(org.springframework.data.repository.query.parser.AbstractQueryCreator) Sort(org.springframework.data.domain.Sort) MatchMode(org.springframework.data.mongodb.core.query.MongoRegexCreator.MatchMode) Point(org.springframework.data.geo.Point) Logger(org.slf4j.Logger) ClassUtils(org.springframework.util.ClassUtils) Iterator(java.util.Iterator) GeoSpatialIndexType(org.springframework.data.mongodb.core.index.GeoSpatialIndexType) PotentiallyConvertingIterator(org.springframework.data.mongodb.repository.query.ConvertingParameterAccessor.PotentiallyConvertingIterator) PartTree(org.springframework.data.repository.query.parser.PartTree) Collection(java.util.Collection) Range(org.springframework.data.domain.Range) Criteria(org.springframework.data.mongodb.core.query.Criteria) Query(org.springframework.data.mongodb.core.query.Query) PersistentPropertyPath(org.springframework.data.mapping.context.PersistentPropertyPath) CriteriaDefinition(org.springframework.data.mongodb.core.query.CriteriaDefinition) Optional(java.util.Optional) PropertyPath(org.springframework.data.mapping.PropertyPath) GeoSpatialIndexed(org.springframework.data.mongodb.core.index.GeoSpatialIndexed) Assert(org.springframework.util.Assert) IgnoreCaseType(org.springframework.data.repository.query.parser.Part.IgnoreCaseType) Type(org.springframework.data.repository.query.parser.Part.Type) GeoSpatialIndexType(org.springframework.data.mongodb.core.index.GeoSpatialIndexType) Point(org.springframework.data.geo.Point) Distance(org.springframework.data.geo.Distance)

Aggregations

Distance (org.springframework.data.geo.Distance)28 Point (org.springframework.data.geo.Point)27 Test (org.junit.Test)25 GeoJsonPoint (org.springframework.data.mongodb.core.geo.GeoJsonPoint)12 Query (org.springframework.data.mongodb.core.query.Query)7 Document (org.springframework.data.mongodb.core.mapping.Document)6 Document (org.bson.Document)5 Circle (org.springframework.data.geo.Circle)5 Metric (org.springframework.data.geo.Metric)5 Sphere (org.springframework.data.mongodb.core.geo.Sphere)5 PartTree (org.springframework.data.repository.query.parser.PartTree)4 Arrays (java.util.Arrays)3 Sort (org.springframework.data.domain.Sort)3 Metrics (org.springframework.data.geo.Metrics)3 Method (java.lang.reflect.Method)2 BlockingQueue (java.util.concurrent.BlockingQueue)2 LinkedBlockingDeque (java.util.concurrent.LinkedBlockingDeque)2 TimeUnit (java.util.concurrent.TimeUnit)2 NoArgsConstructor (lombok.NoArgsConstructor)2 Matchers (org.hamcrest.Matchers)2