lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2026-02-27 16:50:33 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update db.GetConnectionScope

Browse Source
This commit is contained in:
sunkaixuan
2022-04-27 18:10:13 +08:00
parent 436855a8d9
commit 48b608b292
1 changed files with 46 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
Src/Asp.Net/SqlSugar/Abstract/SugarProvider/SqlSugarScopeProvider.cs
View File

@@ -18,31 +18,58 @@ namespace SqlSugar
{ {
this.conn = conn; this.conn = conn;
var key = GetKey(); var key = GetKey();
CallContextAsync<SqlSugarProvider>.SetData(key, conn); this.GetContext(true);
} }
public SqlSugarProvider ScopedContext { get { return GetContext(); } } public SqlSugarProvider ScopedContext { get { return GetContext(); } }
private SqlSugarProvider GetAsyncContext(string key) private SqlSugarProvider GetAsyncContext(bool isInit=false)
{ {
SqlSugarProvider result = CallContextAsync<SqlSugarProvider>.GetData(key); if (isInit)
if (result == null)
{ {
CallContextAsync<SqlSugarProvider>.SetData(key, new SqlSugarProvider(conn.CurrentConnectionConfig)); CallContextAsync<SqlSugarProvider>.SetData(GetKey(), this.conn);
result = CallContextAsync<SqlSugarProvider>.GetData(key); isInit = false;
return conn;
} }
else
{
SqlSugarProvider result = CallContextAsync<SqlSugarProvider>.GetData(GetKey());
if (result == null)
{
SqlSugarProvider db=new SqlSugarProvider(this.conn.CurrentConnectionConfig);
CallContextAsync<SqlSugarProvider>.SetData(GetKey(), db);
return db;
}
else
{
return result; return result;
} }
private SqlSugarProvider GetThreadContext(string key)
{
SqlSugarProvider result = CallContextThread<SqlSugarProvider>.GetData(key);
if (result == null)
{
CallContextThread<SqlSugarProvider>.SetData(key, new SqlSugarProvider (this.CurrentConnectionConfig));
result = CallContextThread<SqlSugarProvider>.GetData(key);
} }
return result;
} }
private SqlSugarProvider GetContext() private SqlSugarProvider GetThreadContext(bool isInit = false)
{
if (isInit)
{
CallContextThread<SqlSugarProvider>.SetData(GetKey(), this.conn);
isInit = false;
return conn;
}
else
{
SqlSugarProvider result = CallContextThread<SqlSugarProvider>.GetData(GetKey());
if (result == null)
{
SqlSugarProvider db = new SqlSugarProvider(this.conn.CurrentConnectionConfig);
CallContextThread<SqlSugarProvider>.SetData(GetKey(), db);
return db;
}
else
{
return result;
}
}
}
private SqlSugarProvider GetContext(bool isInit = false)
{ {
SqlSugarProvider result = null; SqlSugarProvider result = null;
var key = GetKey(); ; var key = GetKey(); ;
@@ -51,11 +78,11 @@ namespace SqlSugar
var isAsync = UtilMethods.IsAnyAsyncMethod(methods); var isAsync = UtilMethods.IsAnyAsyncMethod(methods);
if (isAsync) if (isAsync)
{ {
result= GetAsyncContext(key); result= GetAsyncContext(isInit);
} }
else else
{ {
result= GetThreadContext(key); result= GetThreadContext(isInit);
} }
return result; return result;
} }