Added OrmLiteConfig.ExceptionFilter callback

Author: mythz

src/ServiceStack.OrmLite/OrmLiteConfig.cs

@@ -186,6 +186,8 @@ public static IOrmLiteExecFilter ExecFilter
 
         public static Func<FieldDefinition, object> OnDbNullFilter { get; set; }
 
+        public static Action<IDbCommand, Exception> ExceptionFilter { get; set; }
+
         public static Func<string, string> SanitizeFieldNameForParamNameFn = fieldName =>
             (fieldName ?? "").Replace(" ", "");
 

src/ServiceStack.OrmLite/OrmLiteExecFilter.cs

@@ -61,6 +61,13 @@ public virtual T Exec<T>(IDbConnection dbConn, Func<IDbCommand, T> filter)
                 var ret = filter(dbCmd);
                 return ret;
             }
+            catch (Exception ex)
+            {
+                if (OrmLiteConfig.ExceptionFilter != null)
+                    OrmLiteConfig.ExceptionFilter(dbCmd, ex);
+
+                throw;
+            }
             finally
             {
                 DisposeCommand(dbCmd, dbConn);
@@ -85,6 +92,13 @@ public virtual void Exec(IDbConnection dbConn, Action<IDbCommand> filter)
             {
                 filter(dbCmd);
             }
+            catch (Exception ex)
+            {
+                if (OrmLiteConfig.ExceptionFilter != null)
+                    OrmLiteConfig.ExceptionFilter(dbCmd, ex);
+
+                throw;
+            }
             finally
             {
                 DisposeCommand(dbCmd, dbConn);

tests/ServiceStack.OrmLite.Tests/LoggingTests.cs

@@ -1,4 +1,6 @@
-using NUnit.Framework;
+using System;
+using System.Data;
+using NUnit.Framework;
 using ServiceStack.DataAnnotations;
 using ServiceStack.Logging;
 using ServiceStack.Text;
@@ -31,8 +33,10 @@ public void Does_log_all_statements()
                 db.DropTable<LogTest>();
                 db.CreateTable<LogTest>();
 
-                db.Insert(new LogTest {
-                    CustomerId = 2, Name = "Foo"
+                db.Insert(new LogTest
+                {
+                    CustomerId = 2,
+                    Name = "Foo"
                 });
 
                 var test = db.Single<LogTest>(x => x.CustomerId == 2);
@@ -55,5 +59,33 @@ public void Does_log_all_statements()
                 Assert.That(logs, Is.StringContaining("DELETE FROM"));
             }
         }
+
+        [Test]
+        public void Can_handle_sql_exceptions()
+        {
+            string lastSql = null;
+            IDbDataParameter lastParam = null;
+            OrmLiteConfig.ExceptionFilter = (dbCmd, ex) =>
+            {
+                lastSql = dbCmd.CommandText;
+                lastParam = (IDbDataParameter)dbCmd.Parameters[0];
+            };
+
+            using (var db = OpenDbConnection())
+            {
+                db.DropAndCreateTable<LogTest>();
+
+                try
+                {
+                    var results = db.Select<LogTest>("Unknown = @arg", new { arg = "foo" });
+                    Assert.Fail("Should throw");
+                }
+                catch (Exception ex)
+                {
+                    Assert.That(lastSql, Is.StringContaining("Unknown"));
+                    Assert.That(lastParam.Value, Is.EqualTo("foo"));
+                }
+            }
+        }
     }
 }