graphql.schema.GraphQLDirective Maven / Gradle / Ivy
package graphql.schema;
import graphql.PublicApi;
import graphql.language.DirectiveDefinition;
import graphql.util.TraversalControl;
import graphql.util.TraverserContext;
import java.util.ArrayList;
import java.util.Collections;
import java.util.EnumSet;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Consumer;
import java.util.function.UnaryOperator;
import static graphql.Assert.assertNotNull;
import static graphql.Assert.assertValidName;
import static graphql.introspection.Introspection.DirectiveLocation;
import static graphql.util.FpKit.getByName;
/**
* A directive can be used to modify the behavior of a graphql field or type.
*
* See http://graphql.org/learn/queries/#directives for more details on the concept.
*/
@SuppressWarnings("DeprecatedIsStillUsed") // because the graphql spec still has some of these deprecated fields
@PublicApi
public class GraphQLDirective implements GraphQLNamedSchemaElement {
private final String name;
private final String description;
private final EnumSet locations;
private final List arguments = new ArrayList<>();
private final DirectiveDefinition definition;
public static final String CHILD_ARGUMENTS = "arguments";
/**
* @deprecated Use the Builder
*/
@Deprecated
public GraphQLDirective(String name,
String description,
EnumSet locations,
List arguments) {
this(name, description, locations, arguments, null);
}
private GraphQLDirective(String name,
String description,
EnumSet locations,
List arguments,
DirectiveDefinition definition) {
assertValidName(name);
assertNotNull(arguments, () -> "arguments can't be null");
this.name = name;
this.description = description;
this.locations = locations;
this.arguments.addAll(arguments);
this.definition = definition;
}
@Override
public String getName() {
return name;
}
public List getArguments() {
return new ArrayList<>(arguments);
}
public GraphQLArgument getArgument(String name) {
for (GraphQLArgument argument : arguments) {
if (argument.getName().equals(name)) {
return argument;
}
}
return null;
}
public EnumSet validLocations() {
return locations;
}
public String getDescription() {
return description;
}
public DirectiveDefinition getDefinition() {
return definition;
}
@Override
public String toString() {
return "GraphQLDirective{" +
"name='" + name + '\'' +
", arguments=" + arguments +
", locations=" + locations +
'}';
}
/**
* This helps you transform the current GraphQLDirective into another one by starting a builder with all
* the current values and allows you to transform it how you want.
*
* @param builderConsumer the consumer code that will be given a builder to transform
*
* @return a new field based on calling build on that builder
*/
public GraphQLDirective transform(Consumer builderConsumer) {
Builder builder = newDirective(this);
builderConsumer.accept(builder);
return builder.build();
}
@Override
public TraversalControl accept(TraverserContext context, GraphQLTypeVisitor visitor) {
return visitor.visitGraphQLDirective(this, context);
}
@Override
public List getChildren() {
return new ArrayList<>(arguments);
}
@Override
public SchemaElementChildrenContainer getChildrenWithTypeReferences() {
return SchemaElementChildrenContainer.newSchemaElementChildrenContainer()
.children(CHILD_ARGUMENTS, arguments)
.build();
}
@Override
public GraphQLDirective withNewChildren(SchemaElementChildrenContainer newChildren) {
return transform(builder ->
builder.replaceArguments(newChildren.getChildren(CHILD_ARGUMENTS))
);
}
public static Builder newDirective() {
return new Builder();
}
public static Builder newDirective(GraphQLDirective existing) {
return new Builder(existing);
}
public static class Builder extends GraphqlTypeBuilder {
private EnumSet locations = EnumSet.noneOf(DirectiveLocation.class);
private final Map arguments = new LinkedHashMap<>();
private DirectiveDefinition definition;
public Builder() {
}
@SuppressWarnings("deprecation")
public Builder(GraphQLDirective existing) {
this.name = existing.getName();
this.description = existing.getDescription();
this.locations = existing.validLocations();
this.arguments.putAll(getByName(existing.getArguments(), GraphQLArgument::getName));
}
@Override
public Builder name(String name) {
super.name(name);
return this;
}
@Override
public Builder description(String description) {
super.description(description);
return this;
}
@Override
public Builder comparatorRegistry(GraphqlTypeComparatorRegistry comparatorRegistry) {
super.comparatorRegistry(comparatorRegistry);
return this;
}
public Builder validLocations(DirectiveLocation... validLocations) {
Collections.addAll(locations, validLocations);
return this;
}
public Builder validLocation(DirectiveLocation validLocation) {
locations.add(validLocation);
return this;
}
public Builder clearValidLocations() {
locations = EnumSet.noneOf(DirectiveLocation.class);
return this;
}
public Builder argument(GraphQLArgument argument) {
assertNotNull(argument, () -> "argument must not be null");
arguments.put(argument.getName(), argument);
return this;
}
public Builder replaceArguments(List arguments) {
assertNotNull(arguments, () -> "arguments must not be null");
this.arguments.clear();
for (GraphQLArgument argument : arguments) {
this.arguments.put(argument.getName(), argument);
}
return this;
}
/**
* Take an argument builder in a function definition and apply. Can be used in a jdk8 lambda
* e.g.:
*
* {@code
* argument(a -> a.name("argumentName"))
* }
*
*
* @param builderFunction a supplier for the builder impl
*
* @return this
*/
public Builder argument(UnaryOperator builderFunction) {
GraphQLArgument.Builder builder = GraphQLArgument.newArgument();
builder = builderFunction.apply(builder);
return argument(builder);
}
/**
* Same effect as the argument(GraphQLArgument). Builder.build() is called
* from within
*
* @param builder an un-built/incomplete GraphQLArgument
*
* @return this
*/
public Builder argument(GraphQLArgument.Builder builder) {
return argument(builder.build());
}
/**
* This is used to clear all the arguments in the builder so far.
*
* @return the builder
*/
public Builder clearArguments() {
arguments.clear();
return this;
}
public Builder definition(DirectiveDefinition definition) {
this.definition = definition;
return this;
}
public GraphQLDirective build() {
return new GraphQLDirective(
name,
description,
locations,
sort(arguments, GraphQLDirective.class, GraphQLArgument.class),
definition);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy