// <copyright file="Program.cs" company="OpenTelemetry Authors">
// Copyright The OpenTelemetry Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// </copyright>

using System.Diagnostics.Metrics;
using System.Runtime.CompilerServices;
using OpenTelemetry.Metrics;

namespace OpenTelemetry.Tests.Stress;

public partial class Program
{
    private const int ArraySize = 10;

    // Note: Uncomment the below line if you want to run Histogram stress test
    private const int MaxHistogramMeasurement = 1000;

    private static readonly Meter TestMeter = new(Utils.GetCurrentMethodName());
    private static readonly Counter<long> TestCounter = TestMeter.CreateCounter<long>("TestCounter");
    private static readonly string[] DimensionValues = new string[ArraySize];
    private static readonly ThreadLocal<Random> ThreadLocalRandom = new(() => new Random());

    // Note: Uncomment the below line if you want to run Histogram stress test
    private static readonly Histogram<long> TestHistogram = TestMeter.CreateHistogram<long>("TestHistogram");

    public static void Main()
    {
        for (int i = 0; i < ArraySize; i++)
        {
            DimensionValues[i] = $"DimValue{i}";
        }

        using var meterProvider = Sdk.CreateMeterProviderBuilder()
            .AddMeter(TestMeter.Name)

            // .SetExemplarFilter(new AlwaysOnExemplarFilter())
            .AddPrometheusHttpListener(
                options => options.UriPrefixes = new string[] { $"http://localhost:9185/" })
            .Build();

        Stress(prometheusPort: 9464);
    }

    // Note: Uncomment the below lines if you want to run Counter stress test
    // [MethodImpl(MethodImplOptions.AggressiveInlining)]
    // protected static void Run()
    // {
    //    var random = ThreadLocalRandom.Value;
    //    TestCounter.Add(
    //        100,
    //        new("DimName1", DimensionValues[random.Next(0, ArraySize)]),
    //        new("DimName2", DimensionValues[random.Next(0, ArraySize)]),
    //        new("DimName3", DimensionValues[random.Next(0, ArraySize)]));
    // }

    // Note: Uncomment the below lines if you want to run Histogram stress test
    [MethodImpl(MethodImplOptions.AggressiveInlining)]
    protected static void Run()
    {
        var random = ThreadLocalRandom.Value;
        TestHistogram.Record(
            random.Next(MaxHistogramMeasurement),
            new("DimName1", DimensionValues[random.Next(0, ArraySize)]),
            new("DimName2", DimensionValues[random.Next(0, ArraySize)]),
            new("DimName3", DimensionValues[random.Next(0, ArraySize)]));
    }
}