工具類:
1、ExcelUtil.java
public class ExcelUtil<T> {
/***
* 讀取Excel方法
*/
public static <T extends BaseRowModel> ArrayList<T> readExcel(MultipartFile excel, Class<T> clazz) {
try {
InputStream in = new BufferedInputStream(excel.getInputStream());
GeneralExcelListener<T> listener = new GeneralExcelListener<>();
ExcelReader excelReader = new ExcelReader(in, ExcelTypeEnum.XLS, listener);
excelReader.read(new Sheet(1, 1, clazz));
return listener.getSuccessDatas();
} catch (Exception e) {
return new ArrayList<T>();
}
}
/**
* 導出 Excel :一個 sheet,帶表頭
*
* @param response HttpServletResponse
* @param list 數據 list,每個元素為一個 BaseRowModel
* @param fileName 導出的文件名
* @param sheetName 導入文件的 sheet 名
* @param object 映射實體類,Excel 模型
*/
public static void writeExcel(HttpServletResponse response, List<? extends BaseRowModel> list,
String fileName, String sheetName, BaseRowModel object) {
ExcelWriter writer = new ExcelWriter(getOutputStream(fileName, response), ExcelTypeEnum.XLSX);
Sheet sheet = new Sheet(1, 0, object.getClass());
sheet.setSheetName(sheetName);
writer.write(list, sheet);
writer.finish();
}
/**
* 導出文件時為Writer生成OutputStream
*/
private static OutputStream getOutputStream(String fileName, HttpServletResponse response) {
try {
return response.getOutputStream();
} catch (IOException e) {
throw new RuntimeException("導出文件時為Writer生成OutputStream失敗!");
}
}
/**
* 返回 ExcelReader
*
* @param excel 需要解析的 Excel 文件
* @param excelListener new ExcelListener()
*/
private static ExcelReader getReader(MultipartFile excel,
ExcelListener excelListener) {
String filename = excel.getOriginalFilename();
if (filename == null || (!filename.toLowerCase().endsWith(".xls") && !filename.toLowerCase().endsWith(".xlsx"))) {
throw new RuntimeException("文件格式錯誤!");
}
InputStream inputStream;
try {
inputStream = new BufferedInputStream(excel.getInputStream());
return new ExcelReader(inputStream, null, excelListener, false);
} catch (IOException e) {
e.printStackTrace();
}
return null;
}
}
2、GeneralExcelListener.java
/**
* 解析Excel數據,返回數據的類
**/
public class GeneralExcelListener<T> extends AnalysisEventListener<T> {
private final Logger logger = LoggerFactory.getLogger(this.getClass());
//自定義用於暫時存儲data。
//可以通過實例獲取該值
private ArrayList<T> successDatas = new ArrayList<>();
private ArrayList<T> errorDatas = new ArrayList<>();
public GeneralExcelListener() {
}
@Override
public void invoke(T o, AnalysisContext analysisContext) {
successDatas.add(o);
}
@Override
public void doAfterAllAnalysed(AnalysisContext context) {
}
public ArrayList<T> getErrorDatas() {
return errorDatas;
}
public ArrayList<T> getSuccessDatas() {
return successDatas;
}
}
示例:
Controller層代碼
1、導入示例
@PostMapping(value = "/import")
public ModelAndView import(@RequestParam(value = "file", required = true) MultipartFile file) {
ModelAndView mv = new ModelAndView();
//一行代碼即可獲取解析得到的Excel中的數據
List<Test> datas = ExcelUtil.readExcel(file, Test.class);
//然后對獲取的數據,在service層進行業務處理
ResultMsg result = provShareConfigService.insertList(datas);
return mv;
}
2、導出示例
@PostMapping(value = "/export")
public void export(HttpServletRequest request, HttpServletResponse response, @RequestBody JSONObject ob) {
//前端傳來的參數
String date = ob.getString("date");
String fileName = "退賠導出";
String sheetName = "sheet1";
//從數據庫得到的list集合
List<Test> list = testService.selectAll(date);
//一行代碼即可導出
ExcelUtil.writeExcel(response, list, fileName, sheetName, new Test());
}
