mirror of
https://github.com/dotnet/MQTTnet.git
synced 2026-10-20 19:58:57 +08:00
Refactor code
This commit is contained in:
@@ -2,21 +2,23 @@
|
||||
// The .NET Foundation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using MQTTnet.Exceptions;
|
||||
using MQTTnet.Internal;
|
||||
using System;
|
||||
using System.Runtime.CompilerServices;
|
||||
using System.Runtime.InteropServices;
|
||||
using System.Text;
|
||||
using MQTTnet.Exceptions;
|
||||
using MQTTnet.Internal;
|
||||
#if NETCOREAPP3_0_OR_GREATER
|
||||
using System.Buffers.Binary;
|
||||
#endif
|
||||
|
||||
namespace MQTTnet.Formatter
|
||||
{
|
||||
public sealed class MqttBufferReader
|
||||
{
|
||||
byte[] _buffer = EmptyBuffer.Array;
|
||||
int _maxPosition;
|
||||
int _offset;
|
||||
int _position;
|
||||
int _maxPosition;
|
||||
|
||||
public int BytesLeft => _maxPosition - _position;
|
||||
|
||||
@@ -36,7 +38,7 @@ namespace MQTTnet.Formatter
|
||||
ValidateReceiveBuffer(length);
|
||||
|
||||
var result = new byte[length];
|
||||
Memory.Copy(_buffer, _position, result, 0, length);
|
||||
MqttMemoryHelper.Copy(_buffer, _position, result, 0, length);
|
||||
_position += length;
|
||||
|
||||
return result;
|
||||
@@ -53,11 +55,7 @@ namespace MQTTnet.Formatter
|
||||
ValidateReceiveBuffer(4);
|
||||
|
||||
#if NETCOREAPP3_0_OR_GREATER
|
||||
var value = MemoryMarshal.Read<uint>(_buffer.AsSpan(_position));
|
||||
if (BitConverter.IsLittleEndian)
|
||||
{
|
||||
value = System.Buffers.Binary.BinaryPrimitives.ReverseEndianness(value);
|
||||
}
|
||||
var value = BinaryPrimitives.ReadUInt32BigEndian(_buffer.AsSpan(_position));
|
||||
#else
|
||||
var byte0 = _buffer[_position];
|
||||
var byte1 = _buffer[_position + 1];
|
||||
@@ -80,7 +78,7 @@ namespace MQTTnet.Formatter
|
||||
}
|
||||
|
||||
var buffer = new byte[bufferLength];
|
||||
Memory.Copy(_buffer, _position, buffer, 0, bufferLength);
|
||||
MqttMemoryHelper.Copy(_buffer, _position, buffer, 0, bufferLength);
|
||||
_position += bufferLength;
|
||||
|
||||
return buffer;
|
||||
@@ -102,7 +100,7 @@ namespace MQTTnet.Formatter
|
||||
var result = Encoding.UTF8.GetString(_buffer.AsSpan(_position, length));
|
||||
#else
|
||||
var result = Encoding.UTF8.GetString(_buffer, _position, length);
|
||||
#endif
|
||||
#endif
|
||||
|
||||
_position += length;
|
||||
return result;
|
||||
@@ -113,11 +111,7 @@ namespace MQTTnet.Formatter
|
||||
ValidateReceiveBuffer(2);
|
||||
|
||||
#if NETCOREAPP3_0_OR_GREATER
|
||||
var value = MemoryMarshal.Read<ushort>(_buffer.AsSpan(_position));
|
||||
if (BitConverter.IsLittleEndian)
|
||||
{
|
||||
value = System.Buffers.Binary.BinaryPrimitives.ReverseEndianness(value);
|
||||
}
|
||||
var value = BinaryPrimitives.ReadUInt16BigEndian(_buffer.AsSpan(_position));
|
||||
#else
|
||||
var msb = _buffer[_position];
|
||||
var lsb = _buffer[_position + 1];
|
||||
|
||||
@@ -126,7 +126,7 @@ namespace MQTTnet.Formatter
|
||||
|
||||
EnsureAdditionalCapacity(count);
|
||||
|
||||
Memory.Copy(buffer, offset, _buffer, _position, count);
|
||||
MqttMemoryHelper.Copy(buffer, offset, _buffer, _position, count);
|
||||
IncreasePosition(count);
|
||||
}
|
||||
|
||||
@@ -150,7 +150,7 @@ namespace MQTTnet.Formatter
|
||||
_buffer[_position] = (byte)(valueLength >> 8);
|
||||
_buffer[_position + 1] = (byte)valueLength;
|
||||
|
||||
Memory.Copy(value, 0, _buffer, _position + 2, valueLength);
|
||||
MqttMemoryHelper.Copy(value, 0, _buffer, _position + 2, valueLength);
|
||||
IncreasePosition(valueLength + 2);
|
||||
}
|
||||
}
|
||||
|
||||
@@ -43,8 +43,8 @@ namespace MQTTnet.Formatter
|
||||
}
|
||||
|
||||
var buffer = new byte[Length];
|
||||
Memory.Copy(Packet.Array, Packet.Offset, buffer, 0, Packet.Count);
|
||||
Memory.Copy(Payload.Array, Payload.Offset, buffer, Packet.Count, Payload.Count);
|
||||
MqttMemoryHelper.Copy(Packet.Array, Packet.Offset, buffer, 0, Packet.Count);
|
||||
MqttMemoryHelper.Copy(Payload.Array, Payload.Offset, buffer, Packet.Count, Payload.Count);
|
||||
|
||||
return buffer;
|
||||
}
|
||||
@@ -57,8 +57,8 @@ namespace MQTTnet.Formatter
|
||||
}
|
||||
|
||||
var buffer = new byte[Length];
|
||||
Memory.Copy(Packet.Array, Packet.Offset, buffer, 0, Packet.Count);
|
||||
Memory.Copy(Payload.Array, Payload.Offset, buffer, Packet.Count, Payload.Count);
|
||||
MqttMemoryHelper.Copy(Packet.Array, Packet.Offset, buffer, 0, Packet.Count);
|
||||
MqttMemoryHelper.Copy(Payload.Array, Payload.Offset, buffer, Packet.Count, Payload.Count);
|
||||
|
||||
return new ArraySegment<byte>(buffer);
|
||||
}
|
||||
|
||||
@@ -3,7 +3,7 @@ using System.Runtime.CompilerServices;
|
||||
|
||||
namespace MQTTnet.Internal
|
||||
{
|
||||
public static class Memory
|
||||
public static class MqttMemoryHelper
|
||||
{
|
||||
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
||||
public static void Copy(byte[] source, int sourceIndex, byte[] destination, int destinationIndex, int length)
|
||||
@@ -13,11 +13,11 @@ namespace MQTTnet.Internal
|
||||
#elif NET461_OR_GREATER || NETSTANDARD1_3_OR_GREATER
|
||||
unsafe
|
||||
{
|
||||
fixed (byte* pSoure = &source[sourceIndex])
|
||||
fixed (byte* sourceHandle = &source[sourceIndex])
|
||||
{
|
||||
fixed (byte* pDestination = &destination[destinationIndex])
|
||||
fixed (byte* destinationHandle = &destination[destinationIndex])
|
||||
{
|
||||
System.Buffer.MemoryCopy(pSoure, pDestination, length, length);
|
||||
System.Buffer.MemoryCopy(sourceHandle, destinationHandle, length, length);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user