diff --git a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity 纯净版源码(.Net 4.0).zip b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity 纯净版源码(.Net 4.0).zip deleted file mode 100644 index 6a90cb18..00000000 Binary files a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity 纯净版源码(.Net 4.0).zip and /dev/null differ diff --git a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity 纯净版源码.zip b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity 纯净版源码.zip deleted file mode 100644 index 47a9b4d1..00000000 Binary files a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity 纯净版源码.zip and /dev/null differ diff --git a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity.cs b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity.cs index 829e33c5..a3064ad9 100644 --- a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity.cs +++ b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntity.cs @@ -70,12 +70,12 @@ namespace FreeSql { if (this.Repository == null) return Orm.Update(this as TEntity) - .WithTransaction(UnitOfWork.Current.Value?.GetOrBeginTransaction()) + .WithTransaction(CurrentUnitOfWork?.GetOrBeginTransaction()) .Set(a => (a as BaseEntity).IsDeleted, this.IsDeleted = value).ExecuteAffrows() == 1; this.SetTenantId(); this.IsDeleted = value; - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return this.Repository.Update(this as TEntity) == 1; } /// @@ -89,7 +89,7 @@ namespace FreeSql if (this.Repository == null) return Orm.Delete(this as TEntity).ExecuteAffrows() == 1; //this.SetTenantId(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return this.Repository.Delete(this as TEntity) == 1; } /// @@ -107,11 +107,11 @@ namespace FreeSql this.UpdateTime = DateTime.Now; if (this.Repository == null) return Orm.Update() - .WithTransaction(UnitOfWork.Current.Value?.GetOrBeginTransaction()) + .WithTransaction(CurrentUnitOfWork?.GetOrBeginTransaction()) .SetSource(this as TEntity).ExecuteAffrows() == 1; this.SetTenantId(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return this.Repository.Update(this as TEntity) == 1; } /// @@ -124,7 +124,7 @@ namespace FreeSql this.Repository = Orm.GetRepository(); this.SetTenantId(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return this.Repository.Insert(this as TEntity); } @@ -139,7 +139,7 @@ namespace FreeSql this.Repository = Orm.GetRepository(); this.SetTenantId(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return this.Repository.InsertOrUpdate(this as TEntity); } @@ -152,7 +152,7 @@ namespace FreeSql if (this.Repository == null) this.Repository = Orm.GetRepository(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; this.Repository.SaveMany(this as TEntity, navigatePropertyName); } } diff --git a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityAsync.cs b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityAsync.cs index fb43fcb9..8b0e5ed2 100644 --- a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityAsync.cs +++ b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityAsync.cs @@ -53,11 +53,11 @@ namespace FreeSql { if (this.Repository == null) return await Orm.Update(this as TEntity) - .WithTransaction(UnitOfWork.Current.Value?.GetOrBeginTransaction()) + .WithTransaction(CurrentUnitOfWork?.GetOrBeginTransaction()) .Set(a => (a as BaseEntity).IsDeleted, this.IsDeleted = value).ExecuteAffrowsAsync() == 1; this.IsDeleted = value; - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return await this.Repository.UpdateAsync(this as TEntity) == 1; } /// @@ -71,7 +71,7 @@ namespace FreeSql if (this.Repository == null) return await Orm.Delete(this as TEntity).ExecuteAffrowsAsync() == 1; //this.SetTenantId(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return await this.Repository.DeleteAsync(this as TEntity) == 1; } /// @@ -89,11 +89,11 @@ namespace FreeSql this.UpdateTime = DateTime.Now; if (this.Repository == null) return await Orm.Update() - .WithTransaction(UnitOfWork.Current.Value?.GetOrBeginTransaction()) + .WithTransaction(CurrentUnitOfWork?.GetOrBeginTransaction()) .SetSource(this as TEntity).ExecuteAffrowsAsync() == 1; this.SetTenantId(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return await this.Repository.UpdateAsync(this as TEntity) == 1; } /// @@ -106,7 +106,7 @@ namespace FreeSql this.Repository = Orm.GetRepository(); this.SetTenantId(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return this.Repository.InsertAsync(this as TEntity); } @@ -121,7 +121,7 @@ namespace FreeSql this.Repository = Orm.GetRepository(); this.SetTenantId(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return this.Repository.InsertOrUpdateAsync(this as TEntity); } @@ -134,7 +134,7 @@ namespace FreeSql if (this.Repository == null) this.Repository = Orm.GetRepository(); - this.Repository.UnitOfWork = UnitOfWork.Current.Value; + this.Repository.UnitOfWork = CurrentUnitOfWork; return this.Repository.SaveManyAsync(this as TEntity, navigatePropertyName); } } diff --git a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityReadOnly.cs b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityReadOnly.cs index 68174990..d4efcbff 100644 --- a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityReadOnly.cs +++ b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityReadOnly.cs @@ -106,9 +106,11 @@ namespace FreeSql { var uow = Orm.CreateUnitOfWork(); uow.IsolationLevel = level; + CurrentUnitOfWork = uow; return uow; } + static readonly AsyncLocal _AsyncUnitOfWork = new AsyncLocal(); static readonly AsyncLocal _AsyncTenantId = new AsyncLocal(); /// /// 获取或设置当前租户id @@ -118,6 +120,14 @@ namespace FreeSql get => _AsyncTenantId.Value; set => _AsyncTenantId.Value = value; } + /// + /// 获取或设置当前租户id + /// + public static IUnitOfWork CurrentUnitOfWork + { + get => _AsyncUnitOfWork.Value; + set => _AsyncUnitOfWork.Value = value; + } } /// @@ -143,7 +153,7 @@ namespace FreeSql { var select = Orm.Select() .TrackToList(TrackToList) //自动为每个元素 Attach - .WithTransaction(UnitOfWork.Current.Value?.GetOrBeginTransaction(false)); + .WithTransaction(CurrentUnitOfWork?.GetOrBeginTransaction(false)); if (string.IsNullOrEmpty(CurrentTenantId) == false) select.WhereCascade(a => (a as ITenant).TenantId == CurrentTenantId); return select.WhereCascade(a => (a as BaseEntity).IsDeleted == false); diff --git a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityTree.cs b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityTree.cs index 28be88c8..a2b5231c 100644 --- a/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityTree.cs +++ b/Extensions/FreeSql.Extensions.BaseEntity/BaseEntityTree.cs @@ -79,7 +79,7 @@ namespace FreeSql buf.Add(this as TEntity); buf.AddRange(this.GetAllChilds()); var repo = Orm.GetRepository(); - repo.UnitOfWork = UnitOfWork.Current.Value; + repo.UnitOfWork = CurrentUnitOfWork; buf = repo.Select.WhereDynamic(buf) .Include(a => ((((((((((a as BaseEntityTree).Parent as BaseEntityTree).Parent @@ -110,7 +110,7 @@ namespace FreeSql var childs = GetAllChilds(); childs.Add(this as TEntity); var repo = Orm.GetRepository(); - repo.UnitOfWork = UnitOfWork.Current.Value; + repo.UnitOfWork = CurrentUnitOfWork; repo.Attach(childs); foreach (var item in childs) (item as BaseEntity).IsDeleted = false; diff --git a/FreeSql.DbContext/FreeSql.DbContext.xml b/FreeSql.DbContext/FreeSql.DbContext.xml index f619f03f..9f17feac 100644 --- a/FreeSql.DbContext/FreeSql.DbContext.xml +++ b/FreeSql.DbContext/FreeSql.DbContext.xml @@ -227,6 +227,41 @@ 分表规则,参数:旧表名;返回:新表名 https://github.com/2881099/FreeSql/wiki/Repository + + + 仓储的工作单元管理器 + + + + + 如果当前没有事务,就新建一个事务,如果已存在一个事务中,加入到这个事务中,默认的选择。 + + + + + 支持当前事务,如果没有当前事务,就以非事务方法执行。 + + + + + 使用当前事务,如果没有当前事务,就抛出异常。 + + + + + 以非事务方式执行操作,如果当前存在事务,就把当前事务挂起。 + + + + + 以非事务方式执行操作,如果当前事务存在则抛出异常。 + + + + + 如果当前存在事务,则在嵌套事务内执行。如果当前没有事务,就新建一个事务。 + + 开启过滤器,若使用 using 则使用完后,恢复为原有状态 @@ -299,6 +334,11 @@ 实体对象 属性名 + + + 工作单元 + + 开启事务,或者返回已开启的事务 @@ -326,7 +366,7 @@ - 此工作单元内的实体变化跟踪 + 工作单元内的实体变化跟踪 diff --git a/FreeSql.DbContext/Repository/ContextSet/RepositoryUnitOfWorkManager.cs b/FreeSql.DbContext/Repository/ContextSet/RepositoryUnitOfWorkManager.cs new file mode 100644 index 00000000..a3524b59 --- /dev/null +++ b/FreeSql.DbContext/Repository/ContextSet/RepositoryUnitOfWorkManager.cs @@ -0,0 +1,201 @@ +using System; +using System.Collections.Generic; +using System.Data; +using System.Data.Common; +using System.Linq; +using System.Linq.Expressions; +using System.Threading; + +namespace FreeSql +{ + /// + /// 仓储的工作单元管理器 + /// + public class RepositoryUnitOfWorkManager : IDisposable + { + IFreeSql _fsql; + List _uows = new List(); + bool _isNotSupported = false; + + public RepositoryUnitOfWorkManager(IFreeSql fsql) + { + _fsql = fsql ?? throw new ArgumentNullException($"{nameof(RepositoryUnitOfWorkManager)} 构造参数 {nameof(fsql)} 不能为 null"); + } + + ~RepositoryUnitOfWorkManager() => this.Dispose(); + int _disposeCounter; + public void Dispose() + { + if (Interlocked.Increment(ref _disposeCounter) != 1) return; + try + { + Exception exception = null; + for (var a = _uows.Count - 1; a >= 0; a--) + { + try + { + if (exception == null) _uows[a].Commit(); + else _uows[a].Rollback(); + } + catch (Exception ex) + { + if (exception == null) exception = ex; + } + } + if (exception != null) throw exception; + } + finally + { + _uows.Clear(); + GC.SuppressFinalize(this); + } + } + + public enum Propagation + { + /// + /// 如果当前没有事务,就新建一个事务,如果已存在一个事务中,加入到这个事务中,默认的选择。 + /// + Requierd, + /// + /// 支持当前事务,如果没有当前事务,就以非事务方法执行。 + /// + Supports, + /// + /// 使用当前事务,如果没有当前事务,就抛出异常。 + /// + Mandatory, + /// + /// 以非事务方式执行操作,如果当前存在事务,就把当前事务挂起。 + /// + NotSupported, + /// + /// 以非事务方式执行操作,如果当前事务存在则抛出异常。 + /// + Never, + /// + /// 如果当前存在事务,则在嵌套事务内执行。如果当前没有事务,就新建一个事务。 + /// + Nested + } + + public IRepositoryUnitOfWork Begin(Propagation propagation, IsolationLevel? isolationLevel = null) + { + if (propagation == Propagation.Requierd) + { + if (_isNotSupported == false) + { + for (var a = _uows.Count - 1; a >= 0; a--) + if (_uows[a].GetOrBeginTransaction(false) != null) + return new UnitOfWorkProxy(_uows[a]); + } + var uow = new RepositoryUnitOfWork(_fsql); + if (isolationLevel != null) uow.IsolationLevel = isolationLevel.Value; + try { uow.GetOrBeginTransaction(); } + catch { uow.Dispose(); throw; } + _uows.Add(uow); + return uow; + } + if (propagation == Propagation.Supports) + { + if (_isNotSupported == false) + { + for (var a = _uows.Count - 1; a >= 0; a--) + if (_uows[a].GetOrBeginTransaction(false) != null) + return new UnitOfWorkProxy(_uows[a]); + } + return new UnitOfWorkNothing(_fsql); + } + if (propagation == Propagation.Mandatory) + { + if (_isNotSupported == false) + { + for (var a = _uows.Count - 1; a >= 0; a--) + if (_uows[a].GetOrBeginTransaction(false) != null) + return new UnitOfWorkProxy(_uows[a]); + throw new Exception("Propagation_Mandatory: 使用当前事务,如果没有当前事务,就抛出异常"); + } + throw new Exception("Propagation_Mandatory: 使用当前事务,如果没有当前事务,就抛出异常(NotSupported 事务挂起中)"); + } + if (propagation == Propagation.NotSupported) + { + if (_isNotSupported == false) + { + _isNotSupported = true; + return new UnitOfWorkNothing(_fsql) { OnDispose = () => _isNotSupported = false }; + } + return new UnitOfWorkNothing(_fsql); + } + if (propagation == Propagation.Never) + { + if (_isNotSupported == false) + { + for (var a = _uows.Count - 1; a >= 0; a--) + if (_uows[a].GetOrBeginTransaction(false) != null) + throw new Exception("Propagation_Never: 以非事务方式执行操作,如果当前事务存在则抛出异常"); + } + return new UnitOfWorkNothing(_fsql); + } + if (propagation == Propagation.Nested) + { + var uow = new RepositoryUnitOfWork(_fsql); + if (isolationLevel != null) uow.IsolationLevel = isolationLevel.Value; + try { uow.GetOrBeginTransaction(); } + catch { uow.Dispose(); throw; } + _uows.Add(uow); + return uow; + } + throw new NotImplementedException(); + } + + class UnitOfWorkProxy : IRepositoryUnitOfWork + { + IRepositoryUnitOfWork _baseUow; + public UnitOfWorkProxy(IRepositoryUnitOfWork baseUow) => _baseUow = baseUow; + public IsolationLevel? IsolationLevel { get => _baseUow.IsolationLevel; set { } } + public DbContext.EntityChangeReport EntityChangeReport => _baseUow.EntityChangeReport; + + public bool Enable => _baseUow.Enable; + public void Close() => _baseUow.Close(); + public void Open() => _baseUow.Open(); + + public DbTransaction GetOrBeginTransaction(bool isCreate = true) => _baseUow.GetOrBeginTransaction(isCreate); + public void Commit() => this.Dispose(); + public void Rollback() => _baseUow.Rollback(); + public void Dispose() { } + + public IBaseRepository GetRepository(Expression> filter = null) where TEntity : class => _baseUow.GetRepository(filter); + public IBaseRepository GetRepository(Expression> filter = null) where TEntity : class => _baseUow.GetRepository(filter); + public IBaseRepository GetGuidRepository(Expression> filter = null, Func asTable = null) where TEntity : class => _baseUow.GetGuidRepository(filter); + } + class UnitOfWorkNothing : IRepositoryUnitOfWork + { + internal IFreeSql _fsql; + internal Action OnDispose; + public UnitOfWorkNothing(IFreeSql fsql) => _fsql = fsql; + public IsolationLevel? IsolationLevel { get; set; } + public DbContext.EntityChangeReport EntityChangeReport { get; } = new DbContext.EntityChangeReport(); + + public bool Enable { get; } + public void Close() { } + public void Open() { } + + public DbTransaction GetOrBeginTransaction(bool isCreate = true) => null; + public void Commit() + { + if (EntityChangeReport != null && EntityChangeReport.OnChange != null && EntityChangeReport.Report.Any() == true) + EntityChangeReport.OnChange.Invoke(EntityChangeReport.Report); + this.Dispose(); + } + public void Rollback() => this.Dispose(); + public void Dispose() { + EntityChangeReport?.Report.Clear(); + OnDispose?.Invoke(); + } + + public IBaseRepository GetRepository(Expression> filter = null) where TEntity : class => new DefaultRepository(_fsql, filter); + public IBaseRepository GetRepository(Expression> filter = null) where TEntity : class => new DefaultRepository(_fsql, filter); + public IBaseRepository GetGuidRepository(Expression> filter = null, Func asTable = null) where TEntity : class => new GuidRepository(_fsql, filter, asTable); + } + } +} diff --git a/FreeSql.DbContext/UnitOfWork/IUnitOfWork.cs b/FreeSql.DbContext/UnitOfWork/IUnitOfWork.cs index e6ffb167..dfc5e034 100644 --- a/FreeSql.DbContext/UnitOfWork/IUnitOfWork.cs +++ b/FreeSql.DbContext/UnitOfWork/IUnitOfWork.cs @@ -5,6 +5,9 @@ using System.Data.Common; namespace FreeSql { + /// + /// 工作单元 + /// public interface IUnitOfWork : IDisposable { @@ -40,7 +43,7 @@ namespace FreeSql void Open(); /// - /// 此工作单元内的实体变化跟踪 + /// 工作单元内的实体变化跟踪 /// DbContext.EntityChangeReport EntityChangeReport { get; } } diff --git a/FreeSql.DbContext/UnitOfWork/UnitOfWork.cs b/FreeSql.DbContext/UnitOfWork/UnitOfWork.cs index 4e0aa3de..941cdac8 100644 --- a/FreeSql.DbContext/UnitOfWork/UnitOfWork.cs +++ b/FreeSql.DbContext/UnitOfWork/UnitOfWork.cs @@ -11,10 +11,6 @@ namespace FreeSql { public class UnitOfWork : IUnitOfWork { -#if netcoreapp - public static readonly AsyncLocal Current = new AsyncLocal(); -#endif - static int _seed; /// /// 正在使用中的工作单元(调试) @@ -40,11 +36,7 @@ namespace FreeSql if (_fsql == null) throw new ArgumentNullException(nameof(fsql)); _uowBefore = new Aop.TraceBeforeEventArgs("UnitOfWork", null); - _fsql?.Aop.TraceBeforeHandler?.Invoke(this, _uowBefore); - -#if netcoreapp - Current.Value = this; -#endif + _fsql.Aop.TraceBeforeHandler?.Invoke(this, _uowBefore); } void ReturnObject() @@ -55,9 +47,6 @@ namespace FreeSql _fsql.Ado.MasterPool.Return(_conn); _tran = null; _conn = null; -#if netcoreapp - Current.Value = null; -#endif EntityChangeReport?.Report.Clear(); } @@ -174,7 +163,6 @@ namespace FreeSql try { this.Rollback(); - this.Close(); } finally {