图片变色报错

jero-boot/jero-boot-modules/src/main/java/com/jero/modules/compare/service/impl/SarFileCompareItemCommentServiceImpl.java

@@ -30,6 +30,7 @@ import javax.imageio.ImageIO;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import javax.servlet.http.HttpSession;
+import java.awt.*;
 import java.awt.image.BufferedImage;
 import java.io.*;
 import java.nio.file.Files;
@@ -37,6 +38,7 @@ import java.nio.file.Paths;
 import java.nio.file.StandardOpenOption;
 import java.text.SimpleDateFormat;
 import java.util.*;
+import java.util.List;
 
 /**
  * @Description: 文档对比信息条款评论表
@@ -485,7 +487,19 @@ public class SarFileCompareItemCommentServiceImpl extends ServiceImpl<SarFileCom
             if (file.exists()) {
                 log.info("-------------------------将图片插入excel指定单元格中----------------------------");
                 BufferedImage bufferImg = ImageIO.read(file);
-                ImageIO.write(bufferImg, "jpg", byteArrayOut);
+                //获取文件后缀
+                String fileName = file.getName();
+                String formatName = fileName.substring(fileName.lastIndexOf(".") + 1);
+                //如果是jpg或者是jpeg，需要重画一下，否则会变色
+                if (formatName.equalsIgnoreCase("jpg") || formatName.equalsIgnoreCase("jpeg")) { //重画一下，要么会变色
+                    BufferedImage tag;
+                    tag = new BufferedImage(bufferImg.getWidth(), bufferImg.getHeight(), BufferedImage.TYPE_INT_BGR);
+                    Graphics g = tag.getGraphics();
+                    g.drawImage(bufferImg, 0, 0, null); // 绘制缩小后的图
+                    g.dispose();
+                    bufferImg = tag;
+                }
+                ImageIO.write(bufferImg, formatName, byteArrayOut);
                 //anchor主要用于设置图片的属性
                 HSSFClientAnchor anchor = new HSSFClientAnchor(0, 0, 600, 200, (short) cellNum, j + i, (short) cellNum, j + i);
                 patriarch.createPicture(anchor, workbook.addPicture(byteArrayOut.toByteArray(), HSSFWorkbook.PICTURE_TYPE_JPEG));