Add keyword support for Greater than equal, #144.

Signed-off-by: Pan Li <panli@microsoft.com>

src/main/java/com/microsoft/azure/spring/data/cosmosdb/core/generator/AbstractQueryGenerator.java

@@ -56,6 +56,7 @@ private String generateQueryBody(@NonNull Criteria criteria, @NonNull List<Pair<
             case BEFORE:
             case AFTER:
             case GREATER_THAN:
+            case GREATER_THAN_EQUAL:
                 return this.generateUnaryQuery(criteria, parameters);
             case AND:
             case OR:

src/main/java/com/microsoft/azure/spring/data/cosmosdb/core/query/CriteriaType.java

@@ -23,7 +23,8 @@ public enum CriteriaType {
     AND("AND"),
     BEFORE("<"),
     AFTER(">"),
-    GREATER_THAN(">");
+    GREATER_THAN(">"),
+    GREATER_THAN_EQUAL(">=");
 
     @Getter
     private String sqlKeyword;
@@ -38,6 +39,7 @@ public enum CriteriaType {
         map.put(Part.Type.BEFORE, CriteriaType.BEFORE);
         map.put(Part.Type.AFTER, CriteriaType.AFTER);
         map.put(Part.Type.GREATER_THAN, CriteriaType.GREATER_THAN);
+        map.put(Part.Type.GREATER_THAN_EQUAL, CriteriaType.GREATER_THAN_EQUAL);
 
         PART_TREE_TYPE_TO_CRITERIA = Collections.unmodifiableMap(map);
     }
@@ -100,6 +102,7 @@ public static boolean isUnary(CriteriaType type) {
             case BEFORE:
             case AFTER:
             case GREATER_THAN:
+            case GREATER_THAN_EQUAL:
                 return true;
             default:
                 return false;

src/test/java/com/microsoft/azure/spring/data/cosmosdb/repository/integration/ProjectRepositoryIT.java

@@ -47,12 +47,14 @@ public class ProjectRepositoryIT {
     private static final Long STAR_COUNT_1 = 1L;
     private static final Long STAR_COUNT_2 = 2L;
     private static final Long STAR_COUNT_3 = 3L;
+    private static final Long STAR_COUNT_MAX = 100L;
 
     private static final Long FORK_COUNT_0 = 0L;
     private static final Long FORK_COUNT_1 = 1L;
     private static final Long FORK_COUNT_2 = 2L;
     private static final Long FORK_COUNT_3 = 3L;
     private static final Long FAKE_COUNT = 123234L;
+    private static final Long FORK_COUNT_MAX = 100L;
 
     private static final Project PROJECT_0 = new Project(ID_0, NAME_0, CREATOR_0, true, STAR_COUNT_0, FORK_COUNT_0);
     private static final Project PROJECT_1 = new Project(ID_1, NAME_1, CREATOR_1, true, STAR_COUNT_1, FORK_COUNT_1);
@@ -279,4 +281,38 @@ public void testFindByGreaterThanWithAndOr() {
         Assert.assertEquals(projects.size(), reference.size());
         Assert.assertEquals(projects, reference);
     }
+
+    @Test
+    public void testFindByGreaterThanEqual() {
+        List<Project> projects = repository.findByStarCountGreaterThanEqual(STAR_COUNT_MAX);
+
+        Assert.assertTrue(projects.isEmpty());
+
+        projects = repository.findByStarCountGreaterThanEqual(STAR_COUNT_2);
+
+        final List<Project> reference = Arrays.asList(PROJECT_2, PROJECT_3);
+
+        projects.sort(Comparator.comparing(Project::getId));
+        reference.sort(Comparator.comparing(Project::getId));
+
+        Assert.assertEquals(projects.size(), reference.size());
+        Assert.assertEquals(projects, reference);
+    }
+
+    @Test
+    public void testFindByGreaterThanEqualAnd() {
+        List<Project> projects = repository.findByForkCountGreaterThanEqualAndCreator(FORK_COUNT_MAX, CREATOR_2);
+
+        Assert.assertTrue(projects.isEmpty());
+
+        projects = repository.findByForkCountGreaterThanEqualAndCreator(FORK_COUNT_0, CREATOR_0);
+
+        final List<Project> reference = Arrays.asList(PROJECT_0, PROJECT_4);
+
+        projects.sort(Comparator.comparing(Project::getId));
+        reference.sort(Comparator.comparing(Project::getId));
+
+        Assert.assertEquals(projects.size(), reference.size());
+        Assert.assertEquals(projects, reference);
+    }
 }

src/test/java/com/microsoft/azure/spring/data/cosmosdb/repository/repository/ProjectRepository.java

@@ -41,4 +41,8 @@ List<Project> findByNameOrCreatorAndForkCountOrStarCount(String name, String cre
     List<Project> findByNameAndCreator(String name, String creator, Sort sort);
 
     List<Project> findByForkCount(Long forkCount, Sort sort);
+
+    List<Project> findByStarCountGreaterThanEqual(Long count);
+
+    List<Project> findByForkCountGreaterThanEqualAndCreator(Long count, String creator);
 }