graphql.schema.SchemaElementChildrenContainer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of graphql-java Show documentation
GraphqL Java
There is a newer version: 230521-nf-execution
package graphql.schema;

import graphql.PublicApi;

import java.util.ArrayList;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.function.Consumer;

import static graphql.Assert.assertNotNull;
import static graphql.collect.ImmutableKit.emptyList;

@PublicApi
public class SchemaElementChildrenContainer {

    private final Map> children = new LinkedHashMap<>();

    private SchemaElementChildrenContainer(Map> children) {
        this.children.putAll(assertNotNull(children));
    }

    public  List getChildren(String key) {
        return (List) children.getOrDefault(key, emptyList());
    }

    public  T getChildOrNull(String key) {
        List result = children.getOrDefault(key, new ArrayList<>());
        if (result.size() > 1) {
            throw new IllegalStateException("children " + key + " is not a single value");
        }
        return result.size() > 0 ? (T) result.get(0) : null;
    }

    public Map> getChildren() {
        return new LinkedHashMap<>(children);
    }

    public List getChildrenAsList() {
        List result = new ArrayList<>();
        children.values().forEach(result::addAll);
        return result;
    }

    public static Builder newSchemaElementChildrenContainer() {
        return new Builder();
    }

    public static Builder newSchemaElementChildrenContainer(Map> childrenMap) {
        return new Builder().children(childrenMap);
    }

    public static Builder newSchemaElementChildrenContainer(SchemaElementChildrenContainer existing) {
        return new Builder(existing);
    }

    public SchemaElementChildrenContainer transform(Consumer builderConsumer) {
        Builder builder = new Builder(this);
        builderConsumer.accept(builder);
        return builder.build();
    }

    public boolean isEmpty() {
        return this.children.isEmpty();
    }

    public static class Builder {
        private final Map> children = new LinkedHashMap<>();

        private Builder() {

        }

        private Builder(SchemaElementChildrenContainer other) {
            this.children.putAll(other.children);
        }

        public Builder child(String key, GraphQLSchemaElement child) {
            // we allow null here to make the actual nodes easier
            if (child == null) {
                return this;
            }
            children.computeIfAbsent(key, (k) -> new ArrayList<>());
            children.get(key).add(child);
            return this;
        }

        public Builder children(String key, Collection children) {
            this.children.computeIfAbsent(key, (k) -> new ArrayList<>());
            this.children.get(key).addAll(children);
            return this;
        }

        public Builder children(Map> children) {
            this.children.clear();
            this.children.putAll((Map>) children);
            return this;
        }

        public Builder replaceChild(String key, int index, GraphQLSchemaElement newChild) {
            assertNotNull(newChild);
            this.children.get(key).set(index, newChild);
            return this;
        }

        public Builder removeChild(String key, int index) {
            this.children.get(key).remove(index);
            return this;
        }

        public SchemaElementChildrenContainer build() {
            return new SchemaElementChildrenContainer(this.children);

        }
    }
}