Cover range(DATE)->DATETIME case

craigtaverner · Nov 8, 2024 · d1aa099 · d1aa099
1 parent 7cd46a0
commit d1aa099
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/x-pack/plugin/esql-core/src/main/java/org/elasticsearch/xpack/esql/core/type/DataType.java b/x-pack/plugin/esql-core/src/main/java/org/elasticsearch/xpack/esql/core/type/DataType.java
@@ -588,6 +588,13 @@ public DataType noText() {
         return isString(this) ? KEYWORD : this;
     }
 
+    public boolean isDate() {
+        return switch (this) {
+            case DATETIME, DATE_NANOS -> true;
+            default -> false;
+        };
+    }
+
     /**
      * Named parameters with default values. It's just easier to do this with
      * a builder in java....

diff --git a/...ck/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/enrich/EnrichLookupService.java b/...ck/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/enrich/EnrichLookupService.java
@@ -107,6 +107,7 @@ private static boolean rangeTypesCompatible(RangeType rangeType, DataType inputD
         return switch (rangeType) {
             case INTEGER, LONG -> inputDataType.isWholeNumber();
             case IP -> inputDataType == DataType.IP;
+            case DATE -> inputDataType.isDate();
             default -> rangeType.isNumeric() == inputDataType.isNumeric();
         };
     }