IN query need to return list conditions

Change-Id: I24eec52bc98e456e2e968aae992187e97d725b33
apache · Jan 20, 2022 · 8cd4970 · 8cd4970
1 parent 587670b
commit 8cd4970
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/query/ConditionQuery.java b/hugegraph-core/src/main/java/com/baidu/hugegraph/backend/query/ConditionQuery.java
@@ -242,6 +242,11 @@ public <T> T condition(Object key) {
             T value = (T) valuesEQ.get(0);
             return value;
         }
+        if (valuesEQ.size() == 0 && valuesIN.size() == 1) {
+            @SuppressWarnings("unchecked")
+            T value = (T) valuesIN.get(0);
+            return value;
+        }
 
         Set<Object> intersectValues = InsertionOrderUtil.newSet();
         for (Object value : valuesEQ) {
@@ -263,6 +268,12 @@ public <T> T condition(Object key) {
                                                    valueAsList);
             }
         }
+
+        if (valuesIN.size() > 0) {
+            @SuppressWarnings("unchecked")
+            T value = (T) intersectValues;
+            return value;
+        }
         E.checkState(intersectValues.size() == 1,
                      "Illegal key '%s' with more than one value: %s",
                      key, intersectValues);