FunGame-Core/Api/Transmittal/DataRequest.cs

using System.Collections;
using Milimoe.FunGame.Core.Api.Utility;
using Milimoe.FunGame.Core.Library.Common.Architecture;
using Milimoe.FunGame.Core.Library.Common.Network;
using Milimoe.FunGame.Core.Library.Constant;

namespace Milimoe.FunGame.Core.Api.Transmittal
{
    public class DataRequest
    {
        private static readonly ConcurrentQueue<Request> Queue = new();
        private readonly Request Worker;

        public DataRequest(Socket Socket, DataRequestType RequestType)
        {
            Worker = new(Socket, RequestType);
        }

        public void AddRequestData(string key, object value)
        {
            Worker.RequestData.Add(key, value);
        }

        public async Task SendRequest()
        {
            Queue.AddAsync(Worker);
            if (await Worker.SendRequest() == RequestResult.Success)
            {
                Queue.Delete();
            }
        }

        public object? this[string key] => Worker.ResultData[key];

        public T? GetResult<T>(string key)
        {
            object? obj = this[key];
            if (obj != null)
            {
                return (T)obj;
            }
            return default;
        }

        private class Request : BaseModel
        {
            public Hashtable RequestData { get; } = new();
            public Hashtable ResultData { get; } = new();

            private bool JobFinish = false;
            private readonly Socket? Socket;
            private readonly DataRequestType RequestType;

            public async Task<RequestResult> SendRequest()
            {
                try
                {
                    if (Socket?.Send(SocketMessageType.DataRequest, RequestData) == SocketResult.Success)
                    {
                        await Task.Run(() =>
                        {
                            while (true)
                            {
                                if (JobFinish) break;
                                Thread.Sleep(100);
                            }
                        });
                    }
                }
                catch
                {
                    return RequestResult.Fail;
                }
                return RequestResult.Success;
            }

            public Request(Socket? socket, DataRequestType requestType) : base(socket)
            {
                Socket = socket;
                RequestType = requestType;
            }

            public override void SocketHandler(SocketObject SocketObject)
            {
                if (SocketObject.SocketType == SocketMessageType.DataRequest)
                {
                    DataRequestType type = SocketObject.GetParam<DataRequestType>(0);
                    if (type == RequestType)
                    {
                        Dispose();
                        switch (RequestType)
                        {
                            default:
                                break;
                        }
                        JobFinish = true;
                    }
                }
            }
        }
    }
}