ET9_WebGL_Dev/CSharp/App/BossClient/ENetChannel.cs

using System;
using System.Threading.Tasks;
using BossBase;
using ENet;
using Helper;
using Ionic.Zlib;

namespace BossClient
{
	class ENetChannel: IMessageChannel
	{
		private readonly Peer peer;

		public ENetChannel(Peer peer)
		{
			this.peer = peer;
		}

		public async void Dispose()
		{
			await this.peer.DisconnectLaterAsync();
			this.peer.Dispose();
		}

		public void SendMessage<T>(ushort opcode, T message, byte channelID = 0)
		{
			byte[] protoBytes = ProtobufHelper.ToBytes(message);
			var neworkBytes = new byte[sizeof(ushort) + protoBytes.Length];

			var opcodeBytes = BitConverter.GetBytes(opcode);
			opcodeBytes.CopyTo(neworkBytes, 0);
			protoBytes.CopyTo(neworkBytes, sizeof(ushort));
			this.peer.WriteAsync(channelID, neworkBytes);
		}

		public async Task<Tuple<ushort, byte[]>> RecvMessage()
		{
			using (Packet packet = await this.peer.ReadAsync())
			{
				byte[] bytes = packet.Bytes;
				const int opcodeSize = sizeof(ushort);
				ushort opcode = BitConverter.ToUInt16(bytes, 0);
				byte flag = bytes[2];
				if (flag == 0)
				{
					var messageBytes = new byte[packet.Length - opcodeSize - 1];
					Array.Copy(bytes, opcodeSize + 1, messageBytes, 0, messageBytes.Length);
					return Tuple.Create(opcode, messageBytes);
				}
				else
				{
					var messageBytes = new byte[packet.Length - opcodeSize - 5];
					Array.Copy(bytes, opcodeSize + 5, messageBytes, 0, messageBytes.Length);
					messageBytes = ZlibStream.UncompressBuffer(messageBytes);
					return Tuple.Create(opcode, messageBytes);
				}
			}
		}
	}
}