@@ -13,15 +13,17 @@ public class IncludeVisitor : ChainableMutatingQueryVisitor
13
13
{
14
14
private readonly LuceneQueryParser _parser = new ( ) ;
15
15
private readonly ShouldSkipIncludeFunc _shouldSkipInclude ;
16
+ private readonly string _includeName ;
16
17
17
- public IncludeVisitor ( ShouldSkipIncludeFunc shouldSkipInclude = null )
18
+ public IncludeVisitor ( ShouldSkipIncludeFunc shouldSkipInclude = null , string includeName = "include" )
18
19
{
19
20
_shouldSkipInclude = shouldSkipInclude ;
21
+ _includeName = includeName ;
20
22
}
21
23
22
24
public override async Task < IQueryNode > VisitAsync ( TermNode node , IQueryVisitorContext context )
23
25
{
24
- if ( node . Field != "@include" || ( _shouldSkipInclude != null && _shouldSkipInclude ( node , context ) ) )
26
+ if ( node . Field != "@" + _includeName || ( _shouldSkipInclude != null && _shouldSkipInclude ( node , context ) ) )
25
27
return node ;
26
28
27
29
var includeResolver = context . GetIncludeResolver ( ) ;
@@ -34,7 +36,7 @@ public override async Task<IQueryNode> VisitAsync(TermNode node, IQueryVisitorCo
34
36
var includeStack = context . GetIncludeStack ( ) ;
35
37
if ( includeStack . Contains ( node . Term ) )
36
38
{
37
- context . AddValidationError ( $ "Recursive include ({ node . Term } )") ;
39
+ context . AddValidationError ( $ "Recursive { _includeName } ({ node . Term } )") ;
38
40
return node ;
39
41
}
40
42
@@ -43,7 +45,7 @@ public override async Task<IQueryNode> VisitAsync(TermNode node, IQueryVisitorCo
43
45
string includedQuery = await includeResolver ( node . Term ) . ConfigureAwait ( false ) ;
44
46
if ( includedQuery == null )
45
47
{
46
- context . AddValidationError ( $ "Unresolved include ({ node . Term } )") ;
48
+ context . AddValidationError ( $ "Unresolved { _includeName } ({ node . Term } )") ;
47
49
context . GetValidationResult ( ) . UnresolvedIncludes . Add ( node . Term ) ;
48
50
}
49
51
@@ -62,7 +64,7 @@ public override async Task<IQueryNode> VisitAsync(TermNode node, IQueryVisitorCo
62
64
}
63
65
catch ( Exception ex )
64
66
{
65
- context . AddValidationError ( $ "Error in include resolver callback when resolving include ({ node . Term } ): { ex . Message } ") ;
67
+ context . AddValidationError ( $ "Error in { _includeName } resolver callback when resolving { _includeName } ({ node . Term } ): { ex . Message } ") ;
66
68
context . GetValidationResult ( ) . UnresolvedIncludes . Add ( node . Term ) ;
67
69
68
70
return node ;
0 commit comments