From de12dc25e4ba09850e181bdec998724107098425 Mon Sep 17 00:00:00 2001
From: sunkaixuan <610262374@qq.com>
Date: Tue, 24 May 2022 09:59:02 +0800
Subject: [PATCH] Add ISugarUnitOfWork<T>

---
 Src/Asp.NetCore2/SqlSugar/SugarUnitOfWork.cs | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

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<T> where T : SugarUnitOfWork, new()
+    {
+        public ISqlSugarClient Db { get; set; }
+        T CreateContext(bool isTran);
+    }
+    public class SugarUnitOfWork<T> : ISugarUnitOfWork<T> where T : SugarUnitOfWork, new()
+    {
+        public SugarUnitOfWork(ISqlSugarClient db)
+        {
+            this.Db = db;
+        }
+        public ISqlSugarClient Db { get; set; }
+        public T CreateContext(bool isTran) 
+        {
+            return Db.CreateContext<T>(isTran);
+        }
+    }
     public class SugarUnitOfWork : IDisposable
     {
         public ISqlSugarClient Db { get; internal set; }