Update UtilMethod

This commit is contained in:
sunkaixuan
2023-05-28 21:49:41 +08:00
parent 0fcefc38ad
commit db038b6f1b

View File

@@ -19,7 +19,15 @@ namespace SqlSugar
{ {
internal static void EndCustomSplitTable(ISqlSugarClient context,Type entityType) internal static void EndCustomSplitTable(ISqlSugarClient context,Type entityType)
{ {
if (context == null || entityType == null)
{
return;
}
var splitTableAttribute = entityType.GetCustomAttribute<SplitTableAttribute>(); var splitTableAttribute = entityType.GetCustomAttribute<SplitTableAttribute>();
if (splitTableAttribute == null)
{
return;
}
if (splitTableAttribute.CustomSplitTableService != null) if (splitTableAttribute.CustomSplitTableService != null)
{ {
context.CurrentConnectionConfig.ConfigureExternalServices.SplitTableService = null; context.CurrentConnectionConfig.ConfigureExternalServices.SplitTableService = null;
@@ -28,6 +36,15 @@ namespace SqlSugar
internal static void StartCustomSplitTable(ISqlSugarClient context, Type entityType) internal static void StartCustomSplitTable(ISqlSugarClient context, Type entityType)
{ {
if (context == null || entityType == null)
{
return;
}
var splitTableAttribute = entityType.GetCustomAttribute<SplitTableAttribute>();
if (splitTableAttribute == null)
{
return;
}
var splitTableAttribute = entityType.GetCustomAttribute<SplitTableAttribute>(); var splitTableAttribute = entityType.GetCustomAttribute<SplitTableAttribute>();
if (splitTableAttribute.CustomSplitTableService != null) if (splitTableAttribute.CustomSplitTableService != null)
{ {