NH-3724 - Added support for notification handlers

src/NHibernate.Test/CfgTest/Loquacious/ConfigurationFixture.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Data;
 using System.Data.SqlClient;
 using NHibernate.AdoNet;
@@ -10,12 +11,26 @@
 using NHibernate.Hql.Ast.ANTLR;
 using NHibernate.Type;
 using NUnit.Framework;
+using Environment = NHibernate.Cfg.Environment;
 
 namespace NHibernate.Test.CfgTest.Loquacious
 {
 	[TestFixture]
 	public class ConfigurationFixture
 	{
+		[Test]
+		public void CanSetNotificationHandler()
+		{
+			//NH-3724
+			EventHandler<SqlInfoMessageEventArgs> handler = (s, e) => {};
+			var cfg = new Configuration().DataBaseIntegration(x =>
+			{
+				x.WithNotificationHandler(handler);
+			}).Configure();
+
+			Assert.IsNotNull(cfg.NotificationHandler);
+		}
+
 		[Test]
 		public void CompleteConfiguration()
 		{

src/NHibernate/AdoNet/ConnectionManager.cs

@@ -184,6 +184,11 @@ public IDbConnection GetConnection()
 				if (ownConnection)
 				{
 					connection = Factory.ConnectionProvider.GetConnection();
+					//NH-3724
+					if (Factory.Settings.NotificationHandler != null)
+					{
+						Factory.ConnectionProvider.Driver.AddNotificationHandler(connection, Factory.Settings.NotificationHandler);
+					}
 					if (Factory.Statistics.IsStatisticsEnabled)
 					{
 						Factory.StatisticsImplementor.Connect();

src/NHibernate/Cfg/Configuration.cs

@@ -185,6 +185,7 @@ protected void Reset()
 			tableNameBinding = new Dictionary<string, Mappings.TableDescription>();
 			columnNameBindingPerTable = new Dictionary<Table, Mappings.ColumnNames>();
 			filtersSecondPasses = new Queue<FilterSecondPassArgs>();
+			NotificationHandler = null;
 		}
 		[Serializable]
 		private class Mapping : IMapping
@@ -1337,6 +1338,10 @@ public Configuration SetDefaultNamespace(string newDefaultNamespace)
 			return this;
 		}
 
+		//NH-3724
+		[field: NonSerialized]
+		public Delegate NotificationHandler { get; set; }
+
 		/// <summary>
 		/// Sets the default interceptor for use by all sessions.
 		/// </summary>
@@ -1725,6 +1730,8 @@ private Settings BuildSettings()
 			// NH : Set configuration for IdGenerator SQL logging
 			PersistentIdGeneratorParmsNames.SqlStatementLogger.FormatSql = result.SqlStatementLogger.FormatSql;
 			PersistentIdGeneratorParmsNames.SqlStatementLogger.LogToStdout = result.SqlStatementLogger.LogToStdout;
+			//NH-3724
+			result.NotificationHandler = NotificationHandler;
 			return result;
 		}
 

src/NHibernate/Cfg/Loquacious/DbIntegrationConfigurationProperties.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Data;
 using NHibernate.AdoNet;
 using NHibernate.Connection;
@@ -18,6 +19,12 @@ public DbIntegrationConfigurationProperties(Configuration configuration)
 
 		#region Implementation of IDbIntegrationConfigurationProperties
 
+		public void WithNotificationHandler(Delegate handler)
+		{
+			//NH-3724
+			configuration.NotificationHandler = handler;
+		}
+
 		public void Dialect<TDialect>() where TDialect : Dialect.Dialect
 		{
 			configuration.SetProperty(Environment.Dialect, typeof(TDialect).AssemblyQualifiedName);

src/NHibernate/Cfg/Loquacious/IDbIntegrationConfigurationProperties.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Data;
 using NHibernate.AdoNet;
 using NHibernate.Connection;
@@ -14,6 +15,9 @@ public interface IDbIntegrationConfigurationProperties
 		bool LogSqlInConsole { set; }
 		bool LogFormattedSql { set; }
 
+		//NH-3724
+		void WithNotificationHandler(Delegate handler);
+
 		void ConnectionProvider<TProvider>() where TProvider : IConnectionProvider;
 		void Driver<TDriver>() where TDriver : IDriver;
 		IsolationLevel IsolationLevel { set; }

src/NHibernate/Cfg/Settings.cs

@@ -31,6 +31,9 @@ public Settings()
 		//private bool isJdbcBatchVersionedData;
 
 		#endregion
+
+		public Delegate NotificationHandler { get; internal set; }
+
 		public SqlStatementLogger SqlStatementLogger { get; internal set; }
 
 		public int MaximumFetchDepth { get; internal set; }

src/NHibernate/Driver/DB2400Driver.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Data;
 
 namespace NHibernate.Driver
 {

src/NHibernate/Driver/DB2Driver.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Data;
 
 namespace NHibernate.Driver
 {

src/NHibernate/Driver/DotConnectMySqlDriver.cs

@@ -1,4 +1,7 @@
-namespace NHibernate.Driver
+using System;
+using System.Data;
+
+namespace NHibernate.Driver
 {
     /// <summary>
     /// Provides a database driver for dotConnect for MySQL by DevArt.

src/NHibernate/Driver/DriverBase.cs

@@ -20,6 +20,22 @@ public abstract class DriverBase : IDriver, ISqlParameterFormatter
 		private int commandTimeout;
 		private bool prepareSql;
 
+		public virtual void AddNotificationHandler(IDbConnection con, Delegate handler)
+		{
+			//NH-3724
+			if (handler != null)
+			{
+				var prop = con.GetType().GetEvent("InfoMessage");
+
+				if (prop == null)
+				{
+					throw new NotSupportedException("Current driver does not support notifications");
+				}
+
+				prop.AddEventHandler(con, handler);
+			}
+		}
+
 		public virtual void Configure(IDictionary<string, string> settings)
 		{
 			// Command timeout

src/NHibernate/Driver/FirebirdClientDriver.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Collections.Generic;
 using System.Data;
 using System.Linq;

src/NHibernate/Driver/IDriver.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Collections.Generic;
 using System.Data;
 using NHibernate.Engine;
@@ -35,6 +36,13 @@ public interface IDriver
 		/// </summary>
 		void Configure(IDictionary<string, string> settings);
 
+		//NH-3724
+		/// <summary>
+		/// Attaches an event handler for the notification event (InfoMessage in most ADO.NET drivers).
+		/// </summary>
+		/// <param name="handler"></param>
+		void AddNotificationHandler(IDbConnection con, Delegate handler);
+
 		/// <summary>
 		/// Creates an uninitialized IDbConnection object for the specific Driver
 		/// </summary>

src/NHibernate/Driver/IfxDriver.cs

@@ -1,3 +1,6 @@
+using System;
+using System.Data;
+
 namespace NHibernate.Driver
 {
 	/// <summary>

src/NHibernate/Driver/IngresDriver.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Data;
 
 namespace NHibernate.Driver
 {

src/NHibernate/Driver/MySqlDataDriver.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Data;
 
 namespace NHibernate.Driver
 {

src/NHibernate/Driver/NpgsqlDriver.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Data;
 
 namespace NHibernate.Driver
@@ -39,6 +40,12 @@ public NpgsqlDriver() : base(
 		{
 		}
 
+		public override void AddNotificationHandler(IDbConnection con, Delegate handler)
+		{
+			//NH-3724
+			con.GetType().GetEvent("Notification").AddEventHandler(con, handler);
+		}
+
 		public override bool UseNamedPrefixInSql
 		{
 			get { return true; }

src/NHibernate/Driver/OdbcDriver.cs

@@ -13,6 +13,12 @@ namespace NHibernate.Driver
 	/// </remarks>
 	public class OdbcDriver : DriverBase
 	{
+		public override void AddNotificationHandler(IDbConnection con, Delegate handler)
+		{
+			//NH-3724
+			(con as OdbcConnection).InfoMessage += (OdbcInfoMessageEventHandler)handler;
+		}
+
 		public override IDbConnection CreateConnection()
 		{
 			return new OdbcConnection();

src/NHibernate/Driver/OleDbDriver.cs

@@ -16,6 +16,12 @@ public OleDbDriver()
 		{
 		}
 
+		public override void AddNotificationHandler(IDbConnection con, Delegate handler)
+		{
+			//NH-3724
+			(con as OleDbConnection).InfoMessage += (OleDbInfoMessageEventHandler)handler;
+		}
+
 		public override IDbConnection CreateConnection()
 		{
 			return new OleDbConnection();

src/NHibernate/Driver/OracleClientDriver.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Data;
 using NHibernate.Engine.Query;
 using NHibernate.SqlTypes;

src/NHibernate/Driver/OracleLiteDataClientDriver.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Data;
 using NHibernate.AdoNet;
 using NHibernate.SqlTypes;

src/NHibernate/Driver/SqlClientDriver.cs

@@ -1,3 +1,4 @@
+using System;
 using System.Data;
 using System.Data.SqlClient;
 using NHibernate.AdoNet;
@@ -23,6 +24,12 @@ public class SqlClientDriver : DriverBase, IEmbeddedBatcherFactoryProvider
 		public const byte MaxDateTime2 = 8;
 		public const byte MaxDateTimeOffset = 10;
 
+		public override void AddNotificationHandler(IDbConnection con, Delegate handler)
+		{
+			//NH-3724
+			(con as SqlConnection).InfoMessage += (SqlInfoMessageEventHandler) handler;
+		}
+
 		/// <summary>
 		/// Creates an uninitialized <see cref="IDbConnection" /> object for
 		/// the SqlClientDriver.

src/NHibernate/Driver/SybaseAsaClientDriver.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Data;
 
 namespace NHibernate.Driver
 {

src/NHibernate/Driver/SybaseAseClientDriver.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Data;
 
 namespace NHibernate.Driver
 {
@@ -17,7 +18,7 @@ public class SybaseAseClientDriver : ReflectionBasedDriver
 		public SybaseAseClientDriver() : base("Sybase.AdoNet2.AseClient", "Sybase.Data.AseClient.AseConnection", "Sybase.Data.AseClient.AseCommand")
 		{
 		}
-		
+
 		public override string NamedPrefix
 		{
 			get { return "@"; }

src/NHibernate/Driver/SybaseSQLAnywhereDotNet4Driver.cs

@@ -1,4 +1,7 @@
-namespace NHibernate.Driver
+using System;
+using System.Data;
+
+namespace NHibernate.Driver
 {
 	/// <summary>
 	/// SQL Dialect for SQL Anywhere 12 - for the NHibernate 3.2.0 distribution

src/NHibernate/Driver/SybaseSQLAnywhereDriver.cs

@@ -1,4 +1,7 @@
-namespace NHibernate.Driver
+using System;
+using System.Data;
+
+namespace NHibernate.Driver
 {
 	/// <summary>
 	/// The SybaseSQLAnywhereDriver Driver provides a database driver for Sybase SQL Anywhere 10 and above