poi导入和导出
1.准备工作:导入poi包,创建需要的四个类
2.加载poi包
org.apache.httpcomponents
httpmime
4.5.7
com.alibaba
fastjson
1.2.83
org.apache.poi
poi-ooxml
3.16
3.创建类ExcelUtils
package com.ja.excelie; import java.io.ByteArrayOutputStream; import java.io.File; import java.io.FileInputStream; import java.io.FileOutputStream; import java.io.IOException; import java.io.InputStream; import java.lang.reflect.Field; import java.math.BigDecimal; import java.math.RoundingMode; import java.net.URL; import java.text.NumberFormat; import java.text.SimpleDateFormat; import java.util.*; import java.util.Map.Entry; import javax.servlet.ServletOutputStream; import javax.servlet.http.HttpServletResponse; import org.apache.poi.hssf.usermodel.HSSFDataValidation; import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.apache.poi.poifs.filesystem.POIFSFileSystem; import org.apache.poi.ss.usermodel.Cell; import org.apache.poi.ss.usermodel.CellStyle; import org.apache.poi.ss.usermodel.CellType; import org.apache.poi.ss.usermodel.ClientAnchor.AnchorType; import org.apache.poi.ss.usermodel.DataValidation; import org.apache.poi.ss.usermodel.DataValidationConstraint; import org.apache.poi.ss.usermodel.DataValidationHelper; import org.apache.poi.ss.usermodel.Drawing; import org.apache.poi.ss.usermodel.FillPatternType; import org.apache.poi.ss.usermodel.HorizontalAlignment; import org.apache.poi.ss.usermodel.IndexedColors; import org.apache.poi.ss.usermodel.Row; import org.apache.poi.ss.usermodel.Sheet; import org.apache.poi.ss.usermodel.VerticalAlignment; import org.apache.poi.ss.usermodel.Workbook; import org.apache.poi.ss.util.CellRangeAddress; import org.apache.poi.ss.util.CellRangeAddressList; import org.apache.poi.xssf.streaming.SXSSFWorkbook; import org.apache.poi.xssf.usermodel.XSSFClientAnchor; import org.apache.poi.xssf.usermodel.XSSFWorkbook; import org.springframework.web.multipart.MultipartFile; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; public class ExcelUtils { private static final String XLSX = ".xlsx"; private static final String XLS = ".xls"; public static final String ROW_MERGE = "row_merge"; public static final String COLUMN_MERGE = "column_merge"; private static final String DATE_FORMAT = "yyyy-MM-dd HH:mm:ss"; private static final String ROW_NUM = "rowNum"; private static final String ROW_DATA = "rowData"; private static final String ROW_TIPS = "rowTips"; private static final int CELL_OTHER = 0; private static final int CELL_ROW_MERGE = 1; private static final int CELL_COLUMN_MERGE = 2; private static final int IMG_HEIGHT = 30; private static final int IMG_WIDTH = 30; private static final char LEAN_LINE = '/'; private static final int BYTES_DEFAULT_LENGTH = 10240; private static final NumberFormat NUMBER_FORMAT = NumberFormat.getNumberInstance(); public staticList readFile(File file, Class clazz) throws Exception { JSONArray array = readFile(file); return getBeanList(array, clazz); } public static List readMultipartFile(MultipartFile mFile, Class clazz) throws Exception { JSONArray array = readMultipartFile(mFile); return getBeanList(array, clazz); } public static JSONArray readFile(File file) throws Exception { return readExcel(null, file); } public static JSONArray readMultipartFile(MultipartFile mFile) throws Exception { return readExcel(mFile, null); } private static List getBeanList(JSONArray array, Class clazz) throws Exception { List list = new ArrayList<>(); Map uniqueMap = new HashMap<>(16); for (int i = 0; i < array.size(); i++) { list.add(getBean(clazz, array.getJSONObject(i), uniqueMap)); } return list; } /** * 获取每个对象的数据 */ private static T getBean(Class c, JSONObject obj, Map uniqueMap) throws Exception { T t = c.newInstance(); Field[] fields = c.getDeclaredFields(); List errMsgList = new ArrayList<>(); boolean hasRowTipsField = false; StringBuilder uniqueBuilder = new StringBuilder(); int rowNum = 0; for (Field field : fields) { // 行号 if (field.getName().equals(ROW_NUM)) { rowNum = obj.getInteger(ROW_NUM); field.setAccessible(true); field.set(t, rowNum); continue; } // 是否需要设置异常信息 if (field.getName().equals(ROW_TIPS)) { hasRowTipsField = true; continue; } // 原始数据 if (field.getName().equals(ROW_DATA)) { field.setAccessible(true); field.set(t, obj.toString()); continue; } // 设置对应属性值 setFieldValue(t,field, obj, uniqueBuilder, errMsgList); } // 数据唯一性校验 if (uniqueBuilder.length() > 0) { if (uniqueMap.containsValue(uniqueBuilder.toString())) { Set rowNumKeys = uniqueMap.keySet(); for (Integer num : rowNumKeys) { if (uniqueMap.get(num).equals(uniqueBuilder.toString())) { errMsgList.add(String.format("数据唯一性校验失败,(%s)与第%s行重复)", uniqueBuilder, num)); } } } else { uniqueMap.put(rowNum, uniqueBuilder.toString()); } } // 失败处理 if (errMsgList.isEmpty() && !hasRowTipsField) { return t; } StringBuilder sb = new StringBuilder(); int size = errMsgList.size(); for (int i = 0; i < size; i++) { if (i == size - 1) { sb.append(errMsgList.get(i)); } else { sb.append(errMsgList.get(i)).append(";"); } } // 设置错误信息 for (Field field : fields) { if (field.getName().equals(ROW_TIPS)) { field.setAccessible(true); field.set(t, sb.toString()); } } return t; } private static void setFieldValue(T t, Field field, JSONObject obj, StringBuilder uniqueBuilder, List errMsgList) { // 获取 ExcelImport 注解属性 ExcelImport annotation = field.getAnnotation(ExcelImport.class); if (annotation == null) { return; } String cname = annotation.value(); if (cname.trim().length() == 0) { return; } // 获取具体值 String val = null; if (obj.containsKey(cname)) { val = getString(obj.getString(cname)); } if (val == null) { return; } field.setAccessible(true); // 判断是否必填 boolean require = annotation.required(); if (require && val.isEmpty()) { errMsgList.add(String.format("[%s]不能为空", cname)); return; } // 数据唯一性获取 boolean unique = annotation.unique(); if (unique) { if (uniqueBuilder.length() > 0) { uniqueBuilder.append("--").append(val); } else { uniqueBuilder.append(val); } } // 判断是否超过最大长度 int maxLength = annotation.maxLength(); if (maxLength > 0 && val.length() > maxLength) { errMsgList.add(String.format("[%s]长度不能超过%s个字符(当前%s个字符)", cname, maxLength, val.length())); } // 判断当前属性是否有映射关系 LinkedHashMap kvMap = getKvMap(annotation.kv()); if (!kvMap.isEmpty()) { boolean isMatch = false; for (String key : kvMap.keySet()) { if (kvMap.get(key).equals(val)) { val = key; isMatch = true; break; } } if (!isMatch) { errMsgList.add(String.format("[%s]的值不正确(当前值为%s)", cname, val)); return; } } // 其余情况根据类型赋值 String fieldClassName = field.getType().getSimpleName(); try { if ("String".equalsIgnoreCase(fieldClassName)) { field.set(t, val); } else if ("boolean".equalsIgnoreCase(fieldClassName)) { field.set(t, Boolean.valueOf(val)); } else if ("int".equalsIgnoreCase(fieldClassName) || "Integer".equals(fieldClassName)) { try { field.set(t, Integer.valueOf(val)); } catch (NumberFormatException e) { errMsgList.add(String.format("[%s]的值格式不正确(当前值为%s)", cname, val)); } } else if ("double".equalsIgnoreCase(fieldClassName)) { field.set(t, Double.valueOf(val)); } else if ("long".equalsIgnoreCase(fieldClassName)) { field.set(t, Long.valueOf(val)); } else if ("BigDecimal".equalsIgnoreCase(fieldClassName)) { field.set(t, new BigDecimal(val)); } } catch (Exception e) { e.printStackTrace(); } } private static JSONArray readExcel(MultipartFile mFile, File file) throws IOException { boolean fileNotExist = (file == null || !file.exists()); if (mFile == null && fileNotExist) { return new JSONArray(); } // 解析表格数据 InputStream in; String fileName; if (mFile != null) { // 上传文件解析 in = mFile.getInputStream(); fileName = getString(mFile.getOriginalFilename()).toLowerCase(); } else { // 本地文件解析 in = new FileInputStream(file); fileName = file.getName().toLowerCase(); } Workbook book; if (fileName.endsWith(XLSX)) { book = new XSSFWorkbook(in); } else if (fileName.endsWith(XLS)) { POIFSFileSystem poifsFileSystem = new POIFSFileSystem(in); book = new HSSFWorkbook(poifsFileSystem); } else { return new JSONArray(); } JSONArray array = read(book); book.close(); in.close(); return array; } private static JSONArray read(Workbook book) { // 获取 Excel 文件第一个 Sheet 页面 Sheet sheet = book.getSheetAt(0); return readSheet(sheet); } private static JSONArray readSheet(Sheet sheet) { // 首行下标 int rowStart = sheet.getFirstRowNum(); // 尾行下标 int rowEnd = sheet.getLastRowNum(); // 获取表头行 Row headRow = sheet.getRow(rowStart); if (headRow == null) { return new JSONArray(); } int cellStart = headRow.getFirstCellNum(); int cellEnd = headRow.getLastCellNum(); Map keyMap = new HashMap<>(); for (int j = cellStart; j < cellEnd; j++) { // 获取表头数据 String val = getCellValue(headRow.getCell(j)); if (val != null && val.trim().length() != 0) { keyMap.put(j, val); } } // 如果表头没有数据则不进行解析 if (keyMap.isEmpty()) { return (JSONArray) Collections.emptyList(); } // 获取每行JSON对象的值 JSONArray array = new JSONArray(); // 如果首行与尾行相同,表明只有一行,返回表头数据 if (rowStart == rowEnd) { JSONObject obj = new JSONObject(); // 添加行号 obj.put(ROW_NUM, 1); for (int i : keyMap.keySet()) { obj.put(keyMap.get(i), ""); } array.add(obj); return array; } for (int i = rowStart + 1; i <= rowEnd; i++) { Row eachRow = sheet.getRow(i); JSONObject obj = new JSONObject(); // 添加行号 obj.put(ROW_NUM, i + 1); StringBuilder sb = new StringBuilder(); for (int k = cellStart; k < cellEnd; k++) { if (eachRow != null) { String val = getCellValue(eachRow.getCell(k)); // 所有数据添加到里面,用于判断该行是否为空 sb.append(val); obj.put(keyMap.get(k), val); } } if (sb.length() > 0) { array.add(obj); } } return array; } private static String getCellValue(Cell cell) { // 空白或空 if (cell == null || cell.getCellTypeEnum() == CellType.BLANK) { return ""; } // String类型 if (cell.getCellTypeEnum() == CellType.STRING) { String val = cell.getStringCellValue(); if (val == null || val.trim().length() == 0) { return ""; } return val.trim(); } // 数字类型 if (cell.getCellTypeEnum() == CellType.NUMERIC) { // 科学计数法类型 return NUMBER_FORMAT.format(cell.getNumericCellValue()) + ""; } // 布尔值类型 if (cell.getCellTypeEnum() == CellType.BOOLEAN) { return cell.getBooleanCellValue() + ""; } // 错误类型 return cell.getCellFormula(); } public static void exportTemplate(HttpServletResponse response, String fileName, Class clazz) { exportTemplate(response, fileName, fileName, clazz, false); } public static void exportTemplate(HttpServletResponse response, String fileName, String sheetName, Class clazz) { exportTemplate(response, fileName, sheetName, clazz, false); } public static void exportTemplate(HttpServletResponse response, String fileName, Class clazz, boolean isContainExample) { exportTemplate(response, fileName, fileName, clazz, isContainExample); } public static void exportTemplate(HttpServletResponse response, String fileName, String sheetName, Class clazz, boolean isContainExample) { // 获取表头字段 List headFieldList = getExcelClassFieldList(clazz); // 获取表头数据和示例数据 List > sheetDataList = new ArrayList<>(); List
4.创建接口ExcelImport
package com.ja.excelie; import java.lang.annotation.ElementType; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; @Target(ElementType.FIELD) @Retention(RetentionPolicy.RUNTIME) public @interface ExcelImport { /** 字段名称 */ String value(); /** 导出映射,格式如:0-未知;1-男;2-女 */ String kv() default ""; /** 是否为必填字段(默认为非必填) */ boolean required() default false; /** 最大长度(默认255) */ int maxLength() default 255; /** 导入唯一性验证(多个字段则取联合验证) */ boolean unique() default false; }
5.创建接口ExcelExport
package com.ja.excelie; import java.lang.annotation.ElementType; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; @Target(ElementType.FIELD) @Retention(RetentionPolicy.RUNTIME) public @interface ExcelExport { /** 字段名称 */ String value(); /** 导出排序先后: 数字越小越靠前(默认按Java类字段顺序导出) */ int sort() default 0; /** 导出映射,格式如:0-未知;1-男;2-女 */ String kv() default ""; /** 导出模板示例值(有值的话,直接取该值,不做映射) */ String example() default ""; }
6.创建ExcelClassField
package com.ja.excelie; import java.util.LinkedHashMap; public class ExcelClassField { /** 字段名称 */ private String fieldName; /** 表头名称 */ private String name; /** 映射关系 */ private LinkedHashMapkvMap; /** 示例值 */ private Object example; /** 排序 */ private int sort; /** 是否为注解字段:0-否,1-是 */ private int hasAnnotation; public String getFieldName() { return fieldName; } public void setFieldName(String fieldName) { this.fieldName = fieldName; } public String getName() { return name; } public void setName(String name) { this.name = name; } public LinkedHashMap getKvMap() { return kvMap; } public void setKvMap(LinkedHashMap kvMap) { this.kvMap = kvMap; } public Object getExample() { return example; } public void setExample(Object example) { this.example = example; } public int getSort() { return sort; } public void setSort(int sort) { this.sort = sort; } public int getHasAnnotation() { return hasAnnotation; } public void setHasAnnotation(int hasAnnotation) { this.hasAnnotation = hasAnnotation; } }
7.创建实体(导入和导出的)
package com.ja.entity; import com.baomidou.mybatisplus.annotation.TableField; import com.ja.excelie.ExcelExport; import com.ja.excelie.ExcelImport; import lombok.Data; /** * 说明 * maxLength限制字段长度 * kv:转换 * required = true:必填字段 * unique = true:某个字段是具有唯一性的 */ @Data public class User { //实体中有导入和导出 // 获取行号 // @TableField(exist = false) // private int rowNum; // // @ExcelImport(value = "姓名",required = true) // private String name; // // @ExcelImport("年龄") // private Integer age; // // @ExcelImport(value = "性别",kv = "1-男;2-女") // private String sex; // // @ExcelImport(value = "电话") // private String tel; // // @ExcelImport("城市") // private String city; // // //表中的原始数据,如果某行存在错误,一般我们会将原始的数据拿出来分析,为什么会造成数据错误 // @TableField(exist = false) // private String rowData; // // //数据格式错误提示 // @TableField(exist = false) // private String rowTips; /** * 导出的实体内容说明 *kv:对值进行转换 * sort:表头顺序,越小越前 */ @ExcelExport(value = "姓名",sort = 1) private String name; @ExcelExport(value = "年龄",sort = 2) private Integer age; @ExcelExport(value = "性别",kv = "1-男;2-女",sort = 3) private String sex; @ExcelExport(value = "电话",sort = 5) private String tel; @ExcelExport(value = "城市",sort = 4) private String city; }
8.创建接口
package com.ja.mapper; import com.baomidou.mybatisplus.core.mapper.BaseMapper; import com.ja.entity.User; import org.apache.ibatis.annotations.Mapper; @Mapper public interface UserMapper extends BaseMapper{ }
9.service
package com.ja.service; import com.baomidou.mybatisplus.extension.service.IService; import com.ja.entity.User; public interface UserService extends IService{ }
package com.ja.service.impl; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.ja.entity.User; import com.ja.mapper.UserMapper; import com.ja.service.UserService; import org.springframework.stereotype.Service; @Service public class UserServiceImpl extends ServiceImplimplements UserService { }
10.controller
package com.ja.controller; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.ja.entity.User; import com.ja.excelie.ExcelUtils; import com.ja.service.UserService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; import java.util.ArrayList; import java.util.Arrays; import java.util.Collections; import java.util.List; @Controller public class ExcelController { @Autowired private UserService userService; @RequestMapping("/upload") public String toUpload(){ return "upload"; } /** * 导入 * @param file * @throws Exception */ @PostMapping("/import") @ResponseBody public void inportUser(@RequestParam("file") MultipartFile file) throws Exception { Listusers=ExcelUtils.readMultipartFile(file,User.class); for (User user:users){ String[] split = user.getTel().split(","); StringBuffer phone = new StringBuffer(); for (String s : split){ phone.append(s); } user.setTel(phone.toString()); boolean save = userService.save(user); } } /** * 导出 * @return */ @RequestMapping("/export") public void export(HttpServletResponse response){ // List > allList=new ArrayList<>();
// Listhead=Arrays.asList("编号","姓名","年龄","性别","手机","城市","数据","错误"); // allList.add(head); // allList.add(Collections.singletonList(list)); Listlist = userService.list(); //导出数据 ExcelUtils.export(response,"用户表数据导出",list,User.class); } }
参考文档:https://www.jb51.net/article/236477.htm#_label0
自己的练习路径:E:\myprogram\lx\excelie