YearMonthRangeFormatter.java

spring-boot/src/main/java/com/tutego/ch_09/mappers/YearMonthRangeFormatter.java

package com.tutego.ch_09.mappers;

import org.springframework.format.Formatter;

import java.text.ParseException;
import java.time.YearMonth;
import java.time.format.DateTimeParseException;
import java.util.Locale;
import java.util.regex.Pattern;

class YearMonthRangeFormatter implements Formatter<YearMonthRange> {

    @Override
    public String print(YearMonthRange range, Locale locale) {
        return range.getStart() + "~" + range.getEnd();
    }

    @Override
    public YearMonthRange parse(String s, Locale locale) throws ParseException {
        try {
            var startEnd = Pattern.compile("~").splitAsStream(s)
                    .map(YearMonth::parse)
                    .toArray(YearMonth[]::new);

            var yearMonthRange = new YearMonthRange();
            yearMonthRange.setStart(startEnd[0]);
            yearMonthRange.setEnd(startEnd[1]);

            return yearMonthRange;
        } catch (DateTimeParseException e) {
            throw new ParseException(e.getMessage(), e.getErrorIndex());
        } catch (ArrayIndexOutOfBoundsException e) {
            throw new ParseException("Insufficient number of values", 0);
        }
    }
}