io.gitee.ludii.excel.converts.string.StringNumberReadConverter Maven / Gradle / Ivy

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

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 java.math.BigDecimal;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * String and number converter
 *
 * @author Jiaju Zhuang
 */
public class StringNumberReadConverter extends ReadAndWriteConverter {

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

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

    @Override
    public String convertToJavaData(ReadCellData readCellData) {
        Date date = readCellData.getDateValue();
        if (date != null) {
            return new SimpleDateFormat("yyyyMMdd HH:mm:ss").format(date);
        } else {
            BigDecimal numberValue = readCellData.getNumberValue();
            if (numberValue == null) {
                return null;
            } else {
                return numberValue.toString();
            }
        }
    }

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