SqlDelegatedTransaction.cs source code in C# .NET

Source code for the .NET framework in C#

                        
Code:

                         / FX-1434 / FX-1434 / 1.0 / untmp / whidbey / REDBITS / ndp / fx / src / Data / System / Data / SqlClient / SqlDelegatedTransaction.cs / 5 / SqlDelegatedTransaction.cs
                        

                        
                            //------------------------------------------------------------------------------ 
// 
//     Copyright (c) Microsoft Corporation.  All rights reserved.
// 
// [....] 
// [....]
//----------------------------------------------------------------------------- 
 
namespace System.Data.SqlClient {
 
    using System.Data.Common;
    using System.Data.SqlClient;
    using System.Diagnostics;
    using System.Runtime.CompilerServices; 
    using System.Runtime.ConstrainedExecution;
    using System.Threading; 
    using SysTx = System.Transactions; 

    sealed internal class SqlDelegatedTransaction : SysTx.IPromotableSinglePhaseNotification { 
        private static int _objectTypeCount;
        private readonly int _objectID = Interlocked.Increment(ref _objectTypeCount);
        internal int ObjectID {
            get { 
                return _objectID;
            } 
        } 

        // WARNING!!! Multithreaded object! 
        // Locking strategy: Any potentailly-multithreaded operation must first lock the associated connection, then
        //  validate this object's active state.  Locked activities should ONLY include Sql-transaction state altering activities
        //  or notifications of same. Updates to the connection's association with the transaction or to the connection pool
        //  may be initiated here AFTER the connection lock is released, but should NOT fall under this class's locking strategy. 

        private SqlInternalConnection   _connection;            // the internal connection that is the root of the transaction 
        private IsolationLevel          _isolationLevel;        // the IsolationLevel of the transaction we delegated to the server 
        private SqlInternalTransaction  _internalTransaction;   // the SQL Server transaction we're delegating to
 
        private SysTx.Transaction       _atomicTransaction;

        private bool                    _active;                // Is the transaction active?
 
        internal SqlDelegatedTransaction(SqlInternalConnection connection, SysTx.Transaction tx) {
            Debug.Assert(null != connection, "null connection?"); 
            _connection = connection; 
            _atomicTransaction = tx;
            _active = false; 
            SysTx.IsolationLevel systxIsolationLevel = tx.IsolationLevel;

            // We need to map the System.Transactions IsolationLevel to the one
            // that System.Data uses and communicates to SqlServer.  We could 
            // arguably do that in Initialize when the transaction is delegated,
            // however it is better to do this before we actually begin the process 
            // of delegation, in case System.Transactions adds another isolation 
            // level we don't know about -- we can throw the exception at a better
            // place. 
            switch (systxIsolationLevel) {
                case SysTx.IsolationLevel.ReadCommitted:    _isolationLevel = IsolationLevel.ReadCommitted;     break;
                case SysTx.IsolationLevel.ReadUncommitted:  _isolationLevel = IsolationLevel.ReadUncommitted;   break;
                case SysTx.IsolationLevel.RepeatableRead:   _isolationLevel = IsolationLevel.RepeatableRead;    break; 
                case SysTx.IsolationLevel.Serializable:     _isolationLevel = IsolationLevel.Serializable;      break;
                case SysTx.IsolationLevel.Snapshot:         _isolationLevel = IsolationLevel.Snapshot;          break; 
                default: 
                    throw SQL.UnknownSysTxIsolationLevel(systxIsolationLevel);
            } 
        }

        internal SysTx.Transaction Transaction
        { 
            get { return _atomicTransaction; }
        } 
 
        public void Initialize() {
            // if we get here, then we know for certain that we're the delegated 
            // transaction.
            SqlInternalConnection connection = _connection;
            SqlConnection usersConnection = connection.Connection;
 
            Bid.Trace(" %d#, Connection %d#, delegating transaction.\n", ObjectID, connection.ObjectID);
 
            RuntimeHelpers.PrepareConstrainedRegions(); 
            try {
#if DEBUG 
                object initialReliabilitySlotValue = Thread.GetData(TdsParser.ReliabilitySlot);

                RuntimeHelpers.PrepareConstrainedRegions();
                try { 
                    Thread.SetData(TdsParser.ReliabilitySlot, true);
#endif //DEBUG 
                    if (connection.IsEnlistedInTransaction) { // defect first 
                        Bid.Trace(" %d#, Connection %d#, was enlisted, now defecting.\n", ObjectID, connection.ObjectID);
                        connection.EnlistNull(); 
                    }

                    _internalTransaction = new SqlInternalTransaction(connection, TransactionType.Delegated, null);
 
                    connection.ExecuteTransaction(SqlInternalConnection.TransactionRequest.Begin, null, _isolationLevel, _internalTransaction, true);
                    _active = true; 
#if DEBUG 
                }
                finally { 
                    Thread.SetData(TdsParser.ReliabilitySlot, initialReliabilitySlotValue);
                }
#endif //DEBUG
            } 
            catch (System.OutOfMemoryException e) {
                usersConnection.Abort(e); 
                throw; 
            }
            catch (System.StackOverflowException e) { 
                usersConnection.Abort(e);
                throw;
            }
            catch (System.Threading.ThreadAbortException e)  { 
                usersConnection.Abort(e);
                throw; 
            } 
        }
 
        internal bool IsActive {
            get {
                return _active;
            } 
        }
 
        public Byte [] Promote() { 
            // Operations that might be affected by multi-threaded use MUST be done inside the lock.
            //  Don't read values off of the connection outside the lock unless it doesn't really matter 
            //  from an operational standpoint (i.e. logging connection's ObjectID should be fine,
            //  but the PromotedDTCToken can change over calls. so that must be protected).
            SqlInternalConnection connection = GetValidConnection();
 
            bool succeeded = false;
            byte[] returnValue = null; 
            SqlConnection usersConnection = connection.Connection; 

            Bid.Trace(" %d#, Connection %d#, promoting transaction.\n", ObjectID, connection.ObjectID); 

            RuntimeHelpers.PrepareConstrainedRegions();
            try {
#if DEBUG 
                object initialReliabilitySlotValue = Thread.GetData(TdsParser.ReliabilitySlot);
 
                RuntimeHelpers.PrepareConstrainedRegions(); 
                try {
                    Thread.SetData(TdsParser.ReliabilitySlot, true); 
#endif //DEBUG
                    lock (connection) {
                        // Now that we've acquired the lock, make sure we still have valid state for this operation.
                        ValidateActiveOnConnection(connection); 

                        connection.ExecuteTransaction(SqlInternalConnection.TransactionRequest.Promote, null, IsolationLevel.Unspecified, _internalTransaction, true); 
                        returnValue = _connection.PromotedDTCToken; 
                        succeeded = true;
                    } 
#if DEBUG
                }
                finally {
                    Thread.SetData(TdsParser.ReliabilitySlot, initialReliabilitySlotValue); 
                }
#endif //DEBUG 
            } 
            catch (SqlException e) {
                ADP.TraceExceptionWithoutRethrow(e); 

                // Doom the connection, to make sure that the transaction is
                // eventually rolled back.
                connection.DoomThisConnection(); 

                throw SQL.PromotionFailed(e); 
            } 
            catch (System.OutOfMemoryException e) {
                usersConnection.Abort(e); 
                throw;
            }
            catch (System.StackOverflowException e) {
                usersConnection.Abort(e); 
                throw;
            } 
            catch (System.Threading.ThreadAbortException e)  { 
                usersConnection.Abort(e);
                throw; 
            }

            if (!succeeded) {
                throw SQL.PromotionFailed(null); 
            }
 
            return returnValue; 
        }
 
