Merge pull request #137 from davidroth/transactionscope-support

Add transaction scope support

Author: frankhommers

src/Hangfire.PostgreSql/Hangfire.PostgreSql.csproj

@@ -65,6 +65,7 @@
     <Reference Include="System.Configuration" />
     <Reference Include="System" />
     <Reference Include="Microsoft.CSharp" />
+    <Reference Include="System.Transactions" />
   </ItemGroup>
 
   <ItemGroup>

src/Hangfire.PostgreSql/PostgreSqlStorage.cs

@@ -111,9 +111,13 @@ public PostgreSqlStorage(NpgsqlConnection existingConnection, PostgreSqlStorageO
             if (existingConnection == null) throw new ArgumentNullException(nameof(existingConnection));
             if (options == null) throw new ArgumentNullException(nameof(options));
             var connectionStringBuilder = new NpgsqlConnectionStringBuilder(existingConnection.ConnectionString);
-            if (connectionStringBuilder.Enlist)
-                throw new ArgumentException(
-                    "Npgsql is not fully compatible with TransactionScope yet, only connections without Enlist = true are accepted.");
+
+            if (!options.EnableTransactionScopeEnlistment)
+            {
+                if (connectionStringBuilder.Enlist)
+                    throw new ArgumentException(
+                        $"TransactionScope enlistment must be enabled by setting {nameof(PostgreSqlStorageOptions)}.{nameof(options.EnableTransactionScopeEnlistment)} to `true`.");
+            }
 
             _existingConnection = existingConnection;
             _options = options;
@@ -127,7 +131,7 @@ public PostgreSqlStorage(NpgsqlConnection existingConnection)
             var connectionStringBuilder = new NpgsqlConnectionStringBuilder(existingConnection.ConnectionString);
             if (connectionStringBuilder.Enlist)
                 throw new ArgumentException(
-                    "Npgsql is not fully compatible with TransactionScope yet, only connections without Enlist = true are accepted.");
+                    $"TransactionScope enlistment must be enabled by setting {nameof(PostgreSqlStorageOptions)}.{nameof(PostgreSqlStorageOptions.EnableTransactionScopeEnlistment)} to `true`.");
 
             _existingConnection = existingConnection;
             _options = new PostgreSqlStorageOptions();
@@ -189,10 +193,11 @@ public override string ToString()
 
         internal NpgsqlConnection CreateAndOpenConnection()
         {
-            var connectionStringBuilder = new NpgsqlConnectionStringBuilder(_connectionString)
+            var connectionStringBuilder = new NpgsqlConnectionStringBuilder(_connectionString);
+            if (!_options.EnableTransactionScopeEnlistment)
             {
-                Enlist = false //Npgsql is not fully compatible with TransactionScope yet.
-            };
+                connectionStringBuilder.Enlist = false;
+            }
 
             var connection = new NpgsqlConnection(connectionStringBuilder.ToString());
 

src/Hangfire.PostgreSql/PostgreSqlStorageOptions.cs

@@ -84,6 +84,7 @@ public TimeSpan TransactionSynchronisationTimeout
         public bool UseNativeDatabaseTransactions { get; set; }
         public bool PrepareSchemaIfNecessary { get; set; }
         public string SchemaName { get; set; }
+        public bool EnableTransactionScopeEnlistment { get; set; }
 
         private static void ThrowIfValueIsNotPositive(TimeSpan value, string fieldName)
         {