/// <summary>
/// 带数据的连接
/// </summary>
public void Connect(string ip, int port, IWriteStream writeStream)
{
if (writeStream == null)
{
throw new ArgumentNullException(nameof(writeStream));
}
var endPoint = new IPEndPoint(IPAddress.Parse(ip), port);
var socketService = new SocketService();
socketService.Connection.RemoteAddress = endPoint;
socketService.Connection.LocalAddress = endPoint;
if (Write(socketService, writeStream, SendOption.Connect))
{
if (!_autoResetEvent.WaitOne(_networkConfig.AckInterval))
{
HandleConnect?.Invoke(socketService, null);
}
}
else
{
HandleConnect?.Invoke(socketService, null);
}
}
private void AcknowledgeConnect(IPEndPoint endPoint, IReadStream readStream,
PacketHead packetHead)
{
if (_connections.ContainsKey(endPoint))
{
Logger.Warn($"{endPoint}已经连接，无需重复连接！");
return;
}
var service = new SocketService();
service.Connection.RemoteAddress = endPoint;
service.Connection.LocalAddress = _writer.Client.LocalEndPoint as IPEndPoint;
if (_connections.TryAdd(endPoint, service))
{
_autoResetEvent.Set();
UdpKeepAlive.AddConnection(service,
() => HandleReconnect?.Invoke(service, true),
() => {
HandleReconnect?.Invoke(service, false);
OnDisconnect(service);
});
HandleConnect?.Invoke(service, readStream);
Logger.Info($"{endPoint}连接成功！");
}
else
{
Logger.Error($"添加连接{endPoint}失败!");
}
}
public void Disconnect(SocketService socketService)
{
if (socketService == null)
{
throw new ArgumentNullException($"{nameof(socketService)} 为 null ！");
}
Write(socketService, socketService.SendCounter, SendOption.Disconnect);
}
private void OnDisconnect(SocketService socketService)
{
if (_connections.TryRemove(socketService.Connection.RemoteAddress,
out socketService))
{
UdpKeepAlive.Remove(socketService);
foreach (var id in socketService.PacketIds)
{
_fragmentsTimer.Remove(id);
}
socketService.OnDisconnect();
Logger.Info($"断开{socketService.Connection.RemoteAddress}连接！");
}
else
{
Logger.Warn("是否已经断开?");
}
}
/// <summary>
/// 发送数据
/// </summary>
private bool Write(SocketService socketService, ulong messageId, SendOption sendOption)
{
using (var writeStream = PoolAllocator <IWriteStream> .GetObject()) {
var packetHead = new PacketHead {
Length = (ushort)PacketHead.GetSize(),
PacketId = messageId,
SendOption = sendOption,
TotalBytes = 0,
FragmentId = 0,
TotalFragments = 1
};
Packet.ToBytes(packetHead, writeStream);
var byteFragment = writeStream.ToByteFragment();
var result = _writer.Client.SendTo(byteFragment.Buffer, byteFragment.Offset, byteFragment.Count, SocketFlags.None,
socketService.Connection.RemoteAddress);
socketService.OnWrite(result == byteFragment.Count);
return(result == byteFragment.Count);
}
}
private bool Write(SocketService socketService, IWriteStream writeStream, SendOption sendOption)
{
var fragment = writeStream?.ToByteFragment();
var packetHead = new PacketHead {
Length = fragment.HasValue?(ushort)(PacketHead.GetSize() + fragment?.Count):(ushort)PacketHead.GetSize(),
PacketId = socketService.SendCounter,
SendOption = sendOption,
TotalBytes = fragment.HasValue?(ushort)fragment?.Count:(ushort)0
};
var ws = PoolAllocator <IWriteStream> .GetObject();
if (fragment.HasValue)
{
ws.ShiftRight((ByteFragment)fragment);
}
Packet.ToBytes(packetHead, ws);
var byteFragment = ws.ToByteFragment();
var result = _writer.Client.SendTo(byteFragment.Buffer, byteFragment.Offset, byteFragment.Count, SocketFlags.None,
socketService.Connection.RemoteAddress);
socketService.OnWrite(result == byteFragment.Count);
return(result == byteFragment.Count);
}
private bool Acknowledge(
SocketService socketService,
SendOption sendOption,
PacketHead packetHead,
IWriteStream writeStream = null)
{
var fragment = writeStream?.ToByteFragment();
packetHead.Length = fragment.HasValue ? (ushort)(PacketHead.GetSize() + fragment?.Count) : (ushort)PacketHead.GetSize();
packetHead.SendOption = sendOption;
using (var ws = PoolAllocator <IWriteStream> .GetObject()) {
if (writeStream != null)
{
ws.ShiftRight(writeStream.ToByteFragment());
}
Packet.ToBytes(packetHead, ws);
var bf = ws.ToByteFragment();
var result = _writer.Client.SendTo(bf.Buffer, bf.Offset, bf.Count,
SocketFlags.None, socketService.Connection.RemoteAddress);
socketService.OnWrite(result == bf.Count);
return(result == bf.Count);
}
}
/// <summary>
/// 发送数据
/// </summary>
public bool Write(SocketService socketService, IWriteStream writeStream)
{
if (socketService == null)
{
throw new ArgumentNullException(nameof(socketService));
}
if (writeStream == null)
{
throw new ArgumentNullException(nameof(writeStream));
}
var fragment = writeStream.ToByteFragment();
var dataLengnth = MTU - PacketHead.GetSize();
var messageId = socketService.SendCounter;
// 分包
var splits = Math.Ceiling(fragment.Count * 1.0d / dataLengnth);
for (var i = 0; i < splits; i++)
{
var packetHead = new PacketHead {
Length = (ushort)(Math.Min(fragment.Count - i * dataLengnth, dataLengnth) + PacketHead.GetSize()),
PacketId = messageId,
SendOption = SendOption.Fragment,
TotalBytes = (ushort)Math.Min(fragment.Count - i * dataLengnth, dataLengnth),
FragmentId = (ushort)i,
TotalFragments = (ushort)splits
};
Console.WriteLine(i);
var ws = PoolAllocator <IWriteStream> .GetObject();
ws.ShiftRight(fragment.Buffer, fragment.Offset + i * dataLengnth, packetHead.TotalBytes);
Packet.ToBytes(packetHead, ws);
var bf = ws.ToByteFragment();
_writer.Client.SendTo(bf.Buffer, bf.Offset, bf.Count, SocketFlags.None,
socketService.Connection.RemoteAddress);
void Action()
{
_writer.Client.SendTo(bf.Buffer, bf.Offset, bf.Count, SocketFlags.None,
socketService.Connection.RemoteAddress);
Logger.Info("重发数据包！");
}
void RecycleAction()
{
ws.Dispose();
}
var id = BuildID(packetHead);
if (!_fragmentsTimer.Add(id, Action, RecycleAction))
{
Logger.Error("添加重发消息失败！");
}
socketService.PacketIds.Add(id);
}
return(true);
}
