DriverExtensions.cs

using System.Data;
using System.Data.Common;
using NHibernate.AdoNet;
using NHibernate.SqlTypes;
using NHibernate.Util;

namespace NHibernate.Driver
{
	public static class DriverExtensions
	{
		internal static void AdjustParameterForValue(this IDriver driver, DbParameter parameter, SqlType sqlType, object value)
		{
			var adjustingDriver = driver as IParameterAdjuster;
			adjustingDriver?.AdjustParameterForValue(parameter, sqlType, value);
		}

		//6.0 TODO: inline
		internal static int GetCommandTimeout(this IDriver driver)
		{
			if (driver is DriverBase driverBase)
				return driverBase.CommandTimeout;

#pragma warning disable 618
			return PropertiesHelper.GetInt32(Cfg.Environment.CommandTimeout, Cfg.Environment.Properties, -1);
#pragma warning restore 618
		}

		// 6.0 TODO: merge into IDriver
		/// <summary>
		/// Begin an ADO <see cref="DbTransaction" />.
		/// </summary>
		/// <param name="driver">The driver.</param>
		/// <param name="isolationLevel">The isolation level requested for the transaction.</param>
		/// <param name="connection">The connection on which to start the transaction.</param>
		/// <returns>The started <see cref="DbTransaction" />.</returns>
		public static DbTransaction BeginTransaction(this IDriver driver, IsolationLevel isolationLevel, DbConnection connection)
		{
			if (driver is DriverBase driverBase)
			{
				return driverBase.BeginTransaction(isolationLevel, connection);
			}

			// So long for custom drivers not deriving from DriverBase, they will have to wait for 6.0 if they
			// need the feature.
			if (isolationLevel == IsolationLevel.Unspecified)
			{
				return connection.BeginTransaction();
			}
			return connection.BeginTransaction(isolationLevel);
		}

		// 6.0 TODO: merge into IDriver
		/// <summary>
		/// Unwraps the <see cref="DbCommand"/> in case it is wrapped, otherwise the same instance is returned.
		/// </summary>
		/// <param name="driver">The driver.</param>
		/// <param name="command">The command to unwrap.</param>
		/// <returns>The unwrapped command.</returns>
		public static DbCommand UnwrapDbCommand(this IDriver driver, DbCommand command)
		{
			return driver is DriverBase driverBase ? driverBase.UnwrapDbCommand(command) : command;
		}
	}
}