forked from github/easyexcel
Jiaju Zhuang
5 years ago
26 changed files with 450 additions and 113 deletions
@ -0,0 +1,42 @@
|
||||
package com.alibaba.excel.analysis.v03.handlers; |
||||
|
||||
import org.apache.poi.hssf.record.IndexRecord; |
||||
import org.apache.poi.hssf.record.Record; |
||||
|
||||
import com.alibaba.excel.analysis.v03.AbstractXlsRecordHandler; |
||||
import com.alibaba.excel.context.AnalysisContext; |
||||
|
||||
/** |
||||
* Record handler |
||||
* |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
public class IndexRecordHandler extends AbstractXlsRecordHandler { |
||||
|
||||
private AnalysisContext context; |
||||
|
||||
public IndexRecordHandler(AnalysisContext context) { |
||||
this.context = context; |
||||
} |
||||
|
||||
@Override |
||||
public boolean support(Record record) { |
||||
return record instanceof IndexRecord; |
||||
} |
||||
|
||||
@Override |
||||
public void init() {} |
||||
|
||||
@Override |
||||
public void processRecord(Record record) { |
||||
if (context.readSheetHolder() == null) { |
||||
return; |
||||
} |
||||
context.readSheetHolder().setApproximateTotalRowNumber(((IndexRecord)record).getLastRowAdd1()); |
||||
} |
||||
|
||||
@Override |
||||
public int getOrder() { |
||||
return 1; |
||||
} |
||||
} |
||||
@ -0,0 +1,150 @@
|
||||
package com.alibaba.excel.util; |
||||
|
||||
import java.lang.ref.SoftReference; |
||||
import java.lang.reflect.Field; |
||||
import java.lang.reflect.Modifier; |
||||
import java.util.ArrayList; |
||||
import java.util.Collections; |
||||
import java.util.HashMap; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
import java.util.TreeMap; |
||||
import java.util.concurrent.ConcurrentHashMap; |
||||
|
||||
import com.alibaba.excel.annotation.ExcelIgnore; |
||||
import com.alibaba.excel.annotation.ExcelIgnoreUnannotated; |
||||
import com.alibaba.excel.annotation.ExcelProperty; |
||||
import com.alibaba.excel.exception.ExcelCommonException; |
||||
import com.alibaba.excel.metadata.BaseRowModel; |
||||
|
||||
/** |
||||
* Class utils |
||||
* |
||||
* @author Jiaju Zhuang |
||||
**/ |
||||
public class ClassUtils { |
||||
private static final Map<Class, SoftReference<FieldCache>> FIELD_CACHE = |
||||
new ConcurrentHashMap<Class, SoftReference<FieldCache>>(); |
||||
|
||||
public static void declaredFields(Class clazz, List<Field> defaultFieldList, Map<Integer, Field> customFiledMap, |
||||
Map<String, Field> ignoreMap, Boolean convertAllFiled) { |
||||
FieldCache fieldCache = getFieldCache(clazz, convertAllFiled); |
||||
if (fieldCache != null) { |
||||
defaultFieldList.addAll(fieldCache.getDefaultFieldList()); |
||||
customFiledMap.putAll(fieldCache.getCustomFiledMap()); |
||||
ignoreMap.putAll(fieldCache.getIgnoreMap()); |
||||
} |
||||
} |
||||
|
||||
public static void declaredFields(Class clazz, List<Field> fieldList, Boolean convertAllFiled) { |
||||
FieldCache fieldCache = getFieldCache(clazz, convertAllFiled); |
||||
if (fieldCache != null) { |
||||
fieldList.addAll(fieldCache.getAllFieldList()); |
||||
} |
||||
} |
||||
|
||||
private static FieldCache getFieldCache(Class clazz, Boolean convertAllFiled) { |
||||
if (clazz == null) { |
||||
return null; |
||||
} |
||||
SoftReference<FieldCache> fieldCacheSoftReference = FIELD_CACHE.get(clazz); |
||||
if (fieldCacheSoftReference != null && fieldCacheSoftReference.get() != null) { |
||||
return fieldCacheSoftReference.get(); |
||||
} |
||||
synchronized (clazz) { |
||||
fieldCacheSoftReference = FIELD_CACHE.get(clazz); |
||||
if (fieldCacheSoftReference != null && fieldCacheSoftReference.get() != null) { |
||||
return fieldCacheSoftReference.get(); |
||||
} |
||||
declaredFields(clazz, convertAllFiled); |
||||
} |
||||
return FIELD_CACHE.get(clazz).get(); |
||||
} |
||||
|
||||
private static void declaredFields(Class clazz, Boolean convertAllFiled) { |
||||
List<Field> tempFieldList = new ArrayList<Field>(); |
||||
Class tempClass = clazz; |
||||
// When the parent class is null, it indicates that the parent class (Object class) has reached the top
|
||||
// level.
|
||||
while (tempClass != null && tempClass != BaseRowModel.class) { |
||||
Collections.addAll(tempFieldList, tempClass.getDeclaredFields()); |
||||
// Get the parent class and give it to yourself
|
||||
tempClass = tempClass.getSuperclass(); |
||||
} |
||||
// Screening of field
|
||||
List<Field> defaultFieldList = new ArrayList<Field>(); |
||||
Map<Integer, Field> customFiledMap = new TreeMap<Integer, Field>(); |
||||
List<Field> allFieldList = new ArrayList<Field>(); |
||||
Map<String, Field> ignoreMap = new HashMap<String, Field>(16); |
||||
|
||||
ExcelIgnoreUnannotated excelIgnoreUnannotated = |
||||
(ExcelIgnoreUnannotated)clazz.getAnnotation(ExcelIgnoreUnannotated.class); |
||||
for (Field field : tempFieldList) { |
||||
ExcelIgnore excelIgnore = field.getAnnotation(ExcelIgnore.class); |
||||
if (excelIgnore != null) { |
||||
ignoreMap.put(field.getName(), field); |
||||
continue; |
||||
} |
||||
ExcelProperty excelProperty = field.getAnnotation(ExcelProperty.class); |
||||
boolean noExcelProperty = excelProperty == null |
||||
&& ((convertAllFiled != null && !convertAllFiled) || excelIgnoreUnannotated != null); |
||||
if (noExcelProperty) { |
||||
ignoreMap.put(field.getName(), field); |
||||
continue; |
||||
} |
||||
boolean isStaticFinalOrTransient = |
||||
(Modifier.isStatic(field.getModifiers()) && Modifier.isFinal(field.getModifiers())) |
||||
|| Modifier.isTransient(field.getModifiers()); |
||||
if (excelProperty == null && isStaticFinalOrTransient) { |
||||
ignoreMap.put(field.getName(), field); |
||||
continue; |
||||
} |
||||
if (excelProperty == null || excelProperty.index() < 0) { |
||||
defaultFieldList.add(field); |
||||
allFieldList.add(field); |
||||
continue; |
||||
} |
||||
if (customFiledMap.containsKey(excelProperty.index())) { |
||||
throw new ExcelCommonException("The index of '" + customFiledMap.get(excelProperty.index()).getName() |
||||
+ "' and '" + field.getName() + "' must be inconsistent"); |
||||
} |
||||
customFiledMap.put(excelProperty.index(), field); |
||||
allFieldList.add(field); |
||||
} |
||||
|
||||
FIELD_CACHE.put(clazz, |
||||
new SoftReference<FieldCache>(new FieldCache(defaultFieldList, customFiledMap, allFieldList, ignoreMap))); |
||||
} |
||||
|
||||
private static class FieldCache { |
||||
private List<Field> defaultFieldList; |
||||
private Map<Integer, Field> customFiledMap; |
||||
private List<Field> allFieldList; |
||||
private Map<String, Field> ignoreMap; |
||||
|
||||
public FieldCache(List<Field> defaultFieldList, Map<Integer, Field> customFiledMap, List<Field> allFieldList, |
||||
Map<String, Field> ignoreMap) { |
||||
this.defaultFieldList = defaultFieldList; |
||||
this.customFiledMap = customFiledMap; |
||||
this.allFieldList = allFieldList; |
||||
this.ignoreMap = ignoreMap; |
||||
} |
||||
|
||||
public List<Field> getDefaultFieldList() { |
||||
return defaultFieldList; |
||||
} |
||||
|
||||
public Map<Integer, Field> getCustomFiledMap() { |
||||
return customFiledMap; |
||||
} |
||||
|
||||
public List<Field> getAllFieldList() { |
||||
return allFieldList; |
||||
} |
||||
|
||||
public Map<String, Field> getIgnoreMap() { |
||||
return ignoreMap; |
||||
} |
||||
|
||||
} |
||||
} |
||||
@ -0,0 +1,110 @@
|
||||
package com.alibaba.easyexcel.test.temp; |
||||
|
||||
import java.io.File; |
||||
import java.util.ArrayList; |
||||
import java.util.HashMap; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
import org.junit.Ignore; |
||||
import org.junit.Test; |
||||
|
||||
import com.alibaba.easyexcel.test.demo.fill.FillData; |
||||
import com.alibaba.easyexcel.test.util.TestFileUtil; |
||||
import com.alibaba.excel.EasyExcel; |
||||
import com.alibaba.excel.ExcelWriter; |
||||
import com.alibaba.excel.write.metadata.WriteSheet; |
||||
import com.alibaba.excel.write.metadata.fill.FillConfig; |
||||
import com.alibaba.excel.write.style.row.SimpleRowHeightStyleStrategy; |
||||
|
||||
/** |
||||
* 写的填充写法 |
||||
* |
||||
* @since 2.1.1 |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
@Ignore |
||||
public class FillTempTest { |
||||
|
||||
/** |
||||
* 复杂的填充 |
||||
* |
||||
* @since 2.1.1 |
||||
*/ |
||||
@Test |
||||
public void complexFill() { |
||||
// 模板注意 用{} 来表示你要用的变量 如果本来就有"{","}" 特殊字符 用"\{","\}"代替
|
||||
// {} 代表普通变量 {.} 代表是list的变量
|
||||
String templateFileName = "D:\\test\\complex.xlsx"; |
||||
|
||||
String fileName = TestFileUtil.getPath() + "complexFill" + System.currentTimeMillis() + ".xlsx"; |
||||
ExcelWriter excelWriter = EasyExcel.write(fileName).withTemplate(templateFileName).build(); |
||||
WriteSheet writeSheet = EasyExcel.writerSheet().build(); |
||||
// 这里注意 入参用了forceNewRow 代表在写入list的时候不管list下面有没有空行 都会创建一行，然后下面的数据往后移动。默认 是false，会直接使用下一行，如果没有则创建。
|
||||
// forceNewRow 如果设置了true,有个缺点 就是他会把所有的数据都放到内存了，所以慎用
|
||||
// 简单的说 如果你的模板有list,且list不是最后一行，下面还有数据需要填充 就必须设置 forceNewRow=true 但是这个就会把所有数据放到内存 会很耗内存
|
||||
// 如果数据量大 list不是最后一行 参照下一个
|
||||
FillConfig fillConfig = FillConfig.builder().forceNewRow(Boolean.TRUE).build(); |
||||
excelWriter.fill(data(), fillConfig, writeSheet); |
||||
excelWriter.fill(data(), fillConfig, writeSheet); |
||||
Map<String, Object> map = new HashMap<String, Object>(); |
||||
map.put("date", "2019年10月9日13:28:28"); |
||||
map.put("total", 1000); |
||||
excelWriter.fill(map, writeSheet); |
||||
excelWriter.finish(); |
||||
} |
||||
|
||||
/** |
||||
* 数据量大的复杂填充 |
||||
* <p> |
||||
* 这里的解决方案是 确保模板list为最后一行，然后再拼接table.还有03版没救，只能刚正面加内存。 |
||||
* |
||||
* @since 2.1.1 |
||||
*/ |
||||
@Test |
||||
public void complexFillWithTable() { |
||||
// 模板注意 用{} 来表示你要用的变量 如果本来就有"{","}" 特殊字符 用"\{","\}"代替
|
||||
// {} 代表普通变量 {.} 代表是list的变量
|
||||
// 这里模板 删除了list以后的数据，也就是统计的这一行
|
||||
String templateFileName = "D:\\test\\complex.xlsx"; |
||||
|
||||
String fileName = TestFileUtil.getPath() + "complexFillWithTable" + System.currentTimeMillis() + ".xlsx"; |
||||
ExcelWriter excelWriter = EasyExcel.write(fileName).withTemplate(templateFileName).build(); |
||||
WriteSheet writeSheet = EasyExcel.writerSheet().build(); |
||||
// 直接写入数据
|
||||
excelWriter.fill(data(), writeSheet); |
||||
excelWriter.fill(data(), writeSheet); |
||||
|
||||
// 写入list之前的数据
|
||||
Map<String, Object> map = new HashMap<String, Object>(); |
||||
map.put("date", "2019年10月9日13:28:28"); |
||||
excelWriter.fill(map, writeSheet); |
||||
|
||||
// list 后面还有个统计 想办法手动写入
|
||||
// 这里偷懒直接用list 也可以用对象
|
||||
List<List<String>> totalListList = new ArrayList<List<String>>(); |
||||
List<String> totalList = new ArrayList<String>(); |
||||
totalListList.add(totalList); |
||||
totalList.add(null); |
||||
totalList.add(null); |
||||
totalList.add(null); |
||||
// 第四列
|
||||
totalList.add("统计:1000"); |
||||
// 这里是write 别和fill 搞错了
|
||||
excelWriter.write(totalListList, writeSheet); |
||||
excelWriter.finish(); |
||||
// 总体上写法比较复杂 但是也没有想到好的版本 异步的去写入excel 不支持行的删除和移动，也不支持备注这种的写入，所以也排除了可以
|
||||
// 新建一个 然后一点点复制过来的方案，最后导致list需要新增行的时候，后面的列的数据没法后移，后续会继续想想解决方案
|
||||
} |
||||
|
||||
private List<FillData> data() { |
||||
List<FillData> list = new ArrayList<FillData>(); |
||||
for (int i = 0; i < 10; i++) { |
||||
FillData fillData = new FillData(); |
||||
list.add(fillData); |
||||
fillData.setName("张三"); |
||||
fillData.setNumber(5.2); |
||||
} |
||||
return list; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue