Mercurial > pub > bltoolkit
diff Source/Data/Linq/DataContextTransaction.cs @ 0:f990fcb411a9
Копия текущей версии из github
author | cin |
---|---|
date | Thu, 27 Mar 2014 21:46:09 +0400 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Source/Data/Linq/DataContextTransaction.cs Thu Mar 27 21:46:09 2014 +0400 @@ -0,0 +1,95 @@ +using System; +using System.Data; + +using JetBrains.Annotations; + +namespace BLToolkit.Data.Linq +{ + public class DataContextTransaction : IDisposable + { + public DataContextTransaction([NotNull] DataContext dataContext) + { + if (dataContext == null) throw new ArgumentNullException("dataContext"); + + DataContext = dataContext; + } + + public DataContext DataContext { get; set; } + + int _transactionCounter; + + public void BeginTransaction() + { + var db = DataContext.GetDBManager(); + + db.BeginTransaction(); + + if (_transactionCounter == 0) + DataContext.LockDbManagerCounter++; + + _transactionCounter++; + } + + public void BeginTransaction(IsolationLevel level) + { + var db = DataContext.GetDBManager(); + + db.BeginTransaction(level); + + if (_transactionCounter == 0) + DataContext.LockDbManagerCounter++; + + _transactionCounter++; + } + + public void CommitTransaction() + { + if (_transactionCounter > 0) + { + var db = DataContext.GetDBManager(); + + db.CommitTransaction(); + + _transactionCounter--; + + if (_transactionCounter == 0) + { + DataContext.LockDbManagerCounter--; + DataContext.ReleaseQuery(); + } + } + } + + public void RollbackTransaction() + { + if (_transactionCounter > 0) + { + var db = DataContext.GetDBManager(); + + db.RollbackTransaction(); + + _transactionCounter--; + + if (_transactionCounter == 0) + { + DataContext.LockDbManagerCounter--; + DataContext.ReleaseQuery(); + } + } + } + + public void Dispose() + { + if (_transactionCounter > 0) + { + var db = DataContext.GetDBManager(); + + db.RollbackTransaction(); + + _transactionCounter = 0; + + DataContext.LockDbManagerCounter--; + } + } + } +}