Browse Source

Update Discrimator

master
sunkaixuan 1 year ago
parent
commit
01317cdeb5
  1. 2
      Src/Asp.Net/SqlSugar/Abstract/QueryableProvider/QueryableProvider.cs
  2. 1
      Src/Asp.Net/SqlSugar/Abstract/SugarProvider/SqlSugarProvider.cs
  3. 18
      Src/Asp.Net/SqlSugar/Utilities/UtilMethods.cs

2
Src/Asp.Net/SqlSugar/Abstract/QueryableProvider/QueryableProvider.cs

@ -387,8 +387,10 @@ namespace SqlSugar
{
this.Where(whereString);
}
UtilMethods.AddDiscrimator(type,this);
return this;
}
public virtual ISugarQueryable<T> Mapper(Action<T> mapperAction)
{
this.MapperAction=UtilMethods.IsNullReturnNew(this.MapperAction);

1
Src/Asp.Net/SqlSugar/Abstract/SugarProvider/SqlSugarProvider.cs

@ -127,6 +127,7 @@ namespace SqlSugar
InitMappingInfo<T>();
var result = this.CreateQueryable<T>();
UtilMethods.AddDiscrimator(typeof(T), result);
return result;
}
/// <summary>

18
Src/Asp.Net/SqlSugar/Utilities/UtilMethods.cs

@ -1265,5 +1265,23 @@ namespace SqlSugar
var method = value.GetType().GetMethods().First(it => it.GetParameters().Length == 0 && it.Name == "ToShortDateString");
return method.Invoke(value, new object[] { });
}
internal static void AddDiscrimator<T>(Type type, ISugarQueryable<T> queryable)
{
var entityInfo = queryable.Context?.EntityMaintenance?.GetEntityInfoWithAttr(type);
if (entityInfo!=null&&entityInfo.Discrimator.HasValue())
{
Check.ExceptionEasy(!Regex.IsMatch(entityInfo.Discrimator, @"^(?:\w+:\w+)(?:,\w+:\w+)*$"), "The format should be type:cat for this type, and if there are multiple, it can be FieldName:cat,FieldName2:dog ", "格式错误应该是type:cat这种格式,如果是多个可以FieldName:cat,FieldName2:dog,不要有空格");
var array = entityInfo.Discrimator.Split(',');
foreach (var disItem in array)
{
var name = disItem.Split(':').First();
var value = disItem.Split(':').Last();
queryable.Where(name, "=", value);
}
}
}
}
}

Loading…
Cancel
Save