spring-projects#282 - Add support of 'AFTER/BEFORE' conditions

Author: rchigvintsev

Diff for: src/main/java/org/springframework/data/r2dbc/repository/query/ConditionFactory.java

@@ -66,6 +66,7 @@ public Condition createCondition(Part part) {
                 return Conditions.isGreaterOrEqualTo(pathExpression, firstBindMarker)
                         .and(Conditions.isLessOrEqualTo(pathExpression, secondBindMarker));
             }
+            case AFTER:
             case GREATER_THAN: {
                 Expression pathExpression = createPropertyPathExpression(part.getProperty());
                 BindMarker bindMarker = createBindMarker(parameterMetadataProvider.next(part));
@@ -76,6 +77,7 @@ public Condition createCondition(Part part) {
                 BindMarker bindMarker = createBindMarker(parameterMetadataProvider.next(part));
                 return Conditions.isGreaterOrEqualTo(pathExpression, bindMarker);
             }
+            case BEFORE:
             case LESS_THAN: {
                 Expression pathExpression = createPropertyPathExpression(part.getProperty());
                 BindMarker bindMarker = createBindMarker(parameterMetadataProvider.next(part));

Diff for: src/test/java/org/springframework/data/r2dbc/repository/query/PartTreeR2dbcQueryIntegrationTests.java

@@ -193,6 +193,28 @@ public void createsQueryToFindAllEntitiesByIntegerAttributeGreaterThanEqual() th
         assertThat(bindableQuery.get()).isEqualTo(expectedSql);
     }
 
+    @Test
+    public void createsQueryToFindAllEntitiesByDateAttributeAfter() throws Exception {
+        R2dbcQueryMethod queryMethod = getQueryMethod("findAllByDateOfBirthAfter", Date.class);
+        PartTreeR2dbcQuery r2dbcQuery = new PartTreeR2dbcQuery(queryMethod, databaseClient, r2dbcConverter,
+                dataAccessStrategy);
+        RelationalParametersParameterAccessor accessor = getAccessor(queryMethod, new Object[]{new Date()});
+        BindableQuery bindableQuery = r2dbcQuery.createQuery(accessor);
+        String expectedSql = "SELECT " + ALL_FIELDS + " FROM " + TABLE + " WHERE " + TABLE + ".date_of_birth > ?";
+        assertThat(bindableQuery.get()).isEqualTo(expectedSql);
+    }
+
+    @Test
+    public void createsQueryToFindAllEntitiesByDateAttributeBefore() throws Exception {
+        R2dbcQueryMethod queryMethod = getQueryMethod("findAllByDateOfBirthBefore", Date.class);
+        PartTreeR2dbcQuery r2dbcQuery = new PartTreeR2dbcQuery(queryMethod, databaseClient, r2dbcConverter,
+                dataAccessStrategy);
+        RelationalParametersParameterAccessor accessor = getAccessor(queryMethod, new Object[]{new Date()});
+        BindableQuery bindableQuery = r2dbcQuery.createQuery(accessor);
+        String expectedSql = "SELECT " + ALL_FIELDS + " FROM " + TABLE + " WHERE " + TABLE + ".date_of_birth < ?";
+        assertThat(bindableQuery.get()).isEqualTo(expectedSql);
+    }
+
     private R2dbcQueryMethod getQueryMethod(String methodName, Class<?>... parameterTypes) throws Exception {
         Method method = UserRepository.class.getMethod(methodName, parameterTypes);
         return new R2dbcQueryMethod(method, new DefaultRepositoryMetadata(UserRepository.class),
@@ -221,6 +243,10 @@ private interface UserRepository extends Repository<User, Long> {
         Flux<User> findAllByAgeGreaterThan(Integer age);
 
         Flux<User> findAllByAgeGreaterThanEqual(Integer age);
+
+        Flux<User> findAllByDateOfBirthAfter(Date date);
+
+        Flux<User> findAllByDateOfBirthBefore(Date date);
     }
 
     @Table("users")