图片等比压缩工具类

结合其他同学的案例封装了个图片等比压缩工具类，可以设置预期希望压缩的大小，由于保留了图片宽高比，压缩出来的图片大小实际只是是接近预期值。

package com.ksmart.test.image;

import lombok.extern.log4j.Log4j2;
import org.junit.jupiter.api.Test;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;


/**
 * 图片压缩处理类
 *
 * @author bbkkkk8@163.com
 *
 */
@Log4j2
public class ImageCompressUtil {
    /**
     * 常用文件的文件头如下：(以前六位为准)
     * JPEG (jpg)，文件头：FFD8FF
     * PNG (png)，文件头：89504E47
     * GIF (gif)，文件头：47494638
     * TIFF (tif)，文件头：49492A00
     * Windows Bitmap (bmp)，文件头：424D
     * CAD (dwg)，文件头：41433130
     * Adobe Photoshop (psd)，文件头：38425053
     * Rich Text Format (rtf)，文件头：7B5C727466
     * XML (xml)，文件头：3C3F786D6C
     * HTML (html)，文件头：68746D6C3E
     * Email [thorough only] (eml)，文件头：44656C69766572792D646174653A
     * Outlook Express (dbx)，文件头：CFAD12FEC5FD746F
     * Outlook (pst)，文件头：2142444E
     * MS Word/Excel (xls.or.doc)，文件头：D0CF11E0
     * MS Access (mdb)，文件头：5374616E64617264204A
     * WordPerfect (wpd)，文件头：FF575043
     * Postscript (eps.or.ps)，文件头：252150532D41646F6265
     * Adobe Acrobat (pdf)，文件头：255044462D312E
     * Quicken (qdf)，文件头：AC9EBD8F
     * Windows Password (pwl)，文件头：E3828596
     * ZIP Archive (zip)，文件头：504B0304
     * RAR Archive (rar)，文件头：52617221
     * Wave (wav)，文件头：57415645
     * AVI (avi)，文件头：41564920
     * Real Audio (ram)，文件头：2E7261FD
     * Real Media (rm)，文件头：2E524D46
     * MPEG (mpg)，文件头：000001BA
     * MPEG (mpg)，文件头：000001B3
     * Quicktime (mov)，文件头：6D6F6F76
     * Windows Media (asf)，文件头：3026B2758E66CF11
     * MIDI (mid)，文件头：4D546864
     */


    public static final String TYPE_JPG = "jpg";
    public static final String TYPE_GIF = "gif";
    public static final String TYPE_PNG = "png";
    public static final String TYPE_BMP = "bmp";
    public static final String TYPE_WEBP = "webp";
    public static final String TYPE_TIF = "tif";
    public static final String TYPE_UNKNOWN = "unknown";

    /**
     * byte数组转换成16进制字符串
     *
     * @param src
     * @return
     */
    public static String bytesToHexString(byte[] src) {
        StringBuilder stringBuilder = new StringBuilder();
        if (src == null || src.length <= 0) {
            return null;
        }
        for (int i = 0; i < src.length; i++) {
            int v = src[i] & 0xFF;
            String hv = Integer.toHexString(v);
            if (hv.length() < 2) {
                stringBuilder.append(0);
            }
            stringBuilder.append(hv);
        }
        return stringBuilder.toString();
    }

    /**
     * 根据文件流判断图片类型
     *
     * @param fis
     * @return jpg/png/gif/bmp
     */
    public static String getPicType(FileInputStream fis) {
        //读取文件的前几个字节来判断图片格式
        byte[] b = new byte[4];
        try {
            fis.read(b, 0, b.length);
            String type = bytesToHexString(b).toUpperCase();
            if (type.contains("FFD8FF")) {
                return TYPE_JPG;
            } else if (type.contains("89504E47")) {
                return TYPE_PNG;
            } else if (type.contains("47494638")) {
                return TYPE_GIF;
            } else if (type.contains("424D")) {
                return TYPE_BMP;
            } else if (type.contains("52494646")) {
                return TYPE_WEBP;
            } else if (type.contains("49492A00")) {
                return TYPE_TIF;
            } else {
                return TYPE_UNKNOWN;
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (fis != null) {
                try {
                    fis.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        return TYPE_UNKNOWN;
    }

    /**
     * 生成缩略图
     * 图片压缩格式，支持bmp|gif|jpg|jpeg|png
     *
     * @param srcFile        要被压缩的图片
     * @param targetFilePath 原文件路径
     * @param smallSize      文件压缩倍数
     * @param formatName     图片压缩格式，支持bmp|gif|jpg|jpeg|png
     * @return
     */
    public static boolean imageGenerateSmall(File srcFile, String targetFilePath, double smallSize, String formatName) {
        try {
            Image image = ImageIO.read(srcFile);
            int width = image.getWidth(null);
            int height = image.getHeight(null);
            int widthSmall = (int) (width / smallSize);
            int heightSmall = (int) (height / smallSize);
            BufferedImage bufferedImage = new BufferedImage(widthSmall, heightSmall, BufferedImage.TYPE_INT_RGB);
            Graphics g = bufferedImage.getGraphics();
            g.drawImage(image, 0, 0, widthSmall, heightSmall, null);
            g.dispose();
            //  图片压缩格式，支持bmp|gif|jpg|jpeg|png
            ImageIO.write(bufferedImage, formatName, new File(targetFilePath));
        } catch (IOException e) {
            e.printStackTrace();
            return false;
        }
        return true;
    }

    /**
     * 图片压缩，图片大小超过100K自动按比例压缩到100K以下
     *
     * @param fileOrPath     文件 可以传String文件路径 或者File对象
     * @param kb             期望压缩到多少kb，已经乘了1024；实际输出的值是接近的最优解，因为图片保留了宽高比例
     * @param targetFilePath 压缩后路径
     * @return
     * @throws Exception
     */
    public static boolean imageCompress(Object fileOrPath, int kb, String targetFilePath) throws Exception {
        boolean rs = true;
//        int kb = 100 * 1024;
        long fileSize = 0l;
        File srcFile = null;
        if (fileOrPath instanceof File) {
            srcFile = (File) fileOrPath;
        } else if (fileOrPath instanceof String) {
            srcFile = new File((String) fileOrPath);
        } else {
            return false;
        }
        String formatName = getPicType(new FileInputStream(srcFile));
        if (formatName.equals(TYPE_UNKNOWN) || formatName.equals(TYPE_WEBP)) {
            log.warn("不支持的压缩类型" + formatName);
            return false;
        }
        fileSize = srcFile.length();
        kb = kb * 1024;
        if (fileSize > kb) {
            int smallSize = (int) (fileSize % kb == 0 ? fileSize / kb : fileSize / kb + 1);
            double size = Math.ceil(Math.sqrt(smallSize));
            rs = ImageCompressUtil.imageGenerateSmall(srcFile, targetFilePath, size, formatName);
        }
        return rs;
    }

    @Test
    void testJpg() throws Exception {
        File file = new File("E:\\testImage\\1.jpg");
        System.out.println("--------------文件大小（byte）----------------:" + file.length());
        imageCompress(file, 150, "E:\\testImage\\2.jpg");
    }

    @Test
    void testPng() throws Exception {
        File file = new File("E:\\testImage\\p1.png");
        System.out.println("--------------文件大小（byte）----------------:" + file.length());
        imageCompress("E:\\testImage\\p1.png", 30, "E:\\testImage\\p2.png");
    }
}