fix: add 'CONTAIN' support in query builder

This fix to address the issue #114
linkedin · Jul 14, 2021 · 6ae3731 · 6ae3731
1 parent 77df5d8
commit 6ae3731
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/dao-impl/elasticsearch-dao-7/src/main/java/com/linkedin/metadata/dao/utils/SearchUtils.java b/dao-impl/elasticsearch-dao-7/src/main/java/com/linkedin/metadata/dao/utils/SearchUtils.java
@@ -78,6 +78,8 @@ public static QueryBuilder getQueryBuilderFromCriterion(@Nonnull Criterion crite
       return QueryBuilders.rangeQuery(criterion.getField()).lt(criterion.getValue().trim());
     } else if (condition == Condition.LESS_THAN_OR_EQUAL_TO) {
       return QueryBuilders.rangeQuery(criterion.getField()).lte(criterion.getValue().trim());
+    } else if (condition == Condition.CONTAIN) {
+      return QueryBuilders.termQuery(criterion.getField(), criterion.getValue().trim());
     }
 
     throw new UnsupportedOperationException("Unsupported condition: " + condition);

diff --git a/...l/elasticsearch-dao-7/src/test/java/com/linkedin/metadata/dao/search/ESSearchDAOTest.java b/...l/elasticsearch-dao-7/src/test/java/com/linkedin/metadata/dao/search/ESSearchDAOTest.java
@@ -248,7 +248,7 @@ public void testFilteredQueryUnsupportedCondition() {
     int from = 0;
     int size = 10;
     final Filter filter2 = new Filter().setCriteria(new CriterionArray(Arrays.asList(
-        new Criterion().setField("field_contain").setValue("value_contain").setCondition(Condition.CONTAIN))));
+        new Criterion().setField("field_contain").setValue("value_contain").setCondition(Condition.START_WITH))));
     SortCriterion sortCriterion = new SortCriterion().setOrder(SortOrder.ASCENDING).setField("urn");
     assertThrows(UnsupportedOperationException.class,
         () -> _searchDAO.getFilteredSearchQuery(filter2, sortCriterion, from, size));

diff --git a/dao-impl/elasticsearch-dao/src/main/java/com/linkedin/metadata/dao/utils/SearchUtils.java b/dao-impl/elasticsearch-dao/src/main/java/com/linkedin/metadata/dao/utils/SearchUtils.java
@@ -88,6 +88,8 @@ public static QueryBuilder getQueryBuilderFromCriterion(@Nonnull Criterion crite
       return QueryBuilders.rangeQuery(criterion.getField()).lt(criterion.getValue().trim());
     } else if (condition == Condition.LESS_THAN_OR_EQUAL_TO) {
       return QueryBuilders.rangeQuery(criterion.getField()).lte(criterion.getValue().trim());
+    } else if (condition == Condition.CONTAIN) {
+      return QueryBuilders.termQuery(criterion.getField(), criterion.getValue().trim());
     }
 
     throw new UnsupportedOperationException("Unsupported condition: " + condition);

diff --git a/...mpl/elasticsearch-dao/src/test/java/com/linkedin/metadata/dao/search/ESSearchDAOTest.java b/...mpl/elasticsearch-dao/src/test/java/com/linkedin/metadata/dao/search/ESSearchDAOTest.java
@@ -246,7 +246,7 @@ public void testFilteredQueryUnsupportedCondition() {
     int from = 0;
     int size = 10;
     final Filter filter2 = new Filter().setCriteria(new CriterionArray(Arrays.asList(
-        new Criterion().setField("field_contain").setValue("value_contain").setCondition(Condition.CONTAIN))));
+        new Criterion().setField("field_contain").setValue("value_contain").setCondition(Condition.START_WITH))));
     SortCriterion sortCriterion = new SortCriterion().setOrder(SortOrder.ASCENDING).setField("urn");
     assertThrows(UnsupportedOperationException.class,
         () -> _searchDAO.getFilteredSearchQuery(filter2, sortCriterion, from, size));