Move Slice to use MemorySegment

airlift · Jul 1, 2024 · 3855919 · 3855919
1 parent 86fe760
commit 3855919
Show file tree

Hide file tree

Showing 6 changed files with 197 additions and 321 deletions.
diff --git a/src/main/java/io/airlift/slice/JvmUtils.java b/src/main/java/io/airlift/slice/JvmUtils.java
diff --git a/src/main/java/io/airlift/slice/SizeOf.java b/src/main/java/io/airlift/slice/SizeOf.java
@@ -20,6 +20,7 @@
 import org.openjdk.jol.vm.VM;
 import org.openjdk.jol.vm.VirtualMachine;
 
+import java.lang.foreign.MemorySegment;
 import java.util.AbstractMap;
 import java.util.List;
 import java.util.Map;
@@ -79,6 +80,24 @@ public final class SizeOf
 
     private static final int SIMPLE_ENTRY_INSTANCE_SIZE = instanceSize(AbstractMap.SimpleEntry.class);
 
+    public static long sizeOf(MemorySegment segment)
+    {
+        // final long length;
+        // final boolean readOnly;
+        // final MemorySessionImpl scope (int state);
+        long size = LONG_INSTANCE_SIZE + BOOLEAN_INSTANCE_SIZE + INTEGER_INSTANCE_SIZE;
+        if (segment.isNative()) {
+            return size
+                    + LONG_INSTANCE_SIZE; // addr;
+        }
+
+        return size
+                + LONG_INSTANCE_SIZE // offset
+                + segment.heapBase() // base
+                    .map(value -> sizeOfByteArray(((byte[]) value).length))
+                    .orElse(0L);
+    }
+
     public static long sizeOf(boolean[] array)
     {
         return (array == null) ? 0 : sizeOfBooleanArray(array.length);