v7.3.2018.0614   重构高性能资源池,减少GC压力,增加线程池,让异步任务得到平等竞争CPU的机会
大石头 authored at 2018-06-14 17:56:44
2.83 KiB
X
using System;
using System.ComponentModel;
using System.Net;
using System.Threading.Tasks;
using NewLife.Data;
using NewLife.Messaging;

namespace NewLife.Net
{
    /// <summary>新生命标准网络封包协议</summary>
    [DisplayName("标准封包")]
    public class DefaultPacket : PacketProvider
    {
        /// <summary>实例化标准封包</summary>
        public DefaultPacket()
        {
            Offset = 2;
            Size = 2;
        }

        private Byte _seq;

        /// <summary>创建消息</summary>
        /// <param name="pk">数据包</param>
        /// <returns></returns>
        public override IMessage CreateMessage(Packet pk)
        {
            var msg = new DefaultMessage { Payload = pk };
            // 序列号避开0
            if (_seq == 0) _seq++;
            msg.Sequence = _seq++;

            return msg;
        }

        /// <summary>加载消息</summary>
        /// <param name="pk"></param>
        /// <returns></returns>
        public override IMessage LoadMessage(Packet pk)
        {
            if (pk == null || pk.Count == 0) return null;

            var msg = new DefaultMessage();
            msg.Read(pk);

            return msg;
        }

        /// <summary>加入请求队列</summary>
        /// <param name="request">请求的数据</param>
        /// <param name="remote">远程</param>
        /// <param name="msTimeout">超时取消时间</param>
        public override Task<Packet> Add(Packet request, IPEndPoint remote, Int32 msTimeout)
        {
            if (Queue == null) Queue = new MyQueue();

            return Queue.Add(this, request, remote, msTimeout);
        }
    }

    /// <summary>标准封包工厂</summary>
    [DisplayName("标准封包")]
    public class DefaultPacketFactory : IPacketFactory
    {
        /// <summary>服务端多会话共用</summary>
        private IPacketQueue _queue;

        /// <summary>创建粘包处理实例,内含缓冲区,不同会话不能共用</summary>
        /// <returns></returns>
        public virtual IPacket Create()
        {
            if (_queue == null) _queue = new MyQueue();

            return new DefaultPacket { Queue = _queue };
        }
    }

    class MyQueue : DefaultPacketQueue
    {
        /// <summary>请求和响应是否匹配</summary>
        /// <param name="owner">拥有者</param>
        /// <param name="remote">远程</param>
        /// <param name="request">请求的数据</param>
        /// <param name="response">响应的数据</param>
        /// <returns></returns>
        protected override Boolean IsMatch(Object owner, IPEndPoint remote, Packet request, Packet response)
        {
            if (request.Count < 4 || response.Count < 4) return false;

            // 序号相等
            if (request[1] != response[1]) return false;

            return true;
        }
    }
}