日常在做后臺系統的時候會很頻繁的遇到Excel導入導出的問題,正好這次在做一個后臺系統,就想著寫一個公用工具來進行Excel的導入導出。
一般我們在導出的時候都是導出的前端表格,而前端表格同時也會對應的在后臺有一個映射類。
所以在寫這個工具的時候我們先理一下我們需要實現的效果:
導出方法接收一個list集合,和一個Class類型,和HttpServletResponse 對象
導出是可能會有下拉列表,所以需要一個map存儲下拉列表數據源,傳入參數后只需一行代碼即可導出
導入方法需要傳入file文件,以及一個Class類型,導入之后將會返回一個list集合,里面的對象就是傳入類型的對象,傳入參數后只需一行代碼即可導入
實現過程:
首先需要創建三個注解
一個是EnableExport ,必須有這個注解才能導出
/**
* 設置允許導出
*/
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface EnableExport {
String fileName();
}
然后就是EnableExportField,有這個注解的字段才會導出到Excel里面,并且可以設置列寬
/**
* 設置該字段允許導出
* 并且可以設置寬度
*/
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface EnableExportField {
int colWidth() default 100;
String colName();
}
再就是ImportIndex,導入的時候設置Excel中的列對應的序號
/**
* 導入時索引
*/
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface ImportIndex {
int index() ;
}
注解使用示例
三個注解創建好之后就需要開始操作Excel了
首先,導入方法。在后臺接收到前端上傳的Excel文件之后,使用poi來讀取Excel文件
我們根據傳入的類型上面的字段注解的順序來分別為不同的字段賦值,然后存入集合中,再返回
代碼如下:
/**
* 將Excel轉換為對象集合
* @param excel Excel 文件
* @param clazz pojo類型
* @return
*/
public static List《Object》 parseExcelToList(File excel,Class clazz){
List《Object》 res = new ArrayList《》();
// 創建輸入流,讀取Excel
InputStream is = null;
Sheet sheet = null;
try {
is = new FileInputStream(excel.getAbsolutePath());
if (is != null) {
Workbook workbook = WorkbookFactory.create(is);
//默認只獲取第一個工作表
sheet = workbook.getSheetAt(0);
if (sheet != null) {
//前兩行是標題
int i = 2;
String values[] ;
Row row = sheet.getRow(i);
while (row != null) {
//獲取單元格數目
int cellNum = row.getPhysicalNumberOfCells();
values = new String[cellNum];
for (int j = 0; j 《= cellNum; j++) {
Cell cell = row.getCell(j);
if (cell != null) {
//設置單元格內容類型
cell.setCellType(Cell.CELL_TYPE_STRING );
//獲取單元格值
String value = cell.getStringCellValue() == null ? null : cell.getStringCellValue();
values[j]=value;
}
}
Field[] fields = clazz.getDeclaredFields();
Object obj = clazz.newInstance();
for(Field f : fields){
if(f.isAnnotationPresent(ImportIndex.class)){
ImportIndex annotation = f.getDeclaredAnnotation(ImportIndex.class);
int index = annotation.index();
f.setAccessible(true);
//此處使用了阿里巴巴的fastjson包里面的一個類型轉換工具類
Object val =TypeUtils.cast(values[index],f.getType(),null);
f.set(obj,val);
}
}
res.add(obj);
i++;
row=sheet.getRow(i);
}
}
}
} catch (Exception e) {
e.printStackTrace();
}
return res;
}
接下來就是導出方法。
導出分為幾個步驟:
1.建立一個工作簿,也就是類型新建一個Excel文件
2.建立一張sheet表
3.設置標的行高和列寬
4.繪制標題和表頭
這兩個方法是自定義方法,代碼會貼在后面
5.寫入數據到Excel
6.創建下拉列表
7.寫入文件到response
到這里導出工作就完成了
下面是一些自定義方法的代碼
/**
* 獲取一個基本的帶邊框的單元格
* @param workbook
* @return
*/
private static HSSFCellStyle getBasicCellStyle(HSSFWorkbook workbook){
HSSFCellStyle hssfcellstyle = workbook.createCellStyle();
hssfcellstyle.setBorderLeft(HSSFCellStyle.BORDER_THIN);
hssfcellstyle.setBorderBottom(HSSFCellStyle.BORDER_THIN);
hssfcellstyle.setBorderRight(HSSFCellStyle.BORDER_THIN);
hssfcellstyle.setBorderTop(HSSFCellStyle.BORDER_THIN);
hssfcellstyle.setAlignment(HSSFCellStyle.ALIGN_CENTER);
hssfcellstyle.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
hssfcellstyle.setWrapText(true);
return hssfcellstyle;
}
/**
* 獲取帶有背景色的標題單元格
* @param workbook
* @return
*/
private static HSSFCellStyle getTitleCellStyle(HSSFWorkbook workbook){
HSSFCellStyle hssfcellstyle = getBasicCellStyle(workbook);
hssfcellstyle.setFillForegroundColor((short) HSSFColor.CORNFLOWER_BLUE.index); // 設置背景色
hssfcellstyle.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
return hssfcellstyle;
}
/**
* 創建一個跨列的標題行
* @param workbook
* @param hssfRow
* @param hssfcell
* @param hssfsheet
* @param allColNum
* @param title
*/
private static void createTitle(HSSFWorkbook workbook, HSSFRow hssfRow , HSSFCell hssfcell, HSSFSheet hssfsheet,int allColNum,String title){
//在sheet里增加合并單元格
CellRangeAddress cra = new CellRangeAddress(0, 0, 0, allColNum);
hssfsheet.addMergedRegion(cra);
// 使用RegionUtil類為合并后的單元格添加邊框
RegionUtil.setBorderBottom(1, cra, hssfsheet, workbook); // 下邊框
RegionUtil.setBorderLeft(1, cra, hssfsheet, workbook); // 左邊框
RegionUtil.setBorderRight(1, cra, hssfsheet, workbook); // 有邊框
RegionUtil.setBorderTop(1, cra, hssfsheet, workbook); // 上邊框
//設置表頭
hssfRow = hssfsheet.getRow(0);
hssfcell = hssfRow.getCell(0);
hssfcell.setCellStyle( getTitleCellStyle(workbook));
hssfcell.setCellType(HSSFCell.CELL_TYPE_STRING);
hssfcell.setCellValue(title);
}
/**
* 設置表頭標題欄以及表格高度
* @param workbook
* @param hssfRow
* @param hssfcell
* @param hssfsheet
* @param colNames
*/
private static void createHeadRow(HSSFWorkbook workbook,HSSFRow hssfRow , HSSFCell hssfcell,HSSFSheet hssfsheet,List《String》 colNames){
//插入標題行
hssfRow = hssfsheet.createRow(1);
for (int i = 0; i 《 colNames.size(); i++) {
hssfcell = hssfRow.createCell(i);
hssfcell.setCellStyle(getTitleCellStyle(workbook));
hssfcell.setCellType(HSSFCell.CELL_TYPE_STRING);
hssfcell.setCellValue(colNames.get(i));
}
}
/**
* excel添加下拉數據校驗
* @param sheet 哪個 sheet 頁添加校驗
* @return
*/
public static void createDataValidation(Sheet sheet,Map《Integer,String[]》 selectListMap) {
if(selectListMap!=null) {
selectListMap.forEach(
// 第幾列校驗(0開始)key 數據源數組value
(key, value) -》 {
if(value.length》0) {
CellRangeAddressList cellRangeAddressList = new CellRangeAddressList(2, 65535, key, key);
DataValidationHelper helper = sheet.getDataValidationHelper();
DataValidationConstraint constraint = helper.createExplicitListConstraint(value);
DataValidation dataValidation = helper.createValidation(constraint, cellRangeAddressList);
//處理Excel兼容性問題
if (dataValidation instanceof XSSFDataValidation) {
dataValidation.setSuppressDropDownArrow(true);
dataValidation.setShowErrorBox(true);
} else {
dataValidation.setSuppressDropDownArrow(false);
}
dataValidation.setEmptyCellAllowed(true);
dataValidation.setShowPromptBox(true);
dataValidation.createPromptBox(“提示”, “只能選擇下拉框里面的數據”);
sheet.addValidationData(dataValidation);
}
}
);
}
}
使用實例
導出數據
導入數據(返回對象List)
責任編輯:haq
-
Excel
+關注
關注
4文章
218瀏覽量
55518 -
數據導入
+關注
關注
0文章
3瀏覽量
6932
原文標題:實現 Excel 導入導出
文章出處:【微信號:harmonyos_developer,微信公眾號:harmonyos_developer】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
相關推薦
評論