Merge branch 'dev_' into pr/5083

liunaijie · Oct 13, 2023 · cadf6a4 · cadf6a4
2 parents 1445f96 + b1d66c5
commit cadf6a4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/seatunnel-api/src/main/java/org/apache/seatunnel/api/table/type/SeaTunnelRow.java b/seatunnel-api/src/main/java/org/apache/seatunnel/api/table/type/SeaTunnelRow.java
@@ -255,6 +255,7 @@ private int getBytesForValue(Object v) {
             case "Double[]":
                 return ((Double[]) v).length * 8;
             case "HashMap":
+            case "LinkedHashMap":
                 int size = 0;
                 for (Map.Entry<?, ?> entry : ((Map<?, ?>) v).entrySet()) {
                     size += getBytesForValue(entry.getKey()) + getBytesForValue(entry.getValue());

diff --git a/seatunnel-api/src/test/java/org/apache/seatunnel/api/table/type/SeaTunnelRowTest.java b/seatunnel-api/src/test/java/org/apache/seatunnel/api/table/type/SeaTunnelRowTest.java
@@ -22,6 +22,7 @@
 
 import java.math.BigDecimal;
 import java.util.HashMap;
+import java.util.LinkedHashMap;
 import java.util.Map;
 
 public class SeaTunnelRowTest {
@@ -87,4 +88,12 @@ void testForRowSize() {
                         });
         Assertions.assertEquals(181, row2.getBytesSize());
     }
+
+    @Test
+    void testWithLinkHashMap() {
+        Map<String, String> map = new LinkedHashMap<>();
+        map.put("key", "value");
+        SeaTunnelRow row = new SeaTunnelRow(new Object[] {map});
+        Assertions.assertEquals(8, row.getBytesSize());
+    }
 }