        // Called by transaction to initiate abort sequence
        public void Rollback(SysTx.SinglePhaseEnlistment enlistment) {
            Debug.Assert(null != enlistment, "null enlistment?");
 
            SqlInternalConnection connection = GetValidConnection();
            SqlConnection usersConnection = connection.Connection; 
 
            Bid.Trace(" %d#, Connection %d#, aborting transaction.\n", ObjectID, connection.ObjectID);
 
            RuntimeHelpers.PrepareConstrainedRegions();
            try {
#if DEBUG
                object initialReliabilitySlotValue = Thread.GetData(TdsParser.ReliabilitySlot); 

                RuntimeHelpers.PrepareConstrainedRegions(); 
                try { 
                    Thread.SetData(TdsParser.ReliabilitySlot, true);
#endif //DEBUG 
                    try {
                        lock (connection) {
                            // Now that we've acquired the lock, make sure we still have valid state for this operation.
                            ValidateActiveOnConnection(connection); 
                            _active = false; // set to inactive first, doesn't matter how the execute completes, this transaction is done.
                            _connection = null;  // Set prior to ExecuteTransaction call in case this initiates a TransactionEnd event 
 
                            connection.ExecuteTransaction(SqlInternalConnection.TransactionRequest.Rollback, null, IsolationLevel.Unspecified, _internalTransaction, true);
                        } 
                    }
                    catch (SqlException e) {
                        ADP.TraceExceptionWithoutRethrow(e);
 
                        // Doom the connection, to make sure that the transaction is
                        // eventually rolled back. 
                        connection.DoomThisConnection(); 

                        // Unlike SinglePhaseCommit, a rollback is a rollback, regardless 
                        // of how it happens, so SysTx won't throw an exception, and we
                        // don't want to throw an exception either, because SysTx isn't
                        // handling it and it may create a fail fast scenario. In the end,
                        // there is no way for us to communicate to the consumer that this 
                        // failed for more serious reasons than usual.
                        // 
                        // This is a bit like "should you throw if Close fails", however, 
                        // it only matters when you really need to know.  In that case,
                        // we have the tracing that we're doing to fallback on for the 
                        // investigation.
                    }

                    // it doesn't matter whether the rollback succeeded or not, we presume 
                    // that the transaction is aborted, because it will be eventually.
                    connection.CleanupConnectionOnTransactionCompletion(_atomicTransaction); 
                    enlistment.Aborted(); 
#if DEBUG
                } 
                finally {
                    Thread.SetData(TdsParser.ReliabilitySlot, initialReliabilitySlotValue);
                }
#endif //DEBUG 
            }
            catch (System.OutOfMemoryException e) { 
                usersConnection.Abort(e); 
                throw;
            } 
            catch (System.StackOverflowException e) {
                usersConnection.Abort(e);
                throw;
            } 
            catch (System.Threading.ThreadAbortException e)  {
                usersConnection.Abort(e); 
                throw; 
            }
        } 

