perf[protocol]: memory copy for int list

zfoo-project · Nov 17, 2023 · 4881edb · 4881edb
1 parent 72bc89b
commit 4881edb
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/protocol/src/main/java/com/zfoo/protocol/buffer/CustomByteBuf.java b/protocol/src/main/java/com/zfoo/protocol/buffer/CustomByteBuf.java
@@ -12,10 +12,13 @@
 
 package com.zfoo.protocol.buffer;
 
+import com.zfoo.protocol.collection.ArrayListInt;
+import com.zfoo.protocol.collection.ArrayUtils;
 import com.zfoo.protocol.collection.CollectionUtils;
 import io.netty.buffer.ByteBuf;
 
 import java.nio.ByteBuffer;
+import java.util.List;
 
 /**
  * 自定义协议格式，可以针对某些特定场景对做特定优化
@@ -55,7 +58,7 @@ public static int[] readIntArraySimple(ByteBuf byteBuf) {
     }
 
     // 针对于int数组提高性能的复杂方式
-    public static void writeIntArrayComplex(ByteBuf byteBuf, int[] array) {
+    public static void writeIntArrayMemoryCopy(ByteBuf byteBuf, int[] array) {
         if (array == null) {
             byteBuf.writeByte(0);
             return;
@@ -69,7 +72,7 @@ public static void writeIntArrayComplex(ByteBuf byteBuf, int[] array) {
         byteBuf.writeBytes(byteBuffer);
     }
 
-    public static int[] readIntArrayComplex(ByteBuf byteBuf) {
+    public static int[] readIntArrayMemoryCopy(ByteBuf byteBuf) {
         var length = ByteBufUtils.readInt(byteBuf);
         var byteBuffer = ByteBuffer.allocate(length * 4);
         byteBuf.readBytes(byteBuffer);
@@ -80,4 +83,16 @@ public static int[] readIntArrayComplex(ByteBuf byteBuf) {
         return ints;
     }
 
+    public static void writeIntListMemoryCopy(ByteBuf byteBuf, List<Integer> list) {
+        if (list == null) {
+            byteBuf.writeByte(0);
+            return;
+        }
+        var array = ArrayUtils.intToArray(list);
+        writeIntArrayMemoryCopy(byteBuf, array);
+    }
+
+    public static List<Integer> readIntListMemoryCopy(ByteBuf byteBuf) {
+        return new ArrayListInt(readIntArrayMemoryCopy(byteBuf));
+    }
 }