Java tutorial
/* * Hibernate, Relational Persistence for Idiomatic Java * * Copyright (c) 2012, Red Hat, Inc. and/or its affiliates or third-party contributors as * indicated by the @author tags or express copyright attribution * statements applied by the authors. All third-party contributions are * distributed under license by Red Hat, Inc. * * This copyrighted material is made available to anyone wishing to use, modify, * copy, or redistribute it subject to the terms and conditions of the GNU * Lesser General Public License, as published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License * for more details. * * You should have received a copy of the GNU Lesser General Public License * along with this distribution; if not, write to: * Free Software Foundation, Inc. * 51 Franklin Street, Fifth Floor * Boston, MA 02110-1301 USA */ package org.hibernate.search.test.jpa; import java.util.List; import org.apache.lucene.search.Sort; import org.hibernate.search.jpa.FullTextEntityManager; import org.hibernate.search.jpa.FullTextQuery; import org.hibernate.search.jpa.Search; import org.hibernate.search.query.dsl.QueryBuilder; import org.hibernate.search.query.dsl.Unit; import org.hibernate.search.spatial.impl.DistanceSortField; import org.hibernate.search.spatial.impl.Point; import org.hibernate.search.test.spatial.POI; import org.junit.Assert; import static org.hamcrest.CoreMatchers.is; import static org.hamcrest.CoreMatchers.not; /** * Hibernate Search spatial : unit tests on quering POIs * * @author Nicolas Helleringer <nicolas.helleringer@novacodex.net> * @author Davide Di Somma <davide.disomma@gmail.com> */ public class SpatialQueringJPATest extends JPATestCase { public void testDistanceProjection() throws Exception { POI poi = new POI(1, "Distance to 24,32 : 0", 24.0d, 32.0d, ""); POI poi2 = new POI(2, "Distance to 24,32 : 10.16", 24.0d, 31.9d, ""); POI poi3 = new POI(3, "Distance to 24,32 : 11.12", 23.9d, 32.0d, ""); POI poi4 = new POI(4, "Distance to 24,32 : 15.06", 23.9d, 32.1d, ""); POI poi5 = new POI(5, "Distance to 24,32 : 22.24", 24.2d, 32.0d, ""); POI poi6 = new POI(6, "Distance to 24,32 : 24.45", 24.2d, 31.9d, ""); FullTextEntityManager em = Search.getFullTextEntityManager(factory.createEntityManager()); em.getTransaction().begin(); em.persist(poi); em.persist(poi2); em.persist(poi3); em.getTransaction().commit(); em.clear(); em.getTransaction().begin(); em.persist(poi4); em.persist(poi5); em.persist(poi6); em.getTransaction().commit(); em.getTransaction().begin(); double centerLatitude = 24.0d; double centerLongitude = 32.0d; final QueryBuilder builder = em.getSearchFactory().buildQueryBuilder().forEntity(POI.class).get(); org.apache.lucene.search.Query luceneQuery = builder.spatial().onCoordinates("location") .within(100, Unit.KM).ofLatitude(centerLatitude).andLongitude(centerLongitude).createQuery(); FullTextQuery hibQuery = em.createFullTextQuery(luceneQuery, POI.class); hibQuery.setProjection(FullTextQuery.THIS, FullTextQuery.SPATIAL_DISTANCE); hibQuery.setSpatialParameters(centerLatitude, centerLongitude, "location"); List results = hibQuery.getResultList(); Object[] firstResult = (Object[]) results.get(0); Object[] secondResult = (Object[]) results.get(1); Object[] thirdResult = (Object[]) results.get(2); Object[] fourthResult = (Object[]) results.get(3); Object[] fifthResult = (Object[]) results.get(4); Object[] sixthResult = (Object[]) results.get(5); Assert.assertEquals((Double) firstResult[1], 0.0, 0.0001); Assert.assertEquals((Double) secondResult[1], 10.1582, 0.0001); Assert.assertEquals((Double) thirdResult[1], 11.1195, 0.0001); Assert.assertEquals((Double) fourthResult[1], 15.0636, 0.0001); Assert.assertEquals((Double) fifthResult[1], 22.239, 0.001); Assert.assertEquals((Double) sixthResult[1], 24.446, 0.001); List<?> pois = em.createQuery("from " + POI.class.getName()).getResultList(); for (Object entity : pois) { em.remove(entity); } em.getTransaction().commit(); em.close(); } public void testDistanceSort() throws Exception { POI poi = new POI(1, "Distance to 24,32 : 0", 24.0d, 32.0d, ""); POI poi2 = new POI(2, "Distance to 24,32 : 10.16", 24.0d, 31.9d, ""); POI poi3 = new POI(3, "Distance to 24,32 : 11.12", 23.9d, 32.0d, ""); POI poi4 = new POI(4, "Distance to 24,32 : 15.06", 23.9d, 32.1d, ""); POI poi5 = new POI(5, "Distance to 24,32 : 22.24", 24.2d, 32.0d, ""); POI poi6 = new POI(6, "Distance to 24,32 : 24.45", 24.2d, 31.9d, ""); FullTextEntityManager em = Search.getFullTextEntityManager(factory.createEntityManager()); em.getTransaction().begin(); em.persist(poi); em.persist(poi2); em.persist(poi3); em.getTransaction().commit(); em.getTransaction().begin(); em.persist(poi4); em.persist(poi5); em.persist(poi6); em.getTransaction().commit(); em.getTransaction().begin(); double centerLatitude = 24.0d; double centerLongitude = 32.0d; final QueryBuilder builder = em.getSearchFactory().buildQueryBuilder().forEntity(POI.class).get(); org.apache.lucene.search.Query luceneQuery = builder.spatial().onCoordinates("location") .within(100, Unit.KM).ofLatitude(centerLatitude).andLongitude(centerLongitude).createQuery(); FullTextQuery hibQuery = em.createFullTextQuery(luceneQuery, POI.class); Sort distanceSort = new Sort(new DistanceSortField(centerLatitude, centerLongitude, "location")); hibQuery.setSort(distanceSort); hibQuery.setProjection(FullTextQuery.THIS, FullTextQuery.SPATIAL_DISTANCE); hibQuery.setSpatialParameters(Point.fromDegrees(centerLatitude, centerLongitude), "location"); List results = hibQuery.getResultList(); Object[] firstResult = (Object[]) results.get(0); Object[] secondResult = (Object[]) results.get(1); Object[] thirdResult = (Object[]) results.get(2); Object[] fourthResult = (Object[]) results.get(3); Object[] fifthResult = (Object[]) results.get(4); Object[] sixthResult = (Object[]) results.get(5); Assert.assertEquals((Double) firstResult[1], 0.0, 0.0001); Assert.assertEquals((Double) secondResult[1], 10.1582, 0.0001); Assert.assertEquals((Double) thirdResult[1], 11.1195, 0.0001); Assert.assertEquals((Double) fourthResult[1], 15.0636, 0.0001); Assert.assertEquals((Double) fifthResult[1], 22.239, 0.001); Assert.assertEquals((Double) sixthResult[1], 24.446, 0.001); List<?> pois = em.createQuery("from " + POI.class.getName()).getResultList(); for (Object entity : pois) { em.remove(entity); } em.getTransaction().commit(); em.close(); } public void testDistanceSort2() throws Exception { FullTextEntityManager em = Search.getFullTextEntityManager(factory.createEntityManager()); em.getTransaction().begin(); int cnt = 0; for (double[] c : new double[][] { { 41.04389845, -74.06328534 }, { 40.64383333, -73.75050000 }, { 40.75666667, -73.98650000 }, { 40.69416667, -73.78166667 }, { 40.75802992, -73.98532391 }, { 40.75802992, -73.98532391 }, { 50.77687257, 6.08431213 }, { 50.78361600, 6.07003500 }, { 50.76066667, 6.08866667 }, { 50.77683333, 6.08466667 }, { 50.77650000, 6.08416667 }, }) { em.persist(new POI(cnt, "Test_" + cnt, c[0], c[1], "")); ++cnt; } em.getTransaction().commit(); em.getTransaction().begin(); double centerLatitude = 50.7753455; double centerLongitude = 6.083886799999959; final QueryBuilder builder = em.getSearchFactory().buildQueryBuilder().forEntity(POI.class).get(); org.apache.lucene.search.Query luceneQuery = builder.spatial().onCoordinates("location") .within(1.8097233616663808, Unit.KM).ofLatitude(centerLatitude).andLongitude(centerLongitude) .createQuery(); FullTextQuery hibQuery = em.createFullTextQuery(luceneQuery, POI.class); Sort distanceSort = new Sort(new DistanceSortField(centerLatitude, centerLongitude, "location")); hibQuery.setSort(distanceSort); hibQuery.setMaxResults(1000); hibQuery.setProjection(FullTextQuery.THIS, FullTextQuery.SPATIAL_DISTANCE); hibQuery.setSpatialParameters(Point.fromDegrees(centerLatitude, centerLongitude), "location"); List<Object[]> results = hibQuery.getResultList(); for (Object[] result : results) { POI poi = (POI) result[0]; String message = poi.getName() + " (" + poi.getLatitude() + ", " + poi.getLongitude() + ") is not at " + centerLatitude + ", " + centerLongitude; Assert.assertThat(message, ((Double) result[1]).doubleValue(), is(not(0.0))); } List<?> pois = em.createQuery("from " + POI.class.getName()).getResultList(); for (Object entity : pois) { em.remove(entity); } em.getTransaction().commit(); em.close(); } public void testDistanceSortWithMaxResult() throws Exception { POI poi = new POI(1, "Distance to 24,32 : 0", 24.0d, 32.0d, ""); POI poi2 = new POI(2, "Distance to 24,32 : 10.16", 24.0d, 31.9d, ""); POI poi3 = new POI(3, "Distance to 24,32 : 11.12", 23.9d, 32.0d, ""); POI poi4 = new POI(4, "Distance to 24,32 : 15.06", 23.9d, 32.1d, ""); POI poi5 = new POI(5, "Distance to 24,32 : 22.24", 24.2d, 32.0d, ""); POI poi6 = new POI(6, "Distance to 24,32 : 24.45", 24.2d, 31.9d, ""); FullTextEntityManager em = Search.getFullTextEntityManager(factory.createEntityManager()); em.getTransaction().begin(); em.persist(poi); em.persist(poi2); em.persist(poi3); em.getTransaction().commit(); em.getTransaction().begin(); em.persist(poi4); em.persist(poi5); em.persist(poi6); em.getTransaction().commit(); em.getTransaction().begin(); double centerLatitude = 24.0d; double centerLongitude = 32.0d; final QueryBuilder builder = em.getSearchFactory().buildQueryBuilder().forEntity(POI.class).get(); org.apache.lucene.search.Query luceneQuery = builder.spatial().onCoordinates("location") .within(100, Unit.KM).ofLatitude(centerLatitude).andLongitude(centerLongitude).createQuery(); FullTextQuery hibQuery = em.createFullTextQuery(luceneQuery, POI.class); Sort distanceSort = new Sort(new DistanceSortField(centerLatitude, centerLongitude, "location")); hibQuery.setSort(distanceSort); hibQuery.setProjection(FullTextQuery.THIS, FullTextQuery.SPATIAL_DISTANCE); // Set max results to 3 when 6 documents are stored: hibQuery.setMaxResults(3); hibQuery.setSpatialParameters(Point.fromDegrees(centerLatitude, centerLongitude), "location"); List results = hibQuery.getResultList(); Object[] firstResult = (Object[]) results.get(0); Object[] secondResult = (Object[]) results.get(1); Object[] thirdResult = (Object[]) results.get(2); Assert.assertEquals((Double) firstResult[1], 0.0, 0.0001); Assert.assertEquals((Double) secondResult[1], 10.1582, 0.0001); Assert.assertEquals((Double) thirdResult[1], 11.1195, 0.0001); List<?> pois = em.createQuery("from " + POI.class.getName()).getResultList(); for (Object entity : pois) { em.remove(entity); } em.getTransaction().commit(); em.close(); } @Override public Class[] getAnnotatedClasses() { return new Class<?>[] { POI.class }; } }