org.codhaus.groovy.grails.validation.SizeConstraint.java Source code

Introduction

Here is the source code for org.codhaus.groovy.grails.validation.SizeConstraint.java

Source

/* Copyright 2004-2005 Graeme Rocher
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.codhaus.groovy.grails.validation;

import groovy.lang.IntRange;
import org.codhaus.groovy.grails.validation.ext.ConstrainedPropertyGunn;
import org.springframework.validation.Errors;

import java.lang.reflect.Array;
import java.util.Collection;

/**
 * Validates size of the property, for strings and arrays
 * this is the length, collections the size and numbers the value.
 *
 * @author Graeme Rocher
 * @since 0.4
 */
public class SizeConstraint extends AbstractConstraint {

    private IntRange range;

    /**
     * @return Returns the range.
     */
    public IntRange getRange() {
        return range;
    }

    /* (non-Javadoc)
     * @see org.codehaus.groovy.org.codhaus.groovy.grails.validation.Constraint#supports(java.lang.Class)
     */
    @SuppressWarnings("rawtypes")
    public boolean supports(Class type) {
        return type != null && (String.class.isAssignableFrom(type) || Collection.class.isAssignableFrom(type)
                || type.isArray());
    }

    /* (non-Javadoc)
     * @see org.codehaus.groovy.org.codhaus.groovy.grails.validation.ConstrainedProperty.AbstractConstraint#setParameter(java.lang.Object)
     */
    @Override
    public void setParameter(Object constraintParameter) {
        if (!(constraintParameter instanceof IntRange)) {
            throw new IllegalArgumentException("Parameter for constraint ["
                    + ConstrainedPropertyGunn.SIZE_CONSTRAINT + "] of property [" + constraintPropertyName
                    + "] of class [" + constraintOwningClass + "] must be a of type [groovy.lang.IntRange]");
        }

        range = (IntRange) constraintParameter;
        super.setParameter(constraintParameter);
    }

    public String getName() {
        return ConstrainedPropertyGunn.SIZE_CONSTRAINT;
    }

    @SuppressWarnings("unchecked")
    @Override
    protected void processValidate(Object target, Object propertyValue, Errors errors) {
        Object[] args = { constraintPropertyName, constraintOwningClass, propertyValue, range.getFrom(),
                range.getTo() };

        int size;
        if (propertyValue.getClass().isArray()) {
            size = Array.getLength(propertyValue);
        } else if (propertyValue instanceof Collection<?>) {
            size = ((Collection<?>) propertyValue).size();
        } else { // String
            size = ((String) propertyValue).length();
        }

        if (!range.contains(size)) {
            if (range.getFrom().compareTo(size) == 1) {
                rejectValue(args, errors, target, ConstrainedPropertyGunn.TOOSMALL_SUFFIX);
            } else if (range.getTo().compareTo(size) == -1) {
                rejectValue(args, errors, target, ConstrainedPropertyGunn.TOOBIG_SUFFIX);
            }
        }
    }

    private void rejectValue(Object[] args, Errors errors, Object target, String suffix) {
        rejectValue(target, errors, ConstrainedPropertyGunn.DEFAULT_INVALID_SIZE_MESSAGE_CODE,
                ConstrainedPropertyGunn.SIZE_CONSTRAINT + suffix, args);
    }
}