| | |
|---|
| | | }
|
|---|
| | |
|
|---|
| | | /**
|
|---|
| | | * 扩展：读取Excel中多个指定Sheet的数据
|
|---|
| | | * @param sheetNameList 要读取的Sheet名称列表（null则读取所有Sheet）
|
|---|
| | | * @param is 输入流
|
|---|
| | | * @param titleNum 标题占用行数
|
|---|
| | | * @return Map<Sheet名称, 对应Sheet的数据列表>
|
|---|
| | | */
|
|---|
| | | public Map<String, List<T>> importExcelMultiSheet(List<String> sheetNameList, InputStream is, int titleNum) {
|
|---|
| | | Map<String, List<T>> resultMap = new HashMap<>();
|
|---|
| | | try {
|
|---|
| | | this.type = Type.IMPORT;
|
|---|
| | | this.wb = WorkbookFactory.create(is);
|
|---|
| | |
|
|---|
| | | // 1. 确定要读取的Sheet列表
|
|---|
| | | List<Sheet> sheetsToRead = new ArrayList<>();
|
|---|
| | | if (sheetNameList != null && !sheetNameList.isEmpty()) {
|
|---|
| | | // 读取指定名称的Sheet
|
|---|
| | | for (String sheetName : sheetNameList) {
|
|---|
| | | Sheet sheet = wb.getSheet(sheetName);
|
|---|
| | | if (sheet != null) {
|
|---|
| | | sheetsToRead.add(sheet);
|
|---|
| | | } else {
|
|---|
| | | log.warn("指定的Sheet名称不存在：{}", sheetName);
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | } else {
|
|---|
| | | // 读取所有Sheet
|
|---|
| | | int sheetCount = wb.getNumberOfSheets();
|
|---|
| | | for (int i = 0; i < sheetCount; i++) {
|
|---|
| | | sheetsToRead.add(wb.getSheetAt(i));
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 2. 遍历每个Sheet，复用原有导入逻辑
|
|---|
| | | for (Sheet sheet : sheetsToRead) {
|
|---|
| | | String sheetName = wb.getSheetName(wb.getSheetIndex(sheet));
|
|---|
| | | // 复用原有核心导入逻辑（关键：将原有方法拆分为可复用的内部方法）
|
|---|
| | | List<T> sheetData = importExcelBySheet(sheet, titleNum);
|
|---|
| | | resultMap.put(sheetName, sheetData);
|
|---|
| | | }
|
|---|
| | | } catch (Exception e) {
|
|---|
| | | log.error("导入多Sheet Excel异常{}", e.getMessage());
|
|---|
| | | throw new UtilException(e.getMessage());
|
|---|
| | | } finally {
|
|---|
| | | IOUtils.closeQuietly(is);
|
|---|
| | | }
|
|---|
| | | return resultMap;
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | /**
|
|---|
| | | * 内部复用方法：根据指定Sheet对象读取数据（提取原有importExcel的核心逻辑）
|
|---|
| | | */
|
|---|
| | | private List<T> importExcelBySheet(Sheet sheet, int titleNum) throws Exception {
|
|---|
| | | List<T> list = new ArrayList<T>();
|
|---|
| | | if (sheet == null) {
|
|---|
| | | return list;
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | boolean isXSSFWorkbook = !(wb instanceof HSSFWorkbook);
|
|---|
| | | Map<String, List<PictureData>> pictures = null;
|
|---|
| | | if (isXSSFWorkbook) {
|
|---|
| | | pictures = getSheetPictures07((XSSFSheet) sheet, (XSSFWorkbook) wb);
|
|---|
| | | } else {
|
|---|
| | | pictures = getSheetPictures03((HSSFSheet) sheet, (HSSFWorkbook) wb);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | int rows = sheet.getLastRowNum();
|
|---|
| | | if (rows > 0) {
|
|---|
| | | Map<String, Integer> cellMap = new HashMap<String, Integer>();
|
|---|
| | | Row heard = sheet.getRow(titleNum);
|
|---|
| | | for (int i = 0; i < heard.getPhysicalNumberOfCells(); i++) {
|
|---|
| | | Cell cell = heard.getCell(i);
|
|---|
| | | if (StringUtils.isNotNull(cell)) {
|
|---|
| | | String value = this.getCellValue(heard, i).toString();
|
|---|
| | | cellMap.put(value, i);
|
|---|
| | | } else {
|
|---|
| | | cellMap.put(null, i);
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | List<Object[]> fields = this.getFields();
|
|---|
| | | Map<Integer, Object[]> fieldsMap = new HashMap<Integer, Object[]>();
|
|---|
| | | for (Object[] objects : fields) {
|
|---|
| | | Excel attr = (Excel) objects[1];
|
|---|
| | | Integer column = cellMap.get(attr.name());
|
|---|
| | | if (column != null) {
|
|---|
| | | fieldsMap.put(column, objects);
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | for (int i = titleNum + 1; i <= rows; i++) {
|
|---|
| | | Row row = sheet.getRow(i);
|
|---|
| | | if (isRowEmpty(row)) {
|
|---|
| | | continue;
|
|---|
| | | }
|
|---|
| | | T entity = null;
|
|---|
| | | for (Map.Entry<Integer, Object[]> entry : fieldsMap.entrySet()) {
|
|---|
| | | Object val = this.getCellValue(row, entry.getKey());
|
|---|
| | | entity = (entity == null ? clazz.newInstance() : entity);
|
|---|
| | | Field field = (Field) entry.getValue()[0];
|
|---|
| | | Excel attr = (Excel) entry.getValue()[1];
|
|---|
| | | Class<?> fieldType = field.getType();
|
|---|
| | |
|
|---|
| | | // 以下是原有数据类型转换、字典解析等逻辑（完全复用）
|
|---|
| | | if (String.class == fieldType) {
|
|---|
| | | String s = Convert.toStr(val);
|
|---|
| | | if (s.matches("^\\d+\\.0$")) {
|
|---|
| | | val = StringUtils.substringBefore(s, ".0");
|
|---|
| | | } else {
|
|---|
| | | String dateFormat = field.getAnnotation(Excel.class).dateFormat();
|
|---|
| | | if (StringUtils.isNotEmpty(dateFormat)) {
|
|---|
| | | val = parseDateToStr(dateFormat, val);
|
|---|
| | | } else {
|
|---|
| | | val = Convert.toStr(val);
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | } else if ((Integer.TYPE == fieldType || Integer.class == fieldType) && StringUtils.isNumeric(Convert.toStr(val))) {
|
|---|
| | | val = Convert.toInt(val);
|
|---|
| | | } else if ((Long.TYPE == fieldType || Long.class == fieldType) && StringUtils.isNumeric(Convert.toStr(val))) {
|
|---|
| | | val = Convert.toLong(val);
|
|---|
| | | } else if (Double.TYPE == fieldType || Double.class == fieldType) {
|
|---|
| | | val = Convert.toDouble(val);
|
|---|
| | | } else if (Float.TYPE == fieldType || Float.class == fieldType) {
|
|---|
| | | val = Convert.toFloat(val);
|
|---|
| | | } else if (BigDecimal.class == fieldType) {
|
|---|
| | | val = Convert.toBigDecimal(val);
|
|---|
| | | } else if (Date.class == fieldType) {
|
|---|
| | | if (val instanceof String) {
|
|---|
| | | val = DateUtils.parseDate(val);
|
|---|
| | | } else if (val instanceof Double) {
|
|---|
| | | val = DateUtil.getJavaDate((Double) val);
|
|---|
| | | }
|
|---|
| | | } else if (Boolean.TYPE == fieldType || Boolean.class == fieldType) {
|
|---|
| | | val = Convert.toBool(val, false);
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | if (StringUtils.isNotNull(fieldType)) {
|
|---|
| | | String propertyName = field.getName();
|
|---|
| | | if (StringUtils.isNotEmpty(attr.targetAttr())) {
|
|---|
| | | propertyName = field.getName() + "." + attr.targetAttr();
|
|---|
| | | }
|
|---|
| | | if (StringUtils.isNotEmpty(attr.readConverterExp())) {
|
|---|
| | | val = reverseByExp(Convert.toStr(val), attr.readConverterExp(), attr.separator());
|
|---|
| | | } else if (StringUtils.isNotEmpty(attr.dictType())) {
|
|---|
| | | if (!sysDictMap.containsKey(attr.dictType() + val)) {
|
|---|
| | | String dictValue = reverseDictByExp(Convert.toStr(val), attr.dictType(), attr.separator());
|
|---|
| | | sysDictMap.put(attr.dictType() + val, dictValue);
|
|---|
| | | }
|
|---|
| | | val = sysDictMap.get(attr.dictType() + val);
|
|---|
| | | } else if (!attr.handler().equals(ExcelHandlerAdapter.class)) {
|
|---|
| | | val = dataFormatHandlerAdapter(val, attr, null);
|
|---|
| | | } else if (ColumnType.IMAGE == attr.cellType() && StringUtils.isNotEmpty(pictures)) {
|
|---|
| | | StringBuilder propertyString = new StringBuilder();
|
|---|
| | | List<PictureData> images = pictures.get(row.getRowNum() + "_" + entry.getKey());
|
|---|
| | | for (PictureData picture : images) {
|
|---|
| | | byte[] data = picture.getData();
|
|---|
| | | String fileName = FileUtils.writeImportBytes(data);
|
|---|
| | | propertyString.append(fileName).append(SEPARATOR);
|
|---|
| | | }
|
|---|
| | | val = StringUtils.stripEnd(propertyString.toString(), SEPARATOR);
|
|---|
| | | }
|
|---|
| | | ReflectUtils.invokeSetter(entity, propertyName, val);
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | list.add(entity);
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | return list;
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | /**
|
|---|
| | | * 对list数据源将其里面的数据导入到excel表单
|
|---|
| | | *
|
|---|
| | | * @param list 导出数据集合
|
|---|
