lsm/SqlSugar
1
0
Fork 0
mirror of https://gitee.com/dotnetchina/SqlSugar.git synced 2025-04-24 18:04:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support Shard Same Thread

Browse Source
This commit is contained in:
sunkaixuan 2017-10-27 15:26:01 +08:00
parent 7e6febd8e1
commit 39a8bbc0dc
7 changed files with 152 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

87
Src/Asp.Net/SqlServerTest/Demos/B_SharedConnection.cs Normal file
View File

@ -0,0 +1,87 @@
using OrmTest.Models;
using SqlSugar;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
namespace OrmTest.Demo
{
public class SharedConnection : DemoBase
{
public static void Init()
{
StudentDal studentDal = new StudentDal();
SchoolDal schoolDal = new SchoolDal();
try
{
studentDal.BeginTran();
Console.WriteLine("school Count:"+ schoolDal.GetSchoolCount());//0
studentDal.AddStudent(new Student() { Name = "StudentTest" });
schoolDal.AddSchool(new School() { Name = "SchoolTest" });//1
Console.WriteLine("school Count:" + schoolDal.GetSchoolCount());
throw new Exception("error");
}
catch (Exception ex)
{
studentDal.RollbackTran();
Console.WriteLine("school Count:" + schoolDal.GetSchoolCount());//0
}
}
}
public class StudentDal : BaseDao
{
public void AddStudent(Student sudent)
{
db.Insertable(sudent).ExecuteCommand();
}
}
public class SchoolDal : BaseDao
{
public void AddSchool(School school)
{
db.Insertable(school).ExecuteCommand();
}
public int GetSchoolCount()
{
return db.Queryable<School>().Count();
}
}
public class BaseDao
{
public SqlSugar.SqlSugarClient db { get { return GetInstance(); } }
public void BeginTran()
{
db.Ado.BeginTran();
}
public void CommitTran()
{
db.Ado.CommitTran();
}
public void RollbackTran()
{
db.Ado.RollbackTran();
}
public SqlSugarClient GetInstance()
{
SqlSugarClient db = new SqlSugarClient(
new ConnectionConfig() {
ConnectionString = Config.ConnectionString,
DbType = DbType.SqlServer,
IsAutoCloseConnection = false,
IsShardSameThread= true /*Shard Same Thread*/
});
return db;
}
}
}

1
Src/Asp.Net/SqlServerTest/Program.cs
View File

@ -47,6 +47,7 @@ namespace OrmTest
OrmTest.Demo.CodeFirst.Init();
OrmTest.Demo.Aop.Init();
OrmTest.Demo.MasterSlave.Init();
OrmTest.Demo.SharedConnection.Init();
}
}
}

1
Src/Asp.Net/SqlServerTest/SqlServerTest.csproj
View File

@ -55,6 +55,7 @@
<Compile Include="Demos\5_DbFirst.cs" />
<Compile Include="Demos\4_Delete.cs" />
<Compile Include="Demos\9_Aop.cs" />
<Compile Include="Demos\B_SharedConnection.cs" />
<Compile Include="Demos\DemoBase.cs" />
<Compile Include="Demos\3_Insert.cs" />
<Compile Include="Demos\1_Query.cs" />

2
Src/Asp.Net/SqlSugar/Entities/ConnectionConfig.cs
View File

@ -27,7 +27,7 @@ namespace SqlSugar
/// <summary>
///If true, there is only one connection instance in the same thread within the same connection string
/// </summary>
public bool OneThreadOnlyOneDbConnectionInstance { get; set; }
public bool IsShardSameThread { get; set; }
/// <summary>
/// Configure External Services replace default services,For example, Redis storage
/// </summary>

36
Src/Asp.Net/SqlSugar/Infrastructure/SqlSugarAccessory.cs
View File

@ -10,7 +10,38 @@ namespace SqlSugar
public partial class SqlSugarAccessory
{
#region Properties
public SqlSugarClient Context { get; set; }
public SqlSugarClient Context
{
get
{
var result = _Context; ;
if (CurrentConnectionConfig.IsShardSameThread)
{
if (CallContext.ContextList.Value.IsNullOrEmpty())
{
CallContext.ContextList.Value = new List<SqlSugarClient>();
CallContext.ContextList.Value.Add(_Context);
}
else
{
var cacheContext = CallContext.ContextList.Value.FirstOrDefault(it =>
it.CurrentConnectionConfig.ConnectionString == _Context.CurrentConnectionConfig.ConnectionString &&
it.CurrentConnectionConfig.DbType == _Context.CurrentConnectionConfig.DbType &&
it.CurrentConnectionConfig.IsAutoCloseConnection == _Context.CurrentConnectionConfig.IsAutoCloseConnection &&
it.CurrentConnectionConfig.IsShardSameThread == _Context.CurrentConnectionConfig.IsShardSameThread);
if (cacheContext != null)
{
return cacheContext;
}
}
}
return result;
}
set
{
_Context = value;
}
}
public ConnectionConfig CurrentConnectionConfig { get; set; }
public Dictionary<string, object> TempItems { get; set; }
public bool IsSystemTablesConfig { get { return this.CurrentConnectionConfig.InitKeyType == InitKeyType.SystemTable; } }
@ -22,6 +53,7 @@ namespace SqlSugar
#region Fields
protected ISqlBuilder _SqlBuilder;
public SqlSugarClient _Context { get; set; }
protected EntityMaintenance _EntityProvider;
protected IAdo _Ado;
protected ILambdaExpressions _LambdaExpressions;
@ -287,7 +319,7 @@ namespace SqlSugar
else
{
isJoinType = false;
joinValue += joinValue==null?item:(","+item);
joinValue += joinValue == null ? item : ("," + item);
}
if (isLast)
{

51
Src/Asp.Net/SqlSugar/SqlSugarClient.cs
View File

@ -22,6 +22,7 @@ namespace SqlSugar
{
base.Context = this;
base.CurrentConnectionConfig = config;
base.ContextID = Guid.NewGuid();
Check.ArgumentNullException(config, "config is null");
switch (config.DbType)
{
@ -50,14 +51,14 @@ namespace SqlSugar
{
get
{
if (_Ado == null)
if (base.Context._Ado == null)
{
var reval = InstanceFactory.GetAdo(base.CurrentConnectionConfig);
_Ado = reval;
var reval = InstanceFactory.GetAdo(base.Context.CurrentConnectionConfig);
base.Context._Ado = reval;
reval.Context = base.Context;
return reval;
}
return _Ado;
return base.Context._Ado;
}
}
#endregion
@ -77,14 +78,14 @@ namespace SqlSugar
{
get
{
if (base._RewritableMethods == null)
if (base.Context._RewritableMethods == null)
{
base._RewritableMethods = new ContextMethods();
base._RewritableMethods.Context = base.Context;
base.Context._RewritableMethods = new ContextMethods();
base.Context._RewritableMethods.Context = base.Context;
}
return _RewritableMethods;
return base.Context._RewritableMethods;
}
set { base._RewritableMethods = value; }
set { base.Context._RewritableMethods = value; }
}
#endregion
@ -504,13 +505,13 @@ namespace SqlSugar
{
get
{
if (base._DbMaintenance == null)
if (base.Context._DbMaintenance == null)
{
IDbMaintenance maintenance = InstanceFactory.GetDbMaintenance(base.Context.CurrentConnectionConfig);
base._DbMaintenance = maintenance;
base.Context._DbMaintenance = maintenance;
maintenance.Context = base.Context;
}
return base._DbMaintenance;
return base.Context._DbMaintenance;
}
}
#endregion
@ -526,14 +527,14 @@ namespace SqlSugar
{
get
{
if (base._EntityProvider == null)
if (base.Context._EntityProvider == null)
{
base._EntityProvider = new EntityMaintenance();
base._EntityProvider.Context = base.Context;
base.Context._EntityProvider = new EntityMaintenance();
base.Context._EntityProvider.Context = base.Context;
}
return _EntityProvider;
return base.Context._EntityProvider;
}
set { base._EntityProvider = value; }
set { base.Context._EntityProvider = value; }
}
#endregion
@ -542,14 +543,14 @@ namespace SqlSugar
{
get
{
if (base._QueryFilterProvider == null)
if (base.Context._QueryFilterProvider == null)
{
base._QueryFilterProvider = new QueryFilterProvider();
base._QueryFilterProvider.Context = base.Context;
base.Context._QueryFilterProvider = new QueryFilterProvider();
base.Context._QueryFilterProvider.Context = base.Context;
}
return _QueryFilterProvider;
return base.Context._QueryFilterProvider;
}
set { base._QueryFilterProvider = value; }
set { base.Context._QueryFilterProvider = value; }
}
#endregion
@ -558,9 +559,9 @@ namespace SqlSugar
{
get
{
if (_SimpleClient == null)
_SimpleClient = new SimpleClient(base.Context);
return _SimpleClient;
if (base.Context._SimpleClient == null)
base.Context._SimpleClient = new SimpleClient(base.Context);
return base.Context._SimpleClient;
}
}
#endregion

3
Src/Asp.Net/SqlSugar/Utilities/CallContext.cs
View File

@ -1,5 +1,6 @@
using System;
using System.Collections.Generic;
using System.Data;
using System.Data.Common;
using System.Linq;
using System.Text;
@ -9,6 +10,6 @@ namespace SqlSugar
{
internal class CallContext
{
public static ThreadLocal<List<DbConnection>> ContextList = new ThreadLocal<List<DbConnection>>();
public static ThreadLocal<List<SqlSugarClient>> ContextList = new ThreadLocal<List<SqlSugarClient>>();
}
}