        // Called by the transaction to initiate commit sequence
        public void SinglePhaseCommit(SysTx.SinglePhaseEnlistment enlistment) {
            Debug.Assert(null != enlistment, "null enlistment?"); 

            bool succeeded = false; 
 
            SqlInternalConnection connection = GetValidConnection();
            SqlConnection usersConnection = connection.Connection; 

            Bid.Trace(" %d#, Connection %d#, committing transaction.\n", ObjectID, connection.ObjectID);

            RuntimeHelpers.PrepareConstrainedRegions(); 
            try {
#if DEBUG 
                object initialReliabilitySlotValue = Thread.GetData(TdsParser.ReliabilitySlot); 

                RuntimeHelpers.PrepareConstrainedRegions(); 
                try {
                    Thread.SetData(TdsParser.ReliabilitySlot, true);
#endif //DEBUG
                    // If the connection is dooomed, we can be certain that the 
                    // transaction will eventually be rolled back, and we shouldn't
                    // attempt to commit it. 
                    if (connection.IsConnectionDoomed) { 
                        lock (connection) {
                            _active = false; // set to inactive first, doesn't matter how the rest completes, this transaction is done. 
                            _connection = null;
                        }

                        enlistment.Aborted(SQL.ConnectionDoomed()); 
                    }
                    else { 
                        try { 
                            lock (connection) {
                                // Now that we've acquired the lock, make sure we still have valid state for this operation. 
                                ValidateActiveOnConnection(connection);

                                _active = false; // set to inactive first, doesn't matter how the rest completes, this transaction is done.
                                _connection = null;   // Set prior to ExecuteTransaction call in case this initiates a TransactionEnd event 

                                connection.ExecuteTransaction(SqlInternalConnection.TransactionRequest.Commit, null, IsolationLevel.Unspecified, _internalTransaction, true); 
                                succeeded = true; 
                            }
                        } 
                        catch (SqlException e) {
                            ADP.TraceExceptionWithoutRethrow(e);

 
                            // Doom the connection, to make sure that the transaction is
                            // eventually rolled back. 
                            connection.DoomThisConnection(); 

                            if (_internalTransaction.IsCommitted) { 
                                // Even though we got an exception, the transaction
                                // was committed by the server.
                                enlistment.Committed();
                            } 
                            else if (_internalTransaction.IsAborted) {
                                // The transaction was aborted, report that to 
                                // SysTx. 
                                enlistment.Aborted(e);
                            } 
                            else {
                                // The transaction is still active, we cannot
                                // know the state of the transaction.
                                enlistment.InDoubt(e); 
                            }
 
                            // We eat the exception.  This is called on the SysTx 
                            // thread, not the applications thread.  If we don't
                            // eat the exception an UnhandledException will occur, 
                            // causing the process to FailFast.
                        }

                        connection.CleanupConnectionOnTransactionCompletion(_atomicTransaction); 
                        if (succeeded) {
                            enlistment.Committed(); 
                        } 
                    }
#if DEBUG 
                }
                finally {
                    Thread.SetData(TdsParser.ReliabilitySlot, initialReliabilitySlotValue);
                } 
#endif //DEBUG
            } 
            catch (System.OutOfMemoryException e) { 
                usersConnection.Abort(e);
                throw; 
            }
            catch (System.StackOverflowException e) {
                usersConnection.Abort(e);
                throw; 
            }
            catch (System.Threading.ThreadAbortException e)  { 
                usersConnection.Abort(e); 
                throw;
            } 
        }

        // Event notification that transaction ended. This comes from the subscription to the Transaction's
        //  ended event via the internal connection. If it occurs without a prior Rollback or SinglePhaseCommit call, 
        //  it indicates the transaction was ended externally (generally that one the the DTC participants aborted
        //  the transaction). 
        internal void TransactionEnded(SysTx.Transaction transaction) { 
            SqlInternalConnection connection = _connection;
 
            if (connection != null) {
                Bid.Trace(" %d#, Connection %d#, transaction completed externally.\n", ObjectID, connection.ObjectID);

                lock (connection) { 
                    if (_atomicTransaction.Equals(transaction)) {
                        // No need to validate active on connection, this operation can be called on completed transactions 
                        _active = false; 
                        _connection = null;
                    } 
                }
            }
        }
 
        // Check for connection validity
        private SqlInternalConnection GetValidConnection() { 
            SqlInternalConnection connection = _connection; 
            if (null == connection) {
                throw ADP.ObjectDisposed(this); 
            }

            return connection;
        } 

        // Dooms connection and throws and error if not a valid, active, delegated transaction for the given 
        //  connection. Designed to be called AFTER a lock is placed on the connection, otherwise a normal return 
        //  may not be trusted.
        private void ValidateActiveOnConnection(SqlInternalConnection connection) { 
            bool valid = _active && (connection == _connection) && (connection.DelegatedTransaction == this);

            if (!valid) {
                // Invalid indicates something BAAAD happened (Commit after TransactionEnded, for instance) 
                //  Doom anything remotely involved.
                if (null != connection) { 
                    connection.DoomThisConnection(); 
                }
                if (connection != _connection && null != _connection) { 
                    _connection.DoomThisConnection();
                }

                throw ADP.InternalError(ADP.InternalErrorCode.UnpooledObjectHasWrongOwner);  // 
            }
        } 
 
    }
} 

// File provided for Reference Use Only by Microsoft Corporation (c) 2007.
// Copyright (c) Microsoft Corporation. All rights reserved.

Link Menu

Network programming in C#, Network Programming in VB.NET, Network Programming in .NET
This book is available now!
Buy at Amazon US or
Buy at Amazon UK