private final Function<? super GraphQLSchemaElement, ? extends List<GraphQLSchemaElement>> getChildren;

public SchemaTraverser(Function<? super GraphQLSchemaElement, ? extends List<GraphQLSchemaElement>> getChildren) {

this.getChildren = getChildren;

}

public SchemaTraverser() {

this(GraphQLSchemaElement::getChildren);

}

/**

public TraverserResult depthFirstFullSchema(GraphQLTypeVisitor typeVisitor, GraphQLSchema schema) {

return depthFirstFullSchema(Collections.singletonList(typeVisitor), schema, ImmutableKit.emptyMap());

}

/**

public TraverserResult depthFirstFullSchema(List<GraphQLTypeVisitor> typeVisitors, GraphQLSchema schema, Map<Class<?>, Object> rootVars) {

Set<GraphQLSchemaElement> roots = new LinkedHashSet<>();

roots.add(schema.getQueryType());

if (schema.isSupportingMutations()) {

roots.add(schema.getMutationType());

}

if (schema.isSupportingSubscriptions()) {

roots.add(schema.getSubscriptionType());

}

roots.addAll(schema.getAdditionalTypes());

roots.addAll(schema.getDirectives());

roots.addAll(schema.getSchemaDirectives());

roots.add(schema.getIntrospectionSchemaType());

TraverserDelegateListVisitor traverserDelegateListVisitor = new TraverserDelegateListVisitor(typeVisitors);

return initTraverser().rootVars(rootVars).traverse(roots, traverserDelegateListVisitor);

}

public TraverserResult depthFirst(GraphQLTypeVisitor graphQLTypeVisitor, GraphQLSchemaElement root) {

return depthFirst(graphQLTypeVisitor, Collections.singletonList(root));

}

public TraverserResult depthFirst(final GraphQLTypeVisitor graphQLTypeVisitor, Collection<? extends GraphQLSchemaElement> roots) {

return depthFirst(initTraverser(), new TraverserDelegateVisitor(graphQLTypeVisitor), roots);

}

public TraverserResult depthFirst(final Traverser<GraphQLSchemaElement> traverser,

final TraverserDelegateVisitor traverserDelegateVisitor,

Collection<? extends GraphQLSchemaElement> roots) {

return doTraverse(traverser, roots, traverserDelegateVisitor);

}

private Traverser<GraphQLSchemaElement> initTraverser() {

return Traverser.depthFirst(getChildren);

}

private TraverserResult doTraverse(Traverser<GraphQLSchemaElement> traverser,

Collection<? extends GraphQLSchemaElement> roots,

TraverserDelegateVisitor traverserDelegateVisitor) {

return traverser.traverse(roots, traverserDelegateVisitor);

}

private static class TraverserDelegateVisitor implements TraverserVisitor<GraphQLSchemaElement> {

private final GraphQLTypeVisitor delegate;

TraverserDelegateVisitor(GraphQLTypeVisitor delegate) {

this.delegate = delegate;

}

@Override

public TraversalControl enter(TraverserContext<GraphQLSchemaElement> context) {

return context.thisNode().accept(context, delegate);

}

@Override

public TraversalControl leave(TraverserContext<GraphQLSchemaElement> context) {

return CONTINUE;

}

@Override

public TraversalControl backRef(TraverserContext<GraphQLSchemaElement> context) {

return delegate.visitBackRef(context);

}

}

private static class TraverserDelegateListVisitor implements TraverserVisitor<GraphQLSchemaElement> {

private final List<GraphQLTypeVisitor> typeVisitors;

TraverserDelegateListVisitor(List<GraphQLTypeVisitor> typeVisitors) {

this.typeVisitors = typeVisitors;

}

@Override

public TraversalControl enter(TraverserContext<GraphQLSchemaElement> context) {

for (GraphQLTypeVisitor graphQLTypeVisitor : typeVisitors) {

context.thisNode().accept(context, graphQLTypeVisitor);

}

return CONTINUE;

}

@Override

public TraversalControl leave(TraverserContext<GraphQLSchemaElement> context) {

return CONTINUE;

}

@Override

public TraversalControl backRef(TraverserContext<GraphQLSchemaElement> context) {

for (GraphQLTypeVisitor graphQLTypeVisitor : typeVisitors) {

graphQLTypeVisitor.visitBackRef(context);

}

return CONTINUE;

}

}