0dylan0
diff --git a/‎Blog.Core.Repository/UnitOfWork/UnitOfWork.cs‎
Lines changed: 27 additions & 9 deletions b/‎Blog.Core.Repository/UnitOfWork/UnitOfWork.cs‎
Lines changed: 27 additions & 9 deletions
@@ -8,9 +8,12 @@ public class UnitOfWork : IUnitOfWork
     {
         private readonly ISqlSugarClient _sqlSugarClient;
 
+        private int _tranCount { get; set; }
+
         public UnitOfWork(ISqlSugarClient sqlSugarClient)
         {
             _sqlSugarClient = sqlSugarClient;
+            _tranCount = 0;
         }
 
         /// <summary>
@@ -25,25 +28,40 @@ public SqlSugarScope GetDbClient()
 
         public void BeginTran()
         {
-            GetDbClient().BeginTran();
+            lock (this)
+            {
+                _tranCount++;
+                GetDbClient().BeginTran();
+            }
         }
 
         public void CommitTran()
         {
-            try
+            lock (this)
             {
-                GetDbClient().CommitTran(); //
-            }
-            catch (Exception ex)
-            {
-                Console.WriteLine(ex.Message);
-                GetDbClient().RollbackTran();
+                _tranCount--;
+                if (_tranCount == 0)
+                {
+                    try
+                    {
+                        GetDbClient().CommitTran();
+                    }
+                    catch (Exception ex)
+                    {
+                        Console.WriteLine(ex.Message);
+                        GetDbClient().RollbackTran();
+                    }
+                }
             }
         }
 
         public void RollbackTran()
         {
-            GetDbClient().RollbackTran();
+            lock (this)
+            {
+                _tranCount--;
+                GetDbClient().RollbackTran();
+            }
         }
 
     }
Original file line number	Diff line number	Diff line change
`@@ -8,9 +8,12 @@ public class UnitOfWork : IUnitOfWork`
`8`	`8`	`{`
`9`	`9`	`private readonly ISqlSugarClient _sqlSugarClient;`
`10`	`10`
	`11`	`+ private int _tranCount { get; set; }`
	`12`	`+`
`11`	`13`	`public UnitOfWork(ISqlSugarClient sqlSugarClient)`
`12`	`14`	`{`
`13`	`15`	`_sqlSugarClient = sqlSugarClient;`
	`16`	`+ _tranCount = 0;`
`14`	`17`	`}`
`15`	`18`
`16`	`19`	`/// <summary>`
`@@ -25,25 +28,40 @@ public SqlSugarScope GetDbClient()`
`25`	`28`
`26`	`29`	`public void BeginTran()`
`27`	`30`	`{`
`28`		`- GetDbClient().BeginTran();`
	`31`	`+ lock (this)`
	`32`	`+ {`
	`33`	`+ _tranCount++;`
	`34`	`+ GetDbClient().BeginTran();`
	`35`	`+ }`
`29`	`36`	`}`
`30`	`37`
`31`	`38`	`public void CommitTran()`
`32`	`39`	`{`
`33`		`- try`
	`40`	`+ lock (this)`
`34`	`41`	`{`
`35`		`- GetDbClient().CommitTran(); //`
`36`		`- }`
`37`		`- catch (Exception ex)`
`38`		`- {`
`39`		`- Console.WriteLine(ex.Message);`
`40`		`- GetDbClient().RollbackTran();`
	`42`	`+ _tranCount--;`
	`43`	`+ if (_tranCount == 0)`
	`44`	`+ {`
	`45`	`+ try`
	`46`	`+ {`
	`47`	`+ GetDbClient().CommitTran();`
	`48`	`+ }`
	`49`	`+ catch (Exception ex)`
	`50`	`+ {`
	`51`	`+ Console.WriteLine(ex.Message);`
	`52`	`+ GetDbClient().RollbackTran();`
	`53`	`+ }`
	`54`	`+ }`
`41`	`55`	`}`
`42`	`56`	`}`
`43`	`57`
`44`	`58`	`public void RollbackTran()`
`45`	`59`	`{`
`46`		`- GetDbClient().RollbackTran();`
	`60`	`+ lock (this)`
	`61`	`+ {`
	`62`	`+ _tranCount--;`
	`63`	`+ GetDbClient().RollbackTran();`
	`64`	`+ }`
`47`	`65`	`}`
`48`	`66`
`49`	`67`	`}`