Java 实现文件夹文件指定大小分组压缩的实战案例

2023-10-17 14:55:03

一、前言

在日常工作中，我们经常需要将文件夹中的文件压缩成 ZIP 格式，以便于传输或存储。有时，我们还需要将文件分组压缩，以便更好地管理和组织文件。为了满足这些需求，本文将介绍如何使用 Java 来实现指定大小分组压缩文件夹中文件的实战案例。

二、实现步骤

1. 读取文件夹内容

首先，我们需要使用 Java 来读取文件夹中的文件列表。我们可以使用以下代码来实现：

import java.io.File;

public class FileCompression {

    public static void main(String[] args) {
        // 获取当前工作目录
        String currentDirectory = System.getProperty("user.dir");

        // 创建 File 对象
        File directory = new File(currentDirectory);

        // 获取文件夹中的所有文件
        File[] files = directory.listFiles();

        // 遍历文件列表
        for (File file : files) {
            // 打印文件名称
            System.out.println(file.getName());
        }
    }
}

2. 将文件分组

接下来，我们需要将文件分组。我们可以根据文件的大小、类型或其他属性来分组。例如，我们可以将小于 1MB 的文件分组为一组，将大于 1MB 的文件分组为另一组。

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

public class FileCompression {

    public static void main(String[] args) {
        // 获取当前工作目录
        String currentDirectory = System.getProperty("user.dir");

        // 创建 File 对象
        File directory = new File(currentDirectory);

        // 获取文件夹中的所有文件
        File[] files = directory.listFiles();

        // 创建一个文件组列表
        List<List<File>> fileGroups = new ArrayList<>();

        // 将文件分组
        for (File file : files) {
            // 获取文件大小
            long fileSize = file.length();

            // 根据文件大小将文件分组
            if (fileSize < 1024 * 1024) {
                // 小于 1MB 的文件
                fileGroups.get(0).add(file);
            } else {
                // 大于等于 1MB 的文件
                fileGroups.get(1).add(file);
            }
        }

        // 打印文件组
        for (List<File> fileGroup : fileGroups) {
            for (File file : fileGroup) {
                System.out.println(file.getName());
            }
            System.out.println();
        }
    }
}

3. 压缩文件

最后，我们需要使用 Java 压缩工具来将文件压缩为 ZIP 格式。我们可以使用以下代码来实现：

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class FileCompression {

    public static void main(String[] args) {
        // 获取当前工作目录
        String currentDirectory = System.getProperty("user.dir");

        // 创建 File 对象
        File directory = new File(currentDirectory);

        // 获取文件夹中的所有文件
        File[] files = directory.listFiles();

        // 创建一个文件组列表
        List<List<File>> fileGroups = new ArrayList<>();

        // 将文件分组
        for (File file : files) {
            // 获取文件大小
            long fileSize = file.length();

            // 根据文件大小将文件分组
            if (fileSize < 1024 * 1024) {
                // 小于 1MB 的文件
                fileGroups.get(0).add(file);
            } else {
                // 大于等于 1MB 的文件
                fileGroups.get(1).add(file);
            }
        }

        // 压缩文件
        for (List<File> fileGroup : fileGroups) {
            // 创建一个 ZIP 文件输出流
            ZipOutputStream zipOutputStream = new ZipOutputStream(new FileOutputStream("compressed.zip"));

            // 将文件添加到 ZIP 文件中
            for (File file : fileGroup) {
                // 创建一个 ZIP 文件条目
                ZipEntry zipEntry = new ZipEntry(file.getName());

                // 将文件添加到 ZIP 文件条目中
                zipOutputStream.putNextEntry(zipEntry);

                // 读取文件内容
                FileInputStream fileInputStream = new FileInputStream(file);

                // 将文件内容写入 ZIP 文件条目中
                byte[] buffer = new byte[1024];
                int length;
                while ((length = fileInputStream.read(buffer)) > 0) {
                    zipOutputStream.write(buffer, 0, length);
                }

                // 关闭文件输入流
                fileInputStream.close();
            }

            // 关闭 ZIP 文件输出流
            zipOutputStream.close();
        }

        // 打印压缩成功的消息
        System.out.println("压缩成功！");
    }
}