graphql.ExecutionResultImpl 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;


import com.google.common.collect.ImmutableList;
import graphql.collect.ImmutableKit;

import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import static graphql.collect.ImmutableKit.map;

@Internal
public class ExecutionResultImpl implements ExecutionResult {

    private final List errors;
    private final Object data;
    private final transient Map extensions;
    private final transient boolean dataPresent;

    public ExecutionResultImpl(GraphQLError error) {
        this(false, null, Collections.singletonList(error), null);
    }

    public ExecutionResultImpl(List errors) {
        this(false, null, errors, null);
    }

    public ExecutionResultImpl(Object data, List errors) {
        this(true, data, errors, null);
    }

    public ExecutionResultImpl(Object data, List errors, Map extensions) {
        this(true, data, errors, extensions);
    }

    public ExecutionResultImpl(ExecutionResultImpl other) {
        this(other.dataPresent, other.data, other.errors, other.extensions);
    }

    public > ExecutionResultImpl(Builder builder) {
        this(builder.dataPresent, builder.data, builder.errors, builder.extensions);
    }

    private ExecutionResultImpl(boolean dataPresent, Object data, List errors, Map extensions) {
        this.dataPresent = dataPresent;
        this.data = data;

        if (errors != null) {
            this.errors = ImmutableList.copyOf(errors);
        } else {
            this.errors = ImmutableKit.emptyList();
        }

        this.extensions = extensions;
    }

    public boolean isDataPresent() {
        return dataPresent;
    }

    @Override
    public List getErrors() {
        return errors;
    }

    @Override
    @SuppressWarnings("TypeParameterUnusedInFormals")
    public  T getData() {
        //noinspection unchecked
        return (T) data;
    }

    @Override
    public Map getExtensions() {
        return extensions;
    }

    @Override
    public Map toSpecification() {
        Map result = new LinkedHashMap<>();
        if (errors != null && !errors.isEmpty()) {
            result.put("errors", errorsToSpec(errors));
        }
        if (dataPresent) {
            result.put("data", data);
        }
        if (extensions != null) {
            result.put("extensions", extensions);
        }
        return result;
    }

    private Object errorsToSpec(List errors) {
        return map(errors, GraphQLError::toSpecification);
    }

    @Override
    public String toString() {
        return "ExecutionResultImpl{" +
                "errors=" + errors +
                ", data=" + data +
                ", dataPresent=" + dataPresent +
                ", extensions=" + extensions +
                '}';
    }

    public static > Builder newExecutionResult() {
        return new Builder<>();
    }

    public static class Builder> implements ExecutionResult.Builder {
        private boolean dataPresent;
        private Object data;
        private List errors = new ArrayList<>();
        private Map extensions;

        @Override
        public T from(ExecutionResult executionResult) {
            dataPresent = executionResult.isDataPresent();
            data = executionResult.getData();
            errors = new ArrayList<>(executionResult.getErrors());
            extensions = executionResult.getExtensions();
            return (T) this;
        }

        @Override
        public T data(Object data) {
            dataPresent = true;
            this.data = data;
            return (T) this;
        }

        @Override
        public T errors(List errors) {
            this.errors = errors;
            return (T) this;
        }

        @Override
        public T addErrors(List errors) {
            this.errors.addAll(errors);
            return (T) this;
        }

        @Override
        public T addError(GraphQLError error) {
            this.errors.add(error);
            return (T) this;
        }

        @Override
        public T extensions(Map extensions) {
            this.extensions = extensions;
            return (T) this;
        }

        @Override
        public T addExtension(String key, Object value) {
            this.extensions = (this.extensions == null ? new LinkedHashMap<>() : this.extensions);
            this.extensions.put(key, value);
            return (T) this;
        }

        @Override
        public ExecutionResult build() {
            return new ExecutionResultImpl(dataPresent, data, errors, extensions);
        }
    }
}