DebugConnectionProvider.cs

//------------------------------------------------------------------------------
// <auto-generated>
//     This code was generated by AsyncGenerator.
//
//     Changes to this file may cause incorrect behavior and will be lost if
//     the code is regenerated.
// </auto-generated>
//------------------------------------------------------------------------------


using System;
using System.Collections.Concurrent;
using System.Data;
using System.Data.Common;
using System.Linq;
using NHibernate.Connection;

namespace NHibernate.Test
{
	using System.Threading.Tasks;
	using System.Threading;
	public partial class DebugConnectionProvider : DriverConnectionProvider
	{

		public override async Task<DbConnection> GetConnectionAsync(CancellationToken cancellationToken)
		{
			try
			{
				var connection = await (base.GetConnectionAsync(cancellationToken));
				connections.TryAdd(connection, 0);
				return connection;
			}
			catch (OperationCanceledException) { throw; }
			catch (Exception e)
			{
				throw new HibernateException("Could not open connection to: " + ConnectionString, e);
			}
		}
	}
}