Created
February 27, 2018 08:51
-
-
Save benaadams/d16c873b5927d6f8f7232b2806646dcc to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| public class OwnedMemorySegment<T> : ReadOnlySequenceSegment<T> | |
| { | |
| private OwnedMemory<T> _ownedMemory; | |
| public OwnedMemorySegment(OwnedMemory<T> memory) | |
| { | |
| _ownedMemory = memory; | |
| Memory = memory.Memory; | |
| } | |
| public OwnedMemorySegment<T> Append(OwnedMemory<T> memory) | |
| { | |
| var segment = new OwnedMemorySegment<T>(memory) | |
| { | |
| RunningIndex = RunningIndex + Memory.Length | |
| }; | |
| Next = segment; | |
| return segment; | |
| } | |
| } | |
| public sealed class SafeBufferOwnedMemory<T> : OwnedMemory<T> | |
| { | |
| public static ReadOnlySequence<T> CreateReadOnlyBuffer(SafeBuffer buffer, long offset, long length) | |
| { | |
| int count = (int)Math.Min(length, int.MaxValue); | |
| OwnedMemorySegment<T> first = new OwnedMemorySegment<T>(new SafeBufferOwnedMemory<T>(buffer, offset, count)); | |
| OwnedMemorySegment<T> last = first; | |
| length -= count; | |
| offset += count; | |
| while (length > 0) | |
| { | |
| count = (int)Math.Min(length, int.MaxValue); | |
| last = last.Append(new SafeBufferOwnedMemory<T>(buffer, offset, count)); | |
| length -= count; | |
| offset += count; | |
| } | |
| return new ReadOnlySequence<T>(first, 0, last, last.Memory.Length); | |
| } | |
| private readonly SafeBuffer _buffer; | |
| private readonly long _offset; | |
| private readonly int _length; | |
| private SafeBufferOwnedMemory(SafeBuffer buffer, long offset, int length) | |
| { | |
| _buffer = buffer; | |
| _offset = offset; | |
| _length = length; | |
| } | |
| public override int Length => _length; | |
| protected override void Dispose(bool disposing) { } | |
| public override bool IsDisposed => _buffer.IsClosed; | |
| protected override bool IsRetained => !_buffer.IsClosed; | |
| public override unsafe Span<T> Span => new Span<T>((byte*)_buffer.DangerousGetHandle() + _offset, _length); | |
| public override void Retain() | |
| { | |
| bool success = false; | |
| _buffer.DangerousAddRef(ref success); | |
| } | |
| public override bool Release() | |
| { | |
| _buffer.DangerousRelease(); | |
| return _buffer.IsClosed; | |
| } | |
| public override unsafe MemoryHandle Pin(int byteOffset = 0) => new MemoryHandle(this, (byte*)_buffer.DangerousGetHandle() + _offset); | |
| protected override bool TryGetArray(out ArraySegment<T> arraySegment) | |
| { | |
| arraySegment = null; | |
| return false; | |
| } | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment