Autotest
/
sharding
mirror of https://gitee.com/dotnetchina/sharding-core.git
1
0
Fork 0
Code Issues Releases Wiki Activity

优化动态分库的demo

This commit is contained in:
xuejiaming 2022-11-24 14:01:39 +08:00
parent cce5875191
commit bee07e9ec2
2 changed files with 42 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
samples/Sample.AutoCreateIfPresent/AreaDeviceRoute.cs
View File

@ -19,19 +19,21 @@ namespace Sample.AutoCreateIfPresent
private const string CurrentTableName = nameof(AreaDevice);
private readonly ConcurrentDictionary<string, object?> _tails = new ConcurrentDictionary<string, object?>(StringComparer.OrdinalIgnoreCase);
private readonly ConcurrentDictionary<string, object?> _tails =
new ConcurrentDictionary<string, object?>(StringComparer.OrdinalIgnoreCase);
private readonly object _lock = new object();
private readonly object _initLock = new object();
private bool _inited = false;
public AreaDeviceRoute(IVirtualDataSource virtualDataSource, IShardingTableCreator tableCreator)
{
_virtualDataSource = virtualDataSource;
_tableCreator = tableCreator;
InitTails();
}
private void InitTails()
{
using (var connection = new MySqlConnection(_virtualDataSource.DefaultConnectionString))
{
connection.Open();
@ -45,10 +47,10 @@ namespace Sample.AutoCreateIfPresent
if (database.Equals($"{schema}", StringComparison.OrdinalIgnoreCase))
{
var tableName = dataTable.Rows[i][TABLE_NAME]?.ToString() ?? string.Empty;
if (tableName.StartsWith(CurrentTableName,StringComparison.OrdinalIgnoreCase))
if (tableName.StartsWith(CurrentTableName, StringComparison.OrdinalIgnoreCase))
{
//如果没有下划线那么需要CurrentTableName.Length有下划线就要CurrentTableName.Length+1
_tails.TryAdd(tableName.Substring(CurrentTableName.Length+1), null);
_tails.TryAdd(tableName.Substring(CurrentTableName.Length + 1), null);
}
}
}
@ -61,6 +63,7 @@ namespace Sample.AutoCreateIfPresent
{
return $"{shardingKey}";
}
/// <summary>
/// 如果你是非mysql数据库请自行实现这个方法返回当前类在数据库已经存在的后缀
/// 仅启动时调用
@ -68,6 +71,18 @@ namespace Sample.AutoCreateIfPresent
/// <returns></returns>
public override List<string> GetTails()
{
if (!_inited)
{
lock(_initLock)
{
if (!_inited)
{
InitTails();
_inited = true;
}
}
}
return _tails.Keys.ToList();
}
@ -101,7 +116,6 @@ namespace Sample.AutoCreateIfPresent
{
if (!_tails.TryGetValue(shardingKeyToTail, out var _))
{
try
{
_tableCreator.CreateTable<AreaDevice>(_virtualDataSource.DefaultDataSourceName,

14
samples/Sample.AutoCreateIfPresent/OrderByHourRoute.cs
View File

@ -27,12 +27,13 @@ namespace Sample.AutoCreateIfPresent
private readonly IShardingTableCreator _shardingTableCreator;
private readonly ConcurrentDictionary<string, object?> _tails = new ConcurrentDictionary<string, object?>(StringComparer.OrdinalIgnoreCase);
private readonly object _lock = new object();
private readonly object _initLock = new object();
private bool _inited = false;
public OrderByHourRoute(IVirtualDataSource virtualDataSource, IShardingTableCreator shardingTableCreator)
{
_virtualDataSource = virtualDataSource;
_shardingTableCreator = shardingTableCreator;
InitTails();
}
private void InitTails()
@ -82,6 +83,17 @@ namespace Sample.AutoCreateIfPresent
/// <returns></returns>
public override List<string> GetTails()
{
if (!_inited)
{
lock(_initLock)
{
if (!_inited)
{
InitTails();
_inited = true;
}
}
}
return _tails.Keys.ToList();
}