springboot整合thumbnailator實現圖片壓縮

前言

最近由於首頁產品列表圖片顯示太慢，經過研究發現是用戶上傳的圖片太大。

針對這個問題，想到的解決方案是：

1、 產品上傳時，限定圖片上傳大小不超過2m

2、 上傳成功後將產品圖片進行壓縮，但是保留原圖片，壓縮後的圖片名稱添加後綴”-thumbnail”

3、 對已經上傳的產品圖片全部進行壓縮

4、 前端只有在點擊查看產品大圖時顯示原圖，其他情況均顯示縮略圖

實現

根據需求，找到的解決方法是使用net.coobird.thumbnailator依賴包，實現圖片壓縮和將指定目錄下的圖片全部壓縮的功能

引入maven依賴

<dependency>
    <groupId>net.coobird</groupId>
    <artifactId>thumbnailator</artifactId>
    <version>0.4.8</version>
</dependency>

具體實現

ImageUtil類

import net.coobird.thumbnailator.Thumbnails;
import net.coobird.thumbnailator.name.Rename;

 
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * 圖片壓縮工具類
 *
 * @author lnj
 * createTime 2018-10-19 15:31
 **/
public class ImageUtil {

    // 圖片默認縮放比率
    private static final double DEFAULT_SCALE = 0.8d;

    // 縮略圖後綴
    private static
 
 final String SUFFIX = "-thumbnail";


    /**
     * 生成縮略圖到指定的目錄
     *
     * @param path  目錄
     * @param files 要生成縮略圖的文件列表
     * @throws IOException
     */
    public static List<String> generateThumbnail2Directory(String path, String... files) throws IOException {
        return generateThumbnail2Directory(DEFAULT_SCALE, path, files);
    }

    /**
     * 生成縮略圖到指定的目錄
     *
     * @param scale    圖片縮放率
     * @param pathname 縮略圖保存目錄
     * @param files    要生成縮略圖的文件列表
     * @throws IOException
     */
    public static List<String> generateThumbnail2Directory(double scale, String pathname, String... files) throws IOException {
        Thumbnails.of(files)
                // 圖片縮放率，不能和size()一起使用
                .scale(scale)
                // 縮略圖保存目錄,該目錄需存在，否則報錯
                .toFiles(new File(pathname), Rename.SUFFIX_HYPHEN_THUMBNAIL);
        List<String> list = new ArrayList<>(files.length);
        for (String file : files) {
            list.add(appendSuffix(file, SUFFIX));
        }
        return list;
    }

    /**
     * 將指定目錄下所有圖片生成縮略圖
     *
     * @param pathname 文件目錄
     */
    public static void generateDirectoryThumbnail(String pathname) throws IOException {
        generateDirectoryThumbnail(pathname, DEFAULT_SCALE);
    }

    /**
     * 將指定目錄下所有圖片生成縮略圖
     *
     * @param pathname 文件目錄
     */
    public static void generateDirectoryThumbnail(String pathname, double scale) throws IOException {
        File[] files = new File(pathname).listFiles();
        compressRecurse(files, pathname);
    }

    /**
     * 文件追加後綴
     *
     * @param fileName 原文件名
     * @param suffix   文件後綴
     * @return
     */
    public static String appendSuffix(String fileName, String suffix) {
        String newFileName = "";

        int indexOfDot = fileName.lastIndexOf(‘.‘);

        if (indexOfDot != -1) {
            newFileName = fileName.substring(0, indexOfDot);
            newFileName += suffix;
            newFileName += fileName.substring(indexOfDot);
        } else {
            newFileName = fileName + suffix;
        }

        return newFileName;
    }


    private static void compressRecurse(File[] files, String pathname) throws IOException {
        for (File file : files) {
            // 目錄
            if (file.isDirectory()) {
                File[] subFiles = file.listFiles();
                compressRecurse(subFiles, pathname + File.separator + file.getName());
            } else {
                // 文件包含壓縮文件後綴或非圖片格式，則不再壓縮
                String extension = getFileExtention(file.getName());
                if (!file.getName().contains(SUFFIX) && isImage(extension)) {
                    generateThumbnail2Directory(pathname, file.getAbsolutePath());
                }
            }
        }
    }

    /**
     * 根據文件擴展名判斷文件是否圖片格式
     *
     * @param extension 文件擴展名
     * @return
     */
    public static boolean isImage(String extension) {
        String[] imageExtension = new String[]{"jpeg", "jpg", "gif", "bmp", "png"};

        for (String e : imageExtension) if (extension.toLowerCase().equals(e)) return true;

        return false;
    }

    public static String getFileExtention(String fileName) {
        String extension = fileName.substring(fileName.lastIndexOf(".") + 1);
        return extension;
    }
}

測試

import org.junit.Test;

import java.io.IOException;
import java.util.List;

/**
 * @author lnj
 * createTime 2018-10-19 15:39
 **/
public class ImageUtilTest {

    /**
     * 測試在指定目錄下生成縮略圖
     */
    @Test
    public void testGenerateThumbnail2Directory() throws IOException {
        String path = "D:\\workspace\\idea\\individual\\springboot-learn\\springboot-thumbnail\\image";
        String[] files = new String[]{
                "image/1.jpg",
                "image/2.jpg"
        };

        List<String> list = ImageUtil.generateThumbnail2Directory(path, files);
        System.out.println(list);
    }

    /**
     * 將指定目錄下的圖片生成縮略圖
     */
    @Test
    public void testGenerateDirectoryThumbnail() throws IOException {
        String path = "D:\\workspace\\idea\\individual\\springboot-learn\\springboot-thumbnail\\image";
        ImageUtil.generateDirectoryThumbnail(path);
    }
}

代碼下載地址

https://github.com/linj6/springboot-learn/tree/master/springboot-thumbnail

參考資料

https://juejin.im/post/5b138045f265da6e603934ab#comment

springboot整合thumbnailator實現圖片壓縮