11package graphql .execution ;
22
33import graphql .language .Directive ;
4+ import graphql .language .NodeUtil ;
45
56import java .util .List ;
67import java .util .Map ;
78
89import static graphql .Directives .IncludeDirective ;
910import static graphql .Directives .SkipDirective ;
11+ import static graphql .language .NodeUtil .directivesByName ;
1012
1113
1214public class ConditionalNodes {
@@ -19,14 +21,14 @@ public ConditionalNodes() {
1921
2022 public boolean shouldInclude (Map <String , Object > variables , List <Directive > directives ) {
2123
22- Directive skipDirective = findDirective (directives , SkipDirective .getName ());
24+ Directive skipDirective = getDirectiveByName (directives , SkipDirective .getName ());
2325 if (skipDirective != null ) {
2426 Map <String , Object > argumentValues = valuesResolver .getArgumentValues (SkipDirective .getArguments (), skipDirective .getArguments (), variables );
2527 return !(Boolean ) argumentValues .get ("if" );
2628 }
2729
2830
29- Directive includeDirective = findDirective (directives , IncludeDirective .getName ());
31+ Directive includeDirective = getDirectiveByName (directives , IncludeDirective .getName ());
3032 if (includeDirective != null ) {
3133 Map <String , Object > argumentValues = valuesResolver .getArgumentValues (IncludeDirective .getArguments (), includeDirective .getArguments (), variables );
3234 return (Boolean ) argumentValues .get ("if" );
@@ -35,8 +37,8 @@ public boolean shouldInclude(Map<String, Object> variables, List<Directive> dire
3537 return true ;
3638 }
3739
38- private Directive findDirective (List <Directive > directives , String name ) {
39- return Directive . getDirectivesMap (directives ).get (name );
40+ private Directive getDirectiveByName (List <Directive > directives , String name ) {
41+ return directivesByName (directives ).get (name );
4042 }
4143
4244}
0 commit comments