public void Send(byte[] data) { if (!IsPtP || RemoteAddress == null) { return; } if (RemoteAddress.Protol == 0) { TcpClientSocket tcpClient = new TcpClientSocket() { RemoteHost = RemoteAddress.Address, RemotePort = RemoteAddress.Port }; if (tcpClient.Connect()) { tcpClient.Send(data); } tcpClient.Close(); } else { UDPSocketPack uDP = new UDPSocketPack(); uDP.Send(RemoteAddress.Address, RemoteAddress.Port, data); uDP.Stop(); } }
/// <summary> /// /// </summary> /// <param name="topic"></param> /// <param name="nodeAddr"></param> /// <returns></returns> public bool CopyAddress(string topic, AddressInfo nodeAddr) { TcpClientSocket tcpClient = new TcpClientSocket(); tcpClient.RemoteHost = nodeAddr.Address; tcpClient.RemotePort = nodeAddr.Port; if (tcpClient.Connect()) { tcpClient.Send(Encoding.Default.GetBytes(topic)); } return(true); }
public override bool Equals(object obj) { TcpClientSocket socket = obj as TcpClientSocket; if (socket == null) { return(false); } else if (this.RemoteHost == socket.RemoteHost && socket.RemotePort == RemotePort) { return(true); } return(false); }
/// <summary> /// 添加 /// </summary> /// <param name="socket"></param> /// <returns></returns> internal bool Add(TcpClientSocket socket) { lock (ClientSocket) { if (ClientSocket.Contains(socket)) { socket.Close(); return(false); } else { ClientSocket.Add(socket); return(true); } } }
/// <summary> /// 直接发送订阅信息 /// </summary> /// <param name="topic"></param> /// <param name="addr"></param> /// <returns></returns> public bool SendSubscribeTopic(string topic, AddressInfo addr) { bool isSucess = false; var local = new AddressInfo(); local.Reset(LocalNode.TopicAddress); byte[] tmp = DataPack.PackSubscribeMsg(topic, new AddressInfo[] { local }); if (addr.Protol == 0) { TcpClientSocket tcp = new TcpClientSocket(); tcp.RemoteHost = addr.Address; tcp.RemotePort = addr.Port; if (tcp.Connect()) { tcp.Send(tmp); isSucess = true; } tcp.Close(); } else { UDPSocketPack uDP = new UDPSocketPack(); int num = udpTimes; while (true) { uDP.Send(addr.Address, addr.Port, tmp); byte[] buf = new byte[1024]; Debug.WriteLine("发送订阅信息:" + addr.ToString()); var tsk = Task.Factory.StartNew(() => { return(uDP.Recvice(buf));//只要有接收就确定收到 }); if (tsk.Wait(udpWait)) { isSucess = true; break; } num--; if (num < 0) { break; } } } return(isSucess); }
/// <summary> /// 更新socket /// </summary> /// <param name="topic"></param> /// <param name="socket"></param> public void Update(string topic, TcpClientSocket socket) { TopicTcp tcp = new TopicTcp(); tcp = dictionary.GetOrAdd(topic, tcp); tcp.Refresh(); if (tcp.Add(socket)) { if (Interlocked.Increment(ref SocketNum) > MaxNum * 2) { Task.Factory.StartNew(() => { if (DateTime.Now.Second - lastTime > 3) { Check(); } }); } } }
/// <summary> /// 发布数据 /// </summary> /// <param name="topic"></param> /// <param name="data"></param> public void Publish(string topic, byte[] data) { if (ObserverInit.isInit) { ObserverInit.Init(); } var array = SubscribeList.Subscribe.GetAddresses(topic); if (array == null || array.Length == 0) { //没有订阅看发布 array = PublishList.Publish.GetAddresses(topic); if (array == null || (array.Length == 1 && array[0].ToString() == LocalNode.TopicAddress)) { //没有地址,或者只是本节点发布了数据,当做新增,组播扩展新增 NewTopicPub.Pub.SendNewTopic(topic); PublishList.Publish.AddLocal(topic); array = SubscribeList.Subscribe.GetAddresses(topic); if (array != null) { //再次订阅的地址 PubData(array, topic, data); } else { //丢弃 } } else { //复制订阅地址 List <AddressInfo> lst = null;//地址 foreach (var p in array) { var bytes = DataPack.PackCopyTopic(topic); byte[] buf = new byte[10240]; if (p.Protol == 0) { TcpClientSocket tcpClient = new TcpClientSocket(); tcpClient.RemoteHost = p.Address; tcpClient.RemotePort = p.Port; if (tcpClient.Connect()) { tcpClient.Send(bytes); int r = tcpClient.Recvice(buf); lst = DataPack.UnPackCopyRspTopic(buf); SubscribeList.Subscribe.AddAddress(topic, lst.ToArray()); tcpClient.Close(); break;//成功一个就退出 } tcpClient.Close(); } else { RequestCopy(topic, p, bytes); } } if (lst != null) { PubData(lst.ToArray(), topic, data); } } } else { PubData(array, topic, data); } }
/// <summary> /// 发送数据 /// </summary> /// <param name="lst"></param> /// <param name="dat"></param> private void PubData(AddressInfo[] lst, string topic, byte[] data) { byte[] bytes = DataPack.PackTopicData(topic, data); bool findTcp = false; List <AddressInfo> lstTcp = new List <AddressInfo>(); foreach (var p in lst) { if (p.Protol == 0) { if (findTcp) { if (lstTcp.Contains(p)) { continue;//已经处理 } else { TcpClientSocket tcp = new TcpClientSocket { RemoteHost = p.Address, RemotePort = p.Port }; if (tcp.Connect()) { tcp.Send(BitConverter.GetBytes(bytes.Length)); tcp.Send(bytes); ClientProcess.Instance.Update(topic, tcp); //tcp.Close(); } } continue; } var lstSocket = ClientProcess.Instance.GetTcpClients(topic); if (lstSocket != null) { lock (lstSocket) { //必须锁定防止处理线程正在关闭 foreach (var tcpClient in lstSocket) { try { tcpClient.Send(bytes); lstTcp.Add(new AddressInfo() { Address = tcpClient.RemoteHost, Port = tcpClient.RemotePort, Protol = 0 }); } catch { //有可能正在关闭; } } } } //当前第一个地址 if (!lstTcp.Contains(p)) { TcpClientSocket tcp = new TcpClientSocket { RemoteHost = p.Address, RemotePort = p.Port }; if (tcp.Connect()) { tcp.Send(BitConverter.GetBytes(bytes.Length)); tcp.Send(bytes); ClientProcess.Instance.Update(topic, tcp); //tcp.Close(); } } findTcp = true;//已经完全处理过; } else { // UDPSocket uDP = new UDPSocket(); UDPSocketPack uDP = null; if (udpSocket.TryTake(out uDP)) { uDP.Send(p.Address, p.Port, bytes); } else { if (Interlocked.Decrement(ref UdpSocketNum) > 0) { uDP = new UDPSocketPack(); uDP.Send(p.Address, p.Port, bytes); } else { uDP = udpSocket.Take(); uDP.Send(p.Address, p.Port, bytes); } } if (!udpSocket.TryAdd(uDP)) { uDP.Stop(); } } } }
/// <summary> /// 心跳包 /// </summary> private void Heart() { Task.Factory.StartNew(() => { var msg = DataPack.PackNodeState(LocalNode.NodeId, LocalNode.TopicAddress); Multicast.SendTo(msg); // if (NodeList.LstNodeInfo.Count == 0) { int num = NodeList.dicRefresh.Count; string[] keys = new string[num]; NodeList.dicRefresh.Keys.CopyTo(keys, 0); NodeList.LstNodeInfo.AddRange(keys); //取出ID foreach (var ky in keys) { int index = ky.IndexOf(","); string id = ky.Substring(0, index); NodeList.UpdateListId.Add(long.Parse(id)); } } else { var array = NodeList.LstNodeInfo.ToArray(); foreach (var node in array) { int Sp = 0; NodeList.dicRefresh.TryGetValue(node, out Sp); if (Math.Abs(DateTime.Now.Second - Sp) / WaitTime > 2) { //启动侦测 int index = node.IndexOf(","); string addr = node.Substring(index); string[] address = addr.Split('_'); if (address[0] == "0") { TcpClientSocket tcpClient = new TcpClientSocket(); tcpClient.RemoteHost = address[0]; tcpClient.RemotePort = int.Parse(address[1]); if (tcpClient.Connect()) { //刷新 NodeList.dicRefresh[node] = DateTime.Now.Second; } else { //移除节点信息 string id = node.Substring(0, index); NodeList.LstNodeInfo.Remove(node); NodeList.UpdateListId.Remove(long.Parse(id)); AddressInfo info = new AddressInfo(); info.Reset(addr); SubscribeList.Subscribe.Remove(info); } } else { //UDPSocket uDP = new UDPSocket(); UDPSocketPack uDP = new UDPSocketPack(); byte[] buf = new byte[1024]; int num = WaitTectNum; while (num > 0) { uDP.Send(address[0], int.Parse(address[1]), tectBytes); var tsk = Task.Factory.StartNew(() => { return(uDP.Recvice(buf));//有数据返回就说明收到了; }); if (tsk.Wait(100)) { NodeList.dicRefresh[node] = DateTime.Now.Second; break; } num--; } if (num <= 0) { //移除节点信息 NodeList.LstNodeInfo.Remove(node); AddressInfo info = new AddressInfo(); info.Reset(addr); SubscribeList.Subscribe.Remove(info); string id = node.Substring(0, index); NodeList.UpdateListId.Remove(long.Parse(id)); } uDP.Close(); } } } } }); }