// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Threading.Tasks;
using BenchmarkDotNet.Attributes;
using Microsoft.AspNetCore.Http;
 
namespace Microsoft.AspNetCore.Server.Kestrel.Microbenchmarks;
 
public class Http2ConnectionHeadersBenchmark : Http2ConnectionBenchmarkBase
{
    [Params(1, 4, 32)]
    public int HeadersCount { get; set; }
 
    [Params(true, false)]
    public bool HeadersChange { get; set; }
 
    private int _headerIndex;
    private string[] _headerNames;
 
    [GlobalSetup]
    public override void GlobalSetup()
    {
        base.GlobalSetup();
 
        _headerNames = new string[HeadersCount * (HeadersChange ? 1000 : 1)];
        for (var i = 0; i < _headerNames.Length; i++)
        {
            _headerNames[i] = "CustomHeader" + i;
        }
    }
 
    protected override Task ProcessRequest(HttpContext httpContext)
    {
        for (var i = 0; i < HeadersCount; i++)
        {
            var headerName = _headerNames[_headerIndex % HeadersCount];
            httpContext.Response.Headers[headerName] = "The quick brown fox jumps over the lazy dog.";
            if (HeadersChange)
            {
                _headerIndex++;
            }
        }
 
        return Task.CompletedTask;
    }
}