初始化

UnityGame/Assets/Scripts/ThirdParty/ConsolePro/Remote/LiteNetLib/NetPacketPool.cs

@@ -0,0 +1,76 @@
+using System;
+using System.Threading;
+
+namespace FlyingWormConsole3.LiteNetLib
+{
+    internal sealed class NetPacketPool
+    {
+        private NetPacket _head;
+        private int _count;
+        private readonly object _lock = new object();
+
+        public NetPacket GetWithData(PacketProperty property, byte[] data, int start, int length)
+        {
+            int headerSize = NetPacket.GetHeaderSize(property);
+            NetPacket packet = GetPacket(length + headerSize);
+            packet.Property = property;
+            Buffer.BlockCopy(data, start, packet.RawData, headerSize, length);
+            return packet;
+        }
+
+        //Get packet with size
+        public NetPacket GetWithProperty(PacketProperty property, int size)
+        {
+            NetPacket packet = GetPacket(size + NetPacket.GetHeaderSize(property));
+            packet.Property = property;
+            return packet;
+        }
+
+        public NetPacket GetWithProperty(PacketProperty property)
+        {
+            NetPacket packet = GetPacket(NetPacket.GetHeaderSize(property));
+            packet.Property = property;
+            return packet;
+        }
+
+        public NetPacket GetPacket(int size)
+        {
+            if (size > NetConstants.MaxPacketSize)
+                return new NetPacket(size);
+
+            NetPacket packet;
+            lock (_lock)
+            {
+                packet = _head;
+                if (packet == null)
+                    return new NetPacket(size);
+                _head = _head.Next;
+            }
+
+            Interlocked.Decrement(ref _count);
+            packet.Size = size;
+            if (packet.RawData.Length < size)
+                packet.RawData = new byte[size];
+            return packet;
+        }
+
+        public void Recycle(NetPacket packet)
+        {
+            if (packet.RawData.Length > NetConstants.MaxPacketSize || _count >= NetConstants.PacketPoolSize)
+            {
+                //Don't pool big packets. Save memory
+                return;
+            }
+
+            Interlocked.Increment(ref _count);
+
+            //Clean fragmented flag
+            packet.RawData[0] = 0;
+            lock (_lock)
+            {
+                packet.Next = _head;
+                _head = packet;
+            }
+        }
+    }
+}