54 lines
1.4 KiB
C#
54 lines
1.4 KiB
C#
using System;
|
||
using System.Buffers;
|
||
using System.Collections.Generic;
|
||
using System.Text;
|
||
|
||
namespace System.Net.Sockets.Kcp
|
||
{
|
||
public abstract class KcpOutputWriter : IKcpOutputWriter
|
||
{
|
||
public int UnflushedBytes { get; set; }
|
||
public IMemoryOwner<byte> MemoryOwner { get; set; }
|
||
public void Flush()
|
||
{
|
||
Output(MemoryOwner, UnflushedBytes);
|
||
MemoryOwner = null;
|
||
UnflushedBytes = 0;
|
||
}
|
||
|
||
public void Advance(int count)
|
||
{
|
||
UnflushedBytes += count;
|
||
}
|
||
|
||
public Memory<byte> GetMemory(int sizeHint = 0)
|
||
{
|
||
if (MemoryOwner == null)
|
||
{
|
||
MemoryOwner = MemoryPool<byte>.Shared.Rent(2048);
|
||
}
|
||
return MemoryOwner.Memory.Slice(UnflushedBytes);
|
||
}
|
||
|
||
public Span<byte> GetSpan(int sizeHint = 0)
|
||
{
|
||
if (MemoryOwner == null)
|
||
{
|
||
MemoryOwner = MemoryPool<byte>.Shared.Rent(2048);
|
||
}
|
||
return MemoryOwner.Memory.Span.Slice(UnflushedBytes);
|
||
}
|
||
|
||
/// <summary>
|
||
/// Socket发送是要pin byte[],为了不阻塞KcpFlush，动态缓存是必须的。
|
||
/// </summary>
|
||
/// <param name="buffer"></param>
|
||
/// <param name="avalidLength"></param>
|
||
public abstract void Output(IMemoryOwner<byte> buffer, int avalidLength);
|
||
}
|
||
}
|
||
|
||
|
||
|
||
|