tests/MassTransit.Benchmark/ProgramOptionSet.cs - develop - vshmidt/masstransit

tests/MassTransit.Benchmark/ProgramOptionSet.cs
82 строки3 KB
Chris Patterson
Added SQL transport to latency benchmark
14 июн 2024, 15:13
14 июн 2024, 15:135a0cac2
namespace MassTransitBenchmark
{
    using System;
    using MassTransit.Metadata;
    using NDesk.Options;


    class ProgramOptionSet :
        OptionSet
    {
        [Flags]
        public enum BenchmarkOptions
        {
            Latency = 1,
            Rpc = 2,
            BusOutbox = 4,
        }


        public enum TransportOptions
        {
            RabbitMq,
            AzureServiceBus,
            Mediator,
            InMemory,
            AmazonSqs,
            ActiveMq,
            Kafka,
            Sql
        }


        public ProgramOptionSet()
        {
            Add<string>("v|verbose", "Verbose output", x => Verbose = x != null);
            Add<string>("?|help", "Display this help and exit", x => Help = x != null);
            Add<int>("threads:", "The minimum number of thread pool threads", value => Threads = value);
            Add<string>("traces", "Enable traces capturing to OTel exporter", x => EnableTraces = x != null);
            Add<string>("metrics", "Enable metrics capturing to OTel exporter", x => EnableMetrics = x != null);

            Add<TransportOptions>("t|transport:", "Transport (RabbitMQ, AzureServiceBus, Mediator, AmazonSqs, InMemory)",
                value => Transport = value);
            Add("rabbitmq", "Use RabbitMQ", x => Transport = TransportOptions.RabbitMq);
            Add("kafka", "Use Kafka", x => Transport = TransportOptions.Kafka);
            Add("mediator", "Use Mediator", x => Transport = TransportOptions.Mediator);
            Add("inmemory", "Use InMemory", x => Transport = TransportOptions.InMemory);
            Add("sqs", "Use Amazon SQS", x => Transport = TransportOptions.AmazonSqs);
            Add("servicebus", "Use Azure Service Bus", x => Transport = TransportOptions.AzureServiceBus);
            Add("activemq", "Use ActiveMQ", x => Transport = TransportOptions.ActiveMq);
            Add("sql", "Use SQL Transport", x => Transport = TransportOptions.Sql);

            Add<BenchmarkOptions>("run:", "Run benchmark (All, Latency, RPC)", value => Benchmark = value);
            Add("rpc", "Run the RPC benchmark", x => Benchmark = BenchmarkOptions.Rpc);
            Add("latency", "Run the Latency benchmark", x => Benchmark = BenchmarkOptions.Latency);
            Add("busoutbox", "Run the Bus Outbox benchmark", x => Benchmark = BenchmarkOptions.BusOutbox);

            Benchmark = BenchmarkOptions.Latency | BenchmarkOptions.Rpc;
        }

        public BenchmarkOptions Benchmark { get; private set; }

        public int? Threads { get; set; }
        public bool Verbose { get; set; }
        public bool Help { get; set; }
        public bool EnableTraces { get; set; }
        public bool EnableMetrics { get; set; }

        public TransportOptions Transport { get; private set; }

        public void ShowOptions()
        {
            Console.WriteLine("Transport: {0}", Transport);

            if (Threads.HasValue)
                Console.WriteLine("Threads: {0}", Threads.Value);

            Console.WriteLine("Operating System: {0}", Environment.OSVersion);
            Console.WriteLine("Processor Count: {0}", Environment.ProcessorCount);
            Console.WriteLine("MassTransit Version: {0}", HostMetadataCache.Host.MassTransitVersion);
        }
    }
}