From 284fd4bbeb7f69905431fa2daae52040e3e3a31c Mon Sep 17 00:00:00 2001
From: xiedeyantu <czjourney@163.com>
Date: Sun, 17 Mar 2024 11:06:30 +0800
Subject: [PATCH] data can work when partition column is datetime

---
 .../apache/doris/catalog/PartitionKey.java    | 21 ++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/fe/fe-core/src/main/java/org/apache/doris/catalog/PartitionKey.java b/fe/fe-core/src/main/java/org/apache/doris/catalog/PartitionKey.java
index 5816496c935862d..a1fd8034f9123c7 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/catalog/PartitionKey.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/catalog/PartitionKey.java
@@ -29,6 +29,9 @@
 import org.apache.doris.common.AnalysisException;
 import org.apache.doris.common.io.Text;
 import org.apache.doris.common.io.Writable;
+import org.apache.doris.nereids.trees.expressions.literal.DateTimeLiteral;
+import org.apache.doris.nereids.trees.expressions.literal.DateTimeV2Literal;
+import org.apache.doris.nereids.trees.expressions.literal.Literal;
 import org.apache.doris.qe.SessionVariable;
 
 import com.google.common.base.Joiner;
@@ -90,7 +93,14 @@ public static PartitionKey createPartitionKey(List<PartitionValue> keys, List<Co
         Preconditions.checkArgument(keys.size() <= columns.size());
         int i;
         for (i = 0; i < keys.size(); ++i) {
-            partitionKey.keys.add(keys.get(i).getValue(columns.get(i).getType()));
+            Type keyType = columns.get(i).getType();
+            // If column type is datatime and key type is date, we should convert date to datetime.
+            if (keyType.isDatetime() || keyType.isDatetimeV2()){
+                Literal dateTimeLiteral = getDateTimeLiteral(keys.get(i).getStringValue(), keyType);
+                partitionKey.keys.add(dateTimeLiteral.toLegacyLiteral());
+            } else {
+                partitionKey.keys.add(keys.get(i).getValue(keyType));
+            }
             partitionKey.types.add(columns.get(i).getDataType());
         }
 
@@ -104,6 +114,15 @@ public static PartitionKey createPartitionKey(List<PartitionValue> keys, List<Co
         return partitionKey;
     }
 
+    private static Literal getDateTimeLiteral(String value, Type type) throws AnalysisException {
+        if (type.equals(Type.DATETIME)) {
+            return new DateTimeLiteral(value);
+        } else if (type.equals(Type.DATETIMEV2)) {
+            return new DateTimeV2Literal(value);
+        }
+        throw new AnalysisException("date convert to datetime failed.");
+    }
+
     public static PartitionKey createListPartitionKeyWithTypes(List<PartitionValue> values, List<Type> types,
             boolean isHive)
             throws AnalysisException {