csharp/beetlex-io/HttpClients/src/Request.cs

Request.cs
using BeetleX.Buffers;
using BeetleX.Clients;
using BeetleX.Tasks;
using System;
using System.Collections.Generic;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using System.Diagnostics;
#if NETCOREAPP2_1
using BeetleX.Tracks;
#endif
namespace BeetleX.Http.Clients
{
    public enum LoadedState : int
    {
        None = 1,
        Method = 2,
        Header = 4,
        Completed = 8
    }

#if NETCOREAPP2_1
    public clast Request : IApiObject
#else
    public clast Request
#endif
    {
#if NETCOREAPP2_1

        private CodeTrack mRequestTrack;

        public string Name => this.Url;

        public string[] Group
        {
            get
            {
                return new[] { "HTTPClient", "Request", $"{HttpHost.Host}" };
            }
        }
#endif

        public const string CODE_TREAK_PARENTID = "parent-id";

        public const string POST = "POST";

        public const string GET = "GET";

        public const string DELETE = "DELETE";

        public const string PUT = "PUT";

        public Request()
        {
            Method = GET;
            this.HttpProtocol = "HTTP/1.1";
        }

        public string Boundary { get; set; }

        public IBodyFormater Formater { get; set; } = new JsonFormater();

        public Dictionary QuestryString { get; set; }

        public Header Header { get; set; }

        public string Url { get; set; }

        public string Method { get; set; }

        public string HttpProtocol { get; set; }

        public Response Response { get; set; }

        public Object Body { get; set; }

        public Type BodyType { get; set; }

        public int? TimeOut { get; set; }

        public RequestStatus Status { get; set; }

        public IClient Client { get; set; }

        internal void Execute(PipeStream stream)
        {
            try
            {
                var buffer = HttpParse.GetByteBuffer();
                int offset = 0;
                offset += Encoding.ASCII.GetBytes(Method, 0, Method.Length, buffer, offset);
                buffer[offset] = HeaderTypeFactory._SPACE_BYTE;
                offset++;
                offset += Encoding.ASCII.GetBytes(Url, 0, Url.Length, buffer, offset);
                if (QuestryString != null && QuestryString.Count > 0)
                {
                    int i = 0;
                    foreach (var item in this.QuestryString)
                    {
                        string key = item.Key;
                        string value = item.Value;
                        if (string.IsNullOrEmpty(value))
                            continue;
                        value = System.Net.WebUtility.UrlEncode(value);
                        if (i == 0)
                        {
                            buffer[offset] = HeaderTypeFactory._QMARK;
                            offset++;
                        }
                        else
                        {
                            buffer[offset] = HeaderTypeFactory._AND;
                            offset++;
                        }
                        offset += Encoding.ASCII.GetBytes(key, 0, key.Length, buffer, offset);
                        buffer[offset] = HeaderTypeFactory._EQ;
                        offset++;
                        offset += Encoding.ASCII.GetBytes(value, 0, value.Length, buffer, offset);
                        i++;
                    }
                }
                buffer[offset] = HeaderTypeFactory._SPACE_BYTE;
                offset++;
                offset += Encoding.ASCII.GetBytes(HttpProtocol, 0, HttpProtocol.Length, buffer, offset);
                buffer[offset] = HeaderTypeFactory._LINE_R;
                offset++;
                buffer[offset] = HeaderTypeFactory._LINE_N;
                offset++;
                stream.Write(buffer, 0, offset);

                Formater?.Setting(this);
                if (Header != null)
                    Header.Write(stream);

                if (Method == POST || Method == PUT)
                {
                    if (Body != null)
                    {
                        stream.Write(HeaderTypeFactory.CONTENT_LENGTH_BYTES, 0, 16);
                        MemoryBlockCollection contentLength = stream.Allocate(10);
                        stream.Write(HeaderTypeFactory.TOW_LINE_BYTES, 0, 4);
                        int len = stream.CacheLength;
                        Formater.Serialization(this, Body, stream);
                        int count = stream.CacheLength - len;
                        contentLength.Full(count.ToString().PadRight(10), stream.Encoding);
                    }
                    else
                    {
                        stream.Write(HeaderTypeFactory.NULL_CONTENT_LENGTH_BYTES, 0, HeaderTypeFactory.NULL_CONTENT_LENGTH_BYTES.Length);
                        stream.Write(HeaderTypeFactory.LINE_BYTES, 0, 2);
                    }
                }
                else
                {
                    stream.Write(HeaderTypeFactory.LINE_BYTES, 0, 2);
                }
            }
            catch (Exception e)
            {
                Client?.DisConnect();
                throw new HttpClientException($"http client write data error {e.Message}", e);
            }
        }

        public HttpHost HttpHost { get; set; }

        public Action GetConnection { get; set; }

        public static event Action Executing;

        public async Task Execute()
        {
#if NETCOREAPP2_1
            using (mRequestTrack = CodeTrackFactory.TrackReport(this, CodeTrackLevel.Module, null))
            {
                if (Activity.Current != null)
                    Header[CODE_TREAK_PARENTID] = Activity.Current.Id;
                if (mRequestTrack.Enabled)
                    mRequestTrack.Activity?.AddTag("tag", "BeetleX HttpClient");
#endif
            Executing?.Invoke(this);
            mTaskCompletionSource = CompletionSourceFactory.Create(TimeOut != null ? TimeOut.Value : HttpHost.Pool.TimeOut); // new AnyCompletionSource();
            mTaskCompletionSource.TimeOut = OnRequestTimeout;
            OnExecute();
            return await (Task)mTaskCompletionSource.GetTask();
#if NETCOREAPP2_1
            }
#endif
        }

        private IAnyCompletionSource mTaskCompletionSource;

        private void OnRequestTimeout(IAnyCompletionSource source)
        {
            if (requestResult != null)
            {
                requestResult.TrySetException(new TimeoutException($"request imeout!"));
            }
            else
            {
                Response response = new Response();
                response.Code = "408";
                response.CodeMsg = "Request timeout";
                response.Exception = new HttpClientException(this, HttpHost.Uri, "Request timeout");
                source?.Success(response);
            }
        }

        private void onEventClientError(IClient c, ClientErrorArgs e)
        {
            requestResult.TrySetException(e.Error);
        }

        private void OnEventClientPacketCompleted(IClient client, object message)
        {
            requestResult.TrySetResult(message);
        }

        private TaskCompletionSource requestResult;

        private async Task DisConnect(IClient client)
        {
            client.DisConnect();
            await Task.Delay(1000);
        }

        private async void OnExecute()
        {
            HttpClientHandler client = null;
            Response response;
            bool closeClient = false;
            try
            {
                object result = null;
                requestResult = new TaskCompletionSource();
                client = await HttpHost.Pool.Pop();
                Client = client.Client;
                AsyncTcpClient asyncClient = (AsyncTcpClient)client.Client;
                asyncClient.ClientError = onEventClientError;
                asyncClient.PacketReceive = OnEventClientPacketCompleted;
                GetConnection?.Invoke(asyncClient);
#if NETCOREAPP2_1
                using (CodeTrackFactory.Track(Url, CodeTrackLevel.Function, mRequestTrack?.Activity?.Id, "HTTPClient", "Protocol", "Write"))
                {
                    asyncClient.Send(this);
                    Status = RequestStatus.SendCompleted;
                }
#else
                asyncClient.Send(this);
                Status = RequestStatus.SendCompleted;
#endif

#if NETCOREAPP2_1
                using (CodeTrackFactory.Track(Url, CodeTrackLevel.Function, mRequestTrack?.Activity?.Id, "HTTPClient", "Protocol", "Read"))
                {
                    var a = requestResult.Task;
                    result = await a;
                }
#else
                var a = requestResult.Task;
                result = await a;
#endif
                if (result is Exception error)
                {
                    response = new Response();
                    response.Exception = new HttpClientException(this, HttpHost.Uri, error.Message, error);
                    Status = RequestStatus.Error;
                    closeClient = true;
                }
                else
                {
                    response = (Response)result;
                    Status = RequestStatus.Received;
                }

                if (response.Exception == null)
                {
                    int code = int.Parse(response.Code);
                    if (response.Length > 0)
                    {
                        try
                        {
                            if (code >= 200 && code < 300)
                                response.Body = this.Formater.Deserialization(response, response.Stream, this.BodyType, response.Length);
                            else
                                response.Body = response.Stream.ReadString(response.Length);
                        }
                        finally
                        {
                            response.Stream.ReadFree(response.Length);
                            if (response.Chunked)
                                response.Stream.Dispose();
                            response.Stream = null;
                        }
                    }
                    if (!response.KeepAlive)
                        client.Client.DisConnect();
                    if (code >= 400)
                    {
                        response.Exception = new HttpClientException(this, HttpHost.Uri, $"{Url}({response.Code}) [{response.Body}]");
                        response.Exception.Code = code;
                    }
                    Status = RequestStatus.Completed;
                }
            }
            catch (Exception e_)
            {
                HttpClientException clientException = new HttpClientException(this, HttpHost.Uri, e_.Message, e_);
                response = new Response { Exception = clientException };
                Status = RequestStatus.Error;
                closeClient = true;
            }
            if (response.Exception != null)
                HttpHost.AddError(response.Exception.SocketError);
            else
                HttpHost.AddSuccess();
            Response.Current = response;
            this.Response = response;
            if (client != null)
            {
                if (client.Client is AsyncTcpClient asclient)
                {
                    asclient.ClientError = null;
                    asclient.PacketReceive = null;
                }
                if (closeClient)
                    await DisConnect(client.Client);
                HttpHost.Pool.Push(client);
                client = null;
            }
            await Task.Run(() => mTaskCompletionSource.Success(response));
        }
    }

    public enum RequestStatus
    {
        None,
        SendCompleted,
        Received,
        Completed,
        Error
    }
}