mirror of https://github.com/alibaba/easyexcel
Jiaju Zhuang
5 months ago
15 changed files with 284 additions and 119 deletions
@ -0,0 +1,30 @@
|
||||
package com.alibaba.excel.exception; |
||||
|
||||
/** |
||||
* Throw the exception when you need to stop |
||||
* This exception will only stop the parsing of the current sheet. If you want to stop the entire excel parsing, please |
||||
* use ExcelAnalysisStopException. |
||||
* |
||||
* The com.alibaba.excel.read.listener.ReadListener#doAfterAllAnalysed(com.alibaba.excel.context.AnalysisContext) method |
||||
* is called after the call is stopped. |
||||
* |
||||
* @author Jiaju Zhuang |
||||
* @see ExcelAnalysisStopException |
||||
* @since 3.3.4 |
||||
*/ |
||||
public class ExcelAnalysisStopSheetException extends ExcelAnalysisException { |
||||
|
||||
public ExcelAnalysisStopSheetException() {} |
||||
|
||||
public ExcelAnalysisStopSheetException(String message) { |
||||
super(message); |
||||
} |
||||
|
||||
public ExcelAnalysisStopSheetException(String message, Throwable cause) { |
||||
super(message, cause); |
||||
} |
||||
|
||||
public ExcelAnalysisStopSheetException(Throwable cause) { |
||||
super(cause); |
||||
} |
||||
} |
||||
@ -0,0 +1,52 @@
|
||||
package com.alibaba.easyexcel.test.core.exception; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.HashSet; |
||||
import java.util.List; |
||||
import java.util.Map; |
||||
import java.util.Set; |
||||
|
||||
import com.alibaba.excel.context.AnalysisContext; |
||||
import com.alibaba.excel.event.AnalysisEventListener; |
||||
import com.alibaba.excel.exception.ExcelAnalysisStopException; |
||||
import com.alibaba.excel.exception.ExcelAnalysisStopSheetException; |
||||
import com.alibaba.excel.util.ListUtils; |
||||
import com.alibaba.excel.util.MapUtils; |
||||
import com.alibaba.fastjson2.JSON; |
||||
|
||||
import lombok.Getter; |
||||
import lombok.extern.slf4j.Slf4j; |
||||
import org.apache.commons.collections4.CollectionUtils; |
||||
import org.apache.commons.collections4.SetUtils; |
||||
import org.assertj.core.internal.Maps; |
||||
import org.junit.jupiter.api.Assertions; |
||||
import org.slf4j.Logger; |
||||
import org.slf4j.LoggerFactory; |
||||
|
||||
/** |
||||
* @author Jiaju Zhuang |
||||
*/ |
||||
@Getter |
||||
@Slf4j |
||||
public class ExcelAnalysisStopSheetExceptionDataListener extends AnalysisEventListener<ExceptionData> { |
||||
|
||||
private Map<Integer, List<String>> dataMap = MapUtils.newHashMap(); |
||||
|
||||
|
||||
@Override |
||||
public void invoke(ExceptionData data, AnalysisContext context) { |
||||
List<String> sheetDataList = dataMap.computeIfAbsent(context.readSheetHolder().getSheetNo(), |
||||
key -> ListUtils.newArrayList()); |
||||
sheetDataList.add(data.getName()); |
||||
if (sheetDataList.size() >= 5) { |
||||
throw new ExcelAnalysisStopSheetException(); |
||||
} |
||||
} |
||||
|
||||
@Override |
||||
public void doAfterAllAnalysed(AnalysisContext context) { |
||||
List<String> sheetDataList = dataMap.get(context.readSheetHolder().getSheetNo()); |
||||
Assertions.assertNotNull(sheetDataList); |
||||
Assertions.assertEquals(5, sheetDataList.size()); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue