From 6730e3a3adbf052afe598163dcada3b223462083 Mon Sep 17 00:00:00 2001
From: morrySnow <morrysnow@126.com>
Date: Thu, 29 Aug 2024 17:23:50 +0800
Subject: [PATCH] [fix](Nereids) datetimev2 literal equals should compare
 microsecond (#40121)

pick from master #40121
---
 .../trees/expressions/literal/DateTimeV2Literal.java   |  7 ++++++-
 .../trees/expressions/literal/DateTimeLiteralTest.java | 10 ++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/literal/DateTimeV2Literal.java b/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/literal/DateTimeV2Literal.java
index a3457f2463d65c..0ca19bf2a92715 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/literal/DateTimeV2Literal.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/nereids/trees/expressions/literal/DateTimeV2Literal.java
@@ -104,6 +104,11 @@ public LiteralExpr toLegacyLiteral() {
                 getDataType().toCatalogDataType());
     }
 
+    @Override
+    public double getDouble() {
+        return super.getDouble() + microSecond / 1000000.0;
+    }
+
     @Override
     public String toString() {
         return getStringValue();
@@ -291,6 +296,6 @@ public boolean equals(Object o) {
             return false;
         }
         DateTimeV2Literal literal = (DateTimeV2Literal) o;
-        return Objects.equals(dataType, literal.dataType);
+        return Objects.equals(dataType, literal.dataType) && Objects.equals(microSecond, literal.microSecond);
     }
 }
diff --git a/fe/fe-core/src/test/java/org/apache/doris/nereids/trees/expressions/literal/DateTimeLiteralTest.java b/fe/fe-core/src/test/java/org/apache/doris/nereids/trees/expressions/literal/DateTimeLiteralTest.java
index 820c9ae9ac0b1a..68e2d80826bc84 100644
--- a/fe/fe-core/src/test/java/org/apache/doris/nereids/trees/expressions/literal/DateTimeLiteralTest.java
+++ b/fe/fe-core/src/test/java/org/apache/doris/nereids/trees/expressions/literal/DateTimeLiteralTest.java
@@ -475,4 +475,14 @@ void testRoundCeiling() {
         Assertions.assertEquals(1, literal.roundCeiling(0).month);
         Assertions.assertEquals(2001, literal.roundCeiling(0).year);
     }
+
+    @Test
+    void testEquals() {
+        DateTimeV2Literal l1 = new DateTimeV2Literal(1, 1, 1, 1, 1, 1, 1);
+        DateTimeV2Literal l2 = new DateTimeV2Literal(1, 1, 1, 1, 1, 1, 1);
+        DateTimeV2Literal l3 = new DateTimeV2Literal(1, 1, 1, 1, 1, 1, 2);
+
+        Assertions.assertEquals(l1, l2);
+        Assertions.assertNotEquals(l1, l3);
+    }
 }