Spring/spring-data-elasticsearch
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added Support for IN Query and Added more test cases

Browse Source
This commit is contained in:
Mohsin Husen
2013-01-28 15:34:13 +00:00
parent 1520b4deac
commit 8970785ab8
4 changed files with 177 additions and 184 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/main/java/org/springframework/data/elasticsearch/core/CriteriaQueryProcessor.java
View File

@@ -94,7 +94,13 @@ class CriteriaQueryProcessor {
query = rangeQuery(fieldName).from(ranges[0]).to(ranges[1]); break;
case FUZZY:
query = fuzzyQuery(fieldName, (String) value); break;
case IN:
query = boolQuery();
Iterable<Object> collection = (Iterable<Object>) value;
for(Object item : collection){
((BoolQueryBuilder) query).should(fieldQuery(fieldName, item));
}
break;
}
return query;

10
src/main/java/org/springframework/data/elasticsearch/core/query/Criteria.java
View File

@@ -340,13 +340,7 @@ public class Criteria {
*/
public Criteria in(Iterable<?> values) {
Assert.notNull(values, "Collection of 'in' values must not be null");
for (Object value : values) {
if (value instanceof Collection) {
in((Collection<?>) value);
} else {
is(value);
}
}
criteria.add(new CriteriaEntry(OperationKey.IN, values));
return this;
}
@@ -430,7 +424,7 @@ public class Criteria {
}
public enum OperationKey {
EQUALS, CONTAINS, STARTS_WITH, ENDS_WITH, EXPRESSION, BETWEEN, FUZZY;
EQUALS, CONTAINS, STARTS_WITH, ENDS_WITH, EXPRESSION, BETWEEN, FUZZY, IN;
}
public static class CriteriaEntry {

323
src/test/java/org/springframework/data/elasticsearch/repositories/CustomMethodRepositoryTest.java
View File

@@ -227,170 +227,163 @@ public class CustomMethodRepositoryTest {
assertThat(page.getTotalElements(), is(equalTo(1L)));
}
// @Test
// @Ignore("Test failing due to java.lang.IllegalArgumentException: Invalid order syntax for part Message!")
// public void shouldExecuteCustomMethodForIn(){
// //given
// String documentId = randomNumeric(5);
// SampleEntity sampleEntity = new SampleEntity();
// sampleEntity.setId(documentId);
// sampleEntity.setType("test");
// sampleEntity.setMessage("foo");
// repository.save(sampleEntity);
//
// //given
// String documentId2 = randomNumeric(5);
// SampleEntity sampleEntity2 = new SampleEntity();
// sampleEntity2.setId(documentId2);
// sampleEntity2.setType("test");
// sampleEntity2.setMessage("bar");
// repository.save(sampleEntity2);
//
// List<String> ids = Arrays.asList(documentId,documentId2);
//
//
// //when
// Page<SampleEntity> page = repository.findByIdIn(ids, new PageRequest(1, 10));
// //then
// assertThat(page, is(notNullValue()));
// assertThat(page.getTotalElements(), is(equalTo(2L)));
// }
//
// @Test
// @Ignore("Test failing due to java.lang.IllegalArgumentException: Invalid order syntax for part Message!")
// public void shouldExecuteCustomMethodForNotIn(){
// //given
// String documentId = randomNumeric(5);
// SampleEntity sampleEntity = new SampleEntity();
// sampleEntity.setId(documentId);
// sampleEntity.setType("test");
// sampleEntity.setMessage("foo");
// repository.save(sampleEntity);
//
// //given
// String documentId2 = randomNumeric(5);
// SampleEntity sampleEntity2 = new SampleEntity();
// sampleEntity2.setId(documentId2);
// sampleEntity2.setType("test");
// sampleEntity2.setMessage("bar");
// repository.save(sampleEntity2);
//
// List<String> ids = Arrays.asList(documentId);
//
//
// //when
// Page<SampleEntity> page = repository.findByIdNotIn(ids, new PageRequest(1, 10));
// //then
// assertThat(page, is(notNullValue()));
// assertThat(page.getTotalElements(), is(equalTo(1L)));
// assertThat(page.getContent().get(0).getId(),is(documentId2));
// }
//
// @Test
// @Ignore("Test failing due to java.lang.IllegalArgumentException: Invalid order syntax for part Message!")
// public void shouldExecuteCustomMethodForTrue(){
// //given
// String documentId = randomNumeric(5);
// SampleEntity sampleEntity = new SampleEntity();
// sampleEntity.setId(documentId);
// sampleEntity.setType("test");
// sampleEntity.setMessage("foo");
// sampleEntity.setAvailable(true);
// repository.save(sampleEntity);
//
// //given
// String documentId2 = randomNumeric(5);
// SampleEntity sampleEntity2 = new SampleEntity();
// sampleEntity2.setId(documentId2);
// sampleEntity2.setType("test");
// sampleEntity2.setMessage("bar");
// sampleEntity2.setAvailable(false);
// repository.save(sampleEntity2);
// //when
// Page<SampleEntity> page = repository.findByAvailableTrue(new PageRequest(1, 10));
// //then
// assertThat(page, is(notNullValue()));
// assertThat(page.getTotalElements(), is(equalTo(1L)));
// }
//
// @Test
// @Ignore("Test failing due to java.lang.IllegalArgumentException: Invalid order syntax for part Message!")
// public void shouldExecuteCustomMethodForFalse(){
// //given
// String documentId = randomNumeric(5);
// SampleEntity sampleEntity = new SampleEntity();
// sampleEntity.setId(documentId);
// sampleEntity.setType("test");
// sampleEntity.setMessage("foo");
// sampleEntity.setAvailable(true);
// repository.save(sampleEntity);
//
// //given
// String documentId2 = randomNumeric(5);
// SampleEntity sampleEntity2 = new SampleEntity();
// sampleEntity2.setId(documentId2);
// sampleEntity2.setType("test");
// sampleEntity2.setMessage("bar");
// sampleEntity2.setAvailable(false);
// repository.save(sampleEntity2);
// //when
// Page<SampleEntity> page = repository.findByAvailableFalse(new PageRequest(1, 10));
// //then
// assertThat(page, is(notNullValue()));
// assertThat(page.getTotalElements(), is(equalTo(1L)));
// }
//
// @Test
// @Ignore("Test failing due to java.lang.IllegalArgumentException: Invalid order syntax for part Message!")
// public void shouldExecuteCustomMethodForOrderBy(){
// //given
// String documentId = randomNumeric(5);
// SampleEntity sampleEntity = new SampleEntity();
// sampleEntity.setId(documentId);
// sampleEntity.setType("test");
// sampleEntity.setMessage("foo");
// sampleEntity.setAvailable(true);
// repository.save(sampleEntity);
//
// //given
// String documentId2 = randomNumeric(5);
// SampleEntity sampleEntity2 = new SampleEntity();
// sampleEntity2.setId(documentId2);
// sampleEntity2.setType("test");
// sampleEntity2.setMessage("bar");
// sampleEntity2.setAvailable(false);
// repository.save(sampleEntity2);
// //when
// Page<SampleEntity> page = repository.findByMessageOrderByMessage("foo",new PageRequest(1, 10));
// //then
// assertThat(page, is(notNullValue()));
// assertThat(page.getTotalElements(), is(equalTo(1L)));
// }
//
// @Test
// public void testCustomMethodForBoolean(){
// //given
// String documentId = randomNumeric(5);
// SampleEntity sampleEntity = new SampleEntity();
// sampleEntity.setId(documentId);
// sampleEntity.setType("test");
// sampleEntity.setMessage("foo");
// sampleEntity.setAvailable(true);
// repository.save(sampleEntity);
//
// //given
// String documentId2 = randomNumeric(5);
// SampleEntity sampleEntity2 = new SampleEntity();
// sampleEntity2.setId(documentId2);
// sampleEntity2.setType("test");
// sampleEntity2.setMessage("bar");
// sampleEntity2.setAvailable(false);
// repository.save(sampleEntity2);
// //when
// Page<SampleEntity> page = repository.findByAvailable(false,new PageRequest(1, 10));
// //then
// assertThat(page, is(notNullValue()));
// assertThat(page.getTotalElements(), is(equalTo(1L)));
// }
@Test
public void shouldExecuteCustomMethodForIn(){
//given
String documentId = randomNumeric(5);
SampleEntity sampleEntity = new SampleEntity();
sampleEntity.setId(documentId);
sampleEntity.setType("test");
sampleEntity.setMessage("foo");
repository.save(sampleEntity);
//given
String documentId2 = randomNumeric(5);
SampleEntity sampleEntity2 = new SampleEntity();
sampleEntity2.setId(documentId2);
sampleEntity2.setType("test");
sampleEntity2.setMessage("bar");
repository.save(sampleEntity2);
List<String> ids = Arrays.asList(documentId,documentId2);
//when
Page<SampleEntity> page = repository.findByIdIn(ids, new PageRequest(1, 10));
//then
assertThat(page, is(notNullValue()));
assertThat(page.getTotalElements(), is(equalTo(2L)));
}
@Test
public void shouldExecuteCustomMethodForNotIn(){
//given
String documentId = randomNumeric(5);
SampleEntity sampleEntity = new SampleEntity();
sampleEntity.setId(documentId);
sampleEntity.setType("test");
sampleEntity.setMessage("foo");
repository.save(sampleEntity);
//given
String documentId2 = randomNumeric(5);
SampleEntity sampleEntity2 = new SampleEntity();
sampleEntity2.setId(documentId2);
sampleEntity2.setType("test");
sampleEntity2.setMessage("bar");
repository.save(sampleEntity2);
List<String> ids = Arrays.asList(documentId);
//when
Page<SampleEntity> page = repository.findByIdNotIn(ids, new PageRequest(1, 10));
//then
assertThat(page, is(notNullValue()));
assertThat(page.getTotalElements(), is(equalTo(1L)));
assertThat(page.getContent().get(0).getId(),is(documentId2));
}
@Test
public void shouldExecuteCustomMethodForTrue(){
//given
String documentId = randomNumeric(5);
SampleEntity sampleEntity = new SampleEntity();
sampleEntity.setId(documentId);
sampleEntity.setType("test");
sampleEntity.setMessage("foo");
sampleEntity.setAvailable(true);
repository.save(sampleEntity);
//given
String documentId2 = randomNumeric(5);
SampleEntity sampleEntity2 = new SampleEntity();
sampleEntity2.setId(documentId2);
sampleEntity2.setType("test");
sampleEntity2.setMessage("bar");
sampleEntity2.setAvailable(false);
repository.save(sampleEntity2);
//when
Page<SampleEntity> page = repository.findByAvailableTrue(new PageRequest(1, 10));
//then
assertThat(page, is(notNullValue()));
assertThat(page.getTotalElements(), is(equalTo(1L)));
}
@Test
public void shouldExecuteCustomMethodForFalse(){
//given
String documentId = randomNumeric(5);
SampleEntity sampleEntity = new SampleEntity();
sampleEntity.setId(documentId);
sampleEntity.setType("test");
sampleEntity.setMessage("foo");
sampleEntity.setAvailable(true);
repository.save(sampleEntity);
//given
String documentId2 = randomNumeric(5);
SampleEntity sampleEntity2 = new SampleEntity();
sampleEntity2.setId(documentId2);
sampleEntity2.setType("test");
sampleEntity2.setMessage("bar");
sampleEntity2.setAvailable(false);
repository.save(sampleEntity2);
//when
Page<SampleEntity> page = repository.findByAvailableFalse(new PageRequest(1, 10));
//then
assertThat(page, is(notNullValue()));
assertThat(page.getTotalElements(), is(equalTo(1L)));
}
@Test
public void shouldExecuteCustomMethodForOrderBy(){
//given
String documentId = randomNumeric(5);
SampleEntity sampleEntity = new SampleEntity();
sampleEntity.setId(documentId);
sampleEntity.setType("test");
sampleEntity.setMessage("foo");
sampleEntity.setAvailable(true);
repository.save(sampleEntity);
//given
String documentId2 = randomNumeric(5);
SampleEntity sampleEntity2 = new SampleEntity();
sampleEntity2.setId(documentId2);
sampleEntity2.setType("test");
sampleEntity2.setMessage("bar");
sampleEntity2.setAvailable(false);
repository.save(sampleEntity2);
//when
Page<SampleEntity> page = repository.findByMessageOrderByTypeAsc("foo",new PageRequest(1, 10));
//then
assertThat(page, is(notNullValue()));
assertThat(page.getTotalElements(), is(equalTo(1L)));
}
@Test
public void testCustomMethodForBoolean(){
//given
String documentId = randomNumeric(5);
SampleEntity sampleEntity = new SampleEntity();
sampleEntity.setId(documentId);
sampleEntity.setType("test");
sampleEntity.setMessage("foo");
sampleEntity.setAvailable(true);
repository.save(sampleEntity);
//given
String documentId2 = randomNumeric(5);
SampleEntity sampleEntity2 = new SampleEntity();
sampleEntity2.setId(documentId2);
sampleEntity2.setType("test");
sampleEntity2.setMessage("bar");
sampleEntity2.setAvailable(false);
repository.save(sampleEntity2);
//when
Page<SampleEntity> page = repository.findByAvailable(false,new PageRequest(1, 10));
//then
assertThat(page, is(notNullValue()));
assertThat(page.getTotalElements(), is(equalTo(1L)));
}
}

20
src/test/java/org/springframework/data/elasticsearch/repositories/SampleCustomMethodRepository.java
View File

@@ -18,7 +18,7 @@ public interface SampleCustomMethodRepository extends ElasticsearchRepository<Sa
@Query("{\"bool\" : {\"must\" : {\"field\" : {\"message\" : \"?0\"}}}}")
Page<SampleEntity> findByMessage(String message, Pageable pageable);
// Page<SampleEntity> findByAvailable(boolean available, Pageable pageable);
Page<SampleEntity> findByAvailable(boolean available, Pageable pageable);
Page<SampleEntity> findByRateLessThan(int rate, Pageable pageable);
@@ -34,14 +34,14 @@ public interface SampleCustomMethodRepository extends ElasticsearchRepository<Sa
Page<SampleEntity> findByMessageContaining(String message, Pageable pageable);
// Page<SampleEntity> findByIdIn(List<String> ids, Pageable pageable);
//
// Page<SampleEntity> findByIdNotIn(List<String> messages, Pageable pageable);
//
// Page<SampleEntity> findByAvailableTrue(Pageable pageable);
//
// Page<SampleEntity> findByAvailableFalse(Pageable pageable);
//
// Page<SampleEntity> findByMessageOrderByMessage(String message,Pageable pageable);
Page<SampleEntity> findByIdIn(List<String> ids, Pageable pageable);
Page<SampleEntity> findByIdNotIn(List<String> messages, Pageable pageable);
Page<SampleEntity> findByAvailableTrue(Pageable pageable);
Page<SampleEntity> findByAvailableFalse(Pageable pageable);
Page<SampleEntity> findByMessageOrderByTypeAsc(String message,Pageable pageable);
}