mirror of
https://gitee.com/dotnetchina/SqlSugar.git
synced 2025-09-20 02:29:39 +08:00
433 lines
22 KiB
C#
433 lines
22 KiB
C#
using System;
|
||
using System.Collections;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using System.Linq.Expressions;
|
||
using System.Text;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace SqlSugar
|
||
{
|
||
public class NavigatManager<T>
|
||
{
|
||
public SqlSugarProvider Context { get; set; }
|
||
public Func<ISugarQueryable<object>, List<object>> SelectR1 { get; set; }
|
||
public Func<ISugarQueryable<object>, List<object>> SelectR2 { get; set; }
|
||
public Func<ISugarQueryable<object>, List<object>> SelectR3 { get; set; }
|
||
public Func<ISugarQueryable<object>, List<object>> SelectR4 { get; set; }
|
||
public Func<ISugarQueryable<object>, List<object>> SelectR5 { get; set; }
|
||
public Func<ISugarQueryable<object>, List<object>> SelectR6 { get; set; }
|
||
public Func<ISugarQueryable<object>, List<object>> SelectR7 { get; set; }
|
||
public Func<ISugarQueryable<object>, List<object>> SelectR8 { get; set; }
|
||
public Expression[] Expressions { get; set; }
|
||
public List<T> RootList { get; set; }
|
||
//public QueryableProvider<T> Queryable { get; set; }
|
||
|
||
private List<Expression> _preExpressionList = new List<Expression>();
|
||
private List<object> _preList = new List<object>();
|
||
private List<Expression> _ListCallFunc;
|
||
//private Expression[] _expressions;
|
||
//private List<T> _list;
|
||
//private EntityInfo _entityInfo;
|
||
public void Execute()
|
||
{
|
||
var i = 1;
|
||
foreach (var item in Expressions)
|
||
{
|
||
ExecuteByLay(i, item);
|
||
i++;
|
||
}
|
||
}
|
||
|
||
private void ExecuteByLay(int i, Expression item)
|
||
{
|
||
_ListCallFunc = GetWhereExpression(ref item);
|
||
if (i == 1)
|
||
{
|
||
ExecuteByLay(item, RootList.Select(it => it as object).ToList(), SelectR1);
|
||
}
|
||
else if (i == 2)
|
||
{
|
||
var currentList = RootList;
|
||
if (RootList == null || currentList.Count == 0) return;
|
||
var memberExpression = ((_preExpressionList.Last() as LambdaExpression).Body as MemberExpression);
|
||
var navObjectName = memberExpression.Member.Name;
|
||
var navType = currentList[0].GetType().GetProperty(navObjectName).PropertyType.Name;
|
||
var isList = navType.StartsWith("List`");
|
||
List<object> list = new List<object>();
|
||
if (isList)
|
||
{
|
||
list = currentList.SelectMany(it => (it.GetType().GetProperty(navObjectName).GetValue(it) as IList).Cast<object>()).ToList();
|
||
|
||
}
|
||
else
|
||
{
|
||
list = currentList.Select(it => (it.GetType().GetProperty(navObjectName).GetValue(it))).ToList();
|
||
}
|
||
ExecuteByLay(item, list, SelectR2);
|
||
_preList = list;
|
||
}
|
||
else if (i == 3)
|
||
{
|
||
var currentList = _preList.Where(it => it != null).ToList();
|
||
if (RootList == null || currentList.Count == 0) return;
|
||
List<object> list = ExecuteByLay(currentList);
|
||
ExecuteByLay(item, list, SelectR3);
|
||
_preList = list.ToList();
|
||
}
|
||
else if (i == 4)
|
||
{
|
||
var currentList = _preList.Where(it => it != null).ToList();
|
||
if (RootList == null || currentList.Count == 0) return;
|
||
List<object> list = ExecuteByLay(currentList);
|
||
ExecuteByLay(item, list, SelectR4);
|
||
_preList = list.ToList();
|
||
}
|
||
else if (i == 5)
|
||
{
|
||
var currentList = _preList.Where(it => it != null).ToList();
|
||
if (RootList == null || currentList.Count == 0) return;
|
||
List<object> list = ExecuteByLay(currentList);
|
||
ExecuteByLay(item, list, SelectR5);
|
||
_preList = list.ToList();
|
||
}
|
||
else if (i == 6)
|
||
{
|
||
var currentList = _preList.Where(it => it != null).ToList();
|
||
if (RootList == null || currentList.Count == 0) return;
|
||
List<object> list = ExecuteByLay(currentList);
|
||
ExecuteByLay(item, list, SelectR6);
|
||
_preList = list.ToList();
|
||
}
|
||
else if (i == 7)
|
||
{
|
||
var currentList = _preList.Where(it => it != null).ToList();
|
||
if (RootList == null || currentList.Count == 0) return;
|
||
List<object> list = ExecuteByLay(currentList);
|
||
ExecuteByLay(item, list, SelectR7);
|
||
_preList = list.ToList();
|
||
}
|
||
_preExpressionList.Add(item);
|
||
_ListCallFunc = new List<Expression>();
|
||
}
|
||
|
||
private List<object> ExecuteByLay(List<object> currentList)
|
||
{
|
||
var memberExpression = ((_preExpressionList.Last() as LambdaExpression).Body as MemberExpression);
|
||
var navObjectName = memberExpression.Member.Name;
|
||
var navType = currentList[0].GetType().GetProperty(navObjectName).PropertyType.Name;
|
||
var isList = navType.StartsWith("List`");
|
||
List<object> list = new List<object>();
|
||
if (isList)
|
||
{
|
||
list = currentList.SelectMany(it => (it.GetType().GetProperty(navObjectName).GetValue(it) as IList).Cast<object>()).ToList();
|
||
|
||
}
|
||
else
|
||
{
|
||
list = currentList.Select(it => (it.GetType().GetProperty(navObjectName).GetValue(it))).ToList();
|
||
}
|
||
|
||
return list;
|
||
}
|
||
|
||
private void ExecuteByLay(Expression expression, List<object> list, Func<ISugarQueryable<object>, List<object>> selector)
|
||
{
|
||
if (list == null || list.Count == 0) return;
|
||
list = list.Where(it => it != null).ToList();
|
||
var memberExpression = ((expression as LambdaExpression).Body as MemberExpression);
|
||
|
||
var listItemType = list.Where(it=>it!=null).FirstOrDefault()?.GetType();
|
||
if (listItemType.Name.StartsWith("List`"))
|
||
{
|
||
listItemType = listItemType.GetGenericArguments()[0];
|
||
}
|
||
if (listItemType == null) return;
|
||
|
||
var listItemEntity = this.Context.EntityMaintenance.GetEntityInfo(listItemType);
|
||
var listPkColumn = listItemEntity.Columns.Where(it => it.IsPrimarykey).FirstOrDefault();
|
||
var navObjectName = memberExpression.Member.Name;
|
||
var navObjectNamePropety = listItemType.GetProperty(navObjectName);
|
||
var navObjectNameColumnInfo = listItemEntity.Columns.First(it => it.PropertyName == navObjectName);
|
||
Check.ExceptionEasy(navObjectNameColumnInfo.Navigat == null, $"{navObjectName} not [Navigat(..)] ", $"{navObjectName} 没有导航特性 [Navigat(..)] ");
|
||
|
||
|
||
|
||
if (navObjectNameColumnInfo.Navigat.NavigatType == NavigateType.OneToOne)
|
||
{
|
||
OneToOne(list, selector, listItemEntity, navObjectNamePropety, navObjectNameColumnInfo);
|
||
}
|
||
else if (navObjectNameColumnInfo.Navigat.NavigatType == NavigateType.OneToMany)
|
||
{
|
||
OneToMany(list, selector, listItemEntity, navObjectNamePropety, navObjectNameColumnInfo);
|
||
}
|
||
else if (navObjectNameColumnInfo.Navigat.NavigatType == NavigateType.ManyToOne)
|
||
{
|
||
OneToOne(list, selector, listItemEntity, navObjectNamePropety, navObjectNameColumnInfo);
|
||
}
|
||
else
|
||
{
|
||
ManyToMany(list, selector, listItemEntity, navObjectNamePropety, navObjectNameColumnInfo);
|
||
}
|
||
}
|
||
|
||
private List<Expression> GetWhereExpression(ref Expression expression)
|
||
{
|
||
List<Expression> expressions = new List<Expression>();
|
||
var isCall = (expression as LambdaExpression).Body is MethodCallExpression;
|
||
if (isCall)
|
||
{
|
||
var newexp = (expression as LambdaExpression).Body;
|
||
while (newexp is MethodCallExpression)
|
||
{
|
||
expressions.Add(newexp);
|
||
newexp= (newexp as MethodCallExpression).Arguments[0];
|
||
}
|
||
expression =LambdaExpression.Lambda(newexp);
|
||
}
|
||
return expressions;
|
||
}
|
||
|
||
private void ManyToMany(List<object> list, Func<ISugarQueryable<object>, List<object>> selector, EntityInfo listItemEntity, System.Reflection.PropertyInfo navObjectNamePropety, EntityColumnInfo navObjectNameColumnInfo)
|
||
{
|
||
var bEntity = navObjectNameColumnInfo.PropertyInfo.PropertyType.GetGenericArguments()[0];
|
||
var bEntityInfo = this.Context.EntityMaintenance.GetEntityInfo(bEntity);
|
||
var bPkColumn = bEntityInfo.Columns.FirstOrDefault(it => it.IsPrimarykey);
|
||
|
||
var listItemPkColumn = listItemEntity.Columns.Where(it => it.IsPrimarykey).FirstOrDefault();
|
||
var ids = list.Select(it => it.GetType().GetProperty(listItemPkColumn.PropertyName).GetValue(it)).Select(it => it == null ? "null" : it).Distinct().ToList();
|
||
var mappingEntity = this.Context.EntityMaintenance.GetEntityInfo(navObjectNameColumnInfo.Navigat.MappingType);
|
||
var aColumn = mappingEntity.Columns.First(it => it.PropertyName == navObjectNameColumnInfo.Navigat.MappingAId);
|
||
var bColumn = mappingEntity.Columns.First(it => it.PropertyName == navObjectNameColumnInfo.Navigat.MappingBId);
|
||
List<IConditionalModel> conditionalModels = new List<IConditionalModel>();
|
||
conditionalModels.Add((new ConditionalModel()
|
||
{
|
||
ConditionalType = ConditionalType.In,
|
||
FieldName = aColumn.DbColumnName,
|
||
FieldValue = String.Join(",", ids),
|
||
CSharpTypeName = aColumn.PropertyInfo.PropertyType.Name
|
||
}));
|
||
var abids = this.Context.Queryable<object>().AS(mappingEntity.DbTableName).Where(conditionalModels).Select<SugarAbMapping>($"{aColumn.DbColumnName} as aid,{bColumn.DbColumnName} as bid").ToList();
|
||
|
||
List<IConditionalModel> conditionalModels2 = new List<IConditionalModel>();
|
||
conditionalModels2.Add((new ConditionalModel()
|
||
{
|
||
ConditionalType = ConditionalType.In,
|
||
FieldName = bPkColumn.DbColumnName,
|
||
FieldValue = String.Join(",", abids.Select(it => it.Bid).ToArray()),
|
||
CSharpTypeName = bColumn.PropertyInfo.PropertyType.Name
|
||
}));
|
||
var sql = GetWhereSql();
|
||
var bList = selector(this.Context.Queryable<object>().AS(bEntityInfo.DbTableName).AddParameters(sql.Parameters).Where(conditionalModels2).WhereIF(sql.WhereString.HasValue(),sql.WhereString).Select(sql.SelectString).OrderByIF(sql.OrderByString.HasValue(),sql.OrderByString));
|
||
if (bList.HasValue())
|
||
{
|
||
foreach (var listItem in list)
|
||
{
|
||
if (navObjectNamePropety.GetValue(listItem) == null)
|
||
{
|
||
var instance = Activator.CreateInstance(navObjectNamePropety.PropertyType, true);
|
||
var ilist = instance as IList;
|
||
foreach (var bInfo in bList)
|
||
{
|
||
var pk = listItemPkColumn.PropertyInfo.GetValue(listItem).ObjToString();
|
||
var bid = bPkColumn.PropertyInfo.GetValue(bInfo).ObjToString();
|
||
if (abids.Any(x => x.Aid == pk && x.Bid == bid))
|
||
{
|
||
ilist.Add(bInfo);
|
||
}
|
||
}
|
||
navObjectNamePropety.SetValue(listItem, instance);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
private void OneToOne(List<object> list, Func<ISugarQueryable<object>, List<object>> selector, EntityInfo listItemEntity, System.Reflection.PropertyInfo navObjectNamePropety, EntityColumnInfo navObjectNameColumnInfo)
|
||
{
|
||
var navColumn = listItemEntity.Columns.FirstOrDefault(it => it.PropertyName == navObjectNameColumnInfo.Navigat.Name);
|
||
var navType = navObjectNamePropety.PropertyType;
|
||
var navEntityInfo = this.Context.EntityMaintenance.GetEntityInfo(navType);
|
||
var navPkColumn = navEntityInfo.Columns.Where(it => it.IsPrimarykey).FirstOrDefault();
|
||
Check.ExceptionEasy(navPkColumn==null, navEntityInfo.EntityName+ "need primarykey", navEntityInfo.EntityName + " 需要主键");
|
||
var ids = list.Select(it => it.GetType().GetProperty(navObjectNameColumnInfo.Navigat.Name).GetValue(it)).Select(it => it == null ? "null" : it).Distinct().ToList();
|
||
List<IConditionalModel> conditionalModels = new List<IConditionalModel>();
|
||
conditionalModels.Add((new ConditionalModel()
|
||
{
|
||
ConditionalType = ConditionalType.In,
|
||
FieldName = navPkColumn.DbColumnName,
|
||
FieldValue = String.Join(",", ids),
|
||
CSharpTypeName = navObjectNameColumnInfo.PropertyInfo.PropertyType.Name
|
||
}));
|
||
if (list.Any()&&navObjectNamePropety.GetValue(list.First()) == null)
|
||
{
|
||
var navList = selector(this.Context.Queryable<object>().AS(navEntityInfo.DbTableName).Where(conditionalModels));
|
||
foreach (var item in list)
|
||
{
|
||
|
||
var setValue = navList.FirstOrDefault(x => navPkColumn.PropertyInfo.GetValue(x).ObjToString() == navColumn.PropertyInfo.GetValue(item).ObjToString());
|
||
navObjectNamePropety.SetValue(item, setValue);
|
||
|
||
}
|
||
}
|
||
}
|
||
|
||
private void OneToMany(List<object> list, Func<ISugarQueryable<object>, List<object>> selector, EntityInfo listItemEntity, System.Reflection.PropertyInfo navObjectNamePropety, EntityColumnInfo navObjectNameColumnInfo)
|
||
{
|
||
var navEntity = navObjectNameColumnInfo.PropertyInfo.PropertyType.GetGenericArguments()[0];
|
||
var navEntityInfo = this.Context.EntityMaintenance.GetEntityInfo(navEntity);
|
||
var navColumn = navEntityInfo.Columns.FirstOrDefault(it => it.PropertyName == navObjectNameColumnInfo.Navigat.Name);
|
||
Check.ExceptionEasy(navColumn == null, $"{navEntityInfo.EntityName} not found {navObjectNameColumnInfo.Navigat.Name} ", $"实体 {navEntityInfo.EntityName} 未找到导航配置列 {navObjectNameColumnInfo.Navigat.Name} ");
|
||
//var navType = navObjectNamePropety.PropertyType;
|
||
var listItemPkColumn = listItemEntity.Columns.Where(it => it.IsPrimarykey).FirstOrDefault();
|
||
Check.ExceptionEasy(listItemPkColumn == null, listItemEntity.EntityName + " not primary key", listItemEntity.EntityName + "没有主键");
|
||
var ids = list.Select(it => it.GetType().GetProperty(listItemPkColumn.PropertyName).GetValue(it)).Select(it => it == null ? "null" : it).Distinct().ToList();
|
||
List<IConditionalModel> conditionalModels = new List<IConditionalModel>();
|
||
conditionalModels.Add((new ConditionalModel()
|
||
{
|
||
ConditionalType = ConditionalType.In,
|
||
FieldName = navColumn.DbColumnName,
|
||
FieldValue = String.Join(",", ids),
|
||
CSharpTypeName = listItemPkColumn.PropertyInfo.PropertyType.Name
|
||
}));
|
||
var sqlObj = GetWhereSql();
|
||
|
||
if (list.Any() && navObjectNamePropety.GetValue(list.First()) == null)
|
||
{
|
||
var navList = selector(this.Context.Queryable<object>().AS(navEntityInfo.DbTableName).AddParameters(sqlObj.Parameters).Where(conditionalModels).WhereIF(sqlObj.WhereString.HasValue(), sqlObj.WhereString).Select(sqlObj.SelectString).OrderByIF(sqlObj.OrderByString.HasValue(), sqlObj.OrderByString));
|
||
if (navList.HasValue())
|
||
{
|
||
foreach (var item in list)
|
||
{
|
||
var setValue = navList
|
||
.Where(x => navColumn.PropertyInfo.GetValue(x).ObjToString() == listItemPkColumn.PropertyInfo.GetValue(item).ObjToString()).ToList();
|
||
var instance = Activator.CreateInstance(navObjectNamePropety.PropertyType, true);
|
||
var ilist = instance as IList;
|
||
foreach (var value in setValue)
|
||
{
|
||
ilist.Add(value);
|
||
}
|
||
navObjectNamePropety.SetValue(item, instance);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
private SqlInfo GetWhereSql()
|
||
{
|
||
if (_ListCallFunc == null|| _ListCallFunc.Count==0) return new SqlInfo();
|
||
List<string> where = new List<string>();
|
||
List<string> oredrBy = new List<string>();
|
||
_ListCallFunc.Reverse();
|
||
SqlInfo result = new SqlInfo();
|
||
result.Parameters = new List<SugarParameter>();
|
||
var isList = false;
|
||
foreach (var item in _ListCallFunc)
|
||
{
|
||
var method = item as MethodCallExpression;
|
||
var queryable = this.Context.Queryable<object>();
|
||
if (method.Method.Name == "Where")
|
||
{
|
||
var exp = method.Arguments[1];
|
||
where.Add(" " +queryable.QueryBuilder.GetExpressionValue(exp, ResolveExpressType.WhereSingle).GetString());
|
||
}
|
||
else if (method.Method.Name == "WhereIF")
|
||
{
|
||
var isOk = LambdaExpression.Lambda(method.Arguments[1]).Compile().DynamicInvoke();
|
||
if (isOk.ObjToBool())
|
||
{
|
||
var exp = method.Arguments[2];
|
||
where.Add(" " + queryable.QueryBuilder.GetExpressionValue(exp, ResolveExpressType.WhereSingle).GetString());
|
||
}
|
||
}
|
||
else if (method.Method.Name == "OrderBy")
|
||
{
|
||
var exp = method.Arguments[1];
|
||
oredrBy.Add(" " + queryable.QueryBuilder.GetExpressionValue(exp, ResolveExpressType.WhereSingle).GetString());
|
||
}
|
||
else if (method.Method.Name == "Select")
|
||
{
|
||
var exp = method.Arguments[1];
|
||
var newExp = (exp as LambdaExpression).Body;
|
||
var types = exp.Type.GetGenericArguments();
|
||
if (types != null && types.Length > 0)
|
||
{
|
||
var type = types[0];
|
||
var entityInfo = this.Context.EntityMaintenance.GetEntityInfo(type);
|
||
Check.ExceptionEasy(newExp.Type != entityInfo.Type, $" new {newExp.Type .Name}is error ,use Select(it=>new {entityInfo.Type.Name})",$"new {newExp.Type.Name}是错误的，请使用Select(it=>new {entityInfo.Type.Name})");
|
||
if (entityInfo.Columns.Count(x => x.Navigat != null) == 0)
|
||
{
|
||
result.SelectString = (" " + queryable.QueryBuilder.GetExpressionValue(exp, ResolveExpressType.SelectSingle).GetString());
|
||
}
|
||
else
|
||
{
|
||
var pkInfo = entityInfo.Columns.FirstOrDefault(x => x.IsPrimarykey);
|
||
if (pkInfo != null)
|
||
{
|
||
var pkName = pkInfo.DbColumnName;
|
||
AppColumns(result, queryable, pkName);
|
||
}
|
||
foreach (var nav in entityInfo.Columns.Where(x => x.Navigat != null&&x.Navigat.NavigatType==NavigateType.OneToOne))
|
||
{
|
||
var navColumn = entityInfo.Columns.FirstOrDefault(it => it.PropertyName == nav.Navigat.Name);
|
||
if (navColumn != null)
|
||
{
|
||
AppColumns(result, queryable, navColumn.DbColumnName);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
else if (method.Method.Name == "OrderByDescending")
|
||
{
|
||
var exp = method.Arguments[1];
|
||
oredrBy.Add(" " + queryable.QueryBuilder.GetExpressionValue(exp, ResolveExpressType.WhereSingle).GetString() + " DESC");
|
||
}
|
||
else if (method.Method.Name == "ToList")
|
||
{
|
||
isList = true;
|
||
}
|
||
else
|
||
{
|
||
Check.ExceptionEasy($"no support {item}", $"不支持表达式{item} 不支持方法{method.Method.Name}");
|
||
}
|
||
if(queryable.QueryBuilder.Parameters!=null)
|
||
result.Parameters.AddRange(queryable.QueryBuilder.Parameters);
|
||
}
|
||
if (where.Any())
|
||
{
|
||
Check.Exception(isList == false, $"{_ListCallFunc.First()} need is ToList()", $"{_ListCallFunc.First()} 需要ToList");
|
||
result.WhereString= String.Join(" AND ", where);
|
||
}
|
||
if (oredrBy.Any())
|
||
{
|
||
Check.Exception(isList == false, $"{_ListCallFunc.First()} need is ToList()", $"{_ListCallFunc.First()} 需要ToList");
|
||
result.OrderByString = String.Join(" , ", oredrBy);
|
||
}
|
||
if (result.SelectString.HasValue())
|
||
{
|
||
Check.Exception(isList == false, $"{_ListCallFunc.First()} need is ToList()", $"{_ListCallFunc.First()} 需要ToList");
|
||
result.OrderByString = String.Join(" , ", oredrBy);
|
||
}
|
||
return result;
|
||
}
|
||
|
||
private static void AppColumns(SqlInfo result, ISugarQueryable<object> queryable, string columnName)
|
||
{
|
||
var selectPkName = queryable.SqlBuilder.GetTranslationColumnName(columnName);
|
||
if (result.SelectString.HasValue() && !result.SelectString.ToLower().Contains(selectPkName.ToLower()))
|
||
{
|
||
result.SelectString = result.SelectString + "," + selectPkName;
|
||
}
|
||
}
|
||
|
||
public class SqlInfo
|
||
{
|
||
public string WhereString { get; set; }
|
||
public string OrderByString { get; set; }
|
||
public string SelectString { get; set; }
|
||
public List<SugarParameter> Parameters { get; set; }
|
||
}
|
||
|
||
}
|
||
}
|