io.gitee.ludii.excel.converts.localdate.LocalDateNumberReadConverter Maven / Gradle / Ivy

Go to download
/*
 * Copyright https://yuque.com/easyexcel All rights reserved.
 */
package io.gitee.ludii.excel.converts.localdate;

import io.gitee.ludii.excel.converts.ReadAndWriteConverter;
import io.gitee.ludii.excel.enums.CellDataTypeEnum;
import io.gitee.ludii.excel.read.metadata.data.ReadCellData;
import io.gitee.ludii.excel.write.meatadata.data.ExcelWriteFormat;
import io.gitee.ludii.excel.write.meatadata.data.WriteCellData;
import org.apache.poi.ss.usermodel.DateUtil;

import java.math.BigDecimal;
import java.time.LocalDate;
import java.time.LocalDateTime;

/**
 * LocalDateTime and number converter
 *
 * @author Jiaju Zhuang
 */
public class LocalDateNumberReadConverter extends ReadAndWriteConverter {

    @Override
    public Class supportJavaTypeKey() {
        return LocalDate.class;
    }

    @Override
    public CellDataTypeEnum supportExcelTypeKey() {
        return CellDataTypeEnum.NUMBER;
    }

    @Override
    public LocalDate convertToJavaData(ReadCellData readCellData) {
        BigDecimal numberValue = readCellData.getNumberValue();
        if (numberValue == null) {
            return null;
        } else {
            LocalDateTime localDateTime = DateUtil.getLocalDateTime(numberValue.doubleValue());
            return localDateTime == null ? null : localDateTime.toLocalDate();
        }
    }

    @Override
    public WriteCellData convertToExcelData(LocalDate value, ExcelWriteFormat excelWriteFormat) {
        if (value == null) {
            return new WriteCellData<>(CellDataTypeEnum.DATE);
        }
        return new WriteCellData<>(value);
    }


    @Override
    protected void setDefaultDataFormatPattern(WriteCellData writeCellData) {
        writeCellData.setDataFormatPattern(DEFAULT_LOCAL_DATE_PATTER);
    }
}