Examples with Shape org.springframework.data.geo.Shape used on opensource projects

Example 1 with Shape

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

the class MongoQueryCreatorUnitTests method shouldCreateWithinQueryCorrectly.

// DATAMONGO-1136
@Test
public void shouldCreateWithinQueryCorrectly() {
    Point first = new Point(1, 1);
    Point second = new Point(2, 2);
    Point third = new Point(3, 3);
    Shape shape = new Polygon(first, second, third);
    PartTree tree = new PartTree("findByAddress_GeoWithin", User.class);
    MongoQueryCreator creator = new MongoQueryCreator(tree, getAccessor(converter, shape), context);
    Query query = creator.createQuery();
    assertThat(query, is(query(where("address.geo").within(shape))));
}

Example 2 with Shape

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

the class MongoConvertersUnitTests method convertsCircleToDocumentAndBackCorrectly.

// DATAMONGO-858
@Test
public void convertsCircleToDocumentAndBackCorrectly() {
    Circle circle = new Circle(new Point(1, 2), 3);
    Document document = GeoConverters.CircleToDocumentConverter.INSTANCE.convert(circle);
    Shape shape = GeoConverters.DocumentToCircleConverter.INSTANCE.convert(document);
    assertThat(shape, is((org.springframework.data.geo.Shape) circle));
}

Example 3 with Shape

use of org.springframework.data.geo.Shape 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!");
    }
}

Example 4 with Shape

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

the class MongoConvertersUnitTests method convertsBoxToDocumentAndBackCorrectly.

// DATAMONGO-858
@Test
public void convertsBoxToDocumentAndBackCorrectly() {
    Box box = new Box(new Point(1, 2), new Point(3, 4));
    Document document = GeoConverters.BoxToDocumentConverter.INSTANCE.convert(box);
    Shape shape = GeoConverters.DocumentToBoxConverter.INSTANCE.convert(document);
    assertThat(shape, is((org.springframework.data.geo.Shape) box));
}

Example 5 with Shape

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

the class MongoConvertersUnitTests method convertsPolygonToDocumentAndBackCorrectly.

// DATAMONGO-858
@Test
public void convertsPolygonToDocumentAndBackCorrectly() {
    Polygon polygon = new Polygon(new Point(1, 2), new Point(2, 3), new Point(3, 4), new Point(5, 6));
    Document document = GeoConverters.PolygonToDocumentConverter.INSTANCE.convert(polygon);
    Shape shape = GeoConverters.DocumentToPolygonConverter.INSTANCE.convert(document);
    assertThat(shape, is((org.springframework.data.geo.Shape) polygon));
}