forked from fanruan/easyexcel
Jiaju Zhuang
4 years ago
14 changed files with 585 additions and 122 deletions
@ -0,0 +1,20 @@
|
||||
package com.alibaba.easyexcel.test.core.excludeorinclude; |
||||
|
||||
import com.alibaba.excel.annotation.ExcelProperty; |
||||
|
||||
import lombok.Data; |
||||
|
||||
/** |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
@Data |
||||
public class ExcludeOrIncludeData { |
||||
@ExcelProperty(order = 1) |
||||
private String column1; |
||||
@ExcelProperty(order = 2) |
||||
private String column2; |
||||
@ExcelProperty(order = 3) |
||||
private String column3; |
||||
@ExcelProperty(order = 4) |
||||
private String column4; |
||||
} |
||||
@ -0,0 +1,161 @@
|
||||
package com.alibaba.easyexcel.test.core.excludeorinclude; |
||||
|
||||
import java.io.File; |
||||
import java.util.ArrayList; |
||||
import java.util.Collections; |
||||
import java.util.HashSet; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
import java.util.Set; |
||||
|
||||
import org.junit.Assert; |
||||
import org.junit.BeforeClass; |
||||
import org.junit.FixMethodOrder; |
||||
import org.junit.Test; |
||||
import org.junit.runners.MethodSorters; |
||||
|
||||
import com.alibaba.easyexcel.test.core.sort.SortData; |
||||
import com.alibaba.easyexcel.test.core.sort.SortDataListener; |
||||
import com.alibaba.easyexcel.test.util.TestFileUtil; |
||||
import com.alibaba.excel.EasyExcel; |
||||
|
||||
/** |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
@FixMethodOrder(MethodSorters.NAME_ASCENDING) |
||||
public class ExcludeOrIncludeDataTest { |
||||
|
||||
private static File excludeIndex07; |
||||
private static File excludeIndex03; |
||||
private static File excludeFiledName07; |
||||
private static File excludeFiledName03; |
||||
private static File includeIndex07; |
||||
private static File includeIndex03; |
||||
private static File includeFiledName07; |
||||
private static File includeFiledName03; |
||||
|
||||
@BeforeClass |
||||
public static void init() { |
||||
excludeIndex07 = TestFileUtil.createNewFile("excludeIndex.xlsx"); |
||||
excludeIndex03 = TestFileUtil.createNewFile("excludeIndex.xls"); |
||||
excludeFiledName07 = TestFileUtil.createNewFile("excludeFiledName.xlsx"); |
||||
excludeFiledName03 = TestFileUtil.createNewFile("excludeFiledName.xls"); |
||||
includeIndex07 = TestFileUtil.createNewFile("includeIndex.xlsx"); |
||||
includeIndex03 = TestFileUtil.createNewFile("includeIndex.xls"); |
||||
includeFiledName07 = TestFileUtil.createNewFile("includeFiledName.xlsx"); |
||||
includeFiledName03 = TestFileUtil.createNewFile("includeFiledName.xls"); |
||||
} |
||||
|
||||
@Test |
||||
public void t01ExcludeIndex07() { |
||||
excludeIndex(excludeIndex07); |
||||
} |
||||
|
||||
@Test |
||||
public void t02ExcludeIndex07() { |
||||
excludeIndex(excludeIndex03); |
||||
} |
||||
|
||||
@Test |
||||
public void t03ExcludeFiledName07() { |
||||
excludeFiledName(excludeFiledName07); |
||||
} |
||||
|
||||
@Test |
||||
public void t04ExcludeFiledName07() { |
||||
excludeFiledName(excludeFiledName03); |
||||
} |
||||
|
||||
|
||||
@Test |
||||
public void t05IncludeIndex07() { |
||||
includeIndex(includeIndex07); |
||||
} |
||||
|
||||
@Test |
||||
public void t06IncludeIndex07() { |
||||
includeIndex(includeIndex03); |
||||
} |
||||
|
||||
@Test |
||||
public void t07IncludeFiledName07() { |
||||
includeFiledName(includeFiledName07); |
||||
} |
||||
|
||||
@Test |
||||
public void t08IncludeFiledName07() { |
||||
includeFiledName(includeFiledName03); |
||||
} |
||||
|
||||
|
||||
private void excludeIndex(File file) { |
||||
Set<Integer> excludeColumnIndexes = new HashSet<Integer>(); |
||||
excludeColumnIndexes.add(0); |
||||
excludeColumnIndexes.add(3); |
||||
EasyExcel.write(file, ExcludeOrIncludeData.class).excludeColumnIndexes(excludeColumnIndexes).sheet() |
||||
.doWrite(data()); |
||||
List<Map<Integer, String>> dataMap = EasyExcel.read(file).sheet().doReadSync(); |
||||
Assert.assertEquals(1, dataMap.size()); |
||||
Map<Integer, String> record = dataMap.get(0); |
||||
Assert.assertEquals(2, record.size()); |
||||
Assert.assertEquals("column2", record.get(0)); |
||||
Assert.assertEquals("column3", record.get(1)); |
||||
|
||||
} |
||||
|
||||
private void excludeFiledName(File file) { |
||||
Set<String> excludeColumnFiledNames = new HashSet<String>(); |
||||
excludeColumnFiledNames.add("column1"); |
||||
excludeColumnFiledNames.add("column3"); |
||||
excludeColumnFiledNames.add("column4"); |
||||
EasyExcel.write(file, ExcludeOrIncludeData.class).excludeColumnFiledNames(excludeColumnFiledNames).sheet() |
||||
.doWrite(data()); |
||||
List<Map<Integer, String>> dataMap = EasyExcel.read(file).sheet().doReadSync(); |
||||
Assert.assertEquals(1, dataMap.size()); |
||||
Map<Integer, String> record = dataMap.get(0); |
||||
Assert.assertEquals(1, record.size()); |
||||
Assert.assertEquals("column2", record.get(0)); |
||||
|
||||
} |
||||
|
||||
private void includeIndex(File file) { |
||||
Set<Integer> includeColumnIndexes = new HashSet<Integer>(); |
||||
includeColumnIndexes.add(1); |
||||
includeColumnIndexes.add(2); |
||||
EasyExcel.write(file, ExcludeOrIncludeData.class).includeColumnIndexes(includeColumnIndexes).sheet() |
||||
.doWrite(data()); |
||||
List<Map<Integer, String>> dataMap = EasyExcel.read(file).sheet().doReadSync(); |
||||
Assert.assertEquals(1, dataMap.size()); |
||||
Map<Integer, String> record = dataMap.get(0); |
||||
Assert.assertEquals(2, record.size()); |
||||
Assert.assertEquals("column2", record.get(0)); |
||||
Assert.assertEquals("column3", record.get(1)); |
||||
|
||||
} |
||||
|
||||
private void includeFiledName(File file) { |
||||
Set<String> includeColumnFiledNames = new HashSet<String>(); |
||||
includeColumnFiledNames.add("column2"); |
||||
includeColumnFiledNames.add("column3"); |
||||
EasyExcel.write(file, ExcludeOrIncludeData.class).includeColumnFiledNames(includeColumnFiledNames).sheet() |
||||
.doWrite(data()); |
||||
List<Map<Integer, String>> dataMap = EasyExcel.read(file).sheet().doReadSync(); |
||||
Assert.assertEquals(1, dataMap.size()); |
||||
Map<Integer, String> record = dataMap.get(0); |
||||
Assert.assertEquals(2, record.size()); |
||||
Assert.assertEquals("column2", record.get(0)); |
||||
Assert.assertEquals("column3", record.get(1)); |
||||
} |
||||
|
||||
|
||||
private List<ExcludeOrIncludeData> data() { |
||||
List<ExcludeOrIncludeData> list = new ArrayList<ExcludeOrIncludeData>(); |
||||
ExcludeOrIncludeData excludeOrIncludeData = new ExcludeOrIncludeData(); |
||||
excludeOrIncludeData.setColumn1("column1"); |
||||
excludeOrIncludeData.setColumn2("column2"); |
||||
excludeOrIncludeData.setColumn3("column3"); |
||||
excludeOrIncludeData.setColumn4("column4"); |
||||
list.add(excludeOrIncludeData); |
||||
return list; |
||||
} |
||||
} |
||||
@ -0,0 +1,23 @@
|
||||
package com.alibaba.easyexcel.test.core.sort; |
||||
|
||||
import com.alibaba.excel.annotation.ExcelProperty; |
||||
|
||||
import lombok.Data; |
||||
|
||||
/** |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
@Data |
||||
public class SortData { |
||||
|
||||
private String column5; |
||||
private String column6; |
||||
@ExcelProperty(order = 99) |
||||
private String column3; |
||||
@ExcelProperty(order = 100) |
||||
private String column4; |
||||
@ExcelProperty(value = "column2", index = 1) |
||||
private String column2; |
||||
@ExcelProperty(value = "column1", index = 0) |
||||
private String column1; |
||||
} |
||||
@ -0,0 +1,41 @@
|
||||
package com.alibaba.easyexcel.test.core.sort; |
||||
|
||||
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.easyexcel.test.core.simple.SimpleData; |
||||
import com.alibaba.excel.context.AnalysisContext; |
||||
import com.alibaba.excel.event.AnalysisEventListener; |
||||
import com.alibaba.fastjson.JSON; |
||||
|
||||
/** |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
public class SortDataListener extends AnalysisEventListener<SortData> { |
||||
|
||||
private static final Logger LOGGER = LoggerFactory.getLogger(SortDataListener.class); |
||||
List<SortData> list = new ArrayList<SortData>(); |
||||
|
||||
|
||||
@Override |
||||
public void invoke(SortData data, AnalysisContext context) { |
||||
list.add(data); |
||||
} |
||||
|
||||
@Override |
||||
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
Assert.assertEquals(list.size(), 1); |
||||
SortData sortData = list.get(0); |
||||
Assert.assertEquals("column1", sortData.getColumn1()); |
||||
Assert.assertEquals("column2", sortData.getColumn2()); |
||||
Assert.assertEquals("column3", sortData.getColumn3()); |
||||
Assert.assertEquals("column4", sortData.getColumn4()); |
||||
Assert.assertEquals("column5", sortData.getColumn5()); |
||||
Assert.assertEquals("column6", sortData.getColumn6()); |
||||
} |
||||
} |
||||
@ -0,0 +1,111 @@
|
||||
package com.alibaba.easyexcel.test.core.sort; |
||||
|
||||
import java.io.File; |
||||
import java.util.ArrayList; |
||||
import java.util.Collections; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
|
||||
import org.junit.Assert; |
||||
import org.junit.BeforeClass; |
||||
import org.junit.FixMethodOrder; |
||||
import org.junit.Test; |
||||
import org.junit.runners.MethodSorters; |
||||
|
||||
import com.alibaba.easyexcel.test.util.TestFileUtil; |
||||
import com.alibaba.excel.EasyExcel; |
||||
|
||||
/** |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
@FixMethodOrder(MethodSorters.NAME_ASCENDING) |
||||
public class SortDataTest { |
||||
|
||||
private static File file07; |
||||
private static File file03; |
||||
private static File sortNoHead07; |
||||
private static File sortNoHead03; |
||||
|
||||
@BeforeClass |
||||
public static void init() { |
||||
file07 = TestFileUtil.createNewFile("sort.xlsx"); |
||||
file03 = TestFileUtil.createNewFile("sort.xls"); |
||||
sortNoHead07 = TestFileUtil.createNewFile("sortNoHead.xlsx"); |
||||
sortNoHead03 = TestFileUtil.createNewFile("sortNoHead.xls"); |
||||
} |
||||
|
||||
@Test |
||||
public void t01ReadAndWrite07() { |
||||
readAndWrite(file07); |
||||
} |
||||
|
||||
@Test |
||||
public void t02ReadAndWrite03() { |
||||
readAndWrite(file03); |
||||
} |
||||
|
||||
@Test |
||||
public void t03ReadAndWriteNoHead07() { |
||||
readAndWriteNoHead(sortNoHead07); |
||||
} |
||||
|
||||
@Test |
||||
public void t04ReadAndWriteNoHead03() { |
||||
readAndWriteNoHead(sortNoHead03); |
||||
} |
||||
|
||||
|
||||
private void readAndWrite(File file) { |
||||
EasyExcel.write(file, SortData.class).sheet().doWrite(data()); |
||||
List<Map<Integer, String>> dataMap = EasyExcel.read(file).sheet().doReadSync(); |
||||
Assert.assertEquals(1, dataMap.size()); |
||||
Map<Integer, String> record = dataMap.get(0); |
||||
Assert.assertEquals("column1", record.get(0)); |
||||
Assert.assertEquals("column2", record.get(1)); |
||||
Assert.assertEquals("column3", record.get(2)); |
||||
Assert.assertEquals("column4", record.get(3)); |
||||
Assert.assertEquals("column5", record.get(4)); |
||||
Assert.assertEquals("column6", record.get(5)); |
||||
|
||||
EasyExcel.read(file, SortData.class, new SortDataListener()).sheet().doRead(); |
||||
} |
||||
|
||||
private void readAndWriteNoHead(File file) { |
||||
EasyExcel.write(file).head(head()).sheet().doWrite(data()); |
||||
List<Map<Integer, String>> dataMap = EasyExcel.read(file).sheet().doReadSync(); |
||||
Assert.assertEquals(1, dataMap.size()); |
||||
Map<Integer, String> record = dataMap.get(0); |
||||
Assert.assertEquals("column1", record.get(0)); |
||||
Assert.assertEquals("column2", record.get(1)); |
||||
Assert.assertEquals("column3", record.get(2)); |
||||
Assert.assertEquals("column4", record.get(3)); |
||||
Assert.assertEquals("column5", record.get(4)); |
||||
Assert.assertEquals("column6", record.get(5)); |
||||
EasyExcel.read(file, SortData.class, new SortDataListener()).sheet().doRead(); |
||||
} |
||||
|
||||
private List<List<String>> head() { |
||||
List<List<String>> head = new ArrayList<List<String>>(); |
||||
head.add(Collections.singletonList("column1")); |
||||
head.add(Collections.singletonList("column2")); |
||||
head.add(Collections.singletonList("column3")); |
||||
head.add(Collections.singletonList("column4")); |
||||
head.add(Collections.singletonList("column5")); |
||||
head.add(Collections.singletonList("column6")); |
||||
return head; |
||||
} |
||||
|
||||
|
||||
private List<SortData> data() { |
||||
List<SortData> list = new ArrayList<SortData>(); |
||||
SortData sortData = new SortData(); |
||||
sortData.setColumn1("column1"); |
||||
sortData.setColumn2("column2"); |
||||
sortData.setColumn3("column3"); |
||||
sortData.setColumn4("column4"); |
||||
sortData.setColumn5("column5"); |
||||
sortData.setColumn6("column6"); |
||||
list.add(sortData); |
||||
return list; |
||||
} |
||||
} |
||||
@ -0,0 +1,36 @@
|
||||
package com.alibaba.easyexcel.test.temp.large; |
||||
|
||||
import java.util.Map; |
||||
|
||||
import org.apache.poi.ss.formula.functions.Index; |
||||
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 NoModelLargeDataListener extends AnalysisEventListener<Map<Integer, String>> { |
||||
|
||||
private static final Logger LOGGER = LoggerFactory.getLogger(NoModelLargeDataListener.class); |
||||
private int count = 0; |
||||
|
||||
@Override |
||||
public void invoke(Map<Integer, String> data, AnalysisContext context) { |
||||
if (count == 0) { |
||||
LOGGER.info("First row:{}", JSON.toJSONString(data)); |
||||
} |
||||
count++; |
||||
if (count % 100000 == 0) { |
||||
LOGGER.info("Already read:{}", count); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
LOGGER.info("Large row count:{}", count); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue