RealSys.MotionPlat/Rs.Framework/TcpClientHelper.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net.Sockets;
using System.Runtime.CompilerServices;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

namespace Rs.Framework
{
    public class TcpClientHelper
    {
        private bool m_IsConnected = false;
        private Socket m_Socket;
        private string m_HostName;
        private int m_Port;
        public event Action<Socket> OnConnected;
        public event Action<Socket> OnDisconnected;
        public event Action<Socket, byte[]> DataRecived;


        public bool IsConnected
        {
            get { return m_IsConnected; }
            set { m_IsConnected = value; }
        }

        public TcpClientHelper()
        {

        }

        public TcpClientHelper(string ip,int port)
        {
            m_HostName = ip;
            m_Port = port;
        }

        public void Connect()
        {
            Connect(m_HostName, m_Port);
        }


        /// <summary>
        /// 链接服务的方法
        /// </summary>
        /// <param name="host"></param>
        /// <param name="port"></param>
        public void Connect(string host, int port)
        {
            m_HostName=host;
            m_Port = port;
            m_Socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
            Task.Run(() =>
            {
                while (!m_IsConnected)
                {
                    try
                    {
                        m_Socket.Connect(host, port);
                        m_IsConnected = true;
                        LogHelper.Debug($"ip:{m_HostName},port:{m_Port} connected");
                        OnConnected?.Invoke(m_Socket);
                    }
                    catch (Exception)
                    {
                        Thread.Sleep(3000);
                        LogHelper.Debug($"ip:{host},port:{port} connect fail,will reconnect");
                        m_Socket.Close();
                        m_Socket.Dispose();
                        m_Socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
                    }
                }
                //开启接收数据线程
                Thread tRecive = new Thread(new ParameterizedThreadStart(ReciveData));
                tRecive.IsBackground = true;
                tRecive.Start(m_Socket);
            });
        }

        public void ReciveData(object socket)
        {
            var serverSocket = socket as Socket;
            byte[] data = new byte[1024 * 1024];
            while (m_IsConnected)
            {
                int len = 0;
                try
                {
                    len = serverSocket.Receive(data, 0, data.Length, SocketFlags.None);
                    if(len>0)
                    {
                        DataRecived?.Invoke(serverSocket, data.Take(len).ToArray());
                    }
                    else if(len <= 0)
                    {
                        m_IsConnected = false;
                        LogHelper.Debug($"ip:{m_HostName},port:{m_Port} disconnected");
                        serverSocket.Shutdown(SocketShutdown.Both);
                        serverSocket.Close();
                        serverSocket.Dispose();
                        OnDisconnected?.Invoke(m_Socket);
                        Connect(m_HostName, m_Port);//重新尝试去连接
                        return;
                    }
                }
                catch (Exception ex)
                {
                    m_IsConnected = false;
                    LogHelper.Debug(ex.Message);
                    //异常退出
                    if(serverSocket.Connected)
                    {
                        serverSocket.Shutdown(SocketShutdown.Both);
                    }
                    serverSocket.Close();
                    serverSocket.Dispose();
                    OnDisconnected?.Invoke(m_Socket);
                    Connect(m_HostName,m_Port);//重新尝试去连接
                    return;
                }
            }
        }

        public int Send(string content)
        {
            byte[] sendData = Encoding.ASCII.GetBytes(content);
            if(m_IsConnected)
            {
                return Send(sendData);
            }
            return -1;
        }

        public int Send(string content,Encoding encode)
        {
            byte[] sendData = encode.GetBytes(content);
            if (m_IsConnected)
            {
                return Send(sendData);
            }
            return -1;
        }

        public int Send(byte[] sendData)
        {
            if (m_IsConnected)
            {
                try
                {
                    int len = m_Socket.Send(sendData);
                    //if(len <= 0)
                    //{
                    //    m_IsConnected = false;
                    //    m_Socket.Close();
                    //    m_Socket.Dispose();
                    //    Connect(m_HostName, m_Port);
                    //    OnDisconnected?.Invoke(m_Socket);
                    //}
                    return len;
                }
                catch (Exception)
                {
                    m_IsConnected = false;
                    m_Socket.Close();
                    m_Socket.Dispose();
                    Connect(m_HostName, m_Port);
                    OnDisconnected?.Invoke(m_Socket);
                }
            }
            return -1;
        }

        public void DisConnect()
        {
            m_Socket.Close();
            m_Socket.Dispose();
            m_IsConnected = false;
        }
    }
}