cherry.foundation.type.format.CustomNumberFormatAnnotationFormatterFactory.java Source code

Introduction

Here is the source code for cherry.foundation.type.format.CustomNumberFormatAnnotationFormatterFactory.java

Source

/*
 * Copyright 2015 agwlvssainokuni
 *
 * 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 cherry.foundation.type.format;

import static java.util.Arrays.asList;
import static org.apache.commons.lang3.StringUtils.isNotEmpty;

import java.math.BigDecimal;
import java.math.BigInteger;
import java.math.RoundingMode;
import java.text.ParseException;
import java.util.HashSet;
import java.util.List;
import java.util.Locale;
import java.util.Set;

import org.springframework.format.AnnotationFormatterFactory;
import org.springframework.format.Parser;
import org.springframework.format.Printer;
import org.springframework.format.number.NumberFormatter;

import cherry.goods.util.NumberUtil;

public class CustomNumberFormatAnnotationFormatterFactory
        implements AnnotationFormatterFactory<CustomNumberFormat> {

    private List<NumberFormatter> numberFormatter;

    private final Set<Class<?>> fieldTypes = new HashSet<Class<?>>(asList(Byte.class, Short.class, Integer.class,
            Long.class, Float.class, Double.class, BigInteger.class, BigDecimal.class));

    public void setNumberFormatter(List<NumberFormatter> numberFormatter) {
        this.numberFormatter = numberFormatter;
    }

    @Override
    public Set<Class<?>> getFieldTypes() {
        return fieldTypes;
    }

    @Override
    public Printer<Number> getPrinter(CustomNumberFormat annotation, Class<?> fieldType) {
        int value = adjust(annotation.value());
        NumberFormatter formatter = (isNotEmpty(annotation.pattern()) ? new NumberFormatter(annotation.pattern())
                : numberFormatter.get(value));
        return formatter;
    }

    @Override
    public Parser<Number> getParser(CustomNumberFormat annotation, Class<?> fieldType) {
        int value = adjust(annotation.value());
        final NumberFormatter formatter = (isNotEmpty(annotation.pattern())
                ? new NumberFormatter(annotation.pattern())
                : numberFormatter.get(value));
        final int scale = (annotation.scale() < 0 ? value : annotation.scale());
        return new Parser<Number>() {
            @Override
            public Number parse(String text, Locale locale) throws ParseException {
                return NumberUtil.setScale(formatter.parse(text, locale), scale, RoundingMode.DOWN);
            }
        };
    }

    private int adjust(int value) {
        if (value < 0) {
            return 0;
        } else if (value >= numberFormatter.size()) {
            return numberFormatter.size() - 1;
        } else {
            return value;
        }
    }

}