forked from github/easyexcel
zhuangjiaju
5 years ago
22 changed files with 396 additions and 46 deletions
@ -0,0 +1,17 @@
|
||||
package com.alibaba.excel.enums; |
||||
|
||||
/** |
||||
* The types of write last row |
||||
* |
||||
* @author zhuangjiaju |
||||
**/ |
||||
public enum WriteLastRowType { |
||||
/** |
||||
* Tables are created without templates ,And any data has been written; |
||||
*/ |
||||
EMPTY, |
||||
/** |
||||
* It's supposed to have data in it.Tables are created with templates ,or any data has been written; |
||||
*/ |
||||
HAVE_DATA,; |
||||
} |
||||
@ -0,0 +1,33 @@
|
||||
package com.alibaba.easyexcel.test.core.head; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
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 zhuangjiaju |
||||
*/ |
||||
public class ComplexDataListener extends AnalysisEventListener<ComplexHeadData> { |
||||
private static final Logger LOGGER = LoggerFactory.getLogger(ComplexHeadData.class); |
||||
List<ComplexHeadData> list = new ArrayList<ComplexHeadData>(); |
||||
|
||||
@Override |
||||
public void invoke(ComplexHeadData data, AnalysisContext context) { |
||||
list.add(data); |
||||
} |
||||
|
||||
@Override |
||||
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
Assert.assertEquals(list.size(), 1); |
||||
ComplexHeadData data = list.get(0); |
||||
Assert.assertEquals(data.getString4(), "字符串4"); |
||||
LOGGER.debug("First row:{}", JSON.toJSONString(list.get(0))); |
||||
} |
||||
} |
||||
@ -0,0 +1,22 @@
|
||||
package com.alibaba.easyexcel.test.core.head; |
||||
|
||||
import com.alibaba.excel.annotation.ExcelProperty; |
||||
|
||||
import lombok.Data; |
||||
|
||||
/** |
||||
* @author zhuangjiaju |
||||
*/ |
||||
@Data |
||||
public class ComplexHeadData { |
||||
@ExcelProperty({"顶格", "顶格", "两格"}) |
||||
private String string0; |
||||
@ExcelProperty({"顶格", "顶格", "两格"}) |
||||
private String string1; |
||||
@ExcelProperty({"顶格", "四联", "四联"}) |
||||
private String string2; |
||||
@ExcelProperty({"顶格", "四联", "四联"}) |
||||
private String string3; |
||||
@ExcelProperty({"顶格"}) |
||||
private String string4; |
||||
} |
||||
@ -0,0 +1,54 @@
|
||||
package com.alibaba.easyexcel.test.core.head; |
||||
|
||||
import java.io.File; |
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
import org.junit.BeforeClass; |
||||
import org.junit.Test; |
||||
|
||||
import com.alibaba.easyexcel.test.util.TestFileUtil; |
||||
import com.alibaba.excel.EasyExcelFactory; |
||||
|
||||
/** |
||||
* |
||||
* @author zhuangjiaju |
||||
*/ |
||||
public class ComplexHeadDataTest { |
||||
|
||||
private static File file07; |
||||
private static File file03; |
||||
|
||||
@BeforeClass |
||||
public static void init() { |
||||
file07 = TestFileUtil.createNewFile("complexHead07.xlsx"); |
||||
file03 = TestFileUtil.createNewFile("complexHead03.xls"); |
||||
} |
||||
|
||||
@Test |
||||
public void T01ReadAndWrite07() { |
||||
readAndWrite(file07); |
||||
} |
||||
|
||||
@Test |
||||
public void T02ReadAndWrite03() { |
||||
readAndWrite(file03); |
||||
} |
||||
|
||||
private void readAndWrite(File file) { |
||||
EasyExcelFactory.write(file, ComplexHeadData.class).sheet().doWrite(data()).finish(); |
||||
EasyExcelFactory.read(file, ComplexHeadData.class, new ComplexDataListener()).sheet().doRead().finish(); |
||||
} |
||||
|
||||
private List<ComplexHeadData> data() { |
||||
List<ComplexHeadData> list = new ArrayList<ComplexHeadData>(); |
||||
ComplexHeadData data = new ComplexHeadData(); |
||||
data.setString0("字符串0"); |
||||
data.setString1("字符串1"); |
||||
data.setString2("字符串2"); |
||||
data.setString3("字符串3"); |
||||
data.setString4("字符串4"); |
||||
list.add(data); |
||||
return list; |
||||
} |
||||
} |
||||
@ -0,0 +1,38 @@
|
||||
package com.alibaba.easyexcel.test.core.head; |
||||
|
||||
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 zhuangjiaju |
||||
*/ |
||||
public class ListHeadDataListener extends AnalysisEventListener<Map<Integer, String>> { |
||||
|
||||
private static final Logger LOGGER = LoggerFactory.getLogger(NoHeadData.class); |
||||
List<Map<Integer, String>> list = new ArrayList<Map<Integer, String>>(); |
||||
|
||||
@Override |
||||
public void invoke(Map<Integer, String> data, AnalysisContext context) { |
||||
list.add(data); |
||||
} |
||||
|
||||
@Override |
||||
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
Assert.assertEquals(list.size(), 1); |
||||
Map<Integer, String> data = list.get(0); |
||||
Assert.assertEquals(data.get(0), "字符串0"); |
||||
Assert.assertEquals(data.get(1), "1.0"); |
||||
Assert.assertEquals(data.get(2), "2020-01-01 01:01:01"); |
||||
Assert.assertEquals(data.get(3), "额外数据"); |
||||
LOGGER.debug("First row:{}", JSON.toJSONString(list.get(0))); |
||||
} |
||||
} |
||||
@ -0,0 +1,69 @@
|
||||
package com.alibaba.easyexcel.test.core.head; |
||||
|
||||
import java.io.File; |
||||
import java.text.ParseException; |
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
import org.junit.BeforeClass; |
||||
import org.junit.Test; |
||||
|
||||
import com.alibaba.easyexcel.test.util.TestFileUtil; |
||||
import com.alibaba.excel.EasyExcelFactory; |
||||
import com.alibaba.excel.util.DateUtils; |
||||
|
||||
/** |
||||
* |
||||
* @author zhuangjiaju |
||||
*/ |
||||
public class ListHeadDataTest { |
||||
|
||||
private static File file07; |
||||
private static File file03; |
||||
|
||||
@BeforeClass |
||||
public static void init() { |
||||
file07 = TestFileUtil.createNewFile("listHead07.xlsx"); |
||||
file03 = TestFileUtil.createNewFile("listHead03.xls"); |
||||
} |
||||
|
||||
@Test |
||||
public void T01ReadAndWrite07() throws Exception { |
||||
readAndWrite(file07); |
||||
} |
||||
|
||||
@Test |
||||
public void T02ReadAndWrite03() throws Exception { |
||||
readAndWrite(file03); |
||||
} |
||||
|
||||
private void readAndWrite(File file) throws Exception { |
||||
EasyExcelFactory.write(file).head(head()).sheet().doWrite(data()).finish(); |
||||
EasyExcelFactory.read(file).registerReadListener(new ListHeadDataListener()).sheet().doRead().finish(); |
||||
} |
||||
|
||||
private List<List<String>> head() { |
||||
List<List<String>> list = new ArrayList<List<String>>(); |
||||
List<String> head0 = new ArrayList<String>(); |
||||
head0.add("字符串"); |
||||
List<String> head1 = new ArrayList<String>(); |
||||
head1.add("数字"); |
||||
List<String> head2 = new ArrayList<String>(); |
||||
head2.add("日期"); |
||||
list.add(head0); |
||||
list.add(head1); |
||||
list.add(head2); |
||||
return list; |
||||
} |
||||
|
||||
private List<List<Object>> data() throws ParseException { |
||||
List<List<Object>> list = new ArrayList<List<Object>>(); |
||||
List<Object> data0 = new ArrayList<Object>(); |
||||
data0.add("字符串0"); |
||||
data0.add(1); |
||||
data0.add(DateUtils.parseDate("2020-01-01 01:01:01")); |
||||
data0.add("额外数据"); |
||||
list.add(data0); |
||||
return list; |
||||
} |
||||
} |
||||
@ -0,0 +1,14 @@
|
||||
package com.alibaba.easyexcel.test.core.head; |
||||
|
||||
import com.alibaba.excel.annotation.ExcelProperty; |
||||
|
||||
import lombok.Data; |
||||
|
||||
/** |
||||
* @author zhuangjiaju |
||||
*/ |
||||
@Data |
||||
public class NoHeadData { |
||||
@ExcelProperty("字符串") |
||||
private String string; |
||||
} |
||||
@ -1,20 +0,0 @@
|
||||
package com.alibaba.easyexcel.test.core.head; |
||||
|
||||
import org.junit.Test; |
||||
|
||||
/** |
||||
* Order data test |
||||
* |
||||
* @author zhuangjiaju |
||||
*/ |
||||
public class NoHeadData07Test { |
||||
|
||||
@Test |
||||
public void simple() { |
||||
// ExcelWriter writer = EasyExcelFactory.writerBuilder().outputFile(TestFileUtil.createNewWriteFile("order07.xlsx"))
|
||||
// .head(OrderData.class).build();
|
||||
// Sheet sheet = EasyExcelFactory.writerSheetBuilder().sheetNo(0).sheetName("order").build();
|
||||
// writer.write(createData(10000 * 100), sheet);
|
||||
// writer.finish();
|
||||
} |
||||
} |
||||
@ -0,0 +1,33 @@
|
||||
package com.alibaba.easyexcel.test.core.head; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
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 zhuangjiaju |
||||
*/ |
||||
public class NoHeadDataListener extends AnalysisEventListener<NoHeadData> { |
||||
private static final Logger LOGGER = LoggerFactory.getLogger(NoHeadData.class); |
||||
List<NoHeadData> list = new ArrayList<NoHeadData>(); |
||||
|
||||
@Override |
||||
public void invoke(NoHeadData data, AnalysisContext context) { |
||||
list.add(data); |
||||
} |
||||
|
||||
@Override |
||||
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
Assert.assertEquals(list.size(), 1); |
||||
NoHeadData data = list.get(0); |
||||
Assert.assertEquals(data.getString(), "字符串0"); |
||||
LOGGER.debug("First row:{}", JSON.toJSONString(list.get(0))); |
||||
} |
||||
} |
||||
@ -0,0 +1,51 @@
|
||||
package com.alibaba.easyexcel.test.core.head; |
||||
|
||||
import java.io.File; |
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
import org.junit.BeforeClass; |
||||
import org.junit.Test; |
||||
|
||||
import com.alibaba.easyexcel.test.util.TestFileUtil; |
||||
import com.alibaba.excel.EasyExcelFactory; |
||||
|
||||
/** |
||||
* |
||||
* @author zhuangjiaju |
||||
*/ |
||||
public class NoHeadDataTest { |
||||
|
||||
private static File file07; |
||||
private static File file03; |
||||
|
||||
@BeforeClass |
||||
public static void init() { |
||||
file07 = TestFileUtil.createNewFile("noHead07.xlsx"); |
||||
file03 = TestFileUtil.createNewFile("noHead03.xls"); |
||||
} |
||||
|
||||
@Test |
||||
public void T01ReadAndWrite07() { |
||||
readAndWrite(file07); |
||||
} |
||||
|
||||
@Test |
||||
public void T02ReadAndWrite03() { |
||||
readAndWrite(file03); |
||||
} |
||||
|
||||
private void readAndWrite(File file) { |
||||
EasyExcelFactory.write(file, NoHeadData.class).needHead(Boolean.FALSE).sheet().doWrite(data()).finish(); |
||||
EasyExcelFactory.read(file, NoHeadData.class, new NoHeadDataListener()).headRowNumber(0).sheet().doRead() |
||||
.finish(); |
||||
} |
||||
|
||||
private List<NoHeadData> data() { |
||||
List<NoHeadData> list = new ArrayList<NoHeadData>(); |
||||
NoHeadData data = new NoHeadData(); |
||||
data.setString("字符串0"); |
||||
list.add(data); |
||||
return list; |
||||
} |
||||
} |
||||
Loading…
Reference in new issue