move main method to ut

apache · Apr 7, 2024 · 2a97ec0 · 2a97ec0
1 parent 3f44f9c
commit 2a97ec0
Show file tree

Hide file tree

Showing 2 changed files with 167 additions and 62 deletions.
diff --git a/seatunnel-common/src/main/java/org/apache/seatunnel/common/utils/DateTimeUtils.java b/seatunnel-common/src/main/java/org/apache/seatunnel/common/utils/DateTimeUtils.java
@@ -274,66 +274,4 @@ public static Formatter parse(String format) {
             throw new IllegalArgumentException(errorMsg);
         }
     }
-
-    public static void main(String[] args) {
-        String datetimeStr = "2020-10-10 10:10:10";
-        DateTimeFormatter dateTimeFormatter = matchDateTimeFormatter(datetimeStr);
-        String datetimeStr1 = "20201010101010";
-        DateTimeFormatter dateTimeFormatter1 = matchDateTimeFormatter(datetimeStr1);
-        String datetimeStr2 = "2020.10.10 10:10:10.100";
-        DateTimeFormatter dateTimeFormatter2 = matchDateTimeFormatter(datetimeStr2);
-        String datetimeStr3 = "2020.10.10 10:10:10";
-        DateTimeFormatter dateTimeFormatter3 = matchDateTimeFormatter(datetimeStr3);
-        long t1 = System.currentTimeMillis();
-        for (int i = 0; i < 10000000; i++) {
-            parse(datetimeStr, dateTimeFormatter);
-        }
-        long t2 = System.currentTimeMillis();
-        System.out.println((t2 - t1) + "");
-
-        for (int i = 0; i < 10000000; i++) {
-            parse(datetimeStr);
-        }
-        long t3 = System.currentTimeMillis();
-        System.out.println((t3 - t2) + "");
-
-        long t4 = System.currentTimeMillis();
-        for (int i = 0; i < 10000000; i++) {
-            parse(datetimeStr1, dateTimeFormatter1);
-        }
-        long t5 = System.currentTimeMillis();
-        System.out.println((t5 - t4) + "");
-
-        for (int i = 0; i < 10000000; i++) {
-            parse(datetimeStr1);
-        }
-        long t6 = System.currentTimeMillis();
-        System.out.println((t6 - t5) + "");
-
-        long t7 = System.currentTimeMillis();
-        for (int i = 0; i < 10000000; i++) {
-            parse(datetimeStr2, dateTimeFormatter2);
-        }
-        long t8 = System.currentTimeMillis();
-        System.out.println((t8 - t7) + "");
-
-        for (int i = 0; i < 10000000; i++) {
-            parse(datetimeStr2);
-        }
-        long t9 = System.currentTimeMillis();
-        System.out.println((t9 - t8) + "");
-
-        long t10 = System.currentTimeMillis();
-        for (int i = 0; i < 10000000; i++) {
-            parse(datetimeStr3, dateTimeFormatter3);
-        }
-        long t11 = System.currentTimeMillis();
-        System.out.println((t11 - t10) + "");
-
-        for (int i = 0; i < 10000000; i++) {
-            parse(datetimeStr3);
-        }
-        long t12 = System.currentTimeMillis();
-        System.out.println((t12 - t11) + "");
-    }
 }
diff --git a/seatunnel-common/src/test/java/org/apache/seatunnel/common/utils/DateTimeUtilsTest.java b/seatunnel-common/src/test/java/org/apache/seatunnel/common/utils/DateTimeUtilsTest.java
@@ -24,6 +24,7 @@
 
 import java.time.LocalDateTime;
 import java.time.ZoneId;
