Excel导入导出支持多图片

RuoYi

2025-04-25 79c4711f638585951c4fc4cdfd689736bebb7ddf

 src/main/java/com/ruoyi/common/utils/poi/ExcelUtil.java

@@ -95,6 +95,8 @@
{

    private static final Logger log = LoggerFactory.getLogger(ExcelUtil.class);



    public static final String SEPARATOR = ",";



    public static final String FORMULA_REGEX_STR = "=|-|\\+|@";



    public static final String[] FORMULA_STR = { "=", "-", "+", "@" };

@@ -252,8 +254,6 @@
    {

        if (StringUtils.isNotEmpty(title))

        {

            subMergedFirstRowNum++;

            subMergedLastRowNum++;

            int titleLastCol = this.fields.size() - 1;

            if (isSubList())

            {

@@ -264,7 +264,7 @@
            Cell titleCell = titleRow.createCell(0);

            titleCell.setCellStyle(styles.get("title"));

            titleCell.setCellValue(title);

            sheet.addMergedRegion(new CellRangeAddress(titleRow.getRowNum(), titleRow.getRowNum(), titleRow.getRowNum(), titleLastCol));

            sheet.addMergedRegion(new CellRangeAddress(titleRow.getRowNum(), titleRow.getRowNum(), 0, titleLastCol));

        }

    }



@@ -275,23 +275,31 @@
    {

        if (isSubList())

        {

            subMergedFirstRowNum++;

            subMergedLastRowNum++;

            Row subRow = sheet.createRow(rownum);

            int excelNum = 0;

            int column = 0;

            int subFieldSize = subFields != null ? subFields.size() : 0;

            for (Object[] objects : fields)

            {

                Field field = (Field) objects[0];

                Excel attr = (Excel) objects[1];

                Cell headCell1 = subRow.createCell(excelNum);

                headCell1.setCellValue(attr.name());

                headCell1.setCellStyle(styles.get(StringUtils.format("header_{}_{}", attr.headerColor(), attr.headerBackgroundColor())));

                excelNum++;

            }

            int headFirstRow = excelNum - 1;

            int headLastRow = headFirstRow + subFields.size() - 1;

            if (headLastRow > headFirstRow)

            {

                sheet.addMergedRegion(new CellRangeAddress(rownum, rownum, headFirstRow, headLastRow));

                if (Collection.class.isAssignableFrom(field.getType()))

                {

                    Cell cell = subRow.createCell(column);

                    cell.setCellValue(attr.name());

                    cell.setCellStyle(styles.get(StringUtils.format("header_{}_{}", attr.headerColor(), attr.headerBackgroundColor())));

                    if (subFieldSize > 1)

                    {

                        CellRangeAddress cellAddress = new CellRangeAddress(rownum, rownum, column, column + subFieldSize - 1);

                        sheet.addMergedRegion(cellAddress);

                    }

                    column += subFieldSize;

                }

                else

                {

                    Cell cell = subRow.createCell(column++);

                    cell.setCellValue(attr.name());

                    cell.setCellStyle(styles.get(StringUtils.format("header_{}_{}", attr.headerColor(), attr.headerBackgroundColor())));

                }

            }

            rownum++;

        }

@@ -305,10 +313,22 @@
     */

    public List<T> importExcel(InputStream is)

    {

        return importExcel(is, 0);

    }



    /**

     * 对excel表单默认第一个索引名转换成list

     * 
     * @param is 输入流

     * @param titleNum 标题占用行数

     * @return 转换后集合

     */

    public List<T> importExcel(InputStream is, int titleNum)

    {

        List<T> list = null;

        try

        {

            list = importExcel(is, 0);

            list = importExcel(StringUtils.EMPTY, is, titleNum);

        }

        catch (Exception e)

        {

@@ -320,18 +340,6 @@
            IOUtils.closeQuietly(is);

        }

        return list;

    }



    /**

     * 对excel表单默认第一个索引名转换成list

     * 
     * @param is 输入流

     * @param titleNum 标题占用行数

     * @return 转换后集合

     */

    public List<T> importExcel(InputStream is, int titleNum) throws Exception

    {

        return importExcel(StringUtils.EMPTY, is, titleNum);

    }



    /**

@@ -354,7 +362,7 @@
            throw new IOException("文件sheet不存在");

        }

        boolean isXSSFWorkbook = !(wb instanceof HSSFWorkbook);

        Map<String, PictureData> pictures;

        Map<String, List<PictureData>> pictures = null;

        if (isXSSFWorkbook)

        {

            pictures = getSheetPictures07((XSSFSheet) sheet, (XSSFWorkbook) wb);

@@ -498,16 +506,15 @@
                        }

                        else if (ColumnType.IMAGE == attr.cellType() && StringUtils.isNotEmpty(pictures))

                        {

                            PictureData image = pictures.get(row.getRowNum() + "_" + entry.getKey());

                            if (image == null)

                            StringBuilder propertyString = new StringBuilder();

                            List<PictureData> images = pictures.get(row.getRowNum() + "_" + entry.getKey());

                            for (PictureData picture : images)

                            {

                                val = "";

                                byte[] data = picture.getData();

                                String fileName = FileUtils.writeImportBytes(data);

                                propertyString.append(fileName).append(SEPARATOR);

                            }

                            else

                            {

                                byte[] data = image.getData();

                                val = FileUtils.writeImportBytes(data);

                            }

                            val = StringUtils.stripEnd(propertyString.toString(), SEPARATOR);

                        }

                        ReflectUtils.invokeSetter(entity, propertyName, val);

                    }

@@ -725,64 +732,91 @@
    {

        int startNo = index * sheetSize;

        int endNo = Math.min(startNo + sheetSize, list.size());

        int rowNo = (1 + rownum) - startNo;

        int currentRowNum = rownum + 1; // 从标题行后开始



        for (int i = startNo; i < endNo; i++)

        {

            rowNo = isSubList() ? (i > 1 ? rowNo + 1 : rowNo + i) : i + 1 + rownum - startNo;

            row = sheet.createRow(rowNo);

            // 得到导出对象.

            row = sheet.createRow(currentRowNum);

            T vo = (T) list.get(i);

            Collection<?> subList = null;

            if (isSubList())

            {

                if (isSubListValue(vo))

                {

                    subList = getListCellValue(vo);

                    subMergedLastRowNum = subMergedLastRowNum + subList.size();

                }

                else

                {

                    subMergedFirstRowNum++;

                    subMergedLastRowNum++;

                }

            }

            int column = 0;

            int maxSubListSize = getCurrentMaxSubListSize(vo);

            for (Object[] os : fields)

            {

                Field field = (Field) os[0];

                Excel excel = (Excel) os[1];

                if (Collection.class.isAssignableFrom(field.getType()) && StringUtils.isNotNull(subList))

                if (Collection.class.isAssignableFrom(field.getType()))

                {

                    boolean subFirst = false;

                    for (Object obj : subList)

                    try

                    {

                        if (subFirst)

                        Collection<?> subList = (Collection<?>) getTargetValue(vo, field, excel);

                        if (subList != null && !subList.isEmpty())

                        {

                            rowNo++;

                            row = sheet.createRow(rowNo);

                        }

                        List<Field> subFields = FieldUtils.getFieldsListWithAnnotation(obj.getClass(), Excel.class);

                        int subIndex = 0;

                        for (Field subField : subFields)

                        {

                            if (subField.isAnnotationPresent(Excel.class))

                            int subIndex = 0;

                            for (Object subVo : subList)

                            {

                                subField.setAccessible(true);

                                Excel attr = subField.getAnnotation(Excel.class);

                                this.addCell(attr, row, (T) obj, subField, column + subIndex);

                                Row subRow = sheet.getRow(currentRowNum + subIndex);

                                if (subRow == null)

                                {

                                    subRow = sheet.createRow(currentRowNum + subIndex);

                                }



                                int subColumn = column;

                                for (Field subField : subFields)

                                {

                                    Excel subExcel = subField.getAnnotation(Excel.class);

                                    addCell(subExcel, subRow, (T) subVo, subField, subColumn++);

                                }

                                subIndex++;

                            }

                            subIndex++;

                            column += subFields.size();

                        }

                        subFirst = true;

                    }

                    this.subMergedFirstRowNum = this.subMergedFirstRowNum + subList.size();

                    catch (Exception e)

                    {

                        log.error("填充集合数据失败", e);

                    }

                }

                else

                {

                    this.addCell(excel, row, vo, field, column++);

                    // 创建单元格并设置值

                    addCell(excel, row, vo, field, column);

                    if (maxSubListSize > 1 && excel.needMerge())

                    {

                        sheet.addMergedRegion(new CellRangeAddress(currentRowNum, currentRowNum + maxSubListSize - 1, column, column));

                    }

                    column++;

                }

            }

            currentRowNum += maxSubListSize;

        }

    }



    /**

     * 获取子列表最大数

     */

    private int getCurrentMaxSubListSize(T vo)

    {

        int maxSubListSize = 1;

        for (Object[] os : fields)

        {

            Field field = (Field) os[0];

            if (Collection.class.isAssignableFrom(field.getType()))

            {

                try

                {

                    Collection<?> subList = (Collection<?>) getTargetValue(vo, field, (Excel) os[1]);

                    if (subList != null && !subList.isEmpty())

                    {

                        maxSubListSize = Math.max(maxSubListSize, subList.size());

                    }

                }

                catch (Exception e)

                {

                    log.error("获取集合大小失败", e);

                }

            }

        }

        return maxSubListSize;

    }



    /**

@@ -917,7 +951,7 @@
     */

    public void annotationDataStyles(Map<String, CellStyle> styles, Field field, Excel excel)

    {

        String key = StringUtils.format("data_{}_{}_{}_{}", excel.align(), excel.color(), excel.backgroundColor(), excel.cellType());

        String key = StringUtils.format("data_{}_{}_{}_{}_{}", excel.align(), excel.color(), excel.backgroundColor(), excel.cellType(), excel.wrapText());

        if (!styles.containsKey(key))

        {

            CellStyle style = wb.createCellStyle();

@@ -933,6 +967,7 @@
            style.setBottomBorderColor(IndexedColors.GREY_50_PERCENT.getIndex());

            style.setFillPattern(FillPatternType.SOLID_FOREGROUND);

            style.setFillForegroundColor(excel.backgroundColor().getIndex());

            style.setWrapText(excel.wrapText());

            Font dataFont = wb.createFont();

            dataFont.setFontName("Arial");

            dataFont.setFontHeightInPoints((short) 10);

@@ -961,7 +996,7 @@
        if (isSubList())

        {

            // 填充默认样式，防止合并单元格样式失效

            sheet.setDefaultColumnStyle(column, styles.get(StringUtils.format("data_{}_{}_{}_{}", attr.align(), attr.color(), attr.backgroundColor(), attr.cellType())));

            sheet.setDefaultColumnStyle(column, styles.get(StringUtils.format("data_{}_{}_{}_{}_{}", attr.align(), attr.color(), attr.backgroundColor(), attr.cellType(), attr.wrapText())));

            if (attr.needMerge())

            {

                sheet.addMergedRegion(new CellRangeAddress(rownum - 1, rownum, column, column));

@@ -1003,12 +1038,15 @@
        else if (ColumnType.IMAGE == attr.cellType())

        {

            ClientAnchor anchor = new XSSFClientAnchor(0, 0, 0, 0, (short) cell.getColumnIndex(), cell.getRow().getRowNum(), (short) (cell.getColumnIndex() + 1), cell.getRow().getRowNum() + 1);

            String imagePath = Convert.toStr(value);

            if (StringUtils.isNotEmpty(imagePath))

            String propertyValue = Convert.toStr(value);

            if (StringUtils.isNotEmpty(propertyValue))

            {

                byte[] data = ImageUtils.getImage(imagePath);

                getDrawingPatriarch(cell.getSheet()).createPicture(anchor,

                        cell.getSheet().getWorkbook().addPicture(data, getImageType(data)));

                List<String> imagePaths = StringUtils.str2List(propertyValue, SEPARATOR);

                for (String imagePath : imagePaths)

                {

                    byte[] data = ImageUtils.getImage(imagePath);

                    getDrawingPatriarch(cell.getSheet()).createPicture(anchor, cell.getSheet().getWorkbook().addPicture(data, getImageType(data)));

                }

            }

        }

    }

@@ -1099,10 +1137,12 @@
                cell = row.createCell(column);

                if (isSubListValue(vo) && getListCellValue(vo).size() > 1 && attr.needMerge())

                {

                    CellRangeAddress cellAddress = new CellRangeAddress(subMergedFirstRowNum, subMergedLastRowNum, column, column);

                    sheet.addMergedRegion(cellAddress);

                    if (subMergedLastRowNum >= subMergedFirstRowNum)

                    {

                        sheet.addMergedRegion(new CellRangeAddress(subMergedFirstRowNum, subMergedLastRowNum, column, column));

                    }

                }

                cell.setCellStyle(styles.get(StringUtils.format("data_{}_{}_{}_{}", attr.align(), attr.color(), attr.backgroundColor(), attr.cellType())));

                cell.setCellStyle(styles.get(StringUtils.format("data_{}_{}_{}_{}_{}", attr.align(), attr.color(), attr.backgroundColor(), attr.cellType(), attr.wrapText())));



                // 用于读取对象中的属性

                Object value = getTargetValue(vo, field, attr);

@@ -1112,6 +1152,7 @@
                String dictType = attr.dictType();

                if (StringUtils.isNotEmpty(dateFormat) && StringUtils.isNotNull(value))

                {

                    cell.getCellStyle().setDataFormat(this.wb.getCreationHelper().createDataFormat().getFormat(dateFormat));

                    cell.setCellValue(parseDateToStr(dateFormat, value));

                }

                else if (StringUtils.isNotEmpty(readConverterExp) && StringUtils.isNotNull(value))

@@ -1250,7 +1291,7 @@
    public static String convertByExp(String propertyValue, String converterExp, String separator)

    {

        StringBuilder propertyString = new StringBuilder();

        String[] convertSource = converterExp.split(",");

        String[] convertSource = converterExp.split(SEPARATOR);

        for (String item : convertSource)

        {

            String[] itemArray = item.split("=");

@@ -1287,7 +1328,7 @@
    public static String reverseByExp(String propertyValue, String converterExp, String separator)

    {

        StringBuilder propertyString = new StringBuilder();

        String[] convertSource = converterExp.split(",");

        String[] convertSource = converterExp.split(SEPARATOR);

        for (String item : convertSource)

        {

            String[] itemArray = item.split("=");

@@ -1412,8 +1453,7 @@
     */

    public String encodingFilename(String filename)

    {

        filename = UUID.randomUUID() + "_" + filename + ".xlsx";

        return filename;

        return UUID.randomUUID() + "_" + filename + ".xlsx";

    }



    /**

@@ -1443,6 +1483,7 @@
     */

    private Object getTargetValue(T vo, Field field, Excel excel) throws Exception

    {

        field.setAccessible(true);

        Object o = field.get(vo);

        if (StringUtils.isNotEmpty(excel.targetAttr()))

        {

@@ -1543,7 +1584,6 @@
            Excel attr = field.getAnnotation(Excel.class);

            if (attr != null && (attr.type() == Type.ALL || attr.type() == type))

            {

                field.setAccessible(true);

                fields.add(new Object[] { field, attr });

            }

            if (Collection.class.isAssignableFrom(field.getType()))

@@ -1567,7 +1607,6 @@
                    if (ArrayUtils.contains(this.includeFields, field.getName() + "." + attr.targetAttr())

                            && (attr != null && (attr.type() == Type.ALL || attr.type() == type)))

                    {

                        field.setAccessible(true);

                        fields.add(new Object[] { field, attr });

                    }

                }

@@ -1576,7 +1615,6 @@
                    if (!ArrayUtils.contains(this.excludeFields, field.getName() + "." + attr.targetAttr())

                            && (attr != null && (attr.type() == Type.ALL || attr.type() == type)))

                    {

                        field.setAccessible(true);

                        fields.add(new Object[] { field, attr });

                    }

                }

@@ -1716,30 +1754,24 @@
     * @param workbook 工作簿对象

     * @return Map key:图片单元格索引（1_1）String，value:图片流PictureData

     */

    public static Map<String, PictureData> getSheetPictures03(HSSFSheet sheet, HSSFWorkbook workbook)

    public static Map<String, List<PictureData>> getSheetPictures03(HSSFSheet sheet, HSSFWorkbook workbook)

    {

        Map<String, PictureData> sheetIndexPicMap = new HashMap<String, PictureData>();

        Map<String, List<PictureData>> sheetIndexPicMap = new HashMap<>();

        List<HSSFPictureData> pictures = workbook.getAllPictures();

        if (!pictures.isEmpty())

        if (!pictures.isEmpty() && sheet.getDrawingPatriarch() != null)

        {

            for (HSSFShape shape : sheet.getDrawingPatriarch().getChildren())

            {

                HSSFClientAnchor anchor = (HSSFClientAnchor) shape.getAnchor();

                if (shape instanceof HSSFPicture)

                {

                    HSSFPicture pic = (HSSFPicture) shape;

                    int pictureIndex = pic.getPictureIndex() - 1;

                    HSSFPictureData picData = pictures.get(pictureIndex);

                    HSSFClientAnchor anchor = (HSSFClientAnchor) pic.getAnchor();

                    String picIndex = anchor.getRow1() + "_" + anchor.getCol1();

                    sheetIndexPicMap.put(picIndex, picData);

                    sheetIndexPicMap.computeIfAbsent(picIndex, k -> new ArrayList<>()).add(pic.getPictureData());

                }

            }

            return sheetIndexPicMap;

        }

        else

        {

            return sheetIndexPicMap;

        }

        return sheetIndexPicMap;

    }



    /**

@@ -1749,16 +1781,15 @@
     * @param workbook 工作簿对象

     * @return Map key:图片单元格索引（1_1）String，value:图片流PictureData

     */

    public static Map<String, PictureData> getSheetPictures07(XSSFSheet sheet, XSSFWorkbook workbook)

    public static Map<String, List<PictureData>> getSheetPictures07(XSSFSheet sheet, XSSFWorkbook workbook)

    {

        Map<String, PictureData> sheetIndexPicMap = new HashMap<String, PictureData>();

        Map<String, List<PictureData>> sheetIndexPicMap = new HashMap<>();

        for (POIXMLDocumentPart dr : sheet.getRelations())

        {

            if (dr instanceof XSSFDrawing)

            {

                XSSFDrawing drawing = (XSSFDrawing) dr;

                List<XSSFShape> shapes = drawing.getShapes();

                for (XSSFShape shape : shapes)

                for (XSSFShape shape : drawing.getShapes())

                {

                    if (shape instanceof XSSFPicture)

                    {

@@ -1766,7 +1797,7 @@
                        XSSFClientAnchor anchor = pic.getPreferredSize();

                        CTMarker ctMarker = anchor.getFrom();

                        String picIndex = ctMarker.getRow() + "_" + ctMarker.getCol();

                        sheetIndexPicMap.put(picIndex, pic.getPictureData());

                        sheetIndexPicMap.computeIfAbsent(picIndex, k -> new ArrayList<>()).add(pic.getPictureData());

                    }

                }

            }