ET9_WebGL_Base/Server/Model/Component/ActorComponent.cs

using System;
using System.Collections.Generic;
using System.Threading.Tasks;

namespace Model
{
	public struct ActorMessageInfo
	{
		public Session Session;
		public IActorMessage Message;
	}

	[ObjectEvent]
	public class ActorComponentEvent : ObjectEvent<ActorComponent>, IAwake, IAwake<IEntityActorHandler>, IStart
	{
		public void Awake()
		{
			this.Get().Awake();
		}

		public void Awake(IEntityActorHandler iEntityActorHandler)
		{
			this.Get().Awake(iEntityActorHandler);
		}

		public void Start()
		{
			this.Get().Start();
		}
	}

	/// <summary>
	/// 挂上这个组件表示该Entity是一个Actor, 它会将Entity位置注册到Location Server, 接收的消息将会队列处理
	/// </summary>
	public class ActorComponent: Component
	{
		private IEntityActorHandler entityActorHandler;

		private long actorId;

		// 队列处理消息
		private readonly Queue<ActorMessageInfo> queue = new Queue<ActorMessageInfo>();

		private TaskCompletionSource<ActorMessageInfo> tcs;

		public void Awake()
		{
			this.entityActorHandler = new CommonEntityActorHandler();
		}

		public void Awake(IEntityActorHandler iEntityActorHandler)
		{
			this.entityActorHandler = iEntityActorHandler;
		}
		
		public async void Start()
		{
			this.actorId = this.Owner.Id;
			Game.Scene.GetComponent<ActorManagerComponent>().Add(this.Owner);
			await Game.Scene.GetComponent<LocationProxyComponent>().Add(this.actorId);
			this.HandleAsync();
		}

		public void Add(ActorMessageInfo info)
		{
			this.queue.Enqueue(info);

			if (this.tcs == null)
			{
				return;
			}
			
			this.tcs?.SetResult(this.queue.Dequeue());
			this.tcs = null;
		}

		private Task<ActorMessageInfo> GetAsync()
		{
			if (this.queue.Count > 0)
			{
				return Task.FromResult(this.queue.Dequeue());
			}

			this.tcs = new TaskCompletionSource<ActorMessageInfo>();
			return this.tcs.Task;
		}

		private async void HandleAsync()
		{
			while (true)
			{
				try
				{
					ActorMessageInfo info = await this.GetAsync();
					await this.entityActorHandler.Handle(info.Session, this.Owner, info.Message);
				}
				catch (Exception e)
				{
					Log.Error(e.ToString());
				}
			}
		}

		public override async void Dispose()
		{
			try
			{
				if (this.Id == 0)
				{
					return;
				}

				base.Dispose();

				Game.Scene.GetComponent<ActorManagerComponent>().Remove(actorId);

				await Game.Scene.GetComponent<LocationProxyComponent>().Remove(this.actorId);
			}
			catch (Exception)
			{
				Log.Error($"unregister actor fail: {this.actorId}");
			}
		}
	}
}