46 lines
1.6 KiB
C#
46 lines
1.6 KiB
C#
|
// API consistent with Microsoft's ObjectPool<T>.
|
||
|
|
using System;
|
||
|
|
using System.Runtime.CompilerServices;
|
||
|
|
|
||
|
|
namespace Mirror
|
||
|
|
{
|
||
|
|
/// <summary>Pool of NetworkReaders to avoid allocations.</summary>
|
||
|
|
public static class NetworkReaderPool
|
||
|
|
{
|
||
|
|
// reuse Pool<T>
|
||
|
|
// we still wrap it in NetworkReaderPool.Get/Recyle so we can reset the
|
||
|
|
// position and array before reusing.
|
||
|
|
static readonly Pool<NetworkReaderPooled> Pool = new Pool<NetworkReaderPooled>(
|
||
|
|
// byte[] will be assigned in GetReader
|
||
|
|
() => new NetworkReaderPooled(new byte[]{}),
|
||
|
|
// initial capacity to avoid allocations in the first few frames
|
||
|
|
1000
|
||
|
|
);
|
||
|
|
|
||
|
|
/// <summary>Get the next reader in the pool. If pool is empty, creates a new Reader</summary>
|
||
|
|
public static NetworkReaderPooled Get(byte[] bytes)
|
||
|
|
{
|
||
|
|
// grab from pool & set buffer
|
||
|
|
NetworkReaderPooled reader = Pool.Get();
|
||
|
|
reader.SetBuffer(bytes);
|
||
|
|
return reader;
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>Get the next reader in the pool. If pool is empty, creates a new Reader</summary>
|
||
|
|
public static NetworkReaderPooled Get(ArraySegment<byte> segment)
|
||
|
|
{
|
||
|
|
// grab from pool & set buffer
|
||
|
|
NetworkReaderPooled reader = Pool.Get();
|
||
|
|
reader.SetBuffer(segment);
|
||
|
|
return reader;
|
||
|
|
}
|
||
|
|
|
||
|
|
/// <summary>Returns a reader to the pool.</summary>
|
||
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||
|
|
public static void Return(NetworkReaderPooled reader)
|
||
|
|
{
|
||
|
|
Pool.Return(reader);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|