io.github.vmzakharov.ecdataframe.dataset.IntFormatter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dataframe-ec Show documentation

A tabular data structure based on the Eclipse Collections framework

There is a newer version: 1.0.0

package io.github.vmzakharov.ecdataframe.dataset;

import java.text.DecimalFormat;
import java.text.ParseException;
import java.util.regex.Pattern;

import org.eclipse.collections.api.block.function.primitive.IntFunction;
import org.eclipse.collections.impl.utility.StringIterate;

import static io.github.vmzakharov.ecdataframe.util.ExceptionFactory.exceptionByKey;

public class IntFormatter
{
    private DecimalFormat decimalFormat;
    private IntFunction intParser;
    private Pattern nonNumericPattern;

    public IntFormatter(String pattern)
    {
        if (StringIterate.isEmpty(pattern))
        {
            this.intParser = Integer::parseInt;
        }
        else if (pattern.length() == 1)
        {
            this.nonNumericPattern = Pattern.compile("[^0-9" + pattern + "]");
            this.intParser = s -> Integer.parseInt(this.stripNonNumericCharacters(s));
        }
        else
        {
            this.decimalFormat = new DecimalFormat(pattern);
            this.intParser = s -> {
                try
                {
                    return this.decimalFormat.parse(s.trim()).intValue();
                }
                catch (ParseException e)
                {
                    throw exceptionByKey("CSV_PARSE_ERR")
                            .with("type", "integer")
                            .with("inputString", s)
                            .get(e);
                }
            };
        }
    }

    public int parseAsInt(String aString)
    {
        if (aString == null || aString.isEmpty())
        {
            return 0;
        }

        return this.intParser.applyAsInt(aString);
    }

    public String format(int intValue)
    {
        return this.decimalFormat == null ? Integer.toString(intValue) : this.decimalFormat.format(intValue);
    }

    private String stripNonNumericCharacters(String s)
    {
        return this.nonNumericPattern.matcher(s).replaceAll("");
    }
}