ET9_WebGL_Dev/CSharp/Platform/ENetCS/ClientHost.cs

using System;
using System.Threading.Tasks;
using Log;

namespace ENet
{
	public sealed class ClientHost : Host
	{
		public ClientHost(uint peerLimit = NativeMethods.ENET_PROTOCOL_MAXIMUM_PEER_ID,
				uint channelLimit = 0, uint incomingBandwidth = 0,
				uint outgoingBandwidth = 0, bool enableCrc = true)
		{
			if (peerLimit > NativeMethods.ENET_PROTOCOL_MAXIMUM_PEER_ID)
			{
				throw new ArgumentOutOfRangeException("peerLimit");
			}
			CheckChannelLimit(channelLimit);

			this.host = NativeMethods.enet_host_create(IntPtr.Zero, peerLimit, channelLimit, incomingBandwidth, outgoingBandwidth);

			if (this.host == IntPtr.Zero)
			{
				throw new ENetException(0, "Host creation call failed.");
			}

			if (enableCrc)
			{
				this.EnableCrc();
			}
		}

		public Task<Peer> ConnectAsync(
				Address address, uint channelLimit = NativeMethods.ENET_PROTOCOL_MAXIMUM_CHANNEL_COUNT, uint data = 0)
		{
			CheckChannelLimit(channelLimit);

			var tcs = new TaskCompletionSource<Peer>();
			ENetAddress nativeAddress = address.Struct;
			IntPtr peerPtr = NativeMethods.enet_host_connect(this.host, ref nativeAddress, channelLimit, data);
			if (peerPtr == IntPtr.Zero)
			{
				throw new ENetException(0, "Host connect call failed.");
			}
			var peer = new Peer(peerPtr);
			this.PeersManager.Add(peerPtr, peer);
			peer.PeerEvent.Connected += e =>
			{
				if (e.EventState == EventState.DISCONNECTED)
				{
					throw new ENetException(3, "Connect Disconnected!");
				}
				tcs.TrySetResult(peer);
			};
			return tcs.Task;
		}

		public void RunOnce(int timeout = 0)
		{
			this.OnExecuteEvents();

			if (this.Service(timeout) < 0)
			{
				return;
			}

			Event ev;
			while (this.CheckEvents(out ev) > 0)
			{
				switch (ev.Type)
				{
					case EventType.Connect:
					{
						Logger.Debug("client connect");
						var peer = this.PeersManager[ev.PeerPtr];
						peer.PeerEvent.OnConnected(ev);
						peer.PeerEvent.Connected = null;
						break;
					}
					case EventType.Receive:
					{
						Logger.Debug("client recv");
						var peer = this.PeersManager[ev.PeerPtr];
						peer.PeerEvent.OnReceived(ev);
						peer.PeerEvent.Received = null;
						break;
					}
					case EventType.Disconnect:
					{
						Logger.Debug("client disconnect");

						ev.EventState = EventState.DISCONNECTED;
						var peer = this.PeersManager[ev.PeerPtr];
						if (peer.PeerEvent.Connected != null)
						{
							peer.PeerEvent.OnConnected(ev);
						}
						else if (peer.PeerEvent.Received != null)
						{
							peer.PeerEvent.OnReceived(ev);
						}
						else
						{
							peer.PeerEvent.OnDisconnect(ev);
						}

						this.PeersManager.Remove(ev.PeerPtr);
						break;
					}
				}
			}
		}
		
		public void Run(int timeout = 0)
		{
			while (isRunning)
			{
				RunOnce(timeout);
			}
		}
	}
}