You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Zero/ZeroLevel/Services/Serialization/MemoryStreamWriter.cs

490 lines
12 KiB

6 years ago
using System;
using System.Collections.Concurrent;
6 years ago
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net;
using System.Text;
using ZeroLevel.Services.Extensions;
namespace ZeroLevel.Services.Serialization
{
/// <summary>
6 years ago
/// Wrapper over memorystream for writing
6 years ago
/// </summary>
public sealed class MemoryStreamWriter :
IBinaryWriter
{
public Stream Stream
{
get
{
return _stream;
}
}
private readonly MemoryStream _stream;
public MemoryStreamWriter()
{
_stream = new MemoryStream();
}
6 years ago
/// <summary>
6 years ago
/// Record a boolean value (1 byte)
6 years ago
/// </summary>
public void WriteBoolean(bool val)
{
_stream.WriteByte(BitConverter.GetBytes(val)[0]);
}
6 years ago
/// <summary>
6 years ago
/// Write byte (1 byte)
6 years ago
/// </summary>
public void WriteByte(byte val)
{
_stream.WriteByte(val);
}
/// <summary>
/// Write char (2 bytes)
/// </summary>
public void WriteChar(char val)
{
var data = BitConverter.GetBytes(val);
_stream.Write(data, 0, 2);
}
6 years ago
/// <summary>
6 years ago
/// Write array bytes
6 years ago
/// </summary>
/// <param name="val"></param>
public void WriteBytes(byte[] val)
{
if (val == null)
{
WriteInt32(0);
}
else
{
WriteInt32(val.Length);
_stream.Write(val, 0, val.Length);
}
}
/// <summary>
/// Record a 32-bit integer (4 bytes)
/// </summary>
public void WriteShort(short number)
{
_stream.Write(BitConverter.GetBytes(number), 0, 2);
}
public void WriteUShort(ushort number)
{
_stream.Write(BitConverter.GetBytes(number), 0, 2);
}
6 years ago
/// <summary>
6 years ago
/// Record a 32-bit integer (4 bytes)
6 years ago
/// </summary>
public void WriteInt32(Int32 number)
{
_stream.Write(BitConverter.GetBytes(number), 0, 4);
}
public void WriteUInt32(UInt32 number)
{
_stream.Write(BitConverter.GetBytes(number), 0, 4);
}
6 years ago
/// <summary>
6 years ago
/// Record an integer 64-bit number (8 bytes)
6 years ago
/// </summary>
public void WriteLong(Int64 number)
{
_stream.Write(BitConverter.GetBytes(number), 0, 8);
}
public void WriteULong(UInt64 number)
{
_stream.Write(BitConverter.GetBytes(number), 0, 8);
}
6 years ago
public void WriteTimeSpan(TimeSpan period)
{
WriteLong(period.Ticks);
}
public void WriteDecimal(Decimal number)
{
_stream.Write(BitConverterExt.GetBytes(number), 0, 16);
}
public void WriteDouble(double val)
{
_stream.Write(BitConverter.GetBytes(val), 0, 8);
}
6 years ago
public void WriteFloat(float val)
{
_stream.Write(BitConverter.GetBytes(val), 0, 4);
}
6 years ago
/// <summary>
6 years ago
/// Write string (4 bytes long + Length bytes)
6 years ago
/// </summary>
public void WriteString(string line)
{
if (line == null)
{
WriteInt32(0);
}
else
{
var buffer = Encoding.UTF8.GetBytes(line);
WriteInt32(buffer.Length);
_stream.Write(buffer, 0, buffer.Length);
}
}
6 years ago
/// <summary>
6 years ago
/// GUID record (16 bytes)
6 years ago
/// </summary>
public void WriteGuid(Guid guid)
{
_stream.Write(guid.ToByteArray(), 0, 16);
}
6 years ago
/// <summary>
6 years ago
/// Record the datetime
6 years ago
/// </summary>
/// <param name="datetime"></param>
public void WriteDateTime(DateTime? datetime)
{
if (datetime == null)
{
WriteByte(0);
}
else
{
WriteByte(1);
long serialized = datetime.Value.ToBinary();
byte[] data = BitConverter.GetBytes(serialized);
_stream.Write(data, 0, 8);
}
}
public void WriteIP(IPAddress ip)
{
6 years ago
WriteBytes(ip.GetAddressBytes());
6 years ago
}
public void WriteIPEndpoint(IPEndPoint endpoint)
{
6 years ago
WriteIP(endpoint.Address);
6 years ago
WriteInt32(endpoint.Port);
}
public byte[] Complete()
{
return _stream.ToArray();
}
public void Dispose()
{
_stream.Dispose();
}
#region Extension
6 years ago
public void Write<T>(T item)
where T : IBinarySerializable
{
if (item != null)
{
WriteByte(1);
item.Serialize(this);
}
else
{
WriteByte(0);
}
}
6 years ago
public void WriteCollection<T>(IEnumerable<T> collection)
where T : IBinarySerializable
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
item.Serialize(this);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<string> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteString(item);
}
}
}
public void WriteCollection(IEnumerable<IPAddress> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteIP(item);
}
}
}
public void WriteCollection(IEnumerable<IPEndPoint> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteIPEndpoint(item);
}
}
}
public void WriteCollection(IEnumerable<Guid> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteGuid(item);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<DateTime> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteDateTime(item);
}
}
}
public void WriteCollection(IEnumerable<UInt64> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteULong(item);
}
}
}
public void WriteCollection(IEnumerable<UInt32> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteUInt32(item);
}
}
}
public void WriteCollection(IEnumerable<char> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteChar(item);
}
}
}
public void WriteCollection(IEnumerable<short> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteShort(item);
}
}
}
public void WriteCollection(IEnumerable<ushort> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteUShort(item);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<Int64> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteLong(item);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<Int32> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteInt32(item);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<float> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteFloat(item);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<Double> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteDouble(item);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<bool> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteBoolean(item);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<byte> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteByte(item);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<byte[]> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteBytes(item);
}
}
}
public void WriteCompatible<T>(T item)
{
var buffer = MessageSerializer.SerializeCompatible(item);
_stream.Write(buffer, 0, buffer.Length);
6 years ago
}
6 years ago
public void WriteCollection(IEnumerable<decimal> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteDecimal(item);
}
}
}
public void WriteDictionary<TKey, TValue>(Dictionary<TKey, TValue> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteCompatible(item.Key);
WriteCompatible(item.Value);
}
}
}
public void WriteDictionary<TKey, TValue>(ConcurrentDictionary<TKey, TValue> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteCompatible(item.Key);
WriteCompatible(item.Value);
}
}
}
6 years ago
public void WriteCollection(IEnumerable<TimeSpan> collection)
{
WriteInt32(collection?.Count() ?? 0);
if (collection != null)
{
foreach (var item in collection)
{
WriteTimeSpan(item);
}
}
}
#endregion Extension
6 years ago
}
}

Powered by TurnKey Linux.