Browse Source
We now use the $geoWithin operator for geospatial criteria which requires to run on at least MongoDB 2.4. Original pull request: #263.pull/265/head
9 changed files with 411 additions and 234 deletions
@ -1,216 +1,176 @@ |
|||||||
/* |
/* |
||||||
* Copyright 2010-2014 the original author or authors. |
* Copyright 2015 the original author or authors. |
||||||
* |
* |
||||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||||
* you may not use this file except in compliance with the License. |
* you may not use this file except in compliance with the License. |
||||||
* You may obtain a copy of the License at |
* You may obtain a copy of the License at |
||||||
* |
* |
||||||
* http://www.apache.org/licenses/LICENSE-2.0
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
* |
* |
||||||
* Unless required by applicable law or agreed to in writing, software |
* Unless required by applicable law or agreed to in writing, software |
||||||
* distributed under the License is distributed on an "AS IS" BASIS, |
* distributed under the License is distributed on an "AS IS" BASIS, |
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||||
* See the License for the specific language governing permissions and |
* See the License for the specific language governing permissions and |
||||||
* limitations under the License. |
* limitations under the License. |
||||||
*/ |
*/ |
||||||
|
package org.springframework.data.mongodb.core.geo; |
||||||
package org.springframework.data.mongodb.core.geo; |
|
||||||
|
import static org.hamcrest.Matchers.*; |
||||||
import static org.hamcrest.Matchers.*; |
import static org.junit.Assert.*; |
||||||
import static org.junit.Assert.*; |
import static org.springframework.data.mongodb.core.query.Criteria.*; |
||||||
import static org.springframework.data.mongodb.core.query.Criteria.*; |
import static org.springframework.data.mongodb.core.query.Query.*; |
||||||
import static org.springframework.data.mongodb.core.query.Query.*; |
|
||||||
|
import java.util.List; |
||||||
import java.util.Collection; |
|
||||||
import java.util.List; |
import org.junit.After; |
||||||
|
import org.junit.Before; |
||||||
import org.apache.commons.logging.Log; |
import org.junit.Test; |
||||||
import org.apache.commons.logging.LogFactory; |
import org.junit.runner.RunWith; |
||||||
import org.junit.Before; |
import org.springframework.beans.factory.annotation.Autowired; |
||||||
import org.junit.Test; |
import org.springframework.context.annotation.Configuration; |
||||||
import org.springframework.beans.factory.annotation.Autowired; |
import org.springframework.data.geo.Box; |
||||||
import org.springframework.dao.DataAccessException; |
import org.springframework.data.geo.Circle; |
||||||
import org.springframework.data.domain.Sort.Direction; |
import org.springframework.data.geo.GeoResults; |
||||||
import org.springframework.data.geo.Box; |
import org.springframework.data.geo.Metric; |
||||||
import org.springframework.data.geo.Circle; |
import org.springframework.data.geo.Metrics; |
||||||
import org.springframework.data.geo.GeoResults; |
import org.springframework.data.geo.Point; |
||||||
import org.springframework.data.geo.Metric; |
import org.springframework.data.geo.Polygon; |
||||||
import org.springframework.data.geo.Metrics; |
import org.springframework.data.mongodb.config.AbstractMongoConfiguration; |
||||||
import org.springframework.data.geo.Point; |
import org.springframework.data.mongodb.core.MongoTemplate; |
||||||
import org.springframework.data.geo.Polygon; |
import org.springframework.data.mongodb.core.Venue; |
||||||
import org.springframework.data.mongodb.config.AbstractIntegrationTests; |
import org.springframework.data.mongodb.core.query.NearQuery; |
||||||
import org.springframework.data.mongodb.core.CollectionCallback; |
import org.springframework.data.mongodb.core.query.Query; |
||||||
import org.springframework.data.mongodb.core.IndexOperations; |
import org.springframework.test.context.ContextConfiguration; |
||||||
import org.springframework.data.mongodb.core.MongoTemplate; |
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; |
||||||
import org.springframework.data.mongodb.core.Venue; |
|
||||||
import org.springframework.data.mongodb.core.index.GeospatialIndex; |
import com.mongodb.Mongo; |
||||||
import org.springframework.data.mongodb.core.index.IndexField; |
import com.mongodb.MongoClient; |
||||||
import org.springframework.data.mongodb.core.index.IndexInfo; |
import com.mongodb.WriteConcern; |
||||||
import org.springframework.data.mongodb.core.query.NearQuery; |
|
||||||
import org.springframework.data.mongodb.core.query.Query; |
/** |
||||||
import org.springframework.expression.ExpressionParser; |
* @author Christoph Strobl |
||||||
import org.springframework.expression.spel.standard.SpelExpressionParser; |
*/ |
||||||
|
@RunWith(SpringJUnit4ClassRunner.class) |
||||||
import com.mongodb.DBCollection; |
@ContextConfiguration |
||||||
import com.mongodb.DBObject; |
public abstract class AbstractGeoSpatialTests { |
||||||
import com.mongodb.MongoException; |
|
||||||
import com.mongodb.WriteConcern; |
@Configuration |
||||||
|
static class TestConfig extends AbstractMongoConfiguration { |
||||||
/** |
|
||||||
* Modified from https://github.com/deftlabs/mongo-java-geospatial-example
|
@Override |
||||||
* |
protected String getDatabaseName() { |
||||||
* @author Mark Pollack |
return "database"; |
||||||
* @author Oliver Gierke |
} |
||||||
* @author Thomas Darimont |
|
||||||
*/ |
@Override |
||||||
public class GeoSpatialTests extends AbstractIntegrationTests { |
public Mongo mongo() throws Exception { |
||||||
|
return new MongoClient(); |
||||||
private static final Log LOGGER = LogFactory.getLog(GeoSpatialTests.class); |
} |
||||||
|
} |
||||||
@Autowired MongoTemplate template; |
|
||||||
|
@Autowired MongoTemplate template; |
||||||
ExpressionParser parser = new SpelExpressionParser(); |
|
||||||
|
@Before |
||||||
@Before |
public void setUp() { |
||||||
public void setUp() throws Exception { |
|
||||||
|
template.setWriteConcern(WriteConcern.FSYNC_SAFE); |
||||||
template.setWriteConcern(WriteConcern.FSYNC_SAFE); |
|
||||||
template.indexOps(Venue.class).ensureIndex(new GeospatialIndex("location")); |
createIndex(); |
||||||
|
addVenues(); |
||||||
indexCreated(); |
} |
||||||
addVenues(); |
|
||||||
} |
@After |
||||||
|
public void tearDown() { |
||||||
private void addVenues() { |
|
||||||
|
dropIndex(); |
||||||
template.insert(new Venue("Penn Station", -73.99408, 40.75057)); |
removeVenues(); |
||||||
template.insert(new Venue("10gen Office", -73.99171, 40.738868)); |
} |
||||||
template.insert(new Venue("Flatiron Building", -73.988135, 40.741404)); |
|
||||||
template.insert(new Venue("Players Club", -73.997812, 40.739128)); |
/** |
||||||
template.insert(new Venue("City Bakery ", -73.992491, 40.738673)); |
* Create the index required to run the tests. |
||||||
template.insert(new Venue("Splash Bar", -73.992491, 40.738673)); |
*/ |
||||||
template.insert(new Venue("Momofuku Milk Bar", -73.985839, 40.731698)); |
protected abstract void createIndex(); |
||||||
template.insert(new Venue("Shake Shack", -73.98820, 40.74164)); |
|
||||||
template.insert(new Venue("Penn Station", -73.99408, 40.75057)); |
/** |
||||||
template.insert(new Venue("Empire State Building", -73.98602, 40.74894)); |
* Remove index |
||||||
// template.insert(new Venue("Washington Square Park", -73.99756, 40.73083));
|
*/ |
||||||
template.insert(new Venue("Ulaanbaatar, Mongolia", 106.9154, 47.9245)); |
protected abstract void dropIndex(); |
||||||
template.insert(new Venue("Maplewood, NJ", -74.2713, 40.73137)); |
|
||||||
} |
protected void removeVenues() { |
||||||
|
template.dropCollection(Venue.class); |
||||||
@Test |
} |
||||||
public void geoNear() { |
|
||||||
|
protected void addVenues() { |
||||||
NearQuery geoNear = NearQuery.near(-73, 40, Metrics.KILOMETERS).num(10).maxDistance(150); |
|
||||||
|
template.insert(new Venue("Penn Station", -73.99408, 40.75057)); |
||||||
GeoResults<Venue> result = template.geoNear(geoNear, Venue.class); |
template.insert(new Venue("10gen Office", -73.99171, 40.738868)); |
||||||
|
template.insert(new Venue("Flatiron Building", -73.988135, 40.741404)); |
||||||
assertThat(result.getContent().size(), is(not(0))); |
template.insert(new Venue("Players Club", -73.997812, 40.739128)); |
||||||
assertThat(result.getAverageDistance().getMetric(), is((Metric) Metrics.KILOMETERS)); |
template.insert(new Venue("City Bakery ", -73.992491, 40.738673)); |
||||||
} |
template.insert(new Venue("Splash Bar", -73.992491, 40.738673)); |
||||||
|
template.insert(new Venue("Momofuku Milk Bar", -73.985839, 40.731698)); |
||||||
@Test |
template.insert(new Venue("Shake Shack", -73.98820, 40.74164)); |
||||||
public void withinCenter() { |
template.insert(new Venue("Penn Station", -73.99408, 40.75057)); |
||||||
Circle circle = new Circle(-73.99171, 40.738868, 0.01); |
template.insert(new Venue("Empire State Building", -73.98602, 40.74894)); |
||||||
List<Venue> venues = template.find(query(where("location").within(circle)), Venue.class); |
template.insert(new Venue("Ulaanbaatar, Mongolia", 106.9154, 47.9245)); |
||||||
assertThat(venues.size(), is(7)); |
template.insert(new Venue("Maplewood, NJ", -74.2713, 40.73137)); |
||||||
} |
} |
||||||
|
|
||||||
@Test |
@Test |
||||||
public void withinCenterSphere() { |
public void geoNear() { |
||||||
Circle circle = new Circle(-73.99171, 40.738868, 0.003712240453784); |
|
||||||
List<Venue> venues = template.find(query(where("location").withinSphere(circle)), Venue.class); |
NearQuery geoNear = NearQuery.near(-73, 40, Metrics.KILOMETERS).num(10).maxDistance(150); |
||||||
assertThat(venues.size(), is(11)); |
|
||||||
} |
GeoResults<Venue> result = template.geoNear(geoNear, Venue.class); |
||||||
|
|
||||||
@Test |
assertThat(result.getContent().size(), is(not(0))); |
||||||
public void withinBox() { |
assertThat(result.getAverageDistance().getMetric(), is((Metric) Metrics.KILOMETERS)); |
||||||
|
} |
||||||
Box box = new Box(new Point(-73.99756, 40.73083), new Point(-73.988135, 40.741404)); |
|
||||||
List<Venue> venues = template.find(query(where("location").within(box)), Venue.class); |
@Test |
||||||
assertThat(venues.size(), is(4)); |
public void withinCenter() { |
||||||
} |
|
||||||
|
Circle circle = new Circle(-73.99171, 40.738868, 0.01); |
||||||
@Test |
List<Venue> venues = template.find(query(where("location").within(circle)), Venue.class); |
||||||
public void withinPolygon() { |
assertThat(venues.size(), is(7)); |
||||||
|
} |
||||||
Point first = new Point(-73.99756, 40.73083); |
|
||||||
Point second = new Point(-73.99756, 40.741404); |
@Test |
||||||
Point third = new Point(-73.988135, 40.741404); |
public void withinCenterSphere() { |
||||||
Point fourth = new Point(-73.988135, 40.73083); |
|
||||||
|
Circle circle = new Circle(-73.99171, 40.738868, 0.003712240453784); |
||||||
Polygon polygon = new Polygon(first, second, third, fourth); |
List<Venue> venues = template.find(query(where("location").withinSphere(circle)), Venue.class); |
||||||
|
assertThat(venues.size(), is(11)); |
||||||
List<Venue> venues = template.find(query(where("location").within(polygon)), Venue.class); |
} |
||||||
assertThat(venues.size(), is(4)); |
|
||||||
} |
@Test |
||||||
|
public void withinBox() { |
||||||
@Test |
|
||||||
public void nearPoint() { |
Box box = new Box(new Point(-73.99756, 40.73083), new Point(-73.988135, 40.741404)); |
||||||
Point point = new Point(-73.99171, 40.738868); |
List<Venue> venues = template.find(query(where("location").within(box)), Venue.class); |
||||||
List<Venue> venues = template.find(query(where("location").near(point).maxDistance(0.01)), Venue.class); |
assertThat(venues.size(), is(4)); |
||||||
assertThat(venues.size(), is(7)); |
} |
||||||
} |
|
||||||
|
@Test |
||||||
@Test |
public void withinPolygon() { |
||||||
public void nearSphere() { |
|
||||||
Point point = new Point(-73.99171, 40.738868); |
Point first = new Point(-73.99756, 40.73083); |
||||||
Query query = query(where("location").nearSphere(point).maxDistance(0.003712240453784)); |
Point second = new Point(-73.99756, 40.741404); |
||||||
List<Venue> venues = template.find(query, Venue.class); |
Point third = new Point(-73.988135, 40.741404); |
||||||
assertThat(venues.size(), is(11)); |
Point fourth = new Point(-73.988135, 40.73083); |
||||||
} |
|
||||||
|
Polygon polygon = new Polygon(first, second, third, fourth); |
||||||
@Test |
|
||||||
public void searchAllData() { |
List<Venue> venues = template.find(query(where("location").within(polygon)), Venue.class); |
||||||
|
assertThat(venues.size(), is(4)); |
||||||
Venue foundVenue = template.findOne(query(where("name").is("Penn Station")), Venue.class); |
} |
||||||
assertThat(foundVenue, is(notNullValue())); |
|
||||||
|
@Test |
||||||
List<Venue> venues = template.findAll(Venue.class); |
public void nearSphere() { |
||||||
assertThat(venues.size(), is(12)); |
Point point = new Point(-73.99171, 40.738868); |
||||||
|
Query query = query(where("location").nearSphere(point).maxDistance(0.003712240453784)); |
||||||
Collection<?> names = (Collection<?>) parser.parseExpression("![name]").getValue(venues); |
List<Venue> venues = template.find(query, Venue.class); |
||||||
assertThat(names.size(), is(12)); |
assertThat(venues.size(), is(11)); |
||||||
|
} |
||||||
} |
|
||||||
|
} |
||||||
public void indexCreated() { |
|
||||||
|
|
||||||
List<DBObject> indexInfo = getIndexInfo(Venue.class); |
|
||||||
LOGGER.debug(indexInfo); |
|
||||||
|
|
||||||
assertThat(indexInfo.size(), is(2)); |
|
||||||
assertThat(indexInfo.get(1).get("name").toString(), is("location_2d")); |
|
||||||
assertThat(indexInfo.get(1).get("ns").toString(), is("database.newyork")); |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* @see DATAMONGO-360 |
|
||||||
*/ |
|
||||||
@Test |
|
||||||
public void indexInfoIsCorrect() { |
|
||||||
|
|
||||||
IndexOperations operations = template.indexOps(Venue.class); |
|
||||||
List<IndexInfo> indexInfo = operations.getIndexInfo(); |
|
||||||
|
|
||||||
assertThat(indexInfo.size(), is(2)); |
|
||||||
|
|
||||||
List<IndexField> fields = indexInfo.get(0).getIndexFields(); |
|
||||||
assertThat(fields.size(), is(1)); |
|
||||||
assertThat(fields, hasItem(IndexField.create("_id", Direction.ASC))); |
|
||||||
|
|
||||||
fields = indexInfo.get(1).getIndexFields(); |
|
||||||
assertThat(fields.size(), is(1)); |
|
||||||
assertThat(fields, hasItem(IndexField.geo("location"))); |
|
||||||
} |
|
||||||
|
|
||||||
// TODO move to MongoAdmin
|
|
||||||
public List<DBObject> getIndexInfo(Class<?> clazz) { |
|
||||||
return template.execute(clazz, new CollectionCallback<List<DBObject>>() { |
|
||||||
|
|
||||||
public List<DBObject> doInCollection(DBCollection collection) throws MongoException, DataAccessException { |
|
||||||
return collection.getIndexInfo(); |
|
||||||
} |
|
||||||
}); |
|
||||||
} |
|
||||||
} |
|
||||||
@ -0,0 +1,68 @@ |
|||||||
|
/* |
||||||
|
* Copyright 2010-2015 the original author or authors. |
||||||
|
* |
||||||
|
* 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. |
||||||
|
*/ |
||||||
|
|
||||||
|
package org.springframework.data.mongodb.core.geo; |
||||||
|
|
||||||
|
import static org.hamcrest.Matchers.*; |
||||||
|
import static org.junit.Assert.*; |
||||||
|
|
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
import org.junit.Test; |
||||||
|
import org.springframework.data.domain.Sort.Direction; |
||||||
|
import org.springframework.data.mongodb.core.IndexOperations; |
||||||
|
import org.springframework.data.mongodb.core.Venue; |
||||||
|
import org.springframework.data.mongodb.core.index.GeoSpatialIndexType; |
||||||
|
import org.springframework.data.mongodb.core.index.GeospatialIndex; |
||||||
|
import org.springframework.data.mongodb.core.index.IndexField; |
||||||
|
import org.springframework.data.mongodb.core.index.IndexInfo; |
||||||
|
|
||||||
|
/** |
||||||
|
* @author Christoph Strobl |
||||||
|
*/ |
||||||
|
public class GeoSpatial2DSphereTests extends AbstractGeoSpatialTests { |
||||||
|
|
||||||
|
/** |
||||||
|
* @see DATAMONGO-360 |
||||||
|
*/ |
||||||
|
@Test |
||||||
|
public void indexInfoIsCorrect() { |
||||||
|
|
||||||
|
IndexOperations operations = template.indexOps(Venue.class); |
||||||
|
List<IndexInfo> indexInfo = operations.getIndexInfo(); |
||||||
|
|
||||||
|
assertThat(indexInfo.size(), is(2)); |
||||||
|
|
||||||
|
List<IndexField> fields = indexInfo.get(0).getIndexFields(); |
||||||
|
assertThat(fields.size(), is(1)); |
||||||
|
assertThat(fields, hasItem(IndexField.create("_id", Direction.ASC))); |
||||||
|
|
||||||
|
fields = indexInfo.get(1).getIndexFields(); |
||||||
|
assertThat(fields.size(), is(1)); |
||||||
|
assertThat(fields, hasItem(IndexField.geo("location"))); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
protected void createIndex() { |
||||||
|
template.indexOps(Venue.class).ensureIndex(new GeospatialIndex("location").typed(GeoSpatialIndexType.GEO_2DSPHERE)); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
protected void dropIndex() { |
||||||
|
template.indexOps(Venue.class).dropIndex("location_2dsphere"); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,82 @@ |
|||||||
|
/* |
||||||
|
* Copyright 2010-2015 the original author or authors. |
||||||
|
* |
||||||
|
* 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. |
||||||
|
*/ |
||||||
|
|
||||||
|
package org.springframework.data.mongodb.core.geo; |
||||||
|
|
||||||
|
import static org.hamcrest.Matchers.*; |
||||||
|
import static org.junit.Assert.*; |
||||||
|
import static org.springframework.data.mongodb.core.query.Criteria.*; |
||||||
|
import static org.springframework.data.mongodb.core.query.Query.*; |
||||||
|
|
||||||
|
import java.util.List; |
||||||
|
|
||||||
|
import org.junit.Test; |
||||||
|
import org.springframework.data.domain.Sort.Direction; |
||||||
|
import org.springframework.data.geo.Point; |
||||||
|
import org.springframework.data.mongodb.core.IndexOperations; |
||||||
|
import org.springframework.data.mongodb.core.Venue; |
||||||
|
import org.springframework.data.mongodb.core.index.GeoSpatialIndexType; |
||||||
|
import org.springframework.data.mongodb.core.index.GeospatialIndex; |
||||||
|
import org.springframework.data.mongodb.core.index.IndexField; |
||||||
|
import org.springframework.data.mongodb.core.index.IndexInfo; |
||||||
|
|
||||||
|
/** |
||||||
|
* Modified from https://github.com/deftlabs/mongo-java-geospatial-example
|
||||||
|
* |
||||||
|
* @author Mark Pollack |
||||||
|
* @author Oliver Gierke |
||||||
|
* @author Thomas Darimont |
||||||
|
* @author Christoph Strobl |
||||||
|
*/ |
||||||
|
public class GeoSpatial2DTests extends AbstractGeoSpatialTests { |
||||||
|
|
||||||
|
@Test |
||||||
|
public void nearPoint() { |
||||||
|
Point point = new Point(-73.99171, 40.738868); |
||||||
|
List<Venue> venues = template.find(query(where("location").near(point).maxDistance(0.01)), Venue.class); |
||||||
|
assertThat(venues.size(), is(7)); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* @see DATAMONGO-360 |
||||||
|
*/ |
||||||
|
@Test |
||||||
|
public void indexInfoIsCorrect() { |
||||||
|
|
||||||
|
IndexOperations operations = template.indexOps(Venue.class); |
||||||
|
List<IndexInfo> indexInfo = operations.getIndexInfo(); |
||||||
|
|
||||||
|
assertThat(indexInfo.size(), is(2)); |
||||||
|
|
||||||
|
List<IndexField> fields = indexInfo.get(0).getIndexFields(); |
||||||
|
assertThat(fields.size(), is(1)); |
||||||
|
assertThat(fields, hasItem(IndexField.create("_id", Direction.ASC))); |
||||||
|
|
||||||
|
fields = indexInfo.get(1).getIndexFields(); |
||||||
|
assertThat(fields.size(), is(1)); |
||||||
|
assertThat(fields, hasItem(IndexField.geo("location"))); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
protected void createIndex() { |
||||||
|
template.indexOps(Venue.class).ensureIndex(new GeospatialIndex("location").typed(GeoSpatialIndexType.GEO_2D)); |
||||||
|
} |
||||||
|
|
||||||
|
@Override |
||||||
|
protected void dropIndex() { |
||||||
|
template.indexOps(Venue.class).dropIndex("location_2d"); |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue