/// <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); }