using SafeObjectPool; using System; using System.Collections.Generic; using System.Data.Common; using System.Linq.Expressions; using System.Text; namespace FreeSql { class UnitOfWork : IUnitOfWork { IFreeSql _fsql; Object _conn; DbTransaction _tran; bool _isCommitOrRoolback = false; public UnitOfWork(IFreeSql fsql) { _fsql = fsql; } void ReturnObject() { _fsql.Ado.MasterPool.Return(_conn); _tran = null; _conn = null; } internal DbTransaction GetOrBeginTransaction() { _isCommitOrRoolback = false; if (_tran != null) return _tran; if (_conn != null) _fsql.Ado.MasterPool.Return(_conn); _conn = _fsql.Ado.MasterPool.Get(); try { _tran = _conn.Value.BeginTransaction(); } catch { ReturnObject(); throw; } return _tran; } public void Commit() { if (_tran != null) { try { _tran.Commit(); _isCommitOrRoolback = true; } finally { ReturnObject(); } } } public void Rollback() { _isCommitOrRoolback = true; if (_tran != null) { try { _tran.Rollback(); _isCommitOrRoolback = true; } finally { ReturnObject(); } } } public void Dispose() { if (_isCommitOrRoolback == false) { this.Commit(); } } public DefaultRepository GetRepository(Expression> filter = null) where TEntity : class { var repos = new DefaultRepository(_fsql, filter); repos._unitOfWork = this; return repos; } public GuidRepository GetGuidRepository(Expression> filter = null, Func asTable = null) where TEntity : class { var repos = new GuidRepository(_fsql, filter, asTable); repos._unitOfWork = this; return repos; } } }