当前位置 主页 > 网站技术 > 代码类 >

    ASP.NET Core 3.x 并发限制的实现代码

    栏目:代码类 时间:2019-11-13 12:09

    前言

    Microsoft.AspNetCore.ConcurrencyLimiter AspNetCore3.0后增加的,用于传入的请求进行排队处理,避免线程池的不足.
    我们日常开发中可能常做的给某web服务器配置连接数以及,请求队列大小,那么今天我们看看如何在通过中间件形式实现一个并发量以及队列长度限制.

    Queue策略

    添加Nuget

    Install-Package Microsoft.AspNetCore.ConcurrencyLimiter

        public void ConfigureServices(IServiceCollection services)
        {
          services.AddQueuePolicy(options =>
          {
            //最大并发请求数
            options.MaxConcurrentRequests = 2;
            //请求队列长度限制
            options.RequestQueueLimit = 1;
          });
          services.AddControllers();
        }
        public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
        {
          //添加并发限制中间件
          app.UseConcurrencyLimiter();
          app.Run(async context =>
          {
            Task.Delay(100).Wait(); // 100ms sync-over-async
    
            await context.Response.WriteAsync("Hello World!");
          });
          if (env.IsDevelopment())
          {
            app.UseDeveloperExceptionPage();
          }
    
          app.UseHttpsRedirection();
    
          app.UseRouting();
    
          app.UseAuthorization();
    
          app.UseEndpoints(endpoints =>
          {
            endpoints.MapControllers();
          });
        }   
    
    

    通过上面简单的配置,我们就可以将他引入到我们的代码中,从而做并发量限制,以及队列的长度;那么问题来了,他是怎么实现的呢?

     public static IServiceCollection AddQueuePolicy(this IServiceCollection services, Action<QueuePolicyOptions> configure)
    {
        services.Configure(configure);
        services.AddSingleton<IQueuePolicy, QueuePolicy>();
        return services;
    }
    

    QueuePolicy采用的是SemaphoreSlim信号量设计,SemaphoreSlim、Semaphore(信号量)支持并发多线程进入被保护代码,对象在初始化时会指定 最大任务数量,当线程请求访问资源,信号量递减,而当他们释放时,信号量计数又递增。

       /// <summary>
        ///   构造方法(初始化Queue策略)
        /// </summary>
        /// <param name="options"></param>
        public QueuePolicy(IOptions<QueuePolicyOptions> options)
        {
          _maxConcurrentRequests = options.Value.MaxConcurrentRequests;
          if (_maxConcurrentRequests <= 0)
          {
            throw new ArgumentException(nameof(_maxConcurrentRequests), "MaxConcurrentRequests must be a positive integer.");
          }
    
          _requestQueueLimit = options.Value.RequestQueueLimit;
          if (_requestQueueLimit < 0)
          {
            throw new ArgumentException(nameof(_requestQueueLimit), "The RequestQueueLimit cannot be a negative number.");
          }
          //使用SemaphoreSlim来限制任务最大个数
          _serverSemaphore = new SemaphoreSlim(_maxConcurrentRequests);
        }

    ConcurrencyLimiterMiddleware中间件

        /// <summary>
        /// Invokes the logic of the middleware.
        /// </summary>
        /// <param name="context">The <see cref="HttpContext"/>.</param>
        /// <returns>A <see cref="Task"/> that completes when the request leaves.</returns>
        public async Task Invoke(HttpContext context)
        {
          var waitInQueueTask = _queuePolicy.TryEnterAsync();
    
          // Make sure we only ever call GetResult once on the TryEnterAsync ValueTask b/c it resets.
          bool result;
    
          if (waitInQueueTask.IsCompleted)
          {
            ConcurrencyLimiterEventSource.Log.QueueSkipped();
            result = waitInQueueTask.Result;
          }
          else
          {
            using (ConcurrencyLimiterEventSource.Log.QueueTimer())
            {
              result = await waitInQueueTask;
            }
          }
    
          if (result)
          {
            try
            {
              await _next(context);
            }
            finally
            {
              _queuePolicy.OnExit();
            }
          }
          else
          {
            ConcurrencyLimiterEventSource.Log.RequestRejected();
            ConcurrencyLimiterLog.RequestRejectedQueueFull(_logger);
            context.Response.StatusCode = StatusCodes.Status503ServiceUnavailable;
            await _onRejected(context);
          }
        }