[CALCITE-6867] Druid Adapter transforms an SQL NOT IN filter to a Druid IN filter

Author: xuzifu666

druid/src/main/java/org/apache/calcite/adapter/druid/DruidJsonFilter.java

@@ -276,6 +276,7 @@ abstract class DruidJsonFilter implements DruidJson {
       throw new AssertionError(
           DruidQuery.format("Expecting IN or NOT IN but got [%s]", e.getKind()));
     }
+
     ImmutableList.Builder<String> listBuilder = ImmutableList.builder();
     for (RexNode rexNode : ((RexCall) e).getOperands()) {
       if (rexNode.getKind() == SqlKind.LITERAL) {
@@ -294,7 +295,7 @@ abstract class DruidJsonFilter implements DruidJson {
     if (columnName == null) {
       return null;
     }
-    if (e.getKind() != SqlKind.NOT_IN) {
+    if (e.getKind() != SqlKind.DRUID_NOT_IN) {
       return new DruidJsonFilter.JsonInFilter(columnName, listBuilder.build(), extractionFunction);
     } else {
       return toNotDruidFilter(

druid/src/test/java/org/apache/calcite/adapter/druid/DruidQueryFilterTest.java

@@ -87,6 +87,30 @@ class DruidQueryFilterTest {
             + "\"values\":[\"1\",\"5\",\"value1\"]}"));
   }
 
+  @Test void testNotInFilter() throws IOException {
+    final Fixture f = new Fixture();
+    final List<? extends RexNode> listRexNodes =
+        ImmutableList.of(f.rexBuilder.makeInputRef(f.varcharRowType, 0),
+            f.rexBuilder.makeExactLiteral(BigDecimal.valueOf(1)),
+            f.rexBuilder.makeExactLiteral(BigDecimal.valueOf(5)),
+            f.rexBuilder.makeLiteral("value1"));
+
+    RexNode notInRexNode =
+        f.rexBuilder.makeCall(SqlInternalOperators.DRUID_NOT_IN, listRexNodes);
+    DruidJsonFilter returnValue = DruidJsonFilter
+        .toDruidFilters(notInRexNode, f.varcharRowType, druidQuery, f.rexBuilder);
+    assertThat("Filter is null", returnValue, notNullValue());
+    JsonFactory jsonFactory = new JsonFactory();
+    final StringWriter sw = new StringWriter();
+    JsonGenerator jsonGenerator = jsonFactory.createGenerator(sw);
+    returnValue.write(jsonGenerator);
+    jsonGenerator.close();
+
+    assertThat(sw,
+        hasToString("{\"type\":\"not\",\"field\":{\"type\":\"in\",\"dimension\":"
+            + "\"dimensionName\",\"values\":[\"1\",\"5\",\"value1\"]}}"));
+  }
+
   @Test void testBetweenFilterStringCase() throws IOException {
     final Fixture f = new Fixture();
     final List<RexNode> listRexNodes =