-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathAnonymousColumnRemover.cs
78 lines (65 loc) · 2.25 KB
/
AnonymousColumnRemover.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Reflection;
using System.Text;
namespace ShiftSoftware.ShiftGrid.Core
{
internal class AnonymousColumnRemover<T> : ExpressionVisitor
{
private IQueryable Query;
private List<string> ColumnsToRemove;
public AnonymousColumnRemover(IQueryable<T> query)
{
this.Query = query;
}
public IQueryable<T> RemoveColumns(List<string> columnsToRemove)
{
this.ColumnsToRemove = columnsToRemove;
return Query.Provider.CreateQuery<T>(this.Visit(Query.Expression));
}
public override Expression Visit(Expression node)
{
//System.Diagnostics.Debug.WriteLine(node + "\t" + node?.NodeType);
return base.Visit(node);
}
protected override Expression VisitNew(NewExpression node)
{
var index = 0;
var args = new List<Expression>();
if (node == null || node.Members == null)
return base.VisitNew(node);
foreach (var member in node.Members)
{
if (this.ColumnsToRemove.Any(x => x.Equals(member.Name, StringComparison.InvariantCultureIgnoreCase)))
{
args.Add(GetDefaultExpressionFor(member));
}
else
args.Add(node.Arguments[index]);
index++;
}
var theNewExpression = Expression.New(
node.Constructor, args, node.Members
);
return base.VisitNew(theNewExpression);
}
public static Expression GetDefaultExpressionFor(MemberInfo member)
{
Expression expression = null;
if (member.MemberType == MemberTypes.Property)
{
return GetDefaultExpressionFor(((PropertyInfo)member).PropertyType);
}
return expression;
}
public static Expression GetDefaultExpressionFor(Type type)
{
if (type.IsValueType)
return MemberExpression.New(type);
else
return MemberExpression.Constant(null, type);
}
}
}