graphql.language.OperationDefinition Maven / Gradle / Ivy
package graphql.language;
import com.google.common.collect.ImmutableList;
import graphql.Internal;
import graphql.PublicApi;
import graphql.collect.ImmutableKit;
import graphql.util.TraversalControl;
import graphql.util.TraverserContext;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.function.Consumer;
import static graphql.Assert.assertNotNull;
import static graphql.collect.ImmutableKit.emptyList;
import static graphql.collect.ImmutableKit.emptyMap;
import static graphql.language.NodeChildrenContainer.newNodeChildrenContainer;
@PublicApi
public class OperationDefinition extends AbstractNode implements Definition, SelectionSetContainer, DirectivesContainer, NamedNode {
public enum Operation {
QUERY, MUTATION, SUBSCRIPTION
}
private final String name;
private final Operation operation;
private final ImmutableList variableDefinitions;
private final ImmutableList directives;
private final SelectionSet selectionSet;
public static final String CHILD_VARIABLE_DEFINITIONS = "variableDefinitions";
public static final String CHILD_DIRECTIVES = "directives";
public static final String CHILD_SELECTION_SET = "selectionSet";
@Internal
protected OperationDefinition(String name,
Operation operation,
List variableDefinitions,
List directives,
SelectionSet selectionSet,
SourceLocation sourceLocation,
List comments,
IgnoredChars ignoredChars,
Map additionalData) {
super(sourceLocation, comments, ignoredChars, additionalData);
this.name = name;
this.operation = operation;
this.variableDefinitions = ImmutableList.copyOf(variableDefinitions);
this.directives = ImmutableList.copyOf(directives);
this.selectionSet = selectionSet;
}
public OperationDefinition(String name,
Operation operation) {
this(name, operation, emptyList(), emptyList(), null, null, emptyList(), IgnoredChars.EMPTY, emptyMap());
}
public OperationDefinition(String name) {
this(name, null, emptyList(), emptyList(), null, null, emptyList(), IgnoredChars.EMPTY, emptyMap());
}
@Override
public List getChildren() {
List result = new ArrayList<>();
result.addAll(variableDefinitions);
result.addAll(directives);
result.add(selectionSet);
return result;
}
@Override
public NodeChildrenContainer getNamedChildren() {
return newNodeChildrenContainer()
.children(CHILD_VARIABLE_DEFINITIONS, variableDefinitions)
.children(CHILD_DIRECTIVES, directives)
.child(CHILD_SELECTION_SET, selectionSet)
.build();
}
@Override
public OperationDefinition withNewChildren(NodeChildrenContainer newChildren) {
return transform(builder -> builder
.variableDefinitions(newChildren.getChildren(CHILD_VARIABLE_DEFINITIONS))
.directives(newChildren.getChildren(CHILD_DIRECTIVES))
.selectionSet(newChildren.getChildOrNull(CHILD_SELECTION_SET))
);
}
public String getName() {
return name;
}
public Operation getOperation() {
return operation;
}
public List getVariableDefinitions() {
return variableDefinitions;
}
public List getDirectives() {
return directives;
}
@Override
public SelectionSet getSelectionSet() {
return selectionSet;
}
@Override
public boolean isEqualTo(Node o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
OperationDefinition that = (OperationDefinition) o;
return Objects.equals(this.name, that.name) && operation == that.operation;
}
@Override
public OperationDefinition deepCopy() {
return new OperationDefinition(name,
operation,
deepCopy(variableDefinitions),
deepCopy(directives),
deepCopy(selectionSet),
getSourceLocation(),
getComments(),
getIgnoredChars(),
getAdditionalData());
}
@Override
public String toString() {
return "OperationDefinition{" +
"name='" + name + '\'' +
", operation=" + operation +
", variableDefinitions=" + variableDefinitions +
", directives=" + directives +
", selectionSet=" + selectionSet +
'}';
}
@Override
public TraversalControl accept(TraverserContext context, NodeVisitor visitor) {
return visitor.visitOperationDefinition(this, context);
}
public static Builder newOperationDefinition() {
return new Builder();
}
public OperationDefinition transform(Consumer builderConsumer) {
Builder builder = new Builder(this);
builderConsumer.accept(builder);
return builder.build();
}
public static final class Builder implements NodeDirectivesBuilder {
private SourceLocation sourceLocation;
private ImmutableList comments = emptyList();
private String name;
private Operation operation;
private ImmutableList variableDefinitions = emptyList();
private ImmutableList directives = emptyList();
private SelectionSet selectionSet;
private IgnoredChars ignoredChars = IgnoredChars.EMPTY;
private Map additionalData = new LinkedHashMap<>();
private Builder() {
}
private Builder(OperationDefinition existing) {
this.sourceLocation = existing.getSourceLocation();
this.comments = ImmutableList.copyOf(existing.getComments());
this.name = existing.getName();
this.operation = existing.getOperation();
this.variableDefinitions = ImmutableList.copyOf(existing.getVariableDefinitions());
this.directives = ImmutableList.copyOf(existing.getDirectives());
this.selectionSet = existing.getSelectionSet();
this.ignoredChars = existing.getIgnoredChars();
this.additionalData = new LinkedHashMap<>(existing.getAdditionalData());
}
public Builder sourceLocation(SourceLocation sourceLocation) {
this.sourceLocation = sourceLocation;
return this;
}
public Builder comments(List comments) {
this.comments = ImmutableList.copyOf(comments);
return this;
}
public Builder name(String name) {
this.name = name;
return this;
}
public Builder operation(Operation operation) {
this.operation = operation;
return this;
}
public Builder variableDefinitions(List variableDefinitions) {
this.variableDefinitions = ImmutableList.copyOf(variableDefinitions);
return this;
}
public Builder variableDefinition(VariableDefinition variableDefinition) {
this.variableDefinitions = ImmutableKit.addToList(variableDefinitions, variableDefinition);
return this;
}
@Override
public Builder directives(List directives) {
this.directives = ImmutableList.copyOf(directives);
return this;
}
public Builder directive(Directive directive) {
this.directives = ImmutableKit.addToList(directives, directive);
return this;
}
public Builder selectionSet(SelectionSet selectionSet) {
this.selectionSet = selectionSet;
return this;
}
public Builder ignoredChars(IgnoredChars ignoredChars) {
this.ignoredChars = ignoredChars;
return this;
}
public Builder additionalData(Map additionalData) {
this.additionalData = assertNotNull(additionalData);
return this;
}
public Builder additionalData(String key, String value) {
this.additionalData.put(key, value);
return this;
}
public OperationDefinition build() {
return new OperationDefinition(
name,
operation,
variableDefinitions,
directives,
selectionSet,
sourceLocation,
comments,
ignoredChars,
additionalData);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy