forked from fanruan/easyexcel
Jiaju Zhuang
5 years ago
committed by
GitHub
27 changed files with 439 additions and 116 deletions
@ -0,0 +1,87 @@
|
||||
package com.alibaba.excel.util; |
||||
|
||||
import java.util.HashMap; |
||||
import java.util.Map; |
||||
|
||||
import com.alibaba.excel.converters.Converter; |
||||
import com.alibaba.excel.converters.ConverterKeyBuild; |
||||
import com.alibaba.excel.enums.CellDataTypeEnum; |
||||
import com.alibaba.excel.exception.ExcelDataConvertException; |
||||
import com.alibaba.excel.metadata.CellData; |
||||
import com.alibaba.excel.metadata.GlobalConfiguration; |
||||
import com.alibaba.excel.metadata.property.ExcelContentProperty; |
||||
import com.alibaba.excel.read.metadata.holder.ReadHolder; |
||||
|
||||
/** |
||||
* Converting objects |
||||
* |
||||
* @author Jiaju Zhuang |
||||
**/ |
||||
public class ConverterUtils { |
||||
|
||||
private ConverterUtils() {} |
||||
|
||||
/** |
||||
* Convert it into a String map |
||||
* |
||||
* @param cellDataMap |
||||
* @param readHolder |
||||
* @return |
||||
*/ |
||||
public static Map<Integer, String> convertToStringMap(Map<Integer, CellData> cellDataMap, ReadHolder readHolder) { |
||||
Map<Integer, String> stringMap = new HashMap<Integer, String>(cellDataMap.size() * 4 / 3 + 1); |
||||
for (Map.Entry<Integer, CellData> entry : cellDataMap.entrySet()) { |
||||
CellData cellData = entry.getValue(); |
||||
if (cellData.getType() == CellDataTypeEnum.EMPTY) { |
||||
stringMap.put(entry.getKey(), null); |
||||
continue; |
||||
} |
||||
Converter converter = |
||||
readHolder.converterMap().get(ConverterKeyBuild.buildKey(String.class, cellData.getType())); |
||||
if (converter == null) { |
||||
throw new ExcelDataConvertException( |
||||
"Converter not found, convert " + cellData.getType() + " to String"); |
||||
} |
||||
try { |
||||
stringMap.put(entry.getKey(), |
||||
(String)(converter.convertToJavaData(cellData, null, readHolder.globalConfiguration()))); |
||||
} catch (Exception e) { |
||||
throw new ExcelDataConvertException("Convert data " + cellData + " to String error ", e); |
||||
} |
||||
} |
||||
return stringMap; |
||||
} |
||||
|
||||
/** |
||||
* Convert it into a Java object |
||||
* |
||||
* @param cellData |
||||
* @param clazz |
||||
* @param contentProperty |
||||
* @param converterMap |
||||
* @param globalConfiguration |
||||
* @return |
||||
*/ |
||||
public static Object convertToJavaObject(CellData cellData, Class clazz, ExcelContentProperty contentProperty, |
||||
Map<String, Converter> converterMap, GlobalConfiguration globalConfiguration) { |
||||
if (clazz == CellData.class) { |
||||
return new CellData(cellData); |
||||
} |
||||
Converter converter = null; |
||||
if (contentProperty != null) { |
||||
converter = contentProperty.getConverter(); |
||||
} |
||||
if (converter == null) { |
||||
converter = converterMap.get(ConverterKeyBuild.buildKey(clazz, cellData.getType())); |
||||
} |
||||
if (converter == null) { |
||||
throw new ExcelDataConvertException( |
||||
"Converter not found, convert " + cellData.getType() + " to " + clazz.getName()); |
||||
} |
||||
try { |
||||
return converter.convertToJavaData(cellData, contentProperty, globalConfiguration); |
||||
} catch (Exception e) { |
||||
throw new ExcelDataConvertException("Convert data " + cellData + " to " + clazz + " error ", e); |
||||
} |
||||
} |
||||
} |
||||
@ -0,0 +1,73 @@
|
||||
package com.alibaba.easyexcel.test.demo.read; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
import org.junit.Assert; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
|
||||
import com.alibaba.excel.context.AnalysisContext; |
||||
import com.alibaba.excel.event.AnalysisEventListener; |
||||
import com.alibaba.fastjson.JSON; |
||||
|
||||
/** |
||||
* 读取头 |
||||
* |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
public class DemoHeadDataListener extends AnalysisEventListener<DemoData> { |
||||
private static final Logger LOGGER = LoggerFactory.getLogger(DemoHeadDataListener.class); |
||||
/** |
||||
* 每隔5条存储数据库，实际使用中可以3000条，然后清理list ，方便内存回收 |
||||
*/ |
||||
private static final int BATCH_COUNT = 5; |
||||
List<DemoData> list = new ArrayList<DemoData>(); |
||||
|
||||
/** |
||||
* 在转换异常 获取其他异常下会调用本接口。抛出异常则停止读取。如果这里不抛出异常则 继续读取下一行。 |
||||
* |
||||
* @param exception |
||||
* @param context |
||||
* @throws Exception |
||||
*/ |
||||
@Override |
||||
public void onException(Exception exception, AnalysisContext context) { |
||||
LOGGER.error("解析失败，但是继续解析下一行", exception); |
||||
} |
||||
|
||||
/** |
||||
* 这里会一行行的返回头 |
||||
* |
||||
* @param headMap |
||||
* @param context |
||||
*/ |
||||
@Override |
||||
public void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) { |
||||
LOGGER.info("解析到一条头数据:{}", JSON.toJSONString(headMap)); |
||||
} |
||||
|
||||
@Override |
||||
public void invoke(DemoData data, AnalysisContext context) { |
||||
LOGGER.info("解析到一条数据:{}", JSON.toJSONString(data)); |
||||
if (list.size() >= BATCH_COUNT) { |
||||
saveData(); |
||||
list.clear(); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
saveData(); |
||||
LOGGER.info("所有数据解析完成！"); |
||||
} |
||||
|
||||
/** |
||||
* 加上存储数据库 |
||||
*/ |
||||
private void saveData() { |
||||
LOGGER.info("{}条数据，开始存储数据库！", list.size()); |
||||
LOGGER.info("存储数据库成功！"); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue