[Feature] Support jodatime_format function to partition pruning

fe/fe-core/src/main/java/com/starrocks/sql/optimizer/rewrite/ScalarOperatorFunctions.java

@@ -51,6 +51,8 @@
 import org.apache.commons.lang.StringUtils;
 import org.apache.http.NameValuePair;
 import org.apache.http.client.utils.URLEncodedUtils;
+import org.joda.time.DateTime;
+import org.joda.time.format.DateTimeFormat;
 
 import java.math.BigDecimal;
 import java.math.BigInteger;
@@ -380,6 +382,26 @@ public static ConstantOperator dateFormat(ConstantOperator date, ConstantOperato
         }
     }
 
+    @ConstantFunction.List(list = {
+            @ConstantFunction(name = "jodatime_format", argTypes = {DATETIME, VARCHAR},
+                    returnType = VARCHAR, isMonotonic = true),
+            @ConstantFunction(name = "jodatime_format", argTypes = {DATE, VARCHAR},
+                    returnType = VARCHAR, isMonotonic = true)
+    })
+    public static ConstantOperator jodatimeFormat(ConstantOperator date, ConstantOperator fmtLiteral) {
+        String format = fmtLiteral.getVarchar();
+        if (format.isEmpty()) {
+            return ConstantOperator.createNull(Type.VARCHAR);
+        }
+        org.joda.time.format.DateTimeFormatter formatter = DateTimeFormat.forPattern(format);
+        DateTime jodaDateTime = new DateTime(date.getDatetime()
+                .atZone(ZoneId.systemDefault()) // 将 LocalDateTime 与系统默认时区关联
+                .toInstant() // 转换为 Instant
+                .toEpochMilli());
+        return ConstantOperator.createVarchar(jodaDateTime.toString(formatter));
+    }
+
+
     @ConstantFunction.List(list = {
             @ConstantFunction(name = "to_iso8601", argTypes = {DATETIME}, returnType = VARCHAR, isMonotonic = true),
             @ConstantFunction(name = "to_iso8601", argTypes = {DATE}, returnType = VARCHAR, isMonotonic = true)

fe/fe-core/src/test/java/com/starrocks/connector/parser/trino/TrinoFunctionTransformTest.java

@@ -171,13 +171,13 @@ public void testDateFnTransform() throws Exception {
         assertPlanContains(sql, "week_iso('2023-01-01 00:00:00')");
 
         sql = "select format_datetime(TIMESTAMP '2023-06-25 11:10:20', 'yyyyMMdd HH:mm:ss')";
-        assertPlanContains(sql, "jodatime_format('2023-06-25 11:10:20', 'yyyyMMdd HH:mm:ss')");
+        assertPlanContains(sql, "20230625 11:10:20");
 
         sql = "select format_datetime(date '2023-06-25', 'yyyyMMdd HH:mm:ss');";
-        assertPlanContains(sql, "jodatime_format('2023-06-25', 'yyyyMMdd HH:mm:ss')");
+        assertPlanContains(sql, "20230625 00:00:00");
 
         sql = "select to_char(TIMESTAMP '2023-06-25 11:10:20', 'yyyyMMdd HH:mm:ss');";
-        assertPlanContains(sql, "jodatime_format('2023-06-25 11:10:20', 'yyyyMMdd HH:mm:ss')");
+        assertPlanContains(sql, "20230625 11:10:20");
 
         sql = "select parse_datetime('2023-08-02 14:37:02', 'yyyy-MM-dd HH:mm:ss')";
         assertPlanContains(sql, "str_to_jodatime('2023-08-02 14:37:02', 'yyyy-MM-dd HH:mm:ss')");

fe/fe-core/src/test/java/com/starrocks/sql/optimizer/rewrite/ScalarOperatorFunctionsTest.java

@@ -1462,6 +1462,18 @@ public void testReplace() {
         ).getVarchar());
     }
 
+    @Test
+    public void testJodatimeFormat() {
+        assertEquals("", ScalarOperatorFunctions.jodatimeFormat(
+                new ConstantOperator("2024-08-06", Type.DATE),
+                new ConstantOperator("", Type.VARCHAR)).getVarchar());
+
+        assertEquals("20241109", ScalarOperatorFunctions.jodatimeFormat(
+                new ConstantOperator(LocalDateTime.of(2024, 11, 9, 15, 30, 45),
+                        Type.DATE),
+                new ConstantOperator("yyyyMMdd", Type.VARCHAR)).getVarchar());
+    }
+
     /*
     test cases are generated by the following SQL by capturing:
     1. leap year

test/sql/test_datetime/R/test_jodatime_format

@@ -5,7 +5,7 @@ select jodatime_format('2023-04-15', 'yyyy-MM-dd');
 -- !result
 select jodatime_format('2023-04-15 23:45:12', "yyyy-MM-dd'T'HH:mm:ssZZ");
 -- result:
-2023-04-15T23:45:12
+2023-04-15T23:45:12+08:00
 -- !result
 select jodatime_format('2023-04-15 15:30:00', 'yyyy-MM-dd hh:mm:ssa');
 -- result:
@@ -93,57 +93,57 @@ select jodatime_format('2023-06-25 00:00:00', "012345678901234567890123456789012
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "yyyyMMdd--'012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456'");
 -- result:
-None
+20230625--012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "yyyyMMdd--'012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456");
 -- result:
-None
+20230625--012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "yyyyMMdd--012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456");
 -- result:
-None
+20230625--012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "'012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456'--yyyyMMdd");
 -- result:
-None
+012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456--20230625
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456--yyyyMMdd");
 -- result:
-None
+012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456--20230625
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "'0123456789012345678901234567890123456789'--yyyyMMdd--'234567890123456789012345678901234567890123456789012345678901234567890123456'");
 -- result:
-None
+0123456789012345678901234567890123456789--20230625--234567890123456789012345678901234567890123456789012345678901234567890123456
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "0123456789012345678901234567890123456789--yyyyMMdd--234567890123456789012345678901234567890123456789012345678901234567890123456");
 -- result:
-None
+0123456789012345678901234567890123456789--20230625--234567890123456789012345678901234567890123456789012345678901234567890123456
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "yyyyMMdd--'0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567'");
 -- result:
-None
+20230625--0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "yyyyMMdd--'0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567");
 -- result:
-None
+20230625--0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "yyyyMMdd--0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567");
 -- result:
-None
+20230625--0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "'0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567'--yyyyMMdd");
 -- result:
-None
+0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567--20230625
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567--yyyyMMdd");
 -- result:
-None
+0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567--20230625
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "'0123456789012345678901234567890123456789'--yyyyMMdd--'2345678901234567890123456789012345678901234567890123456789012345678901234567'");
 -- result:
-None
+0123456789012345678901234567890123456789--20230625--2345678901234567890123456789012345678901234567890123456789012345678901234567
 -- !result
 select jodatime_format('2023-06-25 00:00:00', "0123456789012345678901234567890123456789--yyyyMMdd--2345678901234567890123456789012345678901234567890123456789012345678901234567");
 -- result:
-None
+0123456789012345678901234567890123456789--20230625--2345678901234567890123456789012345678901234567890123456789012345678901234567
 -- !result