forked from samus/mongodb-csharp
-
Notifications
You must be signed in to change notification settings - Fork 1
/
AliasesProduced.cs
37 lines (32 loc) · 971 Bytes
/
AliasesProduced.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
using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using System.Text;
using MongoDB.Linq.Expressions;
namespace MongoDB.Linq.Translators
{
/// <summary>
/// returns the set of all aliases produced by a query source
/// </summary>
internal class AliasesProduced : MongoExpressionVisitor
{
private HashSet<Alias> _aliases;
public HashSet<Alias> Gather(Expression source)
{
_aliases = new HashSet<Alias>();
Visit(source);
return _aliases;
}
protected override Expression VisitSelect(SelectExpression select)
{
_aliases.Add(select.Alias);
return select;
}
protected override Expression VisitCollection(CollectionExpression collection)
{
_aliases.Add(collection.Alias);
return collection;
}
}
}