diff --git a/Src/Asp.NetCore2/SqlSugar/SugarUnitOfWork.cs b/Src/Asp.NetCore2/SqlSugar/SugarUnitOfWork.cs index 4cefbdc2b..3ae67e5e0 100644 --- a/Src/Asp.NetCore2/SqlSugar/SugarUnitOfWork.cs +++ b/Src/Asp.NetCore2/SqlSugar/SugarUnitOfWork.cs @@ -7,6 +7,23 @@ using System.Threading.Tasks; namespace SqlSugar { + public interface ISugarUnitOfWork where T : SugarUnitOfWork, new() + { + public ISqlSugarClient Db { get; set; } + T CreateContext(bool isTran); + } + public class SugarUnitOfWork : ISugarUnitOfWork where T : SugarUnitOfWork, new() + { + public SugarUnitOfWork(ISqlSugarClient db) + { + this.Db = db; + } + public ISqlSugarClient Db { get; set; } + public T CreateContext(bool isTran) + { + return Db.CreateContext(isTran); + } + } public class SugarUnitOfWork : IDisposable { public ISqlSugarClient Db { get; internal set; }