Framework_YooAsset_HybridCLR/Assets/Plugins/TouchSocket/TouchSocket.Http/DataAdapter/HttpClientDataHandlingAdapter.cs

//------------------------------------------------------------------------------
//  此代码版权（除特别声明或在XREF结尾的命名空间的代码）归作者本人若汝棋茗所有
//  源代码使用协议遵循本仓库的开源协议及附加协议，若本仓库没有设置，则按MIT开源协议授权
//  CSDN博客：https://blog.csdn.net/qq_40374647
//  哔哩哔哩视频：https://space.bilibili.com/94253567
//  Gitee源代码仓库：https://gitee.com/RRQM_Home
//  Github源代码仓库：https://github.com/RRQM
//  API首页：https://touchsocket.net/
//  交流QQ群：234762506
//  感谢您的下载和使用
//------------------------------------------------------------------------------

using System;
using System.Text;
using System.Threading.Tasks;
using TouchSocket.Core;
using TouchSocket.Sockets;

namespace TouchSocket.Http
{
    /// <summary>
    /// Http客户端数据处理适配器
    /// </summary>
    public class HttpClientDataHandlingAdapter : NormalDataHandlingAdapter
    {
        /// <summary>
        /// 缓存数据，如果需要手动释放，请先判断，然后到调用<see cref="IDisposable.Dispose"/>后，再置空；
        /// </summary>
        protected ByteBlock tempByteBlock;

        private static readonly byte[] m_rnCode = Encoding.UTF8.GetBytes("\r\n");
        private ITcpClientBase m_client;
        private HttpResponse m_httpResponse;

        private long m_surLen;

        private Task m_task;

        /// <summary>
        /// <inheritdoc/>
        /// </summary>
        public override bool CanSplicingSend => false;

        /// <inheritdoc/>
        public override void OnLoaded(object owner)
        {
            if (!(owner is ITcpClientBase clientBase))
            {
                throw new Exception($"此适配器必须适用于{nameof(ITcpClientBase)}");
            }
            this.m_client = clientBase;
            base.OnLoaded(owner);
        }

        /// <summary>
        /// <inheritdoc/>
        /// </summary>
        /// <param name="byteBlock"></param>
        protected override void PreviewReceived(ByteBlock byteBlock)
        {
            if (this.tempByteBlock == null)
            {
                byteBlock.Pos = 0;
                this.Single(byteBlock, false);
            }
            else
            {
                this.tempByteBlock.Write(byteBlock.Buffer, 0, byteBlock.Len);
                var block = this.tempByteBlock;
                this.tempByteBlock = null;
                block.Pos = 0;
                this.Single(block, true);
            }
        }

        private void Cache(ByteBlock byteBlock)
        {
            if (byteBlock.CanReadLen > 0)
            {
                this.tempByteBlock = new ByteBlock();
                this.tempByteBlock.Write(byteBlock.Buffer, byteBlock.Pos, byteBlock.CanReadLen);
                if (this.tempByteBlock.Len > this.MaxPackageSize)
                {
                    this.OnError(default, "缓存的数据长度大于设定值的情况下未收到解析信号", true, true);
                }
            }
        }

        private FilterResult ReadChunk(ByteBlock byteBlock)
        {
            var position = byteBlock.Pos;
            var index = byteBlock.Buffer.IndexOfFirst(byteBlock.Pos, byteBlock.CanReadLen, m_rnCode);
            if (index > 0)
            {
                var headerLength = index - byteBlock.Pos;
                var hex = Encoding.ASCII.GetString(byteBlock.Buffer, byteBlock.Pos, headerLength - 1);
                var count = hex.ByHexStringToInt32();
                byteBlock.Pos += headerLength + 1;

                if (count >= 0)
                {
                    if (count > byteBlock.CanReadLen)
                    {
                        byteBlock.Pos = position;
                        return FilterResult.Cache;
                    }

                    this.m_httpResponse.InternalInput(byteBlock.Buffer, byteBlock.Pos, count);
                    byteBlock.Pos += count;
                    byteBlock.Pos += 2;
                    return FilterResult.GoOn;
                }
                else
                {
                    byteBlock.Pos += 2;
                    return FilterResult.Success;
                }
            }
            else
            {
                return FilterResult.Cache;
            }
        }

        private Task RunGoReceived(HttpResponse response)
        {
            return Task.Run(() =>
             {
                 this.GoReceived(null, response);
             });
        }

        private void Single(ByteBlock byteBlock, bool dis)
        {
            try
            {
                while (byteBlock.CanReadLen > 0)
                {
                    if (this.m_httpResponse == null)
                    {
                        this.m_httpResponse = new HttpResponse(this.m_client);
                        if (this.m_httpResponse.ParsingHeader(byteBlock, byteBlock.CanReadLen))
                        {
                            byteBlock.Pos++;
                            if (this.m_httpResponse.IsChunk || this.m_httpResponse.ContentLength > byteBlock.CanReadLen)
                            {
                                this.m_surLen = this.m_httpResponse.ContentLength;
                                this.m_task = this.RunGoReceived(this.m_httpResponse);
                            }
                            else
                            {
                                byteBlock.Read(out var buffer, (int)this.m_httpResponse.ContentLength);
                                this.m_httpResponse.SetContent(buffer);
                                this.GoReceived(null, this.m_httpResponse);
                                this.m_httpResponse = null;
                            }
                        }
                        else
                        {
                            this.Cache(byteBlock);
                            this.m_httpResponse = null;
                            this.m_task?.Wait();
                            this.m_task = null;
                            return;
                        }
                    }
                    if (this.m_httpResponse != null)
                    {
                        if (this.m_httpResponse.IsChunk)
                        {
                            switch (this.ReadChunk(byteBlock))
                            {
                                case FilterResult.Cache:
                                    this.Cache(byteBlock);
                                    return;

                                case FilterResult.Success:
                                    this.m_httpResponse = null;
                                    this.m_task?.Wait();
                                    this.m_task = null;
                                    break;

                                case FilterResult.GoOn:
                                default:
                                    break;
                            }
                        }
                        else if (this.m_surLen > 0)
                        {
                            if (byteBlock.CanRead)
                            {
                                var len = (int)Math.Min(this.m_surLen, byteBlock.CanReadLen);
                                this.m_httpResponse.InternalInput(byteBlock.Buffer, byteBlock.Pos, len);
                                this.m_surLen -= len;
                                byteBlock.Pos += len;
                                if (this.m_surLen == 0)
                                {
                                    this.m_httpResponse.InternalInput(new byte[0], 0, 0);
                                    this.m_httpResponse = null;
                                    this.m_task?.Wait();
                                    this.m_task = null;
                                }
                            }
                        }
                        else
                        {
                            this.m_httpResponse = null;
                            this.m_task?.Wait();
                            this.m_task = null;
                        }
                    }
                }
            }
            finally
            {
                if (dis)
                {
                    byteBlock.Dispose();
                }
            }
        }
    }
}