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