[Unity网络]异步Socket

1.客户端

using System.Net.Sockets;
using System.Text;
using UnityEngine;
using UnityEngine.UI;
using System;

public class TestSocket2 : MonoBehaviour
{
    private Socket socket;
    public InputField inputField;
    public Text text;

    private byte[] readBytes = new byte[1024];
    private string readStr = "";

    void Start()
    {
        socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
    }

    void Update()
    {
        text.text = readStr;
    }

    public void TestConnect()
    {
        //Connect
        socket.BeginConnect("127.0.0.1", 8888, ConnectCallBack, socket);
    }

    public void TestSend()
    {
        //Send
        byte[] writeBytes = Encoding.Default.GetBytes(inputField.text);
        socket.BeginSend(writeBytes, 0, writeBytes.Length, 0, SendCallBack, socket);
    }

    private void ConnectCallBack(IAsyncResult asyncResult)
    {
        try
        {
            Socket socket = (Socket)asyncResult.AsyncState;
            socket.EndConnect(asyncResult);

            //Receive
            socket.BeginReceive(readBytes, 0, 1024, 0, ReceiveCallBack, socket);
        }
        catch (Exception exception)
        {
            Debug.LogError(exception.ToString());
        }
    }

    private void ReceiveCallBack(IAsyncResult asyncResult)
    {
        try
        {
            Socket socket = (Socket)asyncResult.AsyncState;
            int count = socket.EndReceive(asyncResult);

            //在unity中，只有主线程才可以操作UI组件。因此不能直接对text.text进行赋值
            readStr = Encoding.Default.GetString(readBytes, 0, count);
            //text.text = readStr;

            //Receive
            socket.BeginReceive(readBytes, 0, 1024, 0, ReceiveCallBack, socket);
        }
        catch (Exception exception)
        {
            Debug.LogError(exception.ToString());
        }
    }

    private void SendCallBack(IAsyncResult asyncResult)
    {
        try
        {
            Socket socket = (Socket)asyncResult.AsyncState;
            int count = socket.EndSend(asyncResult);
        }
        catch (Exception exception)
        {
            Debug.LogError(exception.ToString());
        }
    }
}

 

2.服务端

using System;
using System.Collections.Generic;
using System.Net;
using System.Net.Sockets;
using System.Text;

namespace TestSocket2
{
    class ClientState
    {
        public Socket socket;
        public byte[] receiveBytes = new byte[1024];
    }

    class Program
    {
        static Dictionary<Socket, ClientState> clientDic = new Dictionary<Socket, ClientState>();

        static void Main(string[] args)
        {
            Socket socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);

            //Bind
            IPAddress iPAddress = IPAddress.Parse("127.0.0.1");
            IPEndPoint iPEndPoint = new IPEndPoint(iPAddress, 8888);
            socket.Bind(iPEndPoint);

            //Listen
            socket.Listen(0);
            Console.WriteLine("Listen");

            //Accept
            socket.BeginAccept(AcceptCallBack, socket);

            Console.ReadLine();
        }

        private static void AcceptCallBack(IAsyncResult asyncResult)
        {
            try
            {
                Console.WriteLine("Accept");
                Socket serverSocket = (Socket)asyncResult.AsyncState;
                Socket clientSocket = serverSocket.EndAccept(asyncResult);
                ClientState clientState = new ClientState();
                clientState.socket = clientSocket;
                clientDic.Add(clientSocket, clientState);

                //Receive
                clientSocket.BeginReceive(clientState.receiveBytes, 0, 1024, 0, ReceiveCallBack, clientState);

                //Accept
                serverSocket.BeginAccept(AcceptCallBack, serverSocket);
            }
            catch (Exception exception)
            {
                Console.WriteLine(exception.ToString());
            }
        }

        private static void ReceiveCallBack(IAsyncResult asyncResult)
        {
            try
            {
                ClientState clientState = (ClientState)asyncResult.AsyncState;
                Socket clientSocket = clientState.socket;
                int count = clientSocket.EndReceive(asyncResult);

                //客户端关闭
                if (count == 0)
                {
                    clientSocket.Close();
                    clientDic.Remove(clientSocket);
                    return;
                }

                string str = Encoding.Default.GetString(clientState.receiveBytes, 0, count);
                clientSocket.Send(Encoding.Default.GetBytes(str));
                clientSocket.BeginReceive(clientState.receiveBytes, 0, 1024, 0, ReceiveCallBack, clientState);
                Console.WriteLine("Receive:" + str);
            }
            catch (Exception exception)
            {
                Console.WriteLine(exception.ToString());
            }
        }
    }
}