package cn.iocoder.yudao.framework.excel.core.convert;
|
|
import cn.hutool.core.util.StrUtil;
|
import cn.idev.excel.converters.Converter;
|
import cn.idev.excel.enums.CellDataTypeEnum;
|
import cn.idev.excel.metadata.GlobalConfiguration;
|
import cn.idev.excel.metadata.data.ReadCellData;
|
import cn.idev.excel.metadata.data.WriteCellData;
|
import cn.idev.excel.metadata.property.ExcelContentProperty;
|
|
import java.math.BigDecimal;
|
import java.time.LocalTime;
|
import java.time.format.DateTimeFormatter;
|
|
/**
|
* LocalTime 转换器
|
*
|
* FastExcel（cn.idev.excel）内置不支持 LocalTime，未指定 converter 时读写都会抛出
|
* Can not find Converter support class LocalTime。这里补上「HH:mm:ss 字符串」与
|
* 「Excel 时间数值（一天的占比）」两种写法的读写支持。
|
*
|
* @author 超级管理员
|
*/
|
public class LocalTimeConvert implements Converter<LocalTime> {
|
|
private static final DateTimeFormatter TIME_FORMATTER = DateTimeFormatter.ofPattern("HH:mm:ss");
|
|
@Override
|
public Class<LocalTime> supportJavaTypeKey() {
|
return LocalTime.class;
|
}
|
|
@Override
|
public CellDataTypeEnum supportExcelTypeKey() {
|
return CellDataTypeEnum.STRING;
|
}
|
|
@Override
|
public LocalTime convertToJavaData(ReadCellData readCellData, ExcelContentProperty contentProperty,
|
GlobalConfiguration globalConfiguration) {
|
// 情况一：Excel 里填的是时间数值（一天的占比）
|
if (CellDataTypeEnum.NUMBER.equals(readCellData.getType())) {
|
BigDecimal numberValue = readCellData.getNumberValue();
|
if (numberValue == null) {
|
return null;
|
}
|
long secondOfDay = Math.round(numberValue.doubleValue() * 24 * 60 * 60) % (24 * 60 * 60);
|
return LocalTime.ofSecondOfDay(secondOfDay);
|
}
|
// 情况二：Excel 里填的是文本，如 8:30 / 08:30:00
|
String text = readCellData.getStringValue();
|
if (StrUtil.isBlank(text)) {
|
return null;
|
}
|
try {
|
String[] parts = text.trim().replace('：', ':').split(":");
|
int hour = Integer.parseInt(parts[0].trim());
|
int minute = parts.length > 1 ? Integer.parseInt(parts[1].trim()) : 0;
|
int second = parts.length > 2 ? Integer.parseInt(parts[2].trim()) : 0;
|
return LocalTime.of(hour, minute, second);
|
} catch (NumberFormatException | ArrayIndexOutOfBoundsException e) {
|
return null;
|
}
|
}
|
|
@Override
|
public WriteCellData<String> convertToExcelData(LocalTime value, ExcelContentProperty contentProperty,
|
GlobalConfiguration globalConfiguration) {
|
return new WriteCellData<>(value == null ? "" : value.format(TIME_FORMATTER));
|
}
|
|
}
|