+import java.time.format.DateTimeFormatter;
 
 public class DateTimeUtilsTest {
 
@@ -54,4 +55,170 @@ public void testParseTimestamp() {
         Assertions.assertEquals(2023, parse.getYear());
         Assertions.assertEquals(22, parse.getDayOfMonth());
     }
+
+    @Test
+    public void testAutoDateTimeFormatter() {
+        String datetimeStr = "2020-10-10 10:10:10";
+        Assertions.assertEquals("2020-10-10T10:10:10", DateTimeUtils.parse(datetimeStr).toString());
+
+        datetimeStr = "2020-10-10T10:10:10";
+        Assertions.assertEquals("2020-10-10T10:10:10", DateTimeUtils.parse(datetimeStr).toString());
+
+        datetimeStr = "2020/10/10 10:10:10";
+        Assertions.assertEquals("2020-10-10T10:10:10", DateTimeUtils.parse(datetimeStr).toString());
+
+        datetimeStr = "2020年10月10日 10时10分10秒";
+        Assertions.assertEquals("2020-10-10T10:10:10", DateTimeUtils.parse(datetimeStr).toString());
+
+        datetimeStr = "2020.10.10 10:10:10";
+        Assertions.assertEquals("2020-10-10T10:10:10", DateTimeUtils.parse(datetimeStr).toString());
+
+        datetimeStr = "20201010101010";
+        Assertions.assertEquals("2020-10-10T10:10:10", DateTimeUtils.parse(datetimeStr).toString());
+
+        datetimeStr = "2020-10-10 10:10:10.201";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10.201", DateTimeUtils.parse(datetimeStr).toString());
+
+        datetimeStr = "2020-10-10 10:10:10.201111";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10.201111", DateTimeUtils.parse(datetimeStr).toString());
+
+        datetimeStr = "2020-10-10 10:10:10.201111001";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10.201111001", DateTimeUtils.parse(datetimeStr).toString());
+    }
+
+    @Test
+    public void testMatchDateTimeFormatter() {
+        String datetimeStr = "2020-10-10 10:10:10";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10",
+                DateTimeUtils.parse(datetimeStr, DateTimeUtils.matchDateTimeFormatter(datetimeStr))
+                        .toString());
+
+        datetimeStr = "2020-10-10T10:10:10";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10",
+                DateTimeUtils.parse(datetimeStr, DateTimeUtils.matchDateTimeFormatter(datetimeStr))
+                        .toString());
+
+        datetimeStr = "2020/10/10 10:10:10";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10",
+                DateTimeUtils.parse(datetimeStr, DateTimeUtils.matchDateTimeFormatter(datetimeStr))
+                        .toString());
+
+        datetimeStr = "2020年10月10日 10时10分10秒";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10",
+                DateTimeUtils.parse(datetimeStr, DateTimeUtils.matchDateTimeFormatter(datetimeStr))
+                        .toString());
+
+        datetimeStr = "2020.10.10 10:10:10";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10",
+                DateTimeUtils.parse(datetimeStr, DateTimeUtils.matchDateTimeFormatter(datetimeStr))
+                        .toString());
+
+        datetimeStr = "20201010101010";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10",
+                DateTimeUtils.parse(datetimeStr, DateTimeUtils.matchDateTimeFormatter(datetimeStr))
+                        .toString());
+
+        datetimeStr = "2020-10-10 10:10:10.201";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10.201",
+                DateTimeUtils.parse(datetimeStr, DateTimeUtils.matchDateTimeFormatter(datetimeStr))
+                        .toString());
+
+        datetimeStr = "2020-10-10 10:10:10.201111";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10.201111",
+                DateTimeUtils.parse(datetimeStr, DateTimeUtils.matchDateTimeFormatter(datetimeStr))
+                        .toString());
+
+        datetimeStr = "2020-10-10 10:10:10.201111001";
+        Assertions.assertEquals(
+                "2020-10-10T10:10:10.201111001",
+                DateTimeUtils.parse(datetimeStr, DateTimeUtils.matchDateTimeFormatter(datetimeStr))
+                        .toString());
+    }
+
+    @Test
+    public void testPerformance() {
+        String datetimeStr = "2020-10-10 10:10:10";
+        DateTimeFormatter dateTimeFormatter = DateTimeUtils.matchDateTimeFormatter(datetimeStr);
+        String datetimeStr1 = "20201010101010";
+        DateTimeFormatter dateTimeFormatter1 = DateTimeUtils.matchDateTimeFormatter(datetimeStr1);
+        String datetimeStr2 = "2020.10.10 10:10:10.100";
+        DateTimeFormatter dateTimeFormatter2 = DateTimeUtils.matchDateTimeFormatter(datetimeStr2);
+        String datetimeStr3 = "2020.10.10 10:10:10";
+        DateTimeFormatter dateTimeFormatter3 = DateTimeUtils.matchDateTimeFormatter(datetimeStr3);
+        long t1 = System.currentTimeMillis();
+        for (int i = 0; i < 10000000; i++) {
+            DateTimeUtils.parse(datetimeStr, dateTimeFormatter);
+        }
+        long t2 = System.currentTimeMillis();
+        // Use an explicit time format 'yyyy-MM-dd HH:mm:ss' for processing, use time: 4552ms
+        System.out.println((t2 - t1) + "");
+
+        for (int i = 0; i < 10000000; i++) {
+            DateTimeUtils.parse(datetimeStr);
+        }
+        long t3 = System.currentTimeMillis();
+        // If format is not specified, the system automatically obtains the format 'yyyy-MM-dd
+        // HH:mm:ss' for processing, use time: 6082ms
+        System.out.println((t3 - t2) + "");
+
+        long t4 = System.currentTimeMillis();
+        for (int i = 0; i < 10000000; i++) {
+            DateTimeUtils.parse(datetimeStr1, dateTimeFormatter1);
+        }
+        long t5 = System.currentTimeMillis();
+        // Use an explicit time format 'yyyyMMddHHmmss' for processing, use time: 4610ms
+        System.out.println((t5 - t4) + "");
+
+        for (int i = 0; i < 10000000; i++) {
+            DateTimeUtils.parse(datetimeStr1);
+        }
+        long t6 = System.currentTimeMillis();
+        // If format is not specified, the system automatically obtains the format 'yyyy-MM-dd
+        // HH:mm:ss' for processing, use time: 4842ms
+
+        System.out.println((t6 - t5) + "");
+
+        long t7 = System.currentTimeMillis();
+        for (int i = 0; i < 10000000; i++) {
+            DateTimeUtils.parse(datetimeStr2, dateTimeFormatter2);
+        }
+        long t8 = System.currentTimeMillis();
+        // Use an explicit time format 'yyyy-MM-dd HH:mm:ss.SSS' for processing, use time: 8162ms
+        System.out.println((t8 - t7) + "");
+
+        for (int i = 0; i < 10000000; i++) {
+            DateTimeUtils.parse(datetimeStr2);
+        }
+        long t9 = System.currentTimeMillis();
+        // If format is not specified, the system automatically obtains the format 'yyyy-MM-dd
+        // HH:mm:ss.SSS' for processing, use time: 11366ms
+        System.out.println((t9 - t8) + "");
+
+        long t10 = System.currentTimeMillis();
+        for (int i = 0; i < 10000000; i++) {
+            DateTimeUtils.parse(datetimeStr3, dateTimeFormatter3);
+        }
+        long t11 = System.currentTimeMillis();
+        // Use an explicit time format 'yyyy-MM-dd HH:mm:ss' for processing, use time: 4405ms
+        System.out.println((t11 - t10) + "");
+
+        for (int i = 0; i < 10000000; i++) {
+            DateTimeUtils.parse(datetimeStr3);
+        }
+        long t12 = System.currentTimeMillis();
+        // If format is not specified, the system automatically obtains the format 'yyyy-MM-dd
+        // HH:mm:ss' for processing, use time: 7771ms
+        System.out.println((t12 - t11) + "");
+    }
 }