graphql.validation.constraints.standard.AbstractSizeConstraint Maven / Gradle / Ivy

Go to download
package graphql.validation.constraints.standard;

import graphql.GraphQLError;
import graphql.schema.GraphQLDirective;
import graphql.schema.GraphQLInputType;
import graphql.validation.constraints.AbstractDirectiveConstraint;
import graphql.validation.rules.ValidationEnvironment;
import java.util.Collections;
import java.util.List;

public abstract class AbstractSizeConstraint extends AbstractDirectiveConstraint {
    public AbstractSizeConstraint(String name) {
        super(name);
    }

    @Override
    final protected List runConstraint(ValidationEnvironment validationEnvironment) {
        Object validatedValue = validationEnvironment.getValidatedValue();
        GraphQLInputType argType = validationEnvironment.getValidatedType();

        GraphQLDirective directive = validationEnvironment.getContextObject(GraphQLDirective.class);
        int min = getIntArg(directive, "min");
        int max = getIntArg(directive, "max");

        int size = getStringOrIDOrObjectOrMapLength(argType, validatedValue);

        if (size < min || size > max) {
            return mkError(validationEnvironment, "min", min, "max", max, "size", size);
        }

        return Collections.emptyList();
    }
}