tests/MassTransit.Benchmark/BusOutbox/LocalDbConnectionStringProvider.cs - develop - vshmidt/masstransit

tests/MassTransit.Benchmark/BusOutbox/LocalDbConnectionStringProvider.cs
71 строка3 KB
Chris Patterson
Adding new benchmark for bus outbox
19 авг 2022, 16:24
19 авг 2022, 16:2401fa2b7
namespace MassTransitBenchmark.BusOutbox
{
    using System;
    using System.Collections.Generic;
    using Microsoft.Data.SqlClient;


    public static class LocalDbConnectionStringProvider
    {
        /// <summary>
        /// This is a list of the connection strings that we will attempt to find what LocalDb versions
        /// are on the local pc which we can run the unit tests against
        /// </summary>
        static readonly string[] _possibleLocalDbConnectionStrings =
        {
            @"Server=tcp:localhost;Persist Security Info=False;User ID=sa;Password=Password12!;Encrypt=False;TrustServerCertificate=True;", // the linux mssql 2017 installed on appveyor
            @"Server=tcp:mssql;Persist Security Info=False;User ID=sa;Password=Password12!;Encrypt=False;TrustServerCertificate=True;", // the linux mssql 2017 installed on gha
            @"Data Source=(LocalDb)\MSSQLLocalDB;Integrated Security=True;", // the localdb installed with VS 2015
            @"Data Source=(LocalDb)\ProjectsV12;Integrated Security=True;", // the localdb with VS 2013
            @"Data Source=(LocalDb)\v11.0;Integrated Security=True;" // the older version of localdb
        };

        static readonly object _lockConnectionString = new object();
        static string _connectionString;

        /// <summary>
        /// Loops through the array of potential localdb connection strings to find one that we can use for the unit tests
        /// </summary>
        public static string GetLocalDbConnectionString(string initialCatalog = "MassTransitBenchmark;")
        {
            if (!string.IsNullOrWhiteSpace(_connectionString))
                return _connectionString + "Initial Catalog=" + initialCatalog;

            var exceptions = new List<Exception>();
            lock (_lockConnectionString)
            {
                if (!string.IsNullOrWhiteSpace(_connectionString))
                    return _connectionString + "Initial Catalog=" + initialCatalog;

                // Lets find a localdb that we can use for our unit test
                foreach (var connectionString in _possibleLocalDbConnectionStrings)
                {
                    try
                    {
                        using var connection = new SqlConnection(connectionString);
                        connection.Open();

                        // It worked, we can save this as our connection string
                        _connectionString = connectionString;
                        break;
                    }
                    catch (Exception ex)
                    {
                        // Swallow
                        exceptions.Add(ex);
                    }
                }

                // If we looped through all possible localdb connection strings and didn't find one, we fail.
                if (string.IsNullOrWhiteSpace(_connectionString))
                {
                    exceptions.Insert(0, new InvalidOperationException(
                        "Couldn't connect to any of the LocalDB Databases. You might have a version installed that is not in the list. Please check the list and modify as necessary"));
                    throw new AggregateException(exceptions);
                }
            }

            return _connectionString + "Initial Catalog=" + initialCatalog;
        }
    }
}