使用Java代码将文件打包成RAR文件

╰+攻爆jí腚メ 2024-03-23 11:58 70阅读 0赞

该代码实现了将指定文件或目录打包成一个RAR文件的功能。主要使用了Apache Commons Compress框架，该框架提供了多种文件压缩、解压缩的实现方式，并且使用简单。

具体的实现过程：

1.  在compress方法中，首先创建输出流OutputStream对象，用于将打包后的数据输出到指定的RAR文件。
2.  然后使用ArchiveStreamFactory来创建ArchiveOutputStream对象，即RAR归档输出流对象。此处指定RAR格式输入流，并将outputStream对象传入来指定输出位置。
3.  通过compressRecursive方法对指定的源文件或目录进行递归打包。该方法首先判断当前文件是否是目录，如果是，则递归调用compressRecursive方法进行打包。如果是文件，则创建输入流InputStream对象，将输入流的内容按照指定的名称添加到打包输出流ArchiveOutputStream对象中。注意添加完一个文件后要通过closeArchiveEntry方法关闭当前文件的打包输出条目。
4.  最后通过finish方法，关闭RAR归档输出流对象。

需要注意的是，该实现使用了try-with-resources的方式，可以确保在代码块执行完毕后，资源会被自动释放。这样可以避免因为忘记关闭文件流等资源而导致的内存泄漏等问题。同时，在最终释放资源时，我们也需要按照先打开的后关闭的顺序，依次释放相关的资源。

值得注意的是，RAR格式是一种私有格式，压缩、解压缩效率不如zip和gzip等公开格式。在生产环境中，可以根据具体需要选择合适的压缩格式，这也是选择Apache Commons Compress框架的好处之一，它可以支持多种常见格式的压缩与解压缩。

import java.io.*;
    import java.util.*;
    import java.util.zip.*;
    import org.apache.commons.compress.archivers.*;
    import org.apache.commons.compress.archivers.zip.*;
    import org.apache.commons.compress.archivers.rar.*;
    
    public class RarUtil {
     
        private static final int BUFFER_SIZE = 2048;
     
        public static void compress(File sourceFile, File targetFile) throws IOException, ArchiveException {
            // 如果目标文件不存在，创建一个
            if (!targetFile.exists()) {
                targetFile.createNewFile();
            }
     
            OutputStream outputStream = new FileOutputStream(targetFile);
            ArchiveOutputStream archiveOutputStream = null;
            try {
                // 创建 RAR 归档输出流
                archiveOutputStream = new ArchiveStreamFactory().createArchiveOutputStream("rar", outputStream);
     
                // 打包文件
                compressRecursive(sourceFile, "", archiveOutputStream);
            } finally {
                // 关闭归档输出流
                if (archiveOutputStream != null) {
                    archiveOutputStream.finish();
                }
     
                // 关闭输出流
                if (outputStream != null) {
                    outputStream.close();
                }
            }
        }
     
        private static void compressRecursive(File file, String path, ArchiveOutputStream archiveOutputStream) throws IOException {
            String fileName = file.getName();
            String fileFullName = path + fileName;
     
            // 如果是目录，递归打包
            if (file.isDirectory()) {
                ArchiveEntry entry = archiveOutputStream.createArchiveEntry(file, fileFullName + "/");
                archiveOutputStream.putArchiveEntry(entry);
                archiveOutputStream.closeArchiveEntry();
     
                File[] files = file.listFiles();
                for (File subFile : files) {
                    compressRecursive(subFile, fileFullName + "/", archiveOutputStream);
                }
            } else {
                // 如果是文件，直接打包
                InputStream inputStream = new FileInputStream(file);
                ArchiveEntry entry = archiveOutputStream.createArchiveEntry(file, fileFullName);
                archiveOutputStream.putArchiveEntry(entry);
     
                byte[] buffer = new byte[BUFFER_SIZE];
                int length;
                while ((length = inputStream.read(buffer)) != -1) {
                    archiveOutputStream.write(buffer, 0, length);
                }
     
                archiveOutputStream.closeArchiveEntry();
                inputStream.close();
            }
        }
